atlas-init 0.7.0__py3-none-any.whl → 0.8.1__py3-none-any.whl

atlas_init/tf_ext/tf_dep.py

@@ -7,8 +7,8 @@ from typing import Iterable, NamedTuple
 
 import pydot
 from ask_shell import ShellError, new_task, run_and_wait
-from ask_shell.run_pool import run_pool
 from ask_shell._run import stop_runs_and_pool
+from ask_shell.run_pool import run_pool
 from model_lib import Entity, dump
 from pydantic import BaseModel, Field
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed
@@ -17,10 +17,10 @@ from zero_3rdparty.file_utils import ensure_parents_write_text
 from zero_3rdparty.iter_utils import flat_map
 
 from atlas_init.settings.rich_utils import configure_logging
-from atlas_init.tf_ext.args import REPO_PATH_ARG, SKIP_EXAMPLES_DIRS_OPTION
+from atlas_init.tf_ext.args import REPO_PATH_ATLAS_ARG, SKIP_EXAMPLES_DIRS_OPTION
 from atlas_init.tf_ext.constants import ATLAS_PROVIDER_NAME
 from atlas_init.tf_ext.paths import find_variable_resource_type_usages, find_variables, get_example_directories
-from atlas_init.tf_ext.settings import TfDepSettings
+from atlas_init.tf_ext.settings import TfExtSettings
 
 logger = logging.getLogger(__name__)
 v2_grand_parent_dirs = {
@@ -48,10 +48,10 @@ def is_v2_example_dir(example_dir: Path) -> bool:
 
 
 def tf_dep_graph(
-    repo_path: Path = REPO_PATH_ARG,
+    repo_path: Path = REPO_PATH_ATLAS_ARG,
     skip_names: list[str] = SKIP_EXAMPLES_DIRS_OPTION,
 ):
-    settings = TfDepSettings.from_env()
+    settings = TfExtSettings.from_env()
     output_dir = settings.static_root
     logger.info(f"Using output directory: {output_dir}")
     example_dirs = get_example_directories(repo_path, skip_names)

atlas_init/tf_ext/tf_desc_gen.py

@@ -0,0 +1,53 @@
+import logging
+from collections import defaultdict
+from atlas_init.tf_ext.args import TF_CLI_CONFIG_FILE_ARG
+from atlas_init.tf_ext.settings import TfExtSettings
+from atlas_init.tf_ext.provider_schema import ResourceSchema, parse_atlas_schema
+from model_lib import dump, parse_model
+from zero_3rdparty.file_utils import ensure_parents_write_text
+
+logger = logging.getLogger(__name__)
+
+
+def tf_desc_gen(
+    tf_cli_config_file: str = TF_CLI_CONFIG_FILE_ARG,
+):
+    settings = TfExtSettings.from_env()
+    out_path = settings.attribute_description_file_path
+    resource_out_path = settings.attribute_resource_descriptions_file_path
+    assert tf_cli_config_file
+    schema = parse_atlas_schema()
+    descriptions = {}
+    descriptions_by_resource: dict[str, dict[str, str]] = defaultdict(dict)
+    attr_desc_resource = {}
+
+    def add_description(resource_type: str, attr_name: str, description: str | None) -> None:
+        if not description:
+            return
+        descriptions_by_resource[resource_type][attr_name] = description
+        if existing := descriptions.get(attr_name):
+            if existing != description:
+                old_resource_type = attr_desc_resource[attr_name]
+                logger.info(
+                    f"Descriptions differs between '{old_resource_type}' and '{resource_type}' for attribute '{attr_name}':\n{existing}\n{description}"
+                )
+                if len(existing) > len(description):
+                    return
+        descriptions[attr_name] = description
+        attr_desc_resource[attr_name] = resource_type
+
+    for resource_type, resource_schema in schema.raw_resource_schema.items():
+        parsed_schema = parse_model(resource_schema, t=ResourceSchema)
+        schema_block = parsed_schema.block
+        for name, attribute in (schema_block.attributes or {}).items():
+            add_description(resource_type, name, attribute.description)
+        for name, block_type in (schema_block.block_types or {}).items():
+            add_description(resource_type, name, block_type.description)
+    descriptions_yaml = dump(dict(sorted(descriptions.items())), format="yaml")
+    ensure_parents_write_text(out_path, descriptions_yaml)
+    logger.info(f"Generated attribute descriptions to {out_path}")
+    resource_descriptions_yaml = dump(
+        {k: dict(sorted(v.items())) for k, v in sorted(descriptions_by_resource.items())}, format="yaml"
+    )
+    ensure_parents_write_text(resource_out_path, resource_descriptions_yaml)
+    logger.info(f"Generated attribute resource descriptions to {resource_out_path}")

atlas_init/tf_ext/tf_mod_gen.py

@@ -0,0 +1,263 @@
+import difflib
+import logging
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+import typer
+from ask_shell import new_task, run_and_wait, run_pool, text
+from model_lib import parse_model, parse_payload
+from pydantic import DirectoryPath, TypeAdapter
+from zero_3rdparty.file_utils import clean_dir, copy, ensure_parents_write_text
+
+from atlas_init.cli_tf.example_update import UpdateExamples, update_examples
+from atlas_init.tf_ext.args import TF_CLI_CONFIG_FILE_ARG
+from atlas_init.tf_ext.gen_examples import generate_module_examples, read_example_dirs
+from atlas_init.tf_ext.gen_module_readme import generate_readme
+from atlas_init.tf_ext.gen_resource_main import generate_resource_main
+from atlas_init.tf_ext.gen_resource_output import generate_resource_output
+from atlas_init.tf_ext.gen_resource_variables import generate_module_variables
+from atlas_init.tf_ext.gen_versions import dump_versions_tf
+from atlas_init.tf_ext.models_module import (
+    MissingDescriptionError,
+    ModuleGenConfig,
+    import_resource_type_python_module,
+    parse_attribute_descriptions,
+    store_updated_attribute_description,
+)
+from atlas_init.tf_ext.newres import prepare_newres
+from atlas_init.tf_ext.plan_diffs import (
+    ExamplePlanCheck,
+    generate_expected_actual,
+    parse_plan_output,
+    read_variables_path,
+)
+from atlas_init.tf_ext.provider_schema import AtlasSchemaInfo, ResourceSchema, parse_atlas_schema
+from atlas_init.tf_ext.schema_to_dataclass import convert_and_format
+from atlas_init.tf_ext.settings import TfExtSettings
+
+logger = logging.getLogger(__name__)
+
+
+def tf_mod_gen(
+    tf_cli_config_file: str = TF_CLI_CONFIG_FILE_ARG,
+    use_newres: bool = typer.Option(False, "--use-newres", help="Use newres to generate modules"),
+    resource_type: list[str] = typer.Option(
+        ..., "-r", "--resource-type", help="Resource types to generate modules for", default_factory=list
+    ),
+    name: str = typer.Option("", "-n", "--name", help="Name of the module"),
+    in_dir: DirectoryPath = typer.Option(
+        ..., "-i", "--in-dir", help="Parent directory where the module generation files are stored"
+    ),
+    out_dir: DirectoryPath = typer.Option(
+        ...,
+        "-o",
+        "--out-dir",
+        help="Output directory for generated modules, the module will end up in {output_dir}/{name}",
+    ),
+    example_var_file: Path = typer.Option(
+        ..., "-e", "--example-var-file", help="Path to example variable file", envvar="TF_EXT_EXAMPLE_VAR_FILE"
+    ),
+):
+    settings = TfExtSettings.from_env()
+    assert tf_cli_config_file, "tf_cli_config_file is required"
+    if use_newres:
+        prepare_newres(settings.new_res_path)
+    else:
+        settings = TfExtSettings.from_env()
+        logger.info("will use Python generation")
+        config = ModuleGenConfig.from_paths(name, in_dir, out_dir, settings)
+        prepare_out_dir(config)
+        generate_module(config)
+        module_examples_and_readme(config, example_var_file=example_var_file)
+
+
+def prepare_out_dir(config: ModuleGenConfig, *, skip_clean_dir: bool = False):
+    if not skip_clean_dir:
+        clean_dir(config.module_out_path)
+    in_dir = config.in_dir
+    assert in_dir, "in_dir is required"
+    for src_file in in_dir.glob("*"):
+        if config.skip_copy(src_file):
+            continue
+        copy(src_file, config.module_out_path / src_file.name, clean_dest=True)  # also copies directories
+    example_checks = config.example_plan_checks_path
+    if example_checks.exists():
+        example_plan_checks_raw = parse_payload(example_checks)
+        config.example_plan_checks = TypeAdapter(list[ExamplePlanCheck]).validate_python(example_plan_checks_raw)
+
+
+def generate_module(config: ModuleGenConfig) -> Path:
+    with new_task("Reading Atlas Schema"):
+        schema = parse_atlas_schema()
+        assert schema
+    resource_types = config.resource_types
+    with new_task("Generating module files for resource types", total=len(resource_types)) as task:
+        for resource_type in resource_types:
+            generate_resource_module(config, resource_type, schema)
+            task.update(advance=1)
+
+    return finalize_and_validate_module(config)
+
+
+def generate_resource_module(config: ModuleGenConfig, resource_type: str, atlas_schema: AtlasSchemaInfo) -> None:
+    resource_type_schema = atlas_schema.raw_resource_schema.get(resource_type)
+    assert resource_type_schema, f"resource type {resource_type} not found in schema"
+    schema_parsed = parse_model(resource_type_schema, t=ResourceSchema)
+    dataclass_path = config.dataclass_path(resource_type)
+    dataclass_code = convert_and_format(resource_type, schema_parsed, config, existing_path=dataclass_path)
+    logger.info(f"Generated dataclass for {resource_type} to {dataclass_path}")
+    ensure_parents_write_text(dataclass_path, dataclass_code)
+
+    python_module = import_resource_type_python_module(resource_type, dataclass_path)
+    main_tf = generate_resource_main(python_module, config)
+    main_path = config.main_tf_path(resource_type)
+    ensure_parents_write_text(main_path, main_tf)
+
+    variablesx_tf, variables_tf = generate_module_variables(python_module, config.resource_config(resource_type))
+    variables_path = config.variables_path(resource_type)
+    if variablesx_tf and variables_tf:
+        variablesx_path = config.variablesx_path(resource_type)
+        ensure_parents_write_text(variablesx_path, variablesx_tf)
+        ensure_parents_write_text(variables_path, variables_tf)
+    else:
+        ensure_parents_write_text(variables_path, variablesx_tf)
+    if output_tf := generate_resource_output(python_module, config):
+        output_path = config.output_path(resource_type)
+        ensure_parents_write_text(output_path, output_tf)
+    if config.skip_python and dataclass_path.is_relative_to(config.module_out_path):
+        dataclass_path.unlink(missing_ok=True)
+
+
+def finalize_and_validate_module(config: ModuleGenConfig) -> Path:
+    dump_versions_tf(config.module_out_path, skip_python=config.skip_python)
+    logger.info(f"Module dumped to {config.module_out_path}, running checks")
+    validate_module(config.module_out_path, tf_cli_config_file=config.settings.tf_cli_config_file)
+    return config.module_out_path
+
+
+OUT_BINARY_PATH = "tfplan.binary"
+
+
+def validate_module(tf_workdir: Path, *, tf_cli_config_file: Path | None = None):
+    terraform_commands = [
+        "terraform init",
+        "terraform fmt .",
+        "terraform validate .",
+    ]
+    env_extra = {}
+    if tf_cli_config_file:
+        env_extra["TF_CLI_CONFIG_FILE"] = str(tf_cli_config_file)
+    with new_task("Terraform Module Validate Checks", total=len(terraform_commands)) as task:
+        for command in terraform_commands:
+            attempts = 3 if command == "terraform init" else 1  # terraform init can fail due to network issues
+            run_and_wait(command, cwd=tf_workdir, env=env_extra, attempts=attempts)
+            task.update(advance=1)
+
+
+def module_examples_and_readme(config: ModuleGenConfig, *, example_var_file: Path | None = None) -> Path:
+    path = config.module_out_path
+    if (examples_test := config.examples_test_path) and examples_test.exists():
+        with new_task(f"Generating examples from {config.FILENAME_EXAMPLES_TEST}"):
+            assert len(config.resource_types) == 1
+            resource_type = config.resource_types[0]
+            py_module = import_resource_type_python_module(resource_type, config.dataclass_path(resource_type))
+            examples_generated = generate_module_examples(config, py_module, resource_type=resource_type)
+        if examples_generated:
+            with run_pool("Validating examples", total=len(examples_generated), exit_wait_timeout=60) as pool:
+                for example_path in examples_generated:
+                    pool.submit(validate_module, example_path)
+
+    attribute_descriptions = parse_attribute_descriptions(config.settings)
+    settings = config.settings
+
+    def new_description(name: str, old_description: str, path: Path) -> str:
+        resource_type = config.resolve_resource_type(path)
+        try:
+            return attribute_descriptions.resolve_description(name, resource_type)
+        except MissingDescriptionError:
+            if new_text := text(
+                f"Enter description for variable/output {name} in {resource_type} for {path} (empty to skip)",
+                default="",
+            ):
+                store_updated_attribute_description(attribute_descriptions, settings, name, new_text, resource_type)
+            return new_text
+
+    out_event = update_examples(
+        UpdateExamples(
+            examples_base_dir=path,
+            skip_tf_fmt=True,
+            new_description_call=new_description,
+        )
+    )
+    if out_event.changes:
+        logger.info(f"Updated attribute descriptions: {len(out_event.changes)}")
+        run_and_wait("terraform fmt -recursive .", cwd=path, ansi_content=False, allow_non_zero_exit=True)
+    if readme_path := config.readme_path():
+        with new_task("Generating README.md"):
+            readme_content = generate_readme(config)
+            ensure_parents_write_text(readme_path, readme_content)
+    if example_var_file:
+        examples = read_example_dirs(config.module_out_path)
+        if examples:
+            failed_examples: list[Path] = []
+            with run_pool("Running terraform plan on examples", total=len(examples), exit_wait_timeout=60) as pool:
+
+                def run_example(example: Path):
+                    try:
+                        run_and_wait(f"terraform plan -var-file={example_var_file}", cwd=example)
+                    except Exception as e:
+                        logger.error(f"Failed to run terraform plan on {example.name}: {e}")
+                        failed_examples.append(example)
+
+                for example in examples:
+                    pool.submit(run_example, example)
+            if failed_examples:
+                failed_str = ", ".join(sorted(example.name for example in failed_examples))
+                logger.error(f"Failed to run terraform plan on {failed_str} examples")
+    return path
+
+
+def example_plan_checks(config: ModuleGenConfig, timeout_all_seconds: int = 60) -> list[Path]:
+    example_checks = config.example_plan_checks
+    settings = config.settings
+
+    def run_check(check: ExamplePlanCheck):
+        expected_dir = settings.output_plan_dumps / check.expected_output_dir_name
+        variables_path = read_variables_path(expected_dir)
+        with TemporaryDirectory() as temp_dir:
+            stored_plan = Path(temp_dir) / "plan.json"
+            tf_dir = config.example_path(check.example_name)
+            validate_module(tf_dir)
+            var_arg = f" -var-file={variables_path}" if variables_path else ""
+            run_and_wait(f"terraform plan -out={OUT_BINARY_PATH}{var_arg}", cwd=tf_dir)
+            run_and_wait(f"terraform show -json {OUT_BINARY_PATH} > {stored_plan}", cwd=tf_dir)
+            plan_output = parse_plan_output(stored_plan)
+        return generate_expected_actual(settings.output_plan_dumps, check, plan_output)
+
+    with run_pool("Run Examples", total=len(example_checks), exit_wait_timeout=timeout_all_seconds) as pool:
+        futures = {pool.submit(run_check, check): check for check in example_checks}
+    diff_paths: list[Path] = []
+    for future in futures:
+        check = futures[future]
+        try:
+            expected, actual = future.result(timeout=timeout_all_seconds)
+            if expected != actual:
+                diff_path = settings.plan_diff_output_path / f"{config.name}_{check.example_name}.html"
+                dump_html_diff(expected, actual, diff_path)
+                diff_paths.append(diff_path)
+                logger.error(f"Example check failed for {check}")
+        except Exception as e:
+            logger.error(f"Example check failed to run terraform commands for {check}: {e}")
+            raise e
+    return diff_paths
+
+
+def dump_html_diff(expected: str, actual: str, diff_path: Path) -> str:
+    html_text = difflib.HtmlDiff().make_file(
+        expected.splitlines(),
+        actual.splitlines(),
+        "expected",
+        "actual",
+    )
+    ensure_parents_write_text(diff_path, html_text)
+    return html_text

atlas_init/tf_ext/tf_mod_gen_provider.py

@@ -0,0 +1,124 @@
+import logging
+from pathlib import Path
+from ask_shell import confirm, run_pool
+from concurrent.futures import Future
+from ask_shell.rich_live import get_live_console
+from model_lib import copy_and_validate, parse_model
+from rich.markdown import Markdown
+import typer
+from zero_3rdparty.file_utils import clean_dir
+
+from atlas_init.tf_ext.models_module import (
+    ModuleGenConfig,
+    ProviderGenConfig,
+    as_provider_name,
+)
+from atlas_init.tf_ext.provider_schema import parse_atlas_schema_from_settings
+from atlas_init.tf_ext.settings import init_tf_ext_settings
+from atlas_init.tf_ext.tf_mod_gen import finalize_and_validate_module, generate_resource_module
+
+logger = logging.getLogger(__name__)
+ATLAS_PROVIDER_PATH = "mongodb/mongodbatlas"
+
+
+def tf_mod_gen_provider_resource_modules(
+    provider_path: str = typer.Option(
+        ATLAS_PROVIDER_PATH, "--provider-path", help="Provider path name, {owner}/{repo} from terraform registry"
+    ),
+    include_only: list[str] = typer.Option(
+        ..., "-i", "--include-only", help="Only include these resource types", default_factory=list
+    ),
+):
+    settings = init_tf_ext_settings()
+    if provider_path != ATLAS_PROVIDER_PATH:
+        raise NotImplementedError(f"provider_name must be {ATLAS_PROVIDER_PATH}")
+    provider_name = as_provider_name(provider_path)
+    repo_out = settings.repo_out
+    provider_config_path = repo_out.provider_settings_path(provider_name)
+    provider_config = parse_model(provider_config_path, t=ProviderGenConfig)
+
+    atlas_schema = parse_atlas_schema_from_settings(settings, provider_config)
+    include_only_set = set(include_only)
+    deprecated_types = set(atlas_schema.deprecated_resource_types)
+
+    def include_resource_type(resource_type: str) -> bool:
+        return (not include_only or resource_type in include_only_set) and resource_type not in deprecated_types
+
+    resource_types = [
+        resource_type for resource_type in atlas_schema.resource_types if include_resource_type(resource_type)
+    ]
+    if not resource_types:
+        raise ValueError(f"No resource types to generate for provider {provider_name} after filtering")
+
+    def generate_module(module_config: ModuleGenConfig) -> tuple[Path, Path]:
+        resource = module_config.resources[0]
+        generate_resource_module(module_config, resource.name, atlas_schema)
+        module_path = finalize_and_validate_module(module_config)
+
+        config_single = copy_and_validate(
+            module_config,
+            resources=[resource.single_variable_version()],
+            out_dir=module_path.with_name(module_path.stem + "_single"),
+        )
+        generate_resource_module(config_single, resource.name, atlas_schema)
+        module_path_single = finalize_and_validate_module(config_single)
+        return module_path, module_path_single
+
+    with run_pool(
+        "Generating module files for resource types", total=len(resource_types), exit_wait_timeout=60
+    ) as pool:
+        futures: dict[str, Future] = {}
+        for resource_type in resource_types:
+            module_config = ModuleGenConfig.from_repo_out(resource_type, provider_config, repo_out)
+            module_config.skip_python = True
+            futures[resource_type] = pool.submit(generate_module, module_config)
+    summary = ["## Generated Resource Modules"]
+    failures = []
+    generated_module_paths: set[Path] = set()
+    generated_py_files: set[Path] = set()
+    for resource_type, future in futures.items():
+        try:
+            module_path, module_path_single = future.result()
+            logger.info(f"Generated module for resource type = {resource_type} at {module_path} & {module_path_single}")
+            summary.append(f"- {resource_type} -> {module_path}")
+            generated_module_paths.add(module_path)
+            generated_module_paths.add(module_path_single)
+            generated_py_files.add(repo_out.dataclass_path(provider_name, resource_type))
+        except Exception:
+            failures.append(resource_type)
+            logger.exception(f"failed to generate module for resource type = {resource_type}")
+            continue
+    if failures:
+        summary.append("## Failed Resource Modules")
+        for resource_type in failures:
+            summary.append(f"- {resource_type}")
+    get_live_console().print(Markdown("\n".join(summary)))
+    if generated_module_paths:
+        logger.info(f"Generated a total of: {len(generated_module_paths)} modules")
+        if not include_only:
+            clean_extra_modules(repo_out.resource_modules_provider_path(provider_name), generated_module_paths)
+            clean_extra_py_modules(repo_out.py_provider_module(provider_name), generated_py_files)
+
+
+def clean_extra_modules(resource_modules_out_dir: Path, generated_module_paths: set[Path]):
+    if extra_paths := [
+        path for path in resource_modules_out_dir.glob("*") if path.is_dir() and path not in generated_module_paths
+    ]:
+        logger.warning(f"Found extra paths in {resource_modules_out_dir}: {extra_paths}")
+        extra_paths_str = "\n".join(path.name for path in extra_paths)
+        if confirm(f"Can delete extra paths in {resource_modules_out_dir}:\n{extra_paths_str}"):
+            for path in extra_paths:
+                clean_dir(path, recreate=False)
+
+
+def clean_extra_py_modules(py_modules_out_dir: Path, generated_py_files: set[Path]):
+    if extra_paths := [
+        path
+        for path in py_modules_out_dir.glob("*.py")
+        if path.is_file() and not path.name.startswith("_") and path not in generated_py_files
+    ]:
+        logger.warning(f"Found extra paths in {py_modules_out_dir}: {extra_paths}")
+        extra_paths_str = "\n".join(path.name for path in extra_paths)
+        if confirm(f"Can delete extra paths in {py_modules_out_dir}:\n{extra_paths_str}"):
+            for path in extra_paths:
+                path.unlink()

atlas_init/tf_ext/tf_modules.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import logging
 from dataclasses import dataclass
 from pathlib import Path
@@ -13,7 +14,7 @@ from zero_3rdparty.iter_utils import flat_map
 
 from atlas_init.tf_ext.constants import ATLAS_PROVIDER_NAME
 from atlas_init.tf_ext.models import ModuleConfig, ModuleConfigs
-from atlas_init.tf_ext.settings import TfDepSettings
+from atlas_init.tf_ext.settings import TfExtSettings
 from atlas_init.tf_ext.tf_dep import FORCE_INTERNAL_NODES, SKIP_NODES, AtlasGraph, edge_src_dest
 
 logger = logging.getLogger(__name__)
@@ -142,7 +143,7 @@ def tf_modules(
         show_default=True,
     ),
 ):
-    settings = TfDepSettings.from_env()
+    settings = TfExtSettings.from_env()
     atlas_graph = parse_atlas_graph(settings)
     output_dir = settings.static_root
     with new_task("Write graphs"):
@@ -229,7 +230,7 @@ def generate_module_graphs(skipped_module_resource_types, settings, atlas_graph)
     return modules
 
 
-def parse_atlas_graph(settings: TfDepSettings) -> AtlasGraph:
+def parse_atlas_graph(settings: TfExtSettings) -> AtlasGraph:
     atlas_graph = parse_model(settings.atlas_graph_path, t=AtlasGraph)
     deprecated_resources = parse_list(settings.schema_resource_types_deprecated_path, format="yaml")
     atlas_graph.deprecated_resource_types.update(deprecated_resources)
@@ -240,7 +241,7 @@ def parse_atlas_graph(settings: TfDepSettings) -> AtlasGraph:
 
 
 def add_unused_nodes_to_graph(
-    settings: TfDepSettings, atlas_graph: AtlasGraph, color_coder: ColorCoder, internal_graph: pydot.Dot
+    settings: TfExtSettings, atlas_graph: AtlasGraph, color_coder: ColorCoder, internal_graph: pydot.Dot
 ):
     schema_resource_types: list[str] = parse_list(settings.schema_resource_types_path, format="yaml")
     all_nodes = atlas_graph.all_internal_nodes

atlas_init/tf_ext/tf_vars.py

@@ -2,16 +2,15 @@ from __future__ import annotations
 
 import logging
 from pathlib import Path
-from typing import ClassVar, NamedTuple
+from typing import ClassVar
 
-from ask_shell import new_task, run_and_wait
+from ask_shell import new_task
 from model_lib import IgnoreFalsy, dump
 from pydantic import Field, RootModel
 from zero_3rdparty.file_utils import ensure_parents_write_text
 from zero_3rdparty.str_utils import instance_repr
 
-from atlas_init.tf_ext.args import REPO_PATH_ARG, SKIP_EXAMPLES_DIRS_OPTION
-from atlas_init.tf_ext.constants import ATLAS_PROVIDER_NAME
+from atlas_init.tf_ext.args import REPO_PATH_ATLAS_ARG, SKIP_EXAMPLES_DIRS_OPTION
 from atlas_init.tf_ext.paths import (
     ResourceTypes,
     find_resource_types_with_usages,
@@ -19,21 +18,24 @@ from atlas_init.tf_ext.paths import (
     get_example_directories,
     is_variable_name_external,
 )
-from atlas_init.tf_ext.settings import TfDepSettings
+from atlas_init.tf_ext.provider_schema import parse_atlas_schema
+from atlas_init.tf_ext.settings import TfExtSettings
 
 logger = logging.getLogger(__name__)
 
 
 def tf_vars(
-    repo_path: Path = REPO_PATH_ARG,
+    repo_path: Path = REPO_PATH_ATLAS_ARG,
     skip_names: list[str] = SKIP_EXAMPLES_DIRS_OPTION,
 ):
-    settings = TfDepSettings.from_env()
+    settings = TfExtSettings.from_env()
     logger.info(f"Analyzing Terraform variables in repository: {repo_path}")
     example_dirs = get_example_directories(repo_path, skip_names)
     assert example_dirs, "No example directories found. Please check the repository path and skip names."
     with new_task("Parsing provider schema") as task:
-        resource_types, resource_types_deprecated = parse_schema_resource_types(example_dirs[0])
+        atlas_schema = parse_atlas_schema()
+        resource_types = atlas_schema.resource_types
+        resource_types_deprecated = atlas_schema.deprecated_resource_types
         ensure_parents_write_text(settings.schema_resource_types_path, dump(sorted(resource_types), format="yaml"))
         logger.info(f"Provider schema resource types written to {settings.schema_resource_types_path}")
         ensure_parents_write_text(
@@ -51,7 +53,7 @@ def tf_vars(
         logger.warning(f"Missing resource types in examples:\n{'\n'.join(sorted(missing_example_resource_types))}")
 
 
-def parse_provider_resurce_schema(schema: dict, provider_name: str) -> dict:
+def parse_provider_resource_schema(schema: dict, provider_name: str) -> dict:
     schemas = schema.get("provider_schemas", {})
     for provider_url, provider_schema in schemas.items():
         if provider_url.endswith(provider_name):
@@ -104,7 +106,7 @@ def vars_usage_dumping(variables: TfVarsUsage) -> str:
     return dump(vars_model, format="yaml")
 
 
-def update_resource_types(settings: TfDepSettings, example_dirs: list[Path], task: new_task) -> ResourceTypes:
+def update_resource_types(settings: TfExtSettings, example_dirs: list[Path], task: new_task) -> ResourceTypes:
     resource_types = ResourceTypes(root={})
     for example_dir in example_dirs:
         example_resources = find_resource_types_with_usages(example_dir)
@@ -130,7 +132,7 @@ def resource_types_dumping(resource_types: ResourceTypes, with_external: bool =
     return dump(dict(sorted(resource_types_model.items())), format="yaml")
 
 
-def update_variables(settings: TfDepSettings, example_dirs: list[Path], task: new_task):
+def update_variables(settings: TfExtSettings, example_dirs: list[Path], task: new_task):
     variables = parse_all_variables(example_dirs, task)
     logger.info(f"Found {len(variables.root)} variables in the examples.")
     vars_yaml = vars_usage_dumping(variables)
@@ -144,23 +146,6 @@ def update_variables(settings: TfDepSettings, example_dirs: list[Path], task: ne
         logger.info(f"External variables usage written to {settings.vars_external_file_path}")
 
 
-class ResourceTypesSchema(NamedTuple):
-    resource_types: list[str]
-    deprecated_resource_types: list[str]
-
-
-def parse_schema_resource_types(example_dir: Path) -> ResourceTypesSchema:
-    schema_run = run_and_wait("terraform providers schema -json", cwd=example_dir, ansi_content=False)
-    parsed = schema_run.parse_output(dict, output_format="json")
-    resource_schema = parse_provider_resurce_schema(parsed, ATLAS_PROVIDER_NAME)
-
-    def is_deprecated(resource_details: dict) -> bool:
-        return resource_details["block"].get("deprecated", False)
-
-    deprecated_resource_types = [name for name, details in resource_schema.items() if is_deprecated(details)]
-    return ResourceTypesSchema(sorted(resource_schema.keys()), sorted(deprecated_resource_types))
-
-
 def parse_all_variables(examples_dirs: list[Path], task: new_task) -> TfVarsUsage:
     variables_usage = TfVarsUsage(root={})
     for example_dir in examples_dirs:

atlas_init/tf_ext/typer_app.py

@@ -1,11 +1,11 @@
 from ask_shell import configure_logging
 from typer import Typer
 
-from atlas_init.tf_ext import api_call
+from atlas_init.tf_ext import api_call, settings, tf_desc_gen, tf_mod_gen_provider
 
 
 def typer_main():
-    from atlas_init.tf_ext import tf_dep, tf_modules, tf_vars
+    from atlas_init.tf_ext import tf_dep, tf_mod_gen, tf_modules, tf_vars
 
     app = Typer(
         name="tf-ext",
@@ -14,8 +14,12 @@ def typer_main():
     app.command(name="dep-graph")(tf_dep.tf_dep_graph)
     app.command(name="vars")(tf_vars.tf_vars)
     app.command(name="modules")(tf_modules.tf_modules)
+    app.command(name="mod-gen")(tf_mod_gen.tf_mod_gen)
+    app.command(name="desc-gen")(tf_desc_gen.tf_desc_gen)
     app.command(name="api")(api_call.api)
     app.command(name="api-config")(api_call.api_config)
+    app.command(name="mod-gen-provider")(tf_mod_gen_provider.tf_mod_gen_provider_resource_modules)
+    app.command(name="check-env-vars")(settings.init_tf_ext_settings)
     configure_logging(app)
     app()
 

{atlas_init-0.7.0.dist-info → atlas_init-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlas-init
-Version: 0.7.0
+Version: 0.8.1
 Project-URL: Documentation, https://github.com/EspenAlbert/atlas-init#readme
 Project-URL: Issues, https://github.com/EspenAlbert/atlas-init/issues
 Project-URL: Source, https://github.com/EspenAlbert/atlas-init
@@ -12,10 +12,11 @@ Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Requires-Dist: appdirs==1.4.4
-Requires-Dist: ask-shell>=0.0.4
+Requires-Dist: ask-shell>=0.0.5
 Requires-Dist: boto3==1.35.92
 Requires-Dist: gitpython==3.1.42
 Requires-Dist: humanize==4.9.0
+Requires-Dist: inflection==0.5.1
 Requires-Dist: model-lib
 Requires-Dist: motor==3.7.1
 Requires-Dist: mypy-boto3-cloudformation==1.37.22
@@ -30,7 +31,7 @@ Requires-Dist: rich==14.0.0
 Requires-Dist: stringcase==1.2.0
 Requires-Dist: tenacity==9.0.0
 Requires-Dist: typer>=0.15.1
-Requires-Dist: zero-3rdparty
+Requires-Dist: zero-3rdparty==1.0.0b5
 Description-Content-Type: text/markdown
 
 # Atlas Init - A CLI for developing integrations with MongoDB Atlas