feat: add support for durable functions (aws#8479)

Author: bchampp

.github/workflows/integration-tests.yml

@@ -51,6 +51,7 @@ jobs:
           - local-invoke
           - local-start1
           - local-start2
+          - durable-functions
           - other-and-e2e
         exclude:
           # no-container mode only applies to build-integ test suites
@@ -66,6 +67,8 @@ jobs:
             test_suite: local-start1
           - container_runtime: no-container
             test_suite: local-start2
+          - container_runtime: no-container
+            test_suite: durable-functions
           - container_runtime: no-container
             test_suite: other-and-e2e
 
@@ -374,13 +377,16 @@ jobs:
               pytest -vv tests/integration/sync -n 6 --reruns 3 --dist loadscope --json-report --json-report-file=TEST_REPORT-integration-sync-${{ matrix.container_runtime }}.json
               ;;
             "local-invoke")
-              pytest -vv --reruns 3 tests/integration/local/invoke tests/integration/local/generate_event --json-report --json-report-file=TEST_REPORT-integration-local-invoke-${{ matrix.container_runtime }}.json
+              pytest -vv --reruns 3 tests/integration/local/invoke tests/integration/local/generate_event --ignore tests/integration/local/invoke/test_invoke_durable.py --json-report --json-report-file=TEST_REPORT-integration-local-invoke-${{ matrix.container_runtime }}.json
               ;;
             "local-start1")
-              pytest -vv --reruns 3 tests/integration/local/start_api --ignore tests/integration/local/start_api/test_start_api_with_terraform_application.py --json-report --json-report-file=TEST_REPORT-integration-local-start1-${{ matrix.container_runtime }}.json
+              pytest -vv --reruns 3 tests/integration/local/start_api --ignore tests/integration/local/start_api/test_start_api_with_terraform_application.py --ignore tests/integration/local/start_api/test_start_api_durable.py --json-report --json-report-file=TEST_REPORT-integration-local-start1-${{ matrix.container_runtime }}.json
               ;;
             "local-start2")
-              pytest -vv --reruns 3 tests/integration/local/start_lambda tests/integration/local/start_api/test_start_api_with_terraform_application.py --json-report --json-report-file=TEST_REPORT-integration-local-start2-${{ matrix.container_runtime }}.json
+              pytest -vv --reruns 3 tests/integration/local/start_lambda tests/integration/local/start_api/test_start_api_with_terraform_application.py --ignore tests/integration/local/start_lambda/test_start_lambda_durable.py --json-report --json-report-file=TEST_REPORT-integration-local-start2-${{ matrix.container_runtime }}.json
+              ;;
+            "durable-functions")
+              pytest -vv --reruns 3 tests/integration/local/invoke/test_invoke_durable.py tests/integration/local/start_api/test_start_api_durable.py tests/integration/local/start_lambda/test_start_lambda_durable.py tests/integration/local/callback/test_callback.py tests/integration/local/execution/test_execution.py --json-report --json-report-file=TEST_REPORT-integration-durable-functions-${{ matrix.container_runtime }}.json
               ;;
             "other-and-e2e")
               pytest -vv -n 4 --reruns 4 --dist loadgroup tests/integration tests/end_to_end --ignore=tests/integration/buildcmd --ignore=tests/integration/delete --ignore=tests/integration/deploy --ignore=tests/integration/package --ignore=tests/integration/sync --ignore=tests/integration/local --json-report --json-report-file=TEST_REPORT-integration-others-${{ matrix.container_runtime }}.json

.gitignore

@@ -422,4 +422,7 @@ samcli/lib/init/templates/cookiecutter-aws-sam-hello-java-gradle/**/.gradle/
 .build
 
 .kiro
-mise.toml
+mise.toml
+
+# Durable executions
+**/.durable-executions-local/

samcli/cli/main.py

@@ -142,7 +142,7 @@ def cli(ctx):
     if ctx and getattr(ctx, "command_path", None) == "samdev":
         from samcli.cli.import_module_proxy import attach_import_module_proxy
 
-        LOG.info("Attaching import module proxy for analyzing dynamic imports")
+        LOG.debug("Attaching import module proxy for analyzing dynamic imports")
         attach_import_module_proxy()
 
     gc = GlobalConfig()

samcli/cli/types.py

@@ -633,6 +633,23 @@ def convert(
         return {resource_id: [excluded_path]}
 
 
+class DurableExecutionArnType(click.ParamType):
+    """
+    Custom Parameter Type for Durable Execution ARN validation.
+    """
+
+    name = "string"
+    pattern = (
+        r"^arn:([a-zA-Z0-9-]+):lambda:([a-zA-Z0-9-]+):(\d{12}):function:([a-zA-Z0-9_-]+):"
+        r"(\$LATEST(?:\.PUBLISHED)?|[0-9]+)/durable-execution/([a-zA-Z0-9_-]+)/([a-zA-Z0-9_-]+)$"
+    )
+
+    def convert(self, value, param, ctx):
+        if not re.match(self.pattern, value):
+            raise click.BadParameter(f"Invalid Durable Execution ARN format: {value}")
+        return value
+
+
 class TextWithSpaces:
     def __init__(self, text) -> None:
         self.text = text

samcli/commands/common/__init__.py

@@ -0,0 +1 @@
+

samcli/commands/common/callback/fail/options.py

@@ -0,0 +1,13 @@
+"""
+Shared options for callback fail commands
+"""
+
+from typing import Dict, List
+
+# Common options between local and remote callback fail commands
+COMMON_CALLBACK_FAIL_OPTIONS: List[str] = ["error_data", "stack_trace", "error_type", "error_message"]
+
+# Common options info
+COMMON_CALLBACK_FAIL_OPTIONS_INFO: Dict[str, Dict] = {
+    "Callback Options": {"option_names": {opt: {"rank": idx} for idx, opt in enumerate(COMMON_CALLBACK_FAIL_OPTIONS)}}
+}

samcli/commands/common/callback/heartbeat/options.py

@@ -0,0 +1,11 @@
+"""
+Shared options for callback heartbeat commands
+"""
+
+from typing import Dict, List
+
+# Common options between local and remote callback heartbeat commands
+COMMON_CALLBACK_HEARTBEAT_OPTIONS: List[str] = []
+
+# Common options info
+COMMON_CALLBACK_HEARTBEAT_OPTIONS_INFO: Dict[str, Dict] = {}

samcli/commands/common/callback/succeed/options.py

@@ -0,0 +1,15 @@
+"""
+Shared options for callback succeed commands
+"""
+
+from typing import Dict, List
+
+# Common options between local and remote callback succeed commands
+COMMON_CALLBACK_SUCCEED_OPTIONS: List[str] = ["result"]
+
+# Common options info
+COMMON_CALLBACK_SUCCEED_OPTIONS_INFO: Dict[str, Dict] = {
+    "Callback Options": {
+        "option_names": {opt: {"rank": idx} for idx, opt in enumerate(COMMON_CALLBACK_SUCCEED_OPTIONS)}
+    }
+}

samcli/commands/common/execution/__init__.py

@@ -0,0 +1 @@
+

samcli/commands/common/execution/get/options.py

@@ -0,0 +1,18 @@
+"""
+Shared options for execution get commands
+"""
+
+from typing import Dict, List
+
+from samcli.cli.core.options import add_common_options_info
+
+# Common options between local and remote execution get commands
+COMMON_EXECUTION_GET_FORMATTING_OPTIONS: List[str] = ["format"]
+
+# Common options info with common options included
+COMMON_EXECUTION_GET_FORMATTING_OPTIONS_INFO: Dict[str, Dict] = {
+    "Formatting Options": {
+        "option_names": {opt: {"rank": idx} for idx, opt in enumerate(COMMON_EXECUTION_GET_FORMATTING_OPTIONS)}
+    },
+}
+add_common_options_info(COMMON_EXECUTION_GET_FORMATTING_OPTIONS_INFO)