app: add an initializer

vapor-community · Sep 6, 2018 · 2d4d6ab · 2d4d6ab
1 parent 117e10e
commit 2d4d6ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/Sources/OneSignal/OneSignalApp.swift b/Sources/OneSignal/OneSignalApp.swift
@@ -15,4 +15,9 @@ public struct OneSignalApp: Codable {
 
     public var apiKey: String
     public var appId: String
+
+    public init(apiKey: String, appId: String) {
+        self.apiKey = apiKey
+        self.appId = appId
+    }
 }
diff --git a/Sources/OneSignal/OneSignalNotification.swift b/Sources/OneSignal/OneSignalNotification.swift
@@ -57,10 +57,10 @@ public struct OneSignalNotification: Codable {
         self.users = users
     }
 
-    public init(title: String, subtitle: String, message: String, users: [String], sound: String? = nil, category: String? = nil) {
-        self.title = OneSignalMessage(title)
-        self.subtitle = OneSignalMessage(subtitle)
-        self.message = OneSignalMessage(message)
+    public init(title: String?, subtitle: String?, body: String, users: [String], sound: String? = nil, category: String? = nil) {
+        if let title = title { self.title = OneSignalMessage(title) }
+        if let subtitle = subtitle { self.subtitle = OneSignalMessage(subtitle) }
+        self.message = OneSignalMessage(body)
         self.users = users
         self.sound = sound
         self.category = category