Create apns

Author: mono0926

Config.plist

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>KeyPath</key>
+	<string>YOUR_p8_KEY_PATH</string>
+	<key>KeyId</key>
+	<string>YOUR_KEY_ID</string>
+	<key>TeamId</key>
+	<string>YOUR_TEAM_ID</string>
+	<key>Environment</key>
+	<string>sandbox, production or all</string>
+</dict>
+</plist>

Package.resolved

@@ -4,25 +4,26 @@
 import PackageDescription
 
 let package = Package(
-    name: "apns",
+    name: "APNS",
     products: [
         // Products define the executables and libraries produced by a package, and make them visible to other packages.
         .library(
-            name: "apns",
-            targets: ["apns"]),
+            name: "APNS",
+            targets: ["APNS"]),
     ],
     dependencies: [
         // Dependencies declare other packages that this package depends on.
-        .package(url: "https://github.com/vapor/jwt.git", from: "2.2.0")
+        .package(url: "https://github.com/vapor/jwt.git", from: "2.2.0"),
+        .package(url: "https://github.com/vapor/clibressl.git", from: "1.0.0")
     ],
     targets: [
         // Targets are the basic building blocks of a package. A target can define a module or a test suite.
         // Targets can depend on other targets in this package, and on products in packages which this package depends on.
         .target(
-            name: "apns",
-            dependencies: ["JWT"]),
+            name: "APNS",
+            dependencies: ["JWT", "CLibreSSL"]),
         .testTarget(
-            name: "apnsTests",
-            dependencies: ["apns"]),
+            name: "APNSTests",
+            dependencies: ["APNS"]),
     ]
 )

Package.swift

@@ -0,0 +1,6 @@
+import Foundation
+
+// TODO: Error handling
+public enum APNSError: Error {
+    case unknown
+}

Sources/APNS/APNSError.swift

@@ -0,0 +1,19 @@
+/// https://developer.apple.com/library/content/documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/CommunicatingwithAPNs.html#//apple_ref/doc/uid/TP40008194-CH11-SW1
+import Foundation
+
+struct APNSHeader: Encodable {
+    let topic: String
+    let identifier: String?
+    let expiration: Int
+    let priority: Int
+    let collapseIdentifier: String?
+
+    private enum CodingKeys: String, CodingKey {
+        case
+        topic = "apns-topic",
+        identifier = "apns-id",
+        expiration = "apns-expiration",
+        priority = "apns-priority",
+        collapseIdentifier = "apns-collapse-id"
+    }
+}

Sources/APNS/APNSHeader.swift

@@ -0,0 +1,20 @@
+import Foundation
+
+public struct APNSRequest {
+    let header: APNSHeader
+    let payload: Payload
+
+    public init(topic: String,
+                payload: Payload,
+                apnsIdentifier: UUID? = nil,
+                priority: Priority = .immediately,
+                expiration: Date? = nil,
+                collapseIdentifier: String? = nil) {
+        self.payload = payload
+        self.header = APNSHeader(topic: topic,
+                                 identifier: apnsIdentifier?.uuidString,
+                                 expiration: Int(expiration?.timeIntervalSince1970.rounded() ?? 0),
+                                 priority: priority.rawValue,
+                                 collapseIdentifier: collapseIdentifier)
+    }
+}

Sources/APNS/APNSRequest.swift

@@ -0,0 +1,17 @@
+import Foundation
+import Dispatch
+
+class APNSSession {
+    private let session = URLSession(configuration: .default)
+    func sendSync(request: URLRequest) -> (Data?, URLResponse?, Error?) {
+        var result: (Data?, URLResponse?, Error?)! = nil
+        let semaphor = DispatchSemaphore(value: 0)
+        session.dataTask(with: request) { data, response, error in
+            result = (data, response, error)
+            semaphor.signal()
+            }
+            .resume()
+        semaphor.wait()
+        return result
+    }
+}

Sources/APNS/APNSSession.swift

@@ -0,0 +1,47 @@
+import Foundation
+/// https://developer.apple.com/library/content/documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/PayloadKeyReference.html#//apple_ref/doc/uid/TP40008194-CH17-SW1
+public struct Alert: Codable {
+    let title: String?
+    let subtitle: String?
+    let body: String?
+    let titleLocalizationKey: String?
+    let titleLocalizationArguments: [String]?
+    let actionLocalizationKey: String?
+    let bodyLocalizationKey: String?
+    let bodyLocalizationArguments: [String]?
+    let launchImage: String?
+
+    public init(title: String? = nil,
+                subtitle: String? = nil,
+                body: String? = nil,
+                titleLocalizationKey: String? = nil,
+                titleLocalizationArguments: [String]? = nil,
+                actionLocalizationKey: String? = nil,
+                bodyLocalizationKey: String? = nil,
+                bodyLocalizationArguments: [String]? = nil,
+                launchImage: String? = nil) {
+        self.title = title
+        self.subtitle = subtitle
+        self.body = body
+        self.titleLocalizationKey = titleLocalizationKey
+        self.titleLocalizationArguments = titleLocalizationArguments
+        self.actionLocalizationKey = actionLocalizationKey
+        self.bodyLocalizationKey = bodyLocalizationKey
+        self.bodyLocalizationArguments = bodyLocalizationArguments
+        self.launchImage = launchImage
+    }
+
+
+    private enum CodingKeys: String, CodingKey {
+        case
+        title,
+        subtitle,
+        body,
+        titleLocalizationKey = "title-loc-key",
+        titleLocalizationArguments = "title-loc-args",
+        actionLocalizationKey = "action-loc-key",
+        bodyLocalizationKey = "loc-key",
+        bodyLocalizationArguments = "loc-args",
+        launchImage = "launch-image"
+    }
+}

Sources/APNS/Alert.swift

@@ -0,0 +1,34 @@
+import Foundation
+/// https://developer.apple.com/library/content/documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/PayloadKeyReference.html#//apple_ref/doc/uid/TP40008194-CH17-SW1
+public struct Aps: Codable, JsonInitializable {
+    let alert: Alert?
+    let badge: Int?
+    let sound: String?
+    let contentAvailable: Int?
+    let category: String?
+    let threadId: String?
+
+    public init(alert: Alert?,
+         badge: Int? = nil,
+         sound: String? = nil,
+         contentAvailable: Int? = nil,
+         category: String? = nil,
+         threadId: String? = nil) {
+        self.alert = alert
+        self.badge = badge
+        self.sound = sound
+        self.contentAvailable = contentAvailable
+        self.category = category
+        self.threadId = threadId
+    }
+
+    private enum CodingKeys: String, CodingKey {
+        case
+        alert,
+        badge,
+        sound,
+        contentAvailable = "content-available",
+        category,
+        threadId = "thread-id"
+    }
+}

Sources/APNS/Aps.swift

@@ -0,0 +1,92 @@
+import Foundation
+import JWT
+import CLibreSSL
+
+extension JWT {
+    private func time(with key: String) -> Date? {
+        guard let interval = payload[key]?.double else { return nil }
+        return Date(timeIntervalSince1970: interval)
+    }
+    var issuedAtTime: Date? { return time(with: "iat") }
+}
+
+class AuthenticationKey {
+    let privateKey: String
+    let publicKey: String
+    let keyId: String
+    let teamId: String
+    private var cache: (jwt: JWT, token: String)?
+    func getToken() throws -> String {
+        let now = Date()
+        if let cache = cache, let issuedAtTime = cache.jwt.issuedAtTime, issuedAtTime > now.addingTimeInterval(-60*55) {
+            return cache.token
+        }
+        let jwt = try JWT(additionalHeaders: [KeyID(keyId)],
+                        payload: JSON(.object(["iss":.string(teamId), "iat": .number(.int(Int(now.timeIntervalSince1970.rounded())))])),
+                        signer: ES256(key: privateKey.bytes.base64Decoded))
+        let token = try jwt.createToken()
+        let jwtToVerify = try JWT(token: token)
+        try jwtToVerify.verifySignature(using: ES256(key: publicKey.bytes.base64Decoded))
+        self.cache = (jwt, token)
+        return token
+    }
+
+    init(path: String, keyId: String, teamId: String) throws {
+        (self.privateKey, self.publicKey) = try AuthenticationKey.extractKeys(from: path)
+        self.keyId = keyId
+        self.teamId = teamId
+    }
+
+    private static func extractKeys(from path: String) throws -> (privateKey: String, publicKey: String) {
+        guard FileManager.default.fileExists(atPath: path) else {
+            throw APNSError.unknown
+        }
+
+        var pKey = EVP_PKEY_new()
+        let fp = fopen(path, "r")
+        PEM_read_PrivateKey(fp, &pKey, nil, nil)
+        fclose(fp)
+
+        guard let ecKey = EVP_PKEY_get1_EC_KEY(pKey) else {
+            throw APNSError.unknown
+        }
+
+        EC_KEY_set_conv_form(ecKey, POINT_CONVERSION_UNCOMPRESSED)
+
+        var _pub: UnsafeMutablePointer<UInt8>? = nil
+        let pubLen = i2o_ECPublicKey(ecKey, &_pub)
+        guard let pub = _pub else {
+            throw APNSError.unknown
+        }
+
+        guard let priKeyHex = BN_bn2hex(EC_KEY_get0_private_key(ecKey)), let priKeyHexString = String(validatingUTF8:priKeyHex) else {
+            throw APNSError.unknown
+        }
+        let pubKeyHexString = Data(bytes: Bytes((0..<Int(pubLen)).map { Byte(pub[$0]) })).hexString
+
+        let priBase64String = String(bytes: try dataFromHexadecimalString(text: "00\(priKeyHexString)").base64Encoded)
+        let pubBase64String = String(bytes: try dataFromHexadecimalString(text: pubKeyHexString).base64Encoded)
+
+        return (priBase64String, pubBase64String)
+    }
+
+    private static func dataFromHexadecimalString(text: String) throws -> Data {
+        var data = Data(capacity: text.characters.count / 2)
+        let regex = try! NSRegularExpression(pattern: "[0-9a-f]{1,2}", options: .caseInsensitive)
+        regex.enumerateMatches(in: text, options: [], range: NSRange(location: 0, length: text.characters.count)) { match, flags, stop in
+            let range = Range<String.Index>(match!.range, in: text)!
+            let byteString = text[range]
+            var num = UInt8(byteString, radix: 16)
+            data.append(&num!, count: 1)
+        }
+        return data
+    }
+}
+
+struct KeyID: Header {
+    static let name = "kid"
+    var node: Node
+    init(_ keyID: String) {
+        node = Node(keyID)
+    }
+}