terramagic.py

import os
import shutil
from pathlib import Path
import click


MODULES_FOLDER = ["vpc", "cluster", "instance"]
MODULES_FILES = ["main.tf", "variables.tf", "outputs.tf"]
PROJECT_FILES = [
    "main.tf",
    "variables.tf",
    "outputs.tf",
    "backend.tf",
    "main.tfvars",
    ".gitignore",
]


def print_version(ctx, param, value):
    """
    Prints the version number and exits the program.

    Parameters:
        ctx (click.Context): The Click context object.
        param (click.Parameter): The Click parameter object.
        value (bool): The value of the parameter.

    Returns:
        None
    """
    if not value or ctx.resilient_parsing:
        return
    click.echo("Version: 0.3")
    ctx.exit()


def create_provider(base_path):
    """
    Creates a provider file in the specified base path.

    Args:
        base_path (Path): The base path where the provider file will be created.

    Returns:
        None
    """
    provider_path = base_path / "providers"
    provider_path.mkdir(exist_ok=True)
    provider_file = provider_path / "provider.tf"
    with provider_file.open("w", encoding="utf-8"):
        pass


def create_project_files(base_path):
    """
    Create project files in the specified base path.

    Args:
        base_path (str): The base path where the project files will be created.

    Returns:
        None
    """
    for file in PROJECT_FILES:
        file_path = base_path / file
        with file_path.open("w+", encoding="utf-8"):
            pass


def create_modules(base_path):
    """
    Create modules in the specified base path.

    Parameters:
        base_path (Path): The base path where the modules will be created.

    Returns:
        None
    """
    modules_path = base_path / "modules"
    modules_path.mkdir(exist_ok=True)

    for folder in MODULES_FOLDER:
        folder_path = modules_path / folder
        folder_path.mkdir(exist_ok=True)

        with folder_path:
            for file in MODULES_FILES:
                file_path = folder_path / file
                with file_path.open("w+", encoding="utf-8"):
                    pass


def create_env_folders(base_path):
    """
    Create environment folders.

    Args:
        base_path (path-like object): The base path where the environment folders will be created.

    Returns:
        None
    """
    envs_path = base_path / "environments"
    envs_path.mkdir(exist_ok=True)
    envs = ["production", "development", "staging"]
    for e in envs:
        env_path = envs_path / e
        env_path.mkdir(exist_ok=True)

        tfvars_file_path = env_path / f"{e}.tfvars"
        with tfvars_file_path.open("w", encoding="utf-8"):
            pass


@click.group()
@click.option(
    "--version",
    "-v",
    is_flag=True,
    callback=print_version,
    expose_value=False,
    is_eager=True,
    help="Show version",
)
def main():
    """ClI tool to create Terraform project"""


@main.command()
@click.option(
    "--name", "-n", help="Name of the project", prompt=True, confirmation_prompt=True
)
def create(name):
    """Create a new Terraform project with specified name and environment"""

    base_path = Path(name)
    try:
        base_path.mkdir(exist_ok=True)
        create_modules(base_path)
        create_env_folders(base_path)
        create_project_files(base_path)
        create_provider(base_path)
        click.secho(
            (f"Created project {name} successfully, You're ready move to ☁️ !!"),
            fg="blue",
        )
    except FileExistsError:
        click.secho(("The directory already exists!"), bg="red")
        os.chdir("..")


@main.command()
@click.option(
    "--name", "-n", help="Name of the project", prompt=True, confirmation_prompt=True
)
def remove(name):
    """Delete the project"""
    project_dir = Path(name)
    if project_dir.is_dir():
        click.confirm(
            click.style(f"Are you sure you want to delete {name}?", fg="cyan"),
            abort=True,
        )
        shutil.rmtree(project_dir)
        click.secho(f"Project {name} deleted successfully !!!", fg="blue")
    else:
        click.secho(f"Project {name} not found", fg="red")


if __name__ == "__main__":
    main()