Finished scenario

Author: shepazon

swift/example_code/sfn/hello/Sources/entry.swift

@@ -33,6 +33,7 @@ struct ExampleCommand: ParsableCommand {
         let sfnConfig = try await SFNClient.SFNClientConfiguration(region: awsRegion)
         let sfnClient = SFNClient(config: sfnConfig)
 
+        // snippet-start:[swift.sfn.hello.ListStateMachines]
         do {
             let output = try await sfnClient.listStateMachines(
                 input: ListStateMachinesInput(
@@ -52,6 +53,7 @@ struct ExampleCommand: ParsableCommand {
         } catch {
             print("*** Error fetching state machine list: \(error.localizedDescription)")
         }
+        // snippet-end:[swift.sfn.hello.ListStateMachines]
     }
 }
 

swift/example_code/sfn/scenario/Package.swift

@@ -0,0 +1,48 @@
+// swift-tools-version: 5.9
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// (swift-tools-version has two lines here because it needs to be the first
+// line in the file, but it should also appear in the snippet below)
+//
+// snippet-start:[swift.sfn.hello.package]
+// swift-tools-version: 5.9
+//
+// The swift-tools-version declares the minimum version of Swift required to
+// build this package.
+
+import PackageDescription
+
+let package = Package(
+    name: "sfn-scenario",
+    // Let Xcode know the minimum Apple platforms supported.
+    platforms: [
+        .macOS(.v13),
+        .iOS(.v15)
+    ],
+    dependencies: [
+        // Dependencies declare other packages that this package depends on.
+        .package(
+            url: "https://github.com/awslabs/aws-sdk-swift",
+            from: "1.0.0"),
+        .package(
+            url: "https://github.com/apple/swift-argument-parser.git",
+            branch: "main"
+        )
+    ],
+    targets: [
+        // Targets are the basic building blocks of a package, defining a module or a test suite.
+        // Targets can depend on other targets in this package and products
+        // from dependencies.
+        .executableTarget(
+            name: "sfn-scenario",
+            dependencies: [
+                .product(name: "AWSIAM", package: "aws-sdk-swift"),
+                .product(name: "AWSSFN", package: "aws-sdk-swift"),
+                .product(name: "ArgumentParser", package: "swift-argument-parser")
+            ],
+            path: "Sources")
+
+    ]
+)
+// snippet-end:[swift.sfn.hello.package]

swift/example_code/sfn/scenario/Sources/Activity.swift

@@ -0,0 +1,125 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import Foundation
+import AWSSFN
+
+/// Describes errors that occur on Step Functions activities.
+enum ActivityError: Error {
+    /// The ARN is missing from the returned activity.
+    case missingArnError
+    /// The activity list is missing from the response.
+    case missingActivityListError
+    /// No matching activity was found.
+    case activityNotFoundError
+
+    var errorDescription: String {
+        switch self {
+            case .missingArnError:
+                return "The ARN is missing from the returned activity"
+            case .missingActivityListError:
+                return "The activity list is missing from the response"
+            case .activityNotFoundError:
+                return "No activity with the specified name was found"
+        }
+    }
+}
+
+/// Manage a Step Functions activity.
+class Activity {
+    let sfnClient: SFNClient
+    let activityName: String
+    var activityArn = ""
+
+    init(client: SFNClient, name: String) async throws {
+        sfnClient = client
+        self.activityName = name
+
+        try await self.findOrCreateActivity()
+    }
+
+    /// Create a new Step Functions activity.
+    /// 
+    /// - Throws: `ActivityError` and appropriate AWS errors.
+    private func createActivity() async throws {
+        let output = try await sfnClient.createActivity(
+            input: CreateActivityInput(name: activityName)
+        )
+
+        guard let arn = output.activityArn else {
+            throw ActivityError.missingArnError
+        }
+
+        activityArn = arn
+    }
+
+    /// Find an activity with the name specified when initializing the
+    /// `Activity` object.
+    ///
+    /// - Throws: `ActivityError` and appropriate AWS errors.
+    private func findActivity() async throws {
+        let pages = sfnClient.listActivitiesPaginated(
+            input: ListActivitiesInput()
+        )
+
+        for try await page in pages {
+            guard let activities = page.activities else {
+                throw ActivityError.missingActivityListError
+            }
+
+            for activity in activities {
+                if activity.name == activityName {
+                    guard let arn = activity.activityArn else {
+                        throw ActivityError.missingArnError
+                    }
+                    self.activityArn = arn
+                }
+            }
+        }
+        
+        throw ActivityError.activityNotFoundError
+    }
+
+    /// Finds an existing activity with the name given when initializing
+    /// the `Activity`. If one isn't found, a new one is created.
+    /// 
+    /// - Throws: `ActivityError` and appropriate AWS errors.
+    private func findOrCreateActivity() async throws {
+        do {
+            try await findActivity()
+        } catch {
+            try await createActivity()
+        }
+    }
+
+    /// Delete the activity described by this object.
+    public func delete() async {
+        do {
+            _ = try await sfnClient.deleteActivity(
+                input: DeleteActivityInput(activityArn: activityArn)
+            )
+        } catch {
+            print("*** Error deleting the activity: \(error.localizedDescription)")
+        }
+    }
+
+    /// Sends a task success notification to the activity.
+    /// 
+    /// - Parameters:
+    ///   - taskToken: The task's token.
+    ///   - response: The task response.
+    /// 
+    /// - Returns: `ActivityError` and appropriate AWS errors.
+    public func sendTaskSuccess(taskToken: String, response: String) async -> Bool {
+        do {
+            _ = try await sfnClient.sendTaskSuccess(
+                input: SendTaskSuccessInput(output: response, taskToken: taskToken)
+            )
+
+            return true
+        } catch {
+            print("*** Error sending task success: \(error.localizedDescription)")
+            return false
+        }
+    }
+}