localstack 4.13.2.dev3__tar.gz → 4.13.2.dev36__tar.gz

{localstack-4.13.2.dev3 → localstack-4.13.2.dev36}/Makefile

@@ -1,4 +1,4 @@
-.PHONY: install test test-unit test-integration lint clean
+.PHONY: install test test-unit test-integration lint clean entrypoints
 
 VENV_DIR ?= .venv
 PYTHON ?= python3
@@ -29,6 +29,10 @@ format: install
 	$(VENV_DIR)/bin/ruff check --fix localstack_cli tests
 	$(VENV_DIR)/bin/ruff format localstack_cli tests
 
+entrypoints: install
+	$(VENV_DIR)/bin/python -m plux entrypoints
+	@test -s plux.ini || (echo "Entrypoints not created correctly!" && exit 1)
+
 clean:
 	rm -rf $(VENV_DIR)
 	rm -rf build dist *.egg-info

{localstack-4.13.2.dev3/localstack.egg-info → localstack-4.13.2.dev36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: localstack
-Version: 4.13.2.dev3
+Version: 4.13.2.dev36
 Summary: The LocalStack Command Line Interface
 Author-email: LocalStack Contributors <info@localstack.cloud>
 License-Expression: Apache-2.0
@@ -24,7 +24,7 @@ Requires-Dist: cryptography
 Requires-Dist: dnslib>=0.9.10
 Requires-Dist: dnspython>=1.16.0
 Requires-Dist: docker>=6.1.1
-Requires-Dist: plux>=1.10
+Requires-Dist: plux>=1.14.0
 Requires-Dist: psutil>=5.4.8
 Requires-Dist: python-dotenv>=0.19.1
 Requires-Dist: pyyaml>=5.1

{localstack-4.13.2.dev3 → localstack-4.13.2.dev36/localstack.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: localstack
-Version: 4.13.2.dev3
+Version: 4.13.2.dev36
 Summary: The LocalStack Command Line Interface
 Author-email: LocalStack Contributors <info@localstack.cloud>
 License-Expression: Apache-2.0
@@ -24,7 +24,7 @@ Requires-Dist: cryptography
 Requires-Dist: dnslib>=0.9.10
 Requires-Dist: dnspython>=1.16.0
 Requires-Dist: docker>=6.1.1
-Requires-Dist: plux>=1.10
+Requires-Dist: plux>=1.14.0
 Requires-Dist: psutil>=5.4.8
 Requires-Dist: python-dotenv>=0.19.1
 Requires-Dist: pyyaml>=5.1

{localstack-4.13.2.dev3 → localstack-4.13.2.dev36}/localstack.egg-info/SOURCES.txt

@@ -1,6 +1,7 @@
 .gitignore
 Makefile
 README.md
+plux.ini
 pyproject.toml
 ./localstack_cli/__init__.py
 ./localstack_cli/config.py
@@ -26,9 +27,9 @@ pyproject.toml
 ./localstack_cli/pro/core/__init__.py
 ./localstack_cli/pro/core/config.py
 ./localstack_cli/pro/core/constants.py
+./localstack_cli/pro/core/plugins.py
 ./localstack_cli/pro/core/bootstrap/__init__.py
 ./localstack_cli/pro/core/bootstrap/auth.py
-./localstack_cli/pro/core/bootstrap/decryption.py
 ./localstack_cli/pro/core/bootstrap/dns_utils.py
 ./localstack_cli/pro/core/bootstrap/entitlements.py
 ./localstack_cli/pro/core/bootstrap/licensingv2.py
@@ -36,6 +37,7 @@ pyproject.toml
 ./localstack_cli/pro/core/bootstrap/extensions/__init__.py
 ./localstack_cli/pro/core/bootstrap/extensions/__main__.py
 ./localstack_cli/pro/core/bootstrap/extensions/autoinstall.py
+./localstack_cli/pro/core/bootstrap/extensions/bootstrap.py
 ./localstack_cli/pro/core/bootstrap/extensions/repository.py
 ./localstack_cli/pro/core/bootstrap/pods/__init__.py
 ./localstack_cli/pro/core/bootstrap/pods/api_types.py
@@ -59,6 +61,9 @@ pyproject.toml
 ./localstack_cli/pro/core/cli/replicator.py
 ./localstack_cli/pro/core/cli/state.py
 ./localstack_cli/pro/core/cli/tree_view.py
+./localstack_cli/runtime/__init__.py
+./localstack_cli/runtime/exceptions.py
+./localstack_cli/runtime/hooks.py
 ./localstack_cli/utils/__init__.py
 ./localstack_cli/utils/archives.py
 ./localstack_cli/utils/batching.py
@@ -96,13 +101,7 @@ pyproject.toml
 ./localstack_cli/utils/analytics/logger.py
 ./localstack_cli/utils/analytics/metadata.py
 ./localstack_cli/utils/analytics/publisher.py
-./localstack_cli/utils/analytics/service_providers.py
 ./localstack_cli/utils/analytics/service_request_aggregator.py
-./localstack_cli/utils/analytics/metrics/__init__.py
-./localstack_cli/utils/analytics/metrics/api.py
-./localstack_cli/utils/analytics/metrics/counter.py
-./localstack_cli/utils/analytics/metrics/publisher.py
-./localstack_cli/utils/analytics/metrics/registry.py
 ./localstack_cli/utils/container_utils/__init__.py
 ./localstack_cli/utils/container_utils/container_client.py
 ./localstack_cli/utils/container_utils/docker_cmd_client.py
@@ -140,9 +139,9 @@ localstack_cli/pro/__init__.py
 localstack_cli/pro/core/__init__.py
 localstack_cli/pro/core/config.py
 localstack_cli/pro/core/constants.py
+localstack_cli/pro/core/plugins.py
 localstack_cli/pro/core/bootstrap/__init__.py
 localstack_cli/pro/core/bootstrap/auth.py
-localstack_cli/pro/core/bootstrap/decryption.py
 localstack_cli/pro/core/bootstrap/dns_utils.py
 localstack_cli/pro/core/bootstrap/entitlements.py
 localstack_cli/pro/core/bootstrap/licensingv2.py
@@ -150,6 +149,7 @@ localstack_cli/pro/core/bootstrap/pods_client.py
 localstack_cli/pro/core/bootstrap/extensions/__init__.py
 localstack_cli/pro/core/bootstrap/extensions/__main__.py
 localstack_cli/pro/core/bootstrap/extensions/autoinstall.py
+localstack_cli/pro/core/bootstrap/extensions/bootstrap.py
 localstack_cli/pro/core/bootstrap/extensions/repository.py
 localstack_cli/pro/core/bootstrap/pods/__init__.py
 localstack_cli/pro/core/bootstrap/pods/api_types.py
@@ -173,6 +173,9 @@ localstack_cli/pro/core/cli/localstack.py
 localstack_cli/pro/core/cli/replicator.py
 localstack_cli/pro/core/cli/state.py
 localstack_cli/pro/core/cli/tree_view.py
+localstack_cli/runtime/__init__.py
+localstack_cli/runtime/exceptions.py
+localstack_cli/runtime/hooks.py
 localstack_cli/utils/__init__.py
 localstack_cli/utils/archives.py
 localstack_cli/utils/batching.py
@@ -210,13 +213,7 @@ localstack_cli/utils/analytics/events.py
 localstack_cli/utils/analytics/logger.py
 localstack_cli/utils/analytics/metadata.py
 localstack_cli/utils/analytics/publisher.py
-localstack_cli/utils/analytics/service_providers.py
 localstack_cli/utils/analytics/service_request_aggregator.py
-localstack_cli/utils/analytics/metrics/__init__.py
-localstack_cli/utils/analytics/metrics/api.py
-localstack_cli/utils/analytics/metrics/counter.py
-localstack_cli/utils/analytics/metrics/publisher.py
-localstack_cli/utils/analytics/metrics/registry.py
 localstack_cli/utils/container_utils/__init__.py
 localstack_cli/utils/container_utils/container_client.py
 localstack_cli/utils/container_utils/docker_cmd_client.py

localstack-4.13.2.dev36/localstack.egg-info/entry_points.txt

@@ -0,0 +1,17 @@
+[console_scripts]
+localstack = localstack_cli.cli.main:main
+
+[localstack.hooks.configure_localstack_container]
+configure_extensions_dev_container = localstack_cli.pro.core.plugins:configure_extensions_dev_container
+configure_pro_container = localstack_cli.pro.core.plugins:configure_pro_container
+set_analytics_header = localstack_cli.utils.analytics.metadata:set_analytics_header
+
+[localstack.hooks.prepare_host]
+activate_pro_key_on_host = localstack_cli.pro.core.plugins:activate_pro_key_on_host
+configure_extensions_dev_host = localstack_cli.pro.core.plugins:configure_extensions_dev_host
+patch_community_pro_detection = localstack_cli.pro.core.plugins:patch_community_pro_detection
+publish_invocation_metadata = localstack_cli.utils.analytics.metadata:publish_invocation_metadata
+
+[localstack_cli.plugins.cli]
+core = localstack_cli.cli.core_plugin:CoreCliPlugin
+pro = localstack_cli.pro.core.cli.localstack:ProCliPlugins

{localstack-4.13.2.dev3 → localstack-4.13.2.dev36}/localstack.egg-info/requires.txt

@@ -5,7 +5,7 @@ cryptography
 dnslib>=0.9.10
 dnspython>=1.16.0
 docker>=6.1.1
-plux>=1.10
+plux>=1.14.0
 psutil>=5.4.8
 python-dotenv>=0.19.1
 pyyaml>=5.1

localstack-4.13.2.dev36/localstack_cli/pro/core/bootstrap/extensions/bootstrap.py

@@ -0,0 +1,97 @@
+"""Hooks for extension developer mode on the host CLI."""
+
+from __future__ import annotations
+
+import logging
+import os
+from pathlib import Path
+from tempfile import gettempdir
+
+from localstack_cli import config
+from localstack_cli.utils.bootstrap import Container
+from localstack_cli.utils.container_utils.container_client import BindMount
+from localstack_cli.utils.json import FileMappedDocument
+from localstack_cli.utils.strings import md5
+
+LOG = logging.getLogger(__name__)
+
+_ENTRYPOINT_SCRIPT = """#!/bin/bash
+echo "=================================================="
+echo "LocalStack extension developer mode enabled"
+shopt -s nullglob
+
+pkgs=$(echo /opt/code/localstack/.venv/lib/python3*/site-packages)
+
+echo "import localstack_extensions_resolve;localstack_extensions_resolve.resolve()" > ${pkgs}/localstack-extensions-resolve.pth
+
+cat << EOF >> ${pkgs}/localstack_extensions_resolve.py
+import os
+import sys
+import glob
+
+base_dirs_visited = set()
+
+def append_pth_recursively(base_dir):
+    if base_dir in base_dirs_visited:
+        return
+    base_dirs_visited.add(base_dir)
+
+    for f in glob.glob(f"{base_dir}/*.pth", recursive=True):
+        with open(f, "r") as fd:
+            abs_path = os.path.abspath(os.path.dirname(f))
+            lines = fd.readlines()
+            for line in lines:
+                if line := line.strip():
+                    if "import" in line:
+                        continue
+                    module_dir = os.path.abspath(os.path.join(abs_path, line)) if not os.path.isabs(line) else line
+                    if os.path.exists(module_dir) and os.path.isdir(module_dir):
+                        if module_dir not in sys.path:
+                            sys.path.append(module_dir)
+                        append_pth_recursively(module_dir)
+
+
+def resolve():
+    append_pth_recursively(os.path.dirname(__file__))
+EOF
+
+for d in /opt/code/extensions/* ;
+do
+    echo "- mounting extension ${d}"
+    find ${d} -type d -name "site-packages" >> ${pkgs}/localstack-extensions-venv.pth
+    echo ${d} >> ${pkgs}/localstack-extensions-venv.pth
+done
+echo "Resuming normal execution, ..."
+echo "=================================================="
+exec /usr/local/bin/docker-entrypoint.sh
+"""
+
+_host_extension_dirs: list[str] = []
+
+
+def run_on_configure_host_hook():
+    """Load extension directories from ~/.localstack/extensions-dev.json."""
+    doc = FileMappedDocument(os.path.join(config.CONFIG_DIR, "extensions-dev.json"))
+
+    for extensions_spec in doc.get("extensions", []):
+        path = extensions_spec.get("host_path")
+        if path and os.path.exists(path):
+            _host_extension_dirs.append(path)
+
+
+def run_on_configure_localstack_container_hook(container: Container):
+    """Configure container for extension dev mode."""
+    # Create and mount custom entrypoint script
+    h = md5(_ENTRYPOINT_SCRIPT)
+    file = Path(gettempdir(), f"docker-entrypoint-{h}.sh")
+    if not file.exists():
+        file.write_text(_ENTRYPOINT_SCRIPT, newline="\n", encoding="utf-8")
+        file.chmod(0o777)
+
+    container.config.volumes.add(BindMount(str(file), f"/tmp/{file.name}"))
+    container.config.entrypoint = f"/tmp/{file.name}"
+
+    # Mount extension directories
+    for ext_dir in _host_extension_dirs:
+        target = os.path.join("/opt/code/extensions", os.path.basename(ext_dir))
+        container.config.volumes.add(BindMount(ext_dir, target, read_only=True))

{localstack-4.13.2.dev3 → localstack-4.13.2.dev36}/localstack_cli/pro/core/bootstrap/licensingv2.py

@@ -1,8 +1,6 @@
 import base64
-import binascii
 import copy
 import dataclasses
-import hashlib
 import hmac
 import json
 import logging
@@ -26,9 +24,8 @@ from localstack_cli.pro.core.bootstrap.entitlements import (
     ProductInfo,
 )
 from localstack_cli.pro.core.constants import PLATFORM_PLUGIN_NAMESPACE
-from localstack_cli.utils.files import load_file
 from localstack_cli.utils.objects import singleton_factory
-from localstack_cli.utils.strings import md5, to_str
+from localstack_cli.utils.strings import md5
 from plux import Plugin, PluginDisabled, PluginLifecycleListener, PluginSpec
 
 LOG = logging.getLogger(__name__)
@@ -446,20 +443,6 @@ class LicensingClient:
         """
         raise NotImplementedError
 
-    def decode_decryption_key(self, credentials: Credentials, license: License) -> bytes:
-        raise NotImplementedError
-
-
-class LicenseSecretDecoder:
-    """
-    Class to decode the ``license_secret`` field from the given license. The field contains a secret which is
-    the decryption key for the localstack code.
-    """
-
-    def decode_license_secret(self, license: LicenseV1) -> bytes:  # noqa
-        raise NotImplementedError
-
-
 #####################
 # v1 implementation #
 #####################
@@ -518,64 +501,6 @@ class LicenseParser:
             raise LicenseFormatError(f"error parsing license file: {e}") from e
 
 
-class AESLicenseV1SecretDecoder(LicenseSecretDecoder):
-    """
-    Uses a custom AES encryption scheme to decode the source decryption key from the license key. This is
-    mostly obfuscation and not really a trust mechanism.
-
-    The shared key is the sha256 digest of the licensing credentials concatenated with the license signature.
-    """
-
-    scheme = "LS1.0"
-
-    def __init__(self, credentials: Credentials):
-        self.credentials = credentials
-
-    def decode_license_secret(self, license: LicenseV1) -> bytes:  # noqa
-        if not license.signature:
-            raise LicenseFormatError("license is not signed")
-
-        # use the signature of the license as well as the credentials to encode the license key
-        hash_ = hashlib.sha256()
-        hash_.update(self.credentials.to_bytes())
-        hash_.update(binascii.unhexlify(license.signature))  # use the byte value, not the string
-        key = hash_.digest()
-
-        # extract the scheme from the base64 encoded string
-        license_secret = base64.b64decode(license.license_secret)
-        parts = license_secret.split(b":", maxsplit=1)
-        if len(parts) != 2:  # noqa PLR2004
-            raise LicenseFormatError("invalid license key format")
-        scheme, secret = parts
-
-        if scheme.decode("utf-8") != self.scheme:
-            raise ValueError(f"unknown scheme {scheme}")
-
-        iv = secret[:16]
-        encrypted_data = secret[16:]
-
-        return self._aes_decrypt(key, iv, encrypted_data)
-
-    def _aes_decrypt(self, key: bytes, iv: bytes, encrypted_data: bytes):
-        from cryptography.hazmat.primitives import padding
-        from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
-
-        # Create an AES cipher with the provided key and CBC mode
-        cipher = Cipher(algorithms.AES(key), modes.CBC(iv))
-
-        # Create a decryptor object using the AES cipher
-        decryptor = cipher.decryptor()
-
-        # Decrypt the encrypted data
-        decrypted_data = decryptor.update(encrypted_data) + decryptor.finalize()
-
-        # Remove PKCS7 padding from the decrypted data
-        unpadder = padding.PKCS7(128).unpadder()
-        unpadded_data = unpadder.update(decrypted_data)
-
-        return unpadded_data + unpadder.finalize()
-
-
 class LicenseV1ClientBase(LicensingClient):
     def validate_license(self, credentials: Credentials, license: LicenseV1):
         try:
@@ -640,17 +565,6 @@ class LicenseV1Client(LicenseV1ClientBase):
         license_.offline_data["localstack_version"] = VERSION
         return license_
 
-    def decode_decryption_key(self, credentials: Credentials, license: LicenseV1) -> bytes:
-        key = base64.b64decode(license.license_secret)
-        parts = key.split(b":", maxsplit=1)
-        scheme = to_str(parts[0])
-        if scheme == AESLicenseV1SecretDecoder.scheme:
-            decoder = AESLicenseV1SecretDecoder(credentials)
-        else:
-            raise LicenseFormatError(f"unknown license key scheme {scheme}")
-
-        return decoder.decode_license_secret(license)
-
     def _get_machine_data(self) -> dict:
         from localstack_cli.utils.analytics.metadata import get_client_metadata
 
@@ -826,10 +740,9 @@ class LicensedLocalstackEnvironment:
 
     def activate(self, offline_only: bool = False):
         """
-        This high-level method first activates the license using credentials from the environment, and then
-        enables code decryption if necessary. Once completed, it sets ``self.license`` to the license document
-        that was activated. This method will also set the env var ``PRO_ACTIVATED=1`` after successful license
-        activation and code decryption.
+        This high-level method activates the license using credentials from the environment.
+        Once completed, it sets ``self.license`` to the license document that was activated.
+        This method will also set the env var ``PRO_ACTIVATED=1`` after successful license activation.
 
         See ``activate_license`` for the activation algorithm.
 
@@ -854,9 +767,6 @@ class LicensedLocalstackEnvironment:
 
             self.activate_license(offline_only)
 
-            if not self.is_decryption_enabled():
-                self.enable_decryption()
-
             os.environ[constants.ENV_PRO_ACTIVATED] = "1"
 
     def activate_license(self, offline_only: bool = False):
@@ -974,55 +884,6 @@ class LicensedLocalstackEnvironment:
         """
         return self.product_entitlements.has_entitlement(product_name)
 
-    def enable_decryption(self):
-        """
-        Called by ``activate`` to read the source decryption key from the license key and register the
-        ``DecryptionHandler`` into the sys path.
-        """
-        if not self.activated or not self.license:
-            raise ValueError("license not yet activated")
-
-        from localstack_cli.pro.core.bootstrap.decryption import (
-            DecryptionHandler,
-            init_source_decryption,
-        )
-        from localstack_cli.pro.core.config import ROOT_FOLDER
-
-        decryption_handler = DecryptionHandler(
-            self.client.decode_decryption_key(self.require_valid_credentials(), self.license)
-        )
-
-        try:
-            file_name = f"{ROOT_FOLDER}/localstack/pro/core/utils/decryption_check.py.enc"
-            encrypted_file_content = load_file(file_name, mode="rb")
-            if not encrypted_file_content:
-                raise ValueError(
-                    "Decryption check file not found. Are you using localstack pro in host mode?"
-                )
-            file_content = decryption_handler.decrypt(encrypted_file_content)
-            if b"decryption_check" not in file_content:
-                raise ValueError("Decryption resulted in invalid python file")
-        except Exception as e:
-            raise LicenseActivationError(
-                "Error while trying to perform code activation. You may be using a "
-                "version of LocalStack that is not within your license agreement."
-            ) from e
-
-        init_source_decryption(decryption_handler)
-
-    def is_decryption_enabled(self) -> bool:
-        """
-        Checks whether decryption is enabled. Decryption can be enabled in a test environment when the
-        source code is available. So this can return true even if a license has not been activated.
-
-        :return: true if the normally encrypted source code can be accessed.
-        """
-        try:
-            from localstack_cli.pro.core.utils.decryption_check import decryption_check  # noqa
-
-            return True
-        except ImportError:
-            return False
 
     def save_license(self):
         """

localstack-4.13.2.dev36/localstack_cli/pro/core/plugins.py

@@ -0,0 +1,81 @@
+"""
+Pro plugin hooks for the LocalStack CLI.
+
+These hooks are executed on the host machine when running CLI commands like `localstack start`.
+They handle license activation, container configuration, and extension developer mode.
+
+Note: This file was extracted from localstack-pro-core/localstack/pro/core/plugins.py
+and rewritten to contain only CLI-relevant functionality.
+"""
+
+from __future__ import annotations
+
+import logging
+import os
+
+from localstack_cli import config as localstack_config
+from localstack_cli.config import HostAndPort
+from localstack_cli.pro.core import config as pro_config
+from localstack_cli.pro.core.bootstrap import licensingv2
+from localstack_cli.runtime import hooks
+from localstack_cli.runtime.exceptions import LocalstackExit
+from localstack_cli.utils.bootstrap import Container
+
+LOG = logging.getLogger(__name__)
+
+
+def modify_gateway_listen_config(cfg):
+    """
+    Modifies the localstack config to additionally listen to port 443.
+    Needs to be called before any edge URLs are resolved using the config.
+    """
+    if os.getenv("GATEWAY_LISTEN") is None:
+        host = "0.0.0.0" if localstack_config.in_docker() else "127.0.0.1"
+        cfg.GATEWAY_LISTEN.append(HostAndPort(host=host, port=443))
+
+
+@hooks.prepare_host(priority=200)
+def patch_community_pro_detection():
+    """This is currently needed to make localstack core aware of the `localstack auth set-token`
+    functionality, where we set the key into the ``~/.localstack/auth.json`` file that community does not
+    yet know about. ``is_api_key_configured`` is used in the LocalStack CLI to determine whether to start
+    the localstack or localstack-pro container image."""
+    from localstack_cli.utils import bootstrap
+
+    bootstrap.is_auth_token_configured = pro_config.is_auth_token_configured
+
+
+@hooks.prepare_host(priority=100, should_load=pro_config.ACTIVATE_PRO)
+def activate_pro_key_on_host():
+    """Activate license on host (needed for DNS forward and EC2 daemon)."""
+    try:
+        licensingv2.get_licensed_environment().activate()
+    except licensingv2.LicensingError as e:
+        raise LocalstackExit(reason=e.get_user_friendly(), code=55)
+
+
+@hooks.configure_localstack_container(priority=10, should_load=pro_config.ACTIVATE_PRO)
+def configure_pro_container(container: Container):
+    """Configure the LocalStack container for pro features."""
+    modify_gateway_listen_config(localstack_config)
+    container.configure(licensingv2.configure_container_licensing)
+
+
+@hooks.prepare_host(should_load=pro_config.ACTIVATE_PRO and pro_config.EXTENSION_DEV_MODE)
+def configure_extensions_dev_host():
+    """Load extension directories from ~/.localstack/extensions-dev.json."""
+    from localstack_cli.pro.core.bootstrap.extensions.bootstrap import run_on_configure_host_hook
+
+    run_on_configure_host_hook()
+
+
+@hooks.configure_localstack_container(
+    should_load=pro_config.ACTIVATE_PRO and pro_config.EXTENSION_DEV_MODE
+)
+def configure_extensions_dev_container(container: Container):
+    """Configure container for extension developer mode."""
+    from localstack_cli.pro.core.bootstrap.extensions.bootstrap import (
+        run_on_configure_localstack_container_hook,
+    )
+
+    run_on_configure_localstack_container_hook(container)

localstack-4.13.2.dev36/localstack_cli/runtime/__init__.py

@@ -0,0 +1,6 @@
+"""Runtime utilities for the LocalStack CLI."""
+
+from localstack_cli.runtime import hooks  # noqa: F401
+
+# get_current_runtime is not available in the standalone CLI - only in the full LocalStack runtime
+get_current_runtime = None

localstack-4.13.2.dev36/localstack_cli/runtime/exceptions.py

@@ -0,0 +1,7 @@
+class LocalstackExit(Exception):
+    """Raised to gracefully exit LocalStack."""
+
+    def __init__(self, reason: str = None, code: int = 1):
+        self.reason = reason
+        self.code = code
+        super().__init__(reason)

localstack-4.13.2.dev36/localstack_cli/runtime/hooks.py

@@ -0,0 +1,73 @@
+import functools
+
+from plux import PluginManager, plugin
+
+# plugin namespace constants
+HOOKS_CONFIGURE_LOCALSTACK_CONTAINER = "localstack.hooks.configure_localstack_container"
+HOOKS_PREPARE_HOST = "localstack.hooks.prepare_host"
+
+
+def hook(namespace: str, priority: int = 0, **kwargs):
+    """
+    Decorator for creating functional plugins that have a hook_priority attribute. Hooks with a higher priority value
+    will be executed earlier.
+    """
+
+    def wrapper(fn):
+        fn.hook_priority = priority
+        return plugin(namespace=namespace, **kwargs)(fn)
+
+    return wrapper
+
+
+def hook_spec(namespace: str):
+    """
+    Creates a new hook decorator bound to a namespace.
+
+    on_infra_start = hook_spec("localstack.hooks.on_infra_start")
+
+    @on_infra_start()
+    def foo():
+        pass
+
+    # run all hooks in order
+    on_infra_start.run()
+    """
+    fn = functools.partial(hook, namespace=namespace)
+    # attach hook manager and run method to decorator for convenience calls
+    fn.manager = HookManager(namespace)
+    fn.run = fn.manager.run_in_order
+    return fn
+
+
+class HookManager(PluginManager):
+    def load_all_sorted(self, propagate_exceptions=False):
+        """
+        Loads all hook plugins and sorts them by their hook_priority attribute.
+        """
+        plugins = self.load_all(propagate_exceptions)
+        # the hook_priority attribute is part of the function wrapped in the FunctionPlugin
+        plugins.sort(
+            key=lambda _fn_plugin: getattr(_fn_plugin.fn, "hook_priority", 0), reverse=True
+        )
+        return plugins
+
+    def run_in_order(self, *args, **kwargs):
+        """
+        Loads and runs all plugins in order them with the given arguments.
+        """
+        for fn_plugin in self.load_all_sorted():
+            fn_plugin(*args, **kwargs)
+
+    def __str__(self):
+        return f"HookManager({self.namespace})"
+
+    def __repr__(self):
+        return self.__str__()
+
+
+configure_localstack_container = hook_spec(HOOKS_CONFIGURE_LOCALSTACK_CONTAINER)
+"""Hooks to configure the LocalStack container before it starts. Executed on the host when invoking the CLI."""
+
+prepare_host = hook_spec(HOOKS_PREPARE_HOST)
+"""Hooks to prepare the host that's starting LocalStack. Executed on the host when invoking the CLI."""