HTTPCookie is a subtype of HTTPClient (#36)

* HTTPCookie is a subtype of HTTPClient to not collide with Foundations top-level type

Author: Yasumoto

Sources/NIOHTTPClient/HTTPClient+HTTPCookie.swift

@@ -0,0 +1,123 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the SwiftNIOHTTPClient open source project
+//
+// Copyright (c) 2018-2019 Swift Server Working Group and the SwiftNIOHTTPClient project authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE.txt for license information
+// See CONTRIBUTORS.txt for the list of SwiftNIOHTTPClient project authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+//===----------------------------------------------------------------------===//
+
+import Foundation
+import NIOHTTP1
+
+extension HTTPClient {
+    public struct Cookie {
+        public var name: String
+        public var value: String
+        public var path: String
+        public var domain: String?
+        public var expires: Date?
+        public var maxAge: Int?
+        public var httpOnly: Bool
+        public var secure: Bool
+
+        public init?(from string: String, defaultDomain: String) {
+            let components = string.components(separatedBy: ";").map {
+                $0.trimmingCharacters(in: .whitespaces)
+            }
+
+            if components.isEmpty {
+                return nil
+            }
+
+            let nameAndValue = components[0].split(separator: "=", maxSplits: 1).map {
+                $0.trimmingCharacters(in: .whitespaces)
+            }
+
+            guard nameAndValue.count == 2 else {
+                return nil
+            }
+
+            self.name = nameAndValue[0]
+            self.value = nameAndValue[1]
+
+            self.path = "/"
+            self.domain = defaultDomain
+            self.expires = nil
+            self.maxAge = nil
+            self.httpOnly = false
+            self.secure = false
+
+            for component in components[1...] {
+                if component.starts(with: "Path"), let value = parseComponentValue(component) {
+                    self.path = value
+                    continue
+                }
+
+                if component.starts(with: "Domain"), let value = parseComponentValue(component) {
+                    self.domain = value
+                    continue
+                }
+
+                if component.starts(with: "Expires") {
+                    let formatter = DateFormatter()
+                    formatter.locale = Locale(identifier: "en_US")
+                    formatter.timeZone = TimeZone(identifier: "GMT")
+                    formatter.dateFormat = "EEE, dd MMM yyyy HH:mm:ss z"
+                    self.expires = self.parseComponentValue(component).flatMap { formatter.date(from: $0) }
+                    continue
+                }
+
+                if component.starts(with: "Max-Age"), let value = parseComponentValue(component).flatMap(Int.init) {
+                    self.maxAge = value
+                    continue
+                }
+
+                if component == "Secure" {
+                    self.secure = true
+                    continue
+                }
+
+                if component == "HttpOnly" {
+                    self.httpOnly = true
+                    continue
+                }
+            }
+        }
+
+        public init(name: String, value: String, path: String = "/", domain: String? = nil, expires: Date? = nil, maxAge: Int? = nil, httpOnly: Bool = false, secure: Bool = false) {
+            self.name = name
+            self.value = value
+            self.path = path
+            self.domain = domain
+            self.expires = expires
+            self.httpOnly = httpOnly
+            self.secure = secure
+        }
+
+        func parseComponentValue(_ component: String) -> String? {
+            let nameAndValue = component.split(separator: "=", maxSplits: 1).map {
+                $0.trimmingCharacters(in: .whitespaces)
+            }
+            if nameAndValue.count == 2 {
+                return nameAndValue[1]
+            }
+            return nil
+        }
+    }
+}
+
+public extension HTTPClient.Response {
+    internal var cookieHeaders: [HTTPHeaders.Element] {
+        return headers.filter { $0.name.lowercased() == "set-cookie" }
+    }
+
+    var cookies: [HTTPClient.Cookie] {
+        return self.cookieHeaders.compactMap { HTTPClient.Cookie(from: $0.value, defaultDomain: self.host) }
+    }
+}

Sources/NIOHTTPClient/HTTPCookie.swift

@@ -19,7 +19,7 @@ import XCTest
 class HTTPCookieTests: XCTestCase {
     func testCookie() {
         let v = "key=value; Path=/path; Domain=example.com; Expires=Wed, 21 Oct 2015 07:28:00 GMT; Max-Age=42; Secure; HttpOnly"
-        let c = HTTPCookie(from: v, defaultDomain: "exampe.org")!
+        let c = HTTPClient.Cookie(from: v, defaultDomain: "exampe.org")!
         XCTAssertEqual("key", c.name)
         XCTAssertEqual("value", c.value)
         XCTAssertEqual("/path", c.path)
@@ -32,7 +32,7 @@ class HTTPCookieTests: XCTestCase {
 
     func testCookieDefaults() {
         let v = "key=value"
-        let c = HTTPCookie(from: v, defaultDomain: "example.org")!
+        let c = HTTPClient.Cookie(from: v, defaultDomain: "example.org")!
         XCTAssertEqual("key", c.name)
         XCTAssertEqual("value", c.value)
         XCTAssertEqual("/", c.path)