allow sourcing docker images from custom registry (#129)

* update to allow sourcing from internal registry, if one is configured, by adding prefix to image name - leverages existing DOCKER_GLOBAL_IMAGE_PREFIX config

* adding ability to specify image name, including tag, to allow usage when latest tag is blocked

Author: plafond

utils/localstack_extensions/utils/docker.py

@@ -103,7 +103,12 @@ def __init__(
         http2_ports: list[int] | None = None,
         tcp_ports: list[int] | None = None,
     ):
-        self.image_name = image_name
+        try:
+            from localstack.pro.core.utils.container.registry_strategies import CustomizableRegistryStrategy
+            self.image_name = CustomizableRegistryStrategy().resolve(image_name)
+        except ImportError:
+            self.image_name = image_name
+
         if not container_ports:
             raise ValueError("container_ports is required")
         self.container_ports = container_ports

utils/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "localstack-extensions-utils"
-version = "0.1.0"
+version = "0.1.1"
 description = "Utility library for LocalStack Extensions"
 readme = {file = "README.md", content-type = "text/markdown; charset=UTF-8"}
 requires-python = ">=3.10"

wiremock/localstack_wiremock/extension.py

@@ -14,6 +14,10 @@
 ENV_WIREMOCK_API_TOKEN = "WIREMOCK_API_TOKEN"
 # Host path to directory containing .wiremock/ (required for runner mode)
 ENV_WIREMOCK_CONFIG_DIR = "WIREMOCK_CONFIG_DIR"
+# Override the OSS image (default: wiremock/wiremock); accepts full ref with optional tag
+ENV_WIREMOCK_IMAGE = "WIREMOCK_IMAGE"
+# Override the runner image (default: wiremock/wiremock-runner); accepts full ref with optional tag
+ENV_WIREMOCK_IMAGE_RUNNER = "WIREMOCK_IMAGE_RUNNER"
 
 SERVICE_PORT = 8080  # Mock API port
 ADMIN_PORT = 9999  # Admin interface port (runner mode)
@@ -29,7 +33,7 @@ class WireMockExtension(ProxiedDockerContainerExtension):
 
     def __init__(self):
         env_vars = {}
-        image_name = self.DOCKER_IMAGE
+        image_name = os.getenv(ENV_WIREMOCK_IMAGE) or self.DOCKER_IMAGE
         volumes = None
         container_ports = [SERVICE_PORT]
         health_check_path = "/__admin/health"
@@ -41,7 +45,7 @@ def __init__(self):
             env_vars["WMC_ADMIN_PORT"] = str(ADMIN_PORT)
             env_vars["WMC_API_TOKEN"] = api_token
             env_vars["WMC_RUNNER_ENABLED"] = "true"
-            image_name = self.DOCKER_IMAGE_RUNNER
+            image_name = os.getenv(ENV_WIREMOCK_IMAGE_RUNNER) or self.DOCKER_IMAGE_RUNNER
             container_ports = [SERVICE_PORT, ADMIN_PORT]
             health_check_path = "/__/health"
             health_check_retries = 90