supporting badge count (#10)

* supprt badge count

* update PoesTests

* fix failing test

Author: Juanpe

README.md

@@ -31,6 +31,7 @@ OPTIONS:
   --bundle-identifier   The bundle identifier to push to
   --mutable, -m         Adds the mutable-content key to the payload
   --title, -t           The title of the Push Notification
+  --badge               The number to display in a badge on your app’s icon
   --verbose             Show extra logging for debugging purposes
   --help                Display available options
 ```

Sources/PoesCore/Payload.swift

@@ -11,19 +11,21 @@ import Foundation
 struct Payload: Codable {
     let aps: APS
 
-    init(title: String, body: String, isMutable: Bool) {
-        aps = APS(alert: Alert(title: title, body: body), isMutable: isMutable)
+    init(title: String, body: String, isMutable: Bool, badge: Int? = nil) {
+        aps = APS(alert: Alert(title: title, body: body), isMutable: isMutable, badge: badge)
     }
 }
 
 struct APS: Codable {
     enum CodingKeys: String, CodingKey {
         case alert
         case isMutable = "mutable-content"
+        case badge = "badge"
     }
 
     let alert: Alert
     let isMutable: Bool
+    let badge: Int?
 }
 
 struct Alert: Codable {

Sources/PoesCore/PayloadFactory.swift

@@ -13,18 +13,21 @@ struct PayloadFactory {
     private let titleArgument: OptionArgument<String>
     private let bodyArgument: OptionArgument<String>
     private let isMutableArgument: OptionArgument<Bool>
+    private let badgeArgument: OptionArgument<Int>
 
     init(parser: ArgumentParser) {
         titleArgument = parser.add(option: "--title", shortName: "-t", kind: String.self, usage: "The title of the Push Notification")
         bodyArgument = parser.add(option: "--body", shortName: "-b", kind: String.self, usage: "The body of the Push Notification")
         isMutableArgument = parser.add(option: "--mutable", shortName: "-m", kind: Bool.self, usage: "Adds the mutable-content key to the payload")
+        badgeArgument = parser.add(option: "--badge", kind: Int.self, usage: "The number to display in a badge on your app’s icon")
     }
 
     func make(using arguments: ArgumentParser.Result) -> Payload {
         let title = arguments.get(titleArgument) ?? "Default title"
         let body = arguments.get(bodyArgument) ?? "Default body"
         let isMutable = arguments.get(isMutableArgument) ?? false
+        let badge = arguments.get(badgeArgument)
 
-        return Payload(title: title, body: body, isMutable: isMutable)
+        return Payload(title: title, body: body, isMutable: isMutable, badge: badge)
     }
 }

Tests/PoesTests/PoesTests.swift

@@ -22,8 +22,9 @@ final class PoesTests: XCTestCase {
         let expectedBundleIdentifier = "com.example.app"
         let title = "Notification title"
         let body = "Notification body"
+        let badge = 2
         let isMutable = true
-        try Poes.run(arguments: ["poes", "--bundle-identifier", expectedBundleIdentifier, "-t", title, "-b", body, "-m"])
+        try Poes.run(arguments: ["poes", "--bundle-identifier", expectedBundleIdentifier, "-t", title, "-b", body, "-m", "--badge", "\(badge)"])
 
         let command = try XCTUnwrap(MockedShell.executedCommand)
 
@@ -40,6 +41,7 @@ final class PoesTests: XCTestCase {
         XCTAssertEqual(payload.aps.alert.title, title)
         XCTAssertEqual(payload.aps.alert.body, body)
         XCTAssertEqual(payload.aps.isMutable, isMutable)
+        XCTAssertEqual(payload.aps.badge, badge)
 
     }
 }