Merge pull request #186 from ittybittyapps/bundleid/enable-capability

Implement Enable bundleId Capability Command

Author: DechengMa

Sources/AppStoreConnectCLI/AppStoreConnectCLI.swift

@@ -15,6 +15,7 @@ public struct AppStoreConnectCLI: ParsableCommand {
             ReportsCommand.self,
             TestFlightCommand.self,
             UsersCommand.self,
+            CapabilityCommand.self
         ])
 
     public init() {

Sources/AppStoreConnectCLI/Commands/Capability/CapabilityCommand.swift

@@ -0,0 +1,13 @@
+// Copyright 2020 Itty Bitty Apps Pty Ltd
+
+import ArgumentParser
+
+struct CapabilityCommand: ParsableCommand {
+    static var configuration = CommandConfiguration(
+        commandName: "capability",
+        abstract: "Manage the app capabilities for a bundle ID.",
+        subcommands: [
+            EnableBundleIdCapabilityCommand.self
+        ]
+    )
+}

Sources/AppStoreConnectCLI/Commands/Capability/EnableBundleIdCapability.swift

@@ -0,0 +1,31 @@
+// Copyright 2020 Itty Bitty Apps Pty Ltd
+
+import AppStoreConnect_Swift_SDK
+import ArgumentParser
+
+struct EnableBundleIdCapabilityCommand: CommonParsableCommand {
+
+    public static var configuration = CommandConfiguration(
+        commandName: "enable",
+        abstract: "Enable a capability for a bundle ID."
+    )
+
+    @OptionGroup()
+    var common: CommonOptions
+
+    @Argument(help: "The reverse-DNS bundle ID identifier to delete. Must be unique. (eg. com.example.app)")
+    var bundleId: String
+
+    @Argument(help: "Bundle Id capability type. \n \(CapabilityType.allCases)")
+    var capabilityType: CapabilityType
+
+    // TODO: CapabilitySetting
+
+    func run() throws {
+        let service = try makeService()
+
+        try service.enableBundleIdCapability(
+            bundleId: bundleId, capabilityType: capabilityType
+        )
+    }
+}

Sources/AppStoreConnectCLI/Model/API/CapabilityType+ExpressibleByArgument.swift

@@ -0,0 +1,45 @@
+// Copyright 2020 Itty Bitty Apps Pty Ltd
+
+import AppStoreConnect_Swift_SDK
+import ArgumentParser
+import Foundation
+
+extension CapabilityType: CaseIterable, ExpressibleByArgument, CustomStringConvertible {
+    public typealias AllCases = [CapabilityType]
+
+    public static var allCases: AllCases {
+        [
+            .icloud,
+            .inAppPurchase,
+            .gameCenter,
+            .pushNotifications,
+            .wallet,
+            .interAppAudio,
+            .maps,
+            .associatedDomains,
+            .personalVpn,
+            .appGroups,
+            .healthkit,
+            .homekit,
+            .wirelessAccessoryConfiguration,
+            .applePay,
+            .dataProtection,
+            .sirikit,
+            .networkExtensions,
+            .multipath,
+            .hotSpot,
+            .nfcTagReading,
+            .classkit,
+            .autofillCredentialProvider,
+            .accessWifiInformation
+        ]
+    }
+
+    public init?(argument: String) {
+        self.init(rawValue: argument.uppercased())
+    }
+
+    public var description: String {
+        rawValue.lowercased()
+    }
+}

Sources/AppStoreConnectCLI/Services/AppStoreConnectService.swift

@@ -722,6 +722,24 @@ class AppStoreConnectService {
             .await()
     }
 
+    func enableBundleIdCapability(
+        bundleId: String,
+        capabilityType: CapabilityType
+    ) throws {
+        let bundleIdResourceId = try ReadBundleIdOperation(
+                options: .init(bundleId: bundleId)
+            )
+            .execute(with: requestor)
+            .await()
+            .id
+
+        _ = try EnableBundleIdCapabilityOperation(
+                options: .init(bundleIdResourceId: bundleIdResourceId, capabilityType: capabilityType)
+            )
+            .execute(with: requestor)
+            .await()
+    }
+
     /// Make a request for something `Decodable`.
     ///
     /// - Parameters:

Sources/AppStoreConnectCLI/Services/Operations/EnableBundleIdCapabilityOperation.swift

@@ -0,0 +1,31 @@
+// Copyright 2020 Itty Bitty Apps Pty Ltd
+
+import AppStoreConnect_Swift_SDK
+import Combine
+import Foundation
+
+struct EnableBundleIdCapabilityOperation: APIOperation {
+
+    struct Options {
+        let bundleIdResourceId: String
+        let capabilityType: CapabilityType
+    }
+
+    let option: Options
+
+    init(options: Options) {
+        self.option = options
+    }
+
+    func execute(with requestor: EndpointRequestor) -> AnyPublisher<BundleIdCapabilityResponse, Error> {
+        requestor
+            .request(
+                .enableCapability(
+                    id: option.bundleIdResourceId,
+                    capabilityType: option.capabilityType
+                )
+            )
+            .eraseToAnyPublisher()
+    }
+
+}

Sources/AppStoreConnectCLI/Services/Operations/ReadBundleIdOperation.swift

@@ -0,0 +1,52 @@
+// Copyright 2020 Itty Bitty Apps Pty Ltd
+
+import AppStoreConnect_Swift_SDK
+import Combine
+import Foundation
+
+struct ReadBundleIdOperation: APIOperation {
+
+    struct Options {
+        let bundleId: String
+    }
+
+    enum Error: LocalizedError {
+        case couldNotFindBundleId(String)
+        case bundleIdNotUnique(String)
+
+        var errorDescription: String? {
+            switch self {
+            case .couldNotFindBundleId(let bundleId):
+                return "Couldn't find Bundle ID: '\(bundleId)'."
+            case .bundleIdNotUnique(let bundleId):
+                return "The Bundle ID you provided '\(bundleId)' is not unique."
+            }
+        }
+    }
+
+    private let options: Options
+
+    init(options: Options) {
+        self.options = options
+    }
+
+    func execute(with requestor: EndpointRequestor) -> AnyPublisher<BundleId, Swift.Error> {
+        requestor.request(
+            .listBundleIds(
+                filter: [.identifier([options.bundleId])]
+            )
+        )
+        .tryMap {
+            switch $0.data.count {
+            case 0:
+                throw Error.couldNotFindBundleId(self.options.bundleId)
+            case 1:
+                return $0.data.first!
+            default:
+                throw Error.bundleIdNotUnique(self.options.bundleId)
+            }
+        }
+        .eraseToAnyPublisher()
+    }
+
+}