dbt-platform-helper 12.5.0__tar.gz → 12.6.0__tar.gz

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dbt-platform-helper
-Version: 12.5.0
+Version: 12.6.0
 Summary: Set of tools to help transfer applications/services from GOV.UK PaaS to DBT PaaS augmenting AWS Copilot.
 License: MIT
 Author: Department for Business and Trade Platform Team
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: Jinja2 (==3.1.4)
+Requires-Dist: Jinja2 (==3.1.5)
 Requires-Dist: PyYAML (==6.0.1)
 Requires-Dist: aiohttp (>=3.8.4,<4.0.0)
 Requires-Dist: boto3 (>=1.28.24,<2.0.0)

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/COMMANDS.md

@@ -1,40 +1,40 @@
 # Commands Reference
 
 - [platform-helper](#platform-helper)
-- [platform-helper application](#platform-helper-application)
-- [platform-helper application container-stats](#platform-helper-application-container-stats)
-- [platform-helper application task-stats](#platform-helper-application-task-stats)
-- [platform-helper codebase](#platform-helper-codebase)
-- [platform-helper codebase prepare](#platform-helper-codebase-prepare)
-- [platform-helper codebase list](#platform-helper-codebase-list)
-- [platform-helper codebase build](#platform-helper-codebase-build)
-- [platform-helper codebase deploy](#platform-helper-codebase-deploy)
-- [platform-helper conduit](#platform-helper-conduit)
-- [platform-helper config](#platform-helper-config)
-- [platform-helper config validate](#platform-helper-config-validate)
-- [platform-helper config aws](#platform-helper-config-aws)
-- [platform-helper copilot](#platform-helper-copilot)
-- [platform-helper copilot make-addons](#platform-helper-copilot-make-addons)
-- [platform-helper environment](#platform-helper-environment)
-- [platform-helper environment offline](#platform-helper-environment-offline)
-- [platform-helper environment online](#platform-helper-environment-online)
-- [platform-helper environment generate](#platform-helper-environment-generate)
-- [platform-helper environment generate-terraform](#platform-helper-environment-generate-terraform)
-- [platform-helper generate](#platform-helper-generate)
-- [platform-helper pipeline](#platform-helper-pipeline)
-- [platform-helper pipeline generate](#platform-helper-pipeline-generate)
-- [platform-helper secrets](#platform-helper-secrets)
-- [platform-helper secrets copy](#platform-helper-secrets-copy)
-- [platform-helper secrets list](#platform-helper-secrets-list)
-- [platform-helper notify](#platform-helper-notify)
-- [platform-helper notify environment-progress](#platform-helper-notify-environment-progress)
-- [platform-helper notify add-comment](#platform-helper-notify-add-comment)
-- [platform-helper database](#platform-helper-database)
-- [platform-helper database dump](#platform-helper-database-dump)
-- [platform-helper database load](#platform-helper-database-load)
-- [platform-helper database copy](#platform-helper-database-copy)
-- [platform-helper version](#platform-helper-version)
-- [platform-helper version get-platform-helper-for-project](#platform-helper-version-get-platform-helper-for-project)
+    - [platform-helper application](#platform-helper-application)
+        - [platform-helper application container-stats](#platform-helper-application-container-stats)
+        - [platform-helper application task-stats](#platform-helper-application-task-stats)
+    - [platform-helper codebase](#platform-helper-codebase)
+        - [platform-helper codebase prepare](#platform-helper-codebase-prepare)
+        - [platform-helper codebase list](#platform-helper-codebase-list)
+        - [platform-helper codebase build](#platform-helper-codebase-build)
+        - [platform-helper codebase deploy](#platform-helper-codebase-deploy)
+    - [platform-helper conduit](#platform-helper-conduit)
+    - [platform-helper config](#platform-helper-config)
+        - [platform-helper config validate](#platform-helper-config-validate)
+        - [platform-helper config aws](#platform-helper-config-aws)
+    - [platform-helper copilot](#platform-helper-copilot)
+        - [platform-helper copilot make-addons](#platform-helper-copilot-make-addons)
+    - [platform-helper environment](#platform-helper-environment)
+        - [platform-helper environment offline](#platform-helper-environment-offline)
+        - [platform-helper environment online](#platform-helper-environment-online)
+        - [platform-helper environment generate](#platform-helper-environment-generate)
+        - [platform-helper environment generate-terraform](#platform-helper-environment-generate-terraform)
+    - [platform-helper generate](#platform-helper-generate)
+    - [platform-helper pipeline](#platform-helper-pipeline)
+        - [platform-helper pipeline generate](#platform-helper-pipeline-generate)
+    - [platform-helper secrets](#platform-helper-secrets)
+        - [platform-helper secrets copy](#platform-helper-secrets-copy)
+        - [platform-helper secrets list](#platform-helper-secrets-list)
+    - [platform-helper notify](#platform-helper-notify)
+        - [platform-helper notify environment-progress](#platform-helper-notify-environment-progress)
+        - [platform-helper notify add-comment](#platform-helper-notify-add-comment)
+    - [platform-helper database](#platform-helper-database)
+        - [platform-helper database dump](#platform-helper-database-dump)
+        - [platform-helper database load](#platform-helper-database-load)
+        - [platform-helper database copy](#platform-helper-database-copy)
+    - [platform-helper version](#platform-helper-version)
+        - [platform-helper version get-platform-helper-for-project](#platform-helper-version-get-platform-helper-for-project)
 
 # platform-helper
 
@@ -457,19 +457,20 @@ platform-helper environment online --app <application> --env <environment>
 
 [↩ Parent](#platform-helper-environment)
 
+    Gathers various IDs and ARNs from AWS and generates the AWS Copilot
+    environment manifest at copilot/environments/<environment>/manifest.yml.
+
 ## Usage
 
 ```
-platform-helper environment generate --name <name> [--vpc-name <vpc_name>] 
+platform-helper environment generate --name <name> 
 ```
 
 ## Options
 
-- `--vpc-name <text>`
-
 - `--name
 -n <text>`
-
+  - The name of the environment to generate a copilot manifest for.
 - `--help <boolean>` _Defaults to False._
   - Show this message and exit.
 

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/commands/codebase.py

@@ -2,6 +2,7 @@ import click
 
 from dbt_platform_helper.domain.codebase import Codebase
 from dbt_platform_helper.platform_exception import PlatformException
+from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 from dbt_platform_helper.utils.versioning import (
     check_platform_helper_version_needs_update,
@@ -20,8 +21,7 @@ def prepare():
     try:
         Codebase().prepare()
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @codebase.command()
@@ -37,8 +37,7 @@ def list(app, with_images):
     try:
         Codebase().list(app, with_images)
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @codebase.command()
@@ -54,8 +53,7 @@ def build(app, codebase, commit):
     try:
         Codebase().build(app, codebase, commit)
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @codebase.command()
@@ -71,5 +69,4 @@ def deploy(app, env, codebase, commit):
     try:
         Codebase().deploy(app, env, codebase, commit)
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/commands/conduit.py

@@ -4,6 +4,7 @@ from dbt_platform_helper.domain.conduit import Conduit
 from dbt_platform_helper.platform_exception import PlatformException
 from dbt_platform_helper.providers.cloudformation import CloudFormation
 from dbt_platform_helper.providers.ecs import ECS
+from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.providers.secrets import Secrets
 from dbt_platform_helper.utils.application import load_application
 from dbt_platform_helper.utils.click import ClickDocOptCommand
@@ -54,5 +55,4 @@ def conduit(addon_name: str, app: str, env: str, access: str):
             env, addon_name, access
         )
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/commands/config.py

@@ -98,7 +98,7 @@ def deployment():
 
     recommendations = {}
 
-    ConfigProvider.config_file_check()
+    ConfigProvider().config_file_check()
 
     for template_file in addons_templates:
         generated_with_version = maybe

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/commands/copilot.py

@@ -13,12 +13,12 @@ from dbt_platform_helper.constants import PLATFORM_CONFIG_FILE
 from dbt_platform_helper.domain.config_validator import ConfigValidator
 from dbt_platform_helper.domain.copilot_environment import CopilotTemplating
 from dbt_platform_helper.providers.config import ConfigProvider
+from dbt_platform_helper.providers.files import FileProvider
 from dbt_platform_helper.utils.application import get_application_name
 from dbt_platform_helper.utils.application import load_application
 from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 from dbt_platform_helper.utils.files import generate_override_files
-from dbt_platform_helper.utils.files import mkfile
 from dbt_platform_helper.utils.messages import abort_with_error
 from dbt_platform_helper.utils.template import ADDON_TEMPLATE_MAP
 from dbt_platform_helper.utils.template import camel_case
@@ -379,7 +379,9 @@ def _generate_service_addons(
 
             (output_dir / service_path).mkdir(parents=True, exist_ok=True)
             click.echo(
-                mkfile(output_dir, service_path / f"{addon_name}.yml", contents, overwrite=True)
+                FileProvider.mkfile(
+                    output_dir, service_path / f"{addon_name}.yml", contents, overwrite=True
+                )
             )
 
 

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/commands/environment.py

@@ -1,14 +1,17 @@
 import click
-from schema import SchemaError
 
 from dbt_platform_helper.constants import DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION
-from dbt_platform_helper.constants import PLATFORM_CONFIG_FILE
 from dbt_platform_helper.domain.config_validator import ConfigValidator
 from dbt_platform_helper.domain.copilot_environment import CopilotEnvironment
-from dbt_platform_helper.domain.maintenance_page import MaintenancePageProvider
+from dbt_platform_helper.domain.maintenance_page import MaintenancePage
 from dbt_platform_helper.domain.terraform_environment import TerraformEnvironment
 from dbt_platform_helper.platform_exception import PlatformException
+from dbt_platform_helper.providers.cloudformation import CloudFormation
 from dbt_platform_helper.providers.config import ConfigProvider
+from dbt_platform_helper.providers.io import ClickIOProvider
+from dbt_platform_helper.providers.vpc import VpcProvider
+from dbt_platform_helper.utils.application import load_application
+from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 from dbt_platform_helper.utils.versioning import (
     check_platform_helper_version_needs_update,
@@ -36,11 +39,12 @@ def environment():
 @click.option("--vpc", type=str)
 def offline(app, env, svc, template, vpc):
     """Take load-balanced web services offline with a maintenance page."""
+
     try:
-        MaintenancePageProvider().activate(app, env, svc, template, vpc)
+        application = load_application(app)
+        MaintenancePage(application).activate(env, svc, template, vpc)
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @environment.command()
@@ -48,29 +52,37 @@ def offline(app, env, svc, template, vpc):
 @click.option("--env", type=str, required=True)
 def online(app, env):
     """Remove a maintenance page from an environment."""
+
     try:
-        MaintenancePageProvider().deactivate(app, env)
+        application = load_application(app)
+        MaintenancePage(application).deactivate(env)
     except PlatformException as err:
-        click.secho(str(err), fg="red")
-        raise click.Abort
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @environment.command()
-@click.option("--vpc-name", hidden=True)
-@click.option("--name", "-n", required=True)
-def generate(name, vpc_name):
-    if vpc_name:
-        click.secho(
-            f"This option is deprecated. Please add the VPC name for your envs to {PLATFORM_CONFIG_FILE}",
-            fg="red",
-        )
-        raise click.Abort
+@click.option(
+    "--name",
+    "-n",
+    required=True,
+    help="The name of the environment to generate a copilot manifest for.",
+)
+def generate(name):
+    """Gathers various IDs and ARNs from AWS and generates the AWS Copilot
+    environment manifest at copilot/environments/<environment>/manifest.yml."""
+
+    click_io = ClickIOProvider()
     try:
+        session = get_aws_session_or_abort()
         config_provider = ConfigProvider(ConfigValidator())
-        CopilotEnvironment(config_provider).generate(name)
-    except SchemaError as ex:
-        click.secho(f"Invalid `{PLATFORM_CONFIG_FILE}` file: {str(ex)}", fg="red")
-        raise click.Abort
+        vpc_provider = VpcProvider(session)
+        cloudformation_provider = CloudFormation(session.client("cloudformation"))
+
+        CopilotEnvironment(
+            config_provider, vpc_provider, cloudformation_provider, session
+        ).generate(name)
+    except PlatformException as err:
+        click_io.abort_with_error(str(err))
 
 
 @environment.command(help="Generate terraform manifest for the specified environment.")
@@ -82,5 +94,9 @@ def generate(name, vpc_name):
     help=f"Override the default version of terraform-platform-modules. (Default version is '{DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION}').",
 )
 def generate_terraform(name, terraform_platform_modules_version):
-    config_provider = ConfigProvider(ConfigValidator())
-    TerraformEnvironment(config_provider).generate(name, terraform_platform_modules_version)
+
+    try:
+        config_provider = ConfigProvider(ConfigValidator())
+        TerraformEnvironment(config_provider).generate(name, terraform_platform_modules_version)
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))

dbt_platform_helper-12.6.0/dbt_platform_helper/commands/pipeline.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+import click
+
+from dbt_platform_helper.constants import DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION
+from dbt_platform_helper.domain.config_validator import ConfigValidator
+from dbt_platform_helper.domain.pipelines import Pipelines
+from dbt_platform_helper.providers.config import ConfigProvider
+from dbt_platform_helper.utils.aws import get_codestar_connection_arn
+from dbt_platform_helper.utils.click import ClickDocOptGroup
+from dbt_platform_helper.utils.git import git_remote
+from dbt_platform_helper.utils.versioning import (
+    check_platform_helper_version_needs_update,
+)
+
+
+@click.group(chain=True, cls=ClickDocOptGroup)
+def pipeline():
+    """Pipeline commands."""
+    check_platform_helper_version_needs_update()
+
+
+@pipeline.command()
+@click.option(
+    "--terraform-platform-modules-version",
+    help=f"""Override the default version of terraform-platform-modules with a specific version or branch. 
+    Precedence of version used is version supplied via CLI, then the version found in 
+    platform-config.yml/default_versions/terraform-platform-modules. 
+    In absence of these inputs, defaults to version '{DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION}'.""",
+)
+@click.option(
+    "--deploy-branch",
+    help="""Specify the branch of <application>-deploy used to configure the source stage in the environment-pipeline resource. 
+    This is generated from the terraform/environments-pipeline/<aws_account>/main.tf file. 
+    (Default <application>-deploy branch is specified in 
+    <application>-deploy/platform-config.yml/environment_pipelines/<environment-pipeline>/branch).""",
+    default=None,
+)
+def generate(terraform_platform_modules_version, deploy_branch):
+    """
+    Given a platform-config.yml file, generate environment and service
+    deployment pipelines.
+
+    This command does the following in relation to the environment pipelines:
+    - Reads contents of `platform-config.yml/environment-pipelines` configuration.
+      The `terraform/environment-pipelines/<aws_account>/main.tf` file is generated using this configuration.
+      The `main.tf` file is then used to generate Terraform for creating an environment pipeline resource.
+
+    This command does the following in relation to the codebase pipelines:
+    - Generates the copilot pipeline manifest.yml for copilot/pipelines/<codebase_pipeline_name>
+    """
+    pipelines = Pipelines(
+        ConfigProvider(ConfigValidator()),
+        git_remote,
+        get_codestar_connection_arn,
+    )
+    pipelines.generate(terraform_platform_modules_version, deploy_branch)

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/constants.py

@@ -8,6 +8,7 @@ DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION = "5"
 # Keys
 CODEBASE_PIPELINES_KEY = "codebase_pipelines"
 ENVIRONMENTS_KEY = "environments"
+ENVIRONMENT_PIPELINES_KEY = "environment_pipelines"
 
 # Conduit
 CONDUIT_ADDON_TYPES = [

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/domain/codebase.py

@@ -4,12 +4,13 @@ import subprocess
 from collections.abc import Callable
 from pathlib import Path
 
-import click
 import requests
 import yaml
 from boto3 import Session
 
 from dbt_platform_helper.platform_exception import PlatformException
+from dbt_platform_helper.providers.files import FileProvider
+from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.utils.application import Application
 from dbt_platform_helper.utils.application import ApplicationException
 from dbt_platform_helper.utils.application import load_application
@@ -19,7 +20,6 @@ from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.aws import get_build_url_from_arn
 from dbt_platform_helper.utils.aws import list_latest_images
 from dbt_platform_helper.utils.aws import start_build_extraction
-from dbt_platform_helper.utils.files import mkfile
 from dbt_platform_helper.utils.git import check_if_commit_exists
 from dbt_platform_helper.utils.template import setup_templates
 
@@ -27,9 +27,7 @@ from dbt_platform_helper.utils.template import setup_templates
 class Codebase:
     def __init__(
         self,
-        input: Callable[[str], str] = click.prompt,
-        echo: Callable[[str], str] = click.secho,
-        confirm: Callable[[str], bool] = click.confirm,
+        io: ClickIOProvider = ClickIOProvider(),
         load_application: Callable[[str], Application] = load_application,
         get_aws_session_or_abort: Callable[[str], Session] = get_aws_session_or_abort,
         check_codebase_exists: Callable[[str], str] = check_codebase_exists,
@@ -40,9 +38,7 @@ class Codebase:
         check_if_commit_exists: Callable[[str], str] = check_if_commit_exists,
         run_subprocess: Callable[[str], str] = subprocess.run,
     ):
-        self.input = input
-        self.echo = echo
-        self.confirm = confirm
+        self.io = io
         self.load_application = load_application
         self.get_aws_session_or_abort = get_aws_session_or_abort
         self.check_codebase_exists = check_codebase_exists
@@ -91,8 +87,8 @@ class Codebase:
         config_contents = templates.get_template(f".copilot/config.yml").render(
             repository=repository, builder_version=builder_version
         )
-        self.echo(
-            mkfile(
+        self.io.info(
+            FileProvider.mkfile(
                 Path("."), ".copilot/image_build_run.sh", image_build_run_contents, overwrite=True
             )
         )
@@ -100,13 +96,17 @@ class Codebase:
         image_build_run_file = Path(".copilot/image_build_run.sh")
         image_build_run_file.chmod(image_build_run_file.stat().st_mode | stat.S_IEXEC)
 
-        self.echo(mkfile(Path("."), ".copilot/config.yml", config_contents, overwrite=True))
+        self.io.info(
+            FileProvider.mkfile(Path("."), ".copilot/config.yml", config_contents, overwrite=True)
+        )
 
         for phase in ["build", "install", "post_build", "pre_build"]:
             phase_contents = templates.get_template(f".copilot/phases/{phase}.sh").render()
 
-            self.echo(
-                mkfile(Path("./.copilot"), f"phases/{phase}.sh", phase_contents, overwrite=True)
+            self.io.info(
+                FileProvider.mkfile(
+                    Path("./.copilot"), f"phases/{phase}.sh", phase_contents, overwrite=True
+                )
             )
 
     def build(self, app: str, codebase: str, commit: str):
@@ -118,7 +118,6 @@ class Codebase:
 
         codebuild_client = session.client("codebuild")
         build_url = self.__start_build_with_confirmation(
-            self.confirm,
             codebuild_client,
             self.get_build_url_from_arn,
             f'You are about to build "{app}" for "{codebase}" with commit "{commit}". Do you want to continue?',
@@ -130,7 +129,7 @@ class Codebase:
         )
 
         if build_url:
-            return self.echo(
+            return self.io.info(
                 f"Your build has been triggered. Check your build progress in the AWS Console: {build_url}"
             )
 
@@ -150,7 +149,6 @@ class Codebase:
 
         codebuild_client = session.client("codebuild")
         build_url = self.__start_build_with_confirmation(
-            self.confirm,
             codebuild_client,
             self.get_build_url_from_arn,
             f'You are about to deploy "{app}" for "{codebase}" with commit "{commit}" to the "{env}" environment. Do you want to continue?',
@@ -166,7 +164,7 @@ class Codebase:
         )
 
         if build_url:
-            return self.echo(
+            return self.io.info(
                 "Your deployment has been triggered. Check your build progress in the AWS Console: "
                 f"{build_url}",
             )
@@ -181,19 +179,19 @@ class Codebase:
         ecr_client = session.client("ecr")
         codebases = self.__get_codebases(application, ssm_client)
 
-        self.echo("The following codebases are available:")
+        self.io.info("The following codebases are available:")
 
         for codebase in codebases:
-            self.echo(f"- {codebase['name']} (https://github.com/{codebase['repository']})")
+            self.io.info(f"- {codebase['name']} (https://github.com/{codebase['repository']})")
             if with_images:
                 self.list_latest_images(
                     ecr_client,
                     f"{application.name}/{codebase['name']}",
                     codebase["repository"],
-                    self.echo,
+                    self.io.info,
                 )
 
-        self.echo("")
+        self.io.info("")
 
     def __get_codebases(self, application, ssm_client):
         parameters = ssm_client.get_parameters_by_path(
@@ -209,13 +207,12 @@ class Codebase:
 
     def __start_build_with_confirmation(
         self,
-        confirm,
         codebuild_client,
         get_build_url_from_arn,
         confirmation_message,
         build_options,
     ):
-        if confirm(confirmation_message):
+        if self.io.confirm(confirmation_message):
             build_arn = self.start_build_extraction(codebuild_client, build_options)
             return get_build_url_from_arn(build_arn)
         return None

{dbt_platform_helper-12.5.0 → dbt_platform_helper-12.6.0}/dbt_platform_helper/domain/conduit.py

@@ -1,12 +1,10 @@
 import subprocess
-from collections.abc import Callable
-
-import click
 
 from dbt_platform_helper.providers.cloudformation import CloudFormation
 from dbt_platform_helper.providers.copilot import connect_to_addon_client_task
 from dbt_platform_helper.providers.copilot import create_addon_client_task
 from dbt_platform_helper.providers.ecs import ECS
+from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.providers.secrets import Secrets
 from dbt_platform_helper.utils.application import Application
 
@@ -18,7 +16,7 @@ class Conduit:
         secrets_provider: Secrets,
         cloudformation_provider: CloudFormation,
         ecs_provider: ECS,
-        echo: Callable[[str], str] = click.secho,
+        io: ClickIOProvider = ClickIOProvider(),
         subprocess: subprocess = subprocess,
         connect_to_addon_client_task=connect_to_addon_client_task,
         create_addon_client_task=create_addon_client_task,
@@ -29,7 +27,7 @@ class Conduit:
         self.cloudformation_provider = cloudformation_provider
         self.ecs_provider = ecs_provider
         self.subprocess = subprocess
-        self.echo = echo
+        self.io = io
         self.connect_to_addon_client_task = connect_to_addon_client_task
         self.create_addon_client_task = create_addon_client_task
 
@@ -39,10 +37,10 @@ class Conduit:
             addon_name, access
         )
 
-        self.echo(f"Checking if a conduit task is already running for {addon_type}")
+        self.io.info(f"Checking if a conduit task is already running for {addon_type}")
         task_arns = self.ecs_provider.get_ecs_task_arns(cluster_arn, task_name)
         if not task_arns:
-            self.echo("Creating conduit task")
+            self.io.info("Creating conduit task")
             self.create_addon_client_task(
                 clients["iam"],
                 clients["ssm"],
@@ -55,7 +53,7 @@ class Conduit:
                 access,
             )
 
-            self.echo("Updating conduit task")
+            self.io.info("Updating conduit task")
             self._update_stack_resources(
                 self.application.name,
                 env,
@@ -69,13 +67,13 @@ class Conduit:
             task_arns = self.ecs_provider.get_ecs_task_arns(cluster_arn, task_name)
 
         else:
-            self.echo("Conduit task already running")
+            self.io.info("Conduit task already running")
 
-        self.echo(f"Checking if exec is available for conduit task...")
+        self.io.info(f"Checking if exec is available for conduit task...")
 
         self.ecs_provider.ecs_exec_is_available(cluster_arn, task_arns)
 
-        self.echo("Connecting to conduit task")
+        self.io.info("Connecting to conduit task")
         self.connect_to_addon_client_task(
             clients["ecs"], self.subprocess, self.application.name, env, cluster_arn, task_name
         )
@@ -115,7 +113,7 @@ class Conduit:
             parameter_name,
             access,
         )
-        self.echo("Waiting for conduit task update to complete...")
+        self.io.info("Waiting for conduit task update to complete...")
         self.cloudformation_provider.wait_for_cloudformation_to_reach_status(
             "stack_update_complete", stack_name
         )