samgilmore · samgilmore · Jul 25, 2024 · Jul 16, 2024 · Jul 16, 2024 · Jul 18, 2024
diff --git a/Sources/SwiftNetKit/CacheConfiguration.swift b/Sources/SwiftNetKit/CacheConfiguration.swift
@@ -0,0 +1,25 @@
+//
+//  CacheConfiguration.swift
+//
+//
+//  Created by Sam Gilmore on 7/19/24.
+//
+
+import Foundation
+
+public struct CacheConfiguration {
+    var memoryCapacity: Int
+    var diskCapacity: Int
+    var diskPath: String?
+    var cachePolicy: URLRequest.CachePolicy
+
+    public init(memoryCapacity: Int = 20 * 1024 * 1024, // 20 MB
+                diskCapacity: Int = 100 * 1024 * 1024, // 100 MB
+                diskPath: String? = nil,
+                cachePolicy: URLRequest.CachePolicy = .useProtocolCachePolicy) {
+        self.memoryCapacity = memoryCapacity
+        self.diskCapacity = diskCapacity
+        self.diskPath = diskPath
+        self.cachePolicy = cachePolicy
+    }
+}
diff --git a/Sources/SwiftNetKit/CookieManager.swift b/Sources/SwiftNetKit/CookieManager.swift
@@ -0,0 +1,133 @@
+//
+//  CookieManager.swift
+//
+//
+//  Created by Sam Gilmore on 7/22/24.
+//
+
+import Foundation
+
+public class CookieManager {
+    static let shared = CookieManager()
+
+    private static let userDefaultsKey = "SWIFTNETKIT_SAVED_COOKIES"
+    private let userDefaults = UserDefaults(suiteName: "SWIFTNETKIT_COOKIE_SUITE")
+
+    var syncCookiesWithUserDefaults: Bool = true
+
+    private init() {
+        cleanExpiredCookies()
+        syncCookies()
+    }
+
+    func includeCookiesIfNeeded(for urlRequest: inout URLRequest, includeCookies: Bool) {
+        if includeCookies {
+            CookieManager.shared.syncCookies()
+            let cookies = CookieManager.shared.getCookiesForURL(for: urlRequest.url!)
+            let cookieHeader = HTTPCookie.requestHeaderFields(with: cookies)
+
+            if let existingCookieHeader = urlRequest.allHTTPHeaderFields?[HTTPCookie.requestHeaderFields(with: []).keys.first ?? ""] {
+                let mergedCookies = (existingCookieHeader + "; " + (cookieHeader.values.first ?? "")).trimmingCharacters(in: .whitespacesAndNewlines)
+                urlRequest.allHTTPHeaderFields?[HTTPCookie.requestHeaderFields(with: []).keys.first ?? ""] = mergedCookies
+            } else {
+                urlRequest.allHTTPHeaderFields = urlRequest.allHTTPHeaderFields?.merging(cookieHeader) { (existing, new) in
+                    return existing + "; " + new
+                } ?? cookieHeader
+            }
+        }
+    }
+
+    func saveCookiesIfNeeded(from response: URLResponse?, saveResponseCookies: Bool) {
+        guard saveResponseCookies,
+              let httpResponse = response as? HTTPURLResponse,
+              let url = httpResponse.url else { return }
+
+        let setCookieHeaders = httpResponse.allHeaderFields.filter { $0.key as? String == "Set-Cookie" }
+
+        let cookies = HTTPCookie.cookies(withResponseHeaderFields: setCookieHeaders as! [String: String], for: url)
+
+        saveCookiesToSession(cookies)
+
+        if syncCookiesWithUserDefaults {
+            saveCookiesToUserDefaults(cookies)
+        }
+    }
+
+    func getCookiesForURL(for url: URL) -> [HTTPCookie] {
+        return HTTPCookieStorage.shared.cookies(for: url) ?? []
+    }
+
+    func syncCookies() {
+        if syncCookiesWithUserDefaults {
+            // Sync cookies from user defaults to session storage
+            loadCookiesFromUserDefaults()
+
+            // Sync cookies from session storage to user defaults
+            saveCookiesToUserDefaults(getAllCookiesFromSession())
+        }
+    }
+
+    func getAllCookiesFromSession() -> [HTTPCookie] {
+        return HTTPCookieStorage.shared.cookies ?? []
+    }
+
+    func saveCookiesToSession(_ cookies: [HTTPCookie]) {
+        for cookie in cookies {
+            HTTPCookieStorage.shared.setCookie(cookie)
+        }
+    }
+
+    func saveCookiesToUserDefaults(_ cookies: [HTTPCookie]) {
+        var cookieDataArray: [Data] = []
+        for cookie in cookies {
+            if let data = try? NSKeyedArchiver.archivedData(withRootObject: cookie.properties ?? [:], requiringSecureCoding: false) {
+                cookieDataArray.append(data)
+            }
+        }
+        userDefaults?.set(cookieDataArray, forKey: CookieManager.userDefaultsKey)
+    }
+
+    func loadCookiesFromUserDefaults() {
+        guard let cookieDataArray = userDefaults?.array(forKey: CookieManager.userDefaultsKey) as? [Data] else { return }
+
+        let allowedClasses: [AnyClass] = [NSDictionary.self, NSString.self, NSDate.self, NSNumber.self, NSURL.self]
+
+        for cookieData in cookieDataArray {
+            if let cookieProperties = try? NSKeyedUnarchiver.unarchivedObject(ofClasses: allowedClasses, from: cookieData) as? [HTTPCookiePropertyKey: Any],
+               let cookie = HTTPCookie(properties: cookieProperties) {
+                HTTPCookieStorage.shared.setCookie(cookie)
+            }
+        }
+    }
+
+    func deleteAllCookies() {
+        HTTPCookieStorage.shared.cookies?.forEach(HTTPCookieStorage.shared.deleteCookie)
+        if syncCookiesWithUserDefaults {
+            userDefaults?.removeObject(forKey: CookieManager.userDefaultsKey)
+        }
+    }
+
+    func deleteExpiredCookies() {
+        guard let cookieDataArray = userDefaults?.array(forKey: CookieManager.userDefaultsKey) as? [Data] else { return }
+        var validCookieDataArray: [Data] = []
+
+        let allowedClasses: [AnyClass] = [NSDictionary.self, NSString.self, NSDate.self, NSNumber.self, NSURL.self]
+
+        for cookieData in cookieDataArray {
+            if let cookieProperties = try? NSKeyedUnarchiver.unarchivedObject(ofClasses: allowedClasses, from: cookieData) as? [HTTPCookiePropertyKey: Any],
+               let cookie = HTTPCookie(properties: cookieProperties),
+               cookie.expiresDate ?? Date.distantFuture > Date() {
+                validCookieDataArray.append(cookieData)
+            }
+        }
+
+        userDefaults?.set(validCookieDataArray, forKey: CookieManager.userDefaultsKey)
+    }
+
+    func cleanExpiredCookies() {
+        deleteExpiredCookies()
+        if syncCookiesWithUserDefaults {
+            loadCookiesFromUserDefaults()
+        }
+    }
+}
diff --git a/Sources/SwiftNetKit/Models/MethodType.swift b/Sources/SwiftNetKit/Models/MethodType.swift
@@ -0,0 +1,14 @@
+//
+//  MethodType.swift
+//
+//
+//  Created by Sam Gilmore on 7/17/24.
+//
+
+public enum MethodType: String {
+    case get = "GET"
+    case post = "POST"
+    case delete = "DELETE"
+    case put = "PUT"
+    case patch = "PATCH"
+}
diff --git a/Sources/SwiftNetKit/Models/NetworkError.swift b/Sources/SwiftNetKit/Models/NetworkError.swift
@@ -0,0 +1,14 @@
+//
+//  NetworkError.swift
+//  
+//
+//  Created by Sam Gilmore on 7/16/24.
+//
+
+public enum NetworkError: Error {
+    case invalidResponse
+    case decodingFailed
+    case serverError(statusCode: Int)
+    case requestFailed(error: Error)
+    case unknown
+}
diff --git a/Sources/SwiftNetKit/Models/RequestBody.swift b/Sources/SwiftNetKit/Models/RequestBody.swift
@@ -0,0 +1,14 @@
+//
+//  RequestBody.swift
+//
+//
+//  Created by Sam Gilmore on 7/18/24.
+//
+
+import Foundation
+
+public enum RequestBody {
+    case jsonEncodable(Encodable)
+    case data(Data)
+    case string(String)
+}
diff --git a/Sources/SwiftNetKit/Models/SessionConfiguration.swift b/Sources/SwiftNetKit/Models/SessionConfiguration.swift
@@ -0,0 +1,12 @@
+//
+//  SessionConfiguration.swift
+//
+//
+//  Created by Sam Gilmore on 7/17/24.
+//
+
+public enum SessionConfiguration {
+    case `default`
+    case ephemeral
+    case background(String)
+}