Make Application.APNS an APNSwiftClient (#12)

vapor · Jan 9, 2020 · afc9987 · afc9987
1 parent cec35a4
commit afc9987
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/Sources/APNS/Application+APNS.swift b/Sources/APNS/Application+APNS.swift
@@ -50,3 +50,40 @@ extension Application {
         let application: Application
     }
 }
+
+extension Application.APNS: APNSwiftClient {
+    public var logger: Logger? {
+        self.application.logger
+    }
+
+    public var eventLoop: EventLoop {
+        self.application.eventLoopGroup.next()
+    }
+
+    public func send(
+        rawBytes payload: ByteBuffer,
+        pushType: APNSwiftConnection.PushType,
+        to deviceToken: String,
+        expiration: Date?,
+        priority: Int?,
+        collapseIdentifier: String?,
+        topic: String?,
+        logger: Logger?
+    ) -> EventLoopFuture<Void> {
+        self.application.apns.pool.withConnection(
+            logger: logger,
+            on: self.eventLoop
+        ) {
+            $0.send(
+                rawBytes: payload,
+                pushType: pushType,
+                to: deviceToken,
+                expiration: expiration,
+                priority: priority,
+                collapseIdentifier: collapseIdentifier,
+                topic: topic,
+                logger: logger
+            )
+        }
+    }
+}