infrakit-cli 0.1.14__py3-none-any.whl

infrakit_cli/__init__.py

@@ -0,0 +1,114 @@
+"""InfraKit CLI — spec-kit for IaC, with a multi-persona pipeline.
+
+The package's public API is split into focused submodules:
+
+- :mod:`infrakit_cli.cli` — the Typer app, the four user-facing commands
+  (``init``, ``check``, ``mcp``, ``version``), and the ``main()`` entry point.
+- :mod:`infrakit_cli.bootstrap` — ``initialize_iac_config()`` materialises
+  ``.infrakit/`` and ``.infrakit_tracks/`` on a freshly-initialised project.
+- :mod:`infrakit_cli.template_renderer` — runtime per-agent transformer.
+- :mod:`infrakit_cli.skills` — installs prompt files as agent skills.
+- :mod:`infrakit_cli.mcp` — MCP recipe install helpers.
+- :mod:`infrakit_cli.tracker` — :class:`StepTracker` UI primitive.
+- :mod:`infrakit_cli.interactive` — keyboard input + arrow-key menu.
+- :mod:`infrakit_cli.banner` — ASCII banner and Typer group override.
+- :mod:`infrakit_cli.tools` — subprocess + tool-detection helpers.
+- :mod:`infrakit_cli.git_utils` — repo detection + ``git init`` wrapper.
+- :mod:`infrakit_cli.github_api` — token/auth/rate-limit helpers for the
+  optional ``infrakit version`` check.
+- :mod:`infrakit_cli.console` — shared :class:`rich.console.Console` instance.
+
+The top-level :func:`main` function (re-exported below) is the
+``project.scripts`` entry point: ``infrakit = "infrakit_cli:main"``.
+
+The other re-exports preserve a stable import surface so that existing tests
+and downstream callers continue to use ``from infrakit_cli import X``.
+"""
+
+from .banner import BannerGroup, show_banner
+from .bootstrap import initialize_iac_config
+from .cli import app, callback, check, init, main, mcp, version
+from .console import console
+from .git_utils import init_git_repo, is_git_repo
+from .github_api import (
+    _format_rate_limit_error,
+    _github_auth_headers,
+    _github_token,
+    _parse_rate_limit_headers,
+    client,
+    ssl_context,
+)
+from .interactive import get_key, select_with_arrows
+from .mcp import (
+    _build_mcp_markdown_block,
+    _build_mcp_server_entry,
+    _read_mcp_json,
+    _update_mcp_use_table,
+)
+from .skills import (
+    AGENT_SKILLS_DIR_OVERRIDES,
+    DEFAULT_SKILLS_DIR,
+    SKILL_DESCRIPTIONS,
+    _get_skills_dir,
+    ensure_project_context_from_template,
+    install_ai_skills,
+)
+from .tools import (
+    CLAUDE_LOCAL_PATH,
+    SCRIPT_TYPE_CHOICES,
+    check_tool,
+    find_project_root,
+    run_command,
+)
+from .tracker import StepTracker
+
+__all__ = [
+    # Typer app + commands
+    "app",
+    "callback",
+    "check",
+    "init",
+    "main",
+    "mcp",
+    "version",
+    # UI primitives
+    "BannerGroup",
+    "console",
+    "show_banner",
+    "StepTracker",
+    "select_with_arrows",
+    "get_key",
+    # Project bootstrap
+    "initialize_iac_config",
+    "ensure_project_context_from_template",
+    # Tools / git
+    "check_tool",
+    "find_project_root",
+    "init_git_repo",
+    "is_git_repo",
+    "run_command",
+    "CLAUDE_LOCAL_PATH",
+    "SCRIPT_TYPE_CHOICES",
+    # MCP
+    "_build_mcp_markdown_block",
+    "_build_mcp_server_entry",
+    "_read_mcp_json",
+    "_update_mcp_use_table",
+    # Skills
+    "AGENT_SKILLS_DIR_OVERRIDES",
+    "DEFAULT_SKILLS_DIR",
+    "SKILL_DESCRIPTIONS",
+    "_get_skills_dir",
+    "install_ai_skills",
+    # GitHub API helpers (used by the version command)
+    "_format_rate_limit_error",
+    "_github_auth_headers",
+    "_github_token",
+    "_parse_rate_limit_headers",
+    "client",
+    "ssl_context",
+]
+
+
+if __name__ == "__main__":
+    main()

infrakit_cli/agent_config.py

@@ -0,0 +1,95 @@
+"""
+Agent configurations for infrakit-cli.
+
+Five AI coding agents are supported. Each entry's runtime-rendering keys
+mirror what :mod:`template_renderer` consumes:
+
+- ``folder``: agent-specific root directory under the project (e.g. ``.claude/``).
+- ``commands_subdir``: subdir under ``folder`` where rendered commands live.
+- ``command_format``: ``markdown`` | ``toml`` | ``agent.md``.
+- ``command_args``: the token that replaces ``{ARGS}`` in command bodies.
+- ``command_extension``: the file extension applied to ``infrakit:<name>``.
+- ``supports_subagents``: whether the agent has a built-in subagent
+  invocation primitive (Claude Code's ``Task`` tool). The multi-persona
+  commands check this to decide whether to delegate review phases.
+- ``extras`` (optional list): post-render hooks. Recognised values:
+    * ``"copilot_prompts"`` — for every ``infrakit:*.agent.md`` produced,
+      also emit a companion ``infrakit:*.prompt.md`` with ``agent:``
+      frontmatter.
+    * ``"vscode_settings"`` — copy ``templates/vscode-settings.json`` to
+      ``.vscode/settings.json``.
+
+InfraKit historically supported 19 agents but the per-agent surface area
+(layout differences, MCP install paths, command-format quirks) was
+unmaintainable. The supported set is now the five agents we actively
+test: Claude Code, Codex CLI, Gemini CLI, GitHub Copilot, plus a
+``generic`` fallback for bring-your-own-agent setups.
+"""
+
+# Default values for agent metadata.
+DEFAULT_FORMAT = "markdown"
+DEFAULT_ARGS = "$ARGUMENTS"
+DEFAULT_EXTENSION = ".md"
+
+AGENT_CONFIG = {
+    "claude": {
+        "name": "Claude Code",
+        "folder": ".claude/",
+        "commands_subdir": "commands",
+        "install_url": "https://docs.anthropic.com/en/docs/claude-code/setup",
+        "requires_cli": True,
+        "command_format": DEFAULT_FORMAT,
+        "command_args": DEFAULT_ARGS,
+        "command_extension": DEFAULT_EXTENSION,
+        "mcp_install_path": ".mcp.json",
+        "supports_subagents": True,
+        # Render persona files into .claude/agents/ so the Task tool can
+        # invoke them by subagent_type matching the frontmatter `name:`.
+        "extras": ["claude_subagents"],
+    },
+    "codex": {
+        "name": "Codex CLI",
+        "folder": ".codex/",
+        "commands_subdir": "prompts",
+        "install_url": "https://github.com/openai/codex",
+        "requires_cli": True,
+        "command_format": DEFAULT_FORMAT,
+        "command_args": DEFAULT_ARGS,
+        "command_extension": DEFAULT_EXTENSION,
+        "supports_subagents": False,
+    },
+    "gemini": {
+        "name": "Gemini CLI",
+        "folder": ".gemini/",
+        "commands_subdir": "commands",
+        "install_url": "https://github.com/google-gemini/gemini-cli",
+        "requires_cli": True,
+        "command_format": "toml",
+        "command_args": "{{args}}",
+        "command_extension": ".toml",
+        "supports_subagents": False,
+    },
+    "copilot": {
+        "name": "GitHub Copilot",
+        "folder": ".github/",
+        "commands_subdir": "agents",
+        "install_url": None,
+        "requires_cli": False,
+        "command_format": "agent.md",
+        "command_args": DEFAULT_ARGS,
+        "command_extension": ".agent.md",
+        "extras": ["copilot_prompts", "vscode_settings"],
+        "supports_subagents": False,
+    },
+    "generic": {
+        "name": "Generic (bring your own agent)",
+        "folder": ".infrakit/",
+        "commands_subdir": "commands",
+        "install_url": None,
+        "requires_cli": False,
+        "command_format": DEFAULT_FORMAT,
+        "command_args": DEFAULT_ARGS,
+        "command_extension": DEFAULT_EXTENSION,
+        "supports_subagents": False,
+    },
+}

infrakit_cli/banner.py

@@ -0,0 +1,47 @@
+"""ASCII banner displayed at the top of every ``infrakit`` command run.
+
+:class:`BannerGroup` overrides Typer's group class so that ``infrakit --help``
+also prints the banner before the help text.
+"""
+
+from __future__ import annotations
+
+from rich.align import Align
+from rich.text import Text
+from typer.core import TyperGroup
+
+from .console import console
+
+BANNER = """
+██╗███╗   ██╗███████╗██████╗  █████╗ ██╗  ██╗██╗████████╗
+██║████╗  ██║██╔════╝██╔══██╗██╔══██╗██║ ██╔╝██║╚══██╔══╝
+██║██╔██╗ ██║█████╗  ██████╔╝███████║█████╔╝ ██║   ██║
+██║██║╚██╗██║██╔══╝  ██╔══██╗██╔══██║██╔═██╗ ██║   ██║
+██║██║ ╚████║██║     ██║  ██║██║  ██║██║  ██╗██║   ██║
+╚═╝╚═╝  ╚═══╝╚═╝     ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝   ╚═╝
+"""
+
+TAGLINE = "InfraKit — spec-kit for IaC, with a multi-persona pipeline"
+
+
+def show_banner():
+    """Render the ASCII banner with a gradient and the project tagline beneath."""
+    banner_lines = BANNER.strip().split("\n")
+    colors = ["bright_blue", "blue", "cyan", "bright_cyan", "white", "bright_white"]
+
+    styled_banner = Text()
+    for i, line in enumerate(banner_lines):
+        color = colors[i % len(colors)]
+        styled_banner.append(line + "\n", style=color)
+
+    console.print(Align.center(styled_banner))
+    console.print(Align.center(Text(TAGLINE, style="italic bright_yellow")))
+    console.print()
+
+
+class BannerGroup(TyperGroup):
+    """Typer group that prints the banner ahead of ``--help`` output."""
+
+    def format_help(self, ctx, formatter):
+        show_banner()
+        super().format_help(ctx, formatter)

infrakit_cli/bootstrap.py

@@ -0,0 +1,184 @@
+"""IaC project bootstrap — writes ``.infrakit/`` and ``.infrakit_tracks/``.
+
+:func:`initialize_iac_config` is the single function called by
+``infrakit init`` (and the corresponding tests) to materialise everything a
+freshly-bootstrapped InfraKit project needs:
+
+- ``.infrakit/config.yaml`` recording the selected IaC tool + AI assistant
+- ``.infrakit/context.md`` + ``.infrakit/coding-style.md`` (from IaC-specific assets)
+- ``.infrakit/tagging-standard.md`` (shared template)
+- ``.infrakit/mcp-use.md`` (empty installed-MCP index)
+- ``.infrakit/memory/`` (project memory directory)
+- ``.infrakit_tracks/tracks.md`` (master resource registry)
+- ``.infrakit_tracks/tracks/`` (per-track working directories)
+- Rendered slash commands + personas via
+  :func:`infrakit_cli.template_renderer.materialize_project`.
+"""
+
+from __future__ import annotations
+
+import shutil
+from pathlib import Path
+from typing import Optional
+
+import yaml
+
+from .agent_config import AGENT_CONFIG
+from .iac_config import IAC_CONFIG
+from .template_renderer import materialize_project, templates_root
+from .tracker import StepTracker
+
+
+def initialize_iac_config(
+    project_path: Path,
+    iac_tool: str,
+    ai_assistant: str,
+    *,
+    tracker: Optional[StepTracker] = None,
+) -> None:
+    """Set up IaC-specific configuration, commands, agents, and documentation.
+
+    Creates:
+
+    - ``.infrakit/config.yaml`` — selected IaC tool and AI agent
+    - ``.infrakit/context.md`` — project context template (copied if absent)
+    - ``.infrakit/coding-style.md`` — default coding standards (copied if absent)
+    - ``.infrakit/tagging-standard.md`` — shared tagging template (copied if absent)
+    - ``.infrakit/mcp-use.md`` — installed MCP server index (created if absent)
+    - ``.infrakit/memory/`` — per-project memory directory
+    - ``.infrakit_tracks/tracks.md`` — master resource registry
+    - ``.infrakit_tracks/tracks/`` — track working directories
+    - Rendered IaC-native commands in the agent's commands directory
+    - Rendered IaC + generic personas under ``.infrakit/agent_personas/``
+    """
+    iac_cfg = IAC_CONFIG.get(iac_tool, {})
+    if not iac_cfg:
+        if tracker:
+            tracker.error("iac-config", f"unknown IaC tool: {iac_tool}")
+        return
+
+    tpl_root = templates_root()
+    iac_templates_dir = tpl_root / "iac" / iac_tool
+
+    # --- 1. Create .infrakit/ configuration directory --------------------
+
+    if tracker:
+        tracker.start("iac-config")
+
+    infrakit_dir = project_path / ".infrakit"
+    infrakit_dir.mkdir(parents=True, exist_ok=True)
+
+    infrakit_tracks_dir = project_path / ".infrakit_tracks"
+    infrakit_tracks_dir.mkdir(parents=True, exist_ok=True)
+
+    # config.yaml — never overwritten on subsequent init runs.
+    config_data = {
+        "iac_tool": iac_tool,
+        "iac_name": iac_cfg.get("name", iac_tool),
+        "ai_assistant": ai_assistant,
+        "resource_term": iac_cfg.get("resource_term", "composition"),
+    }
+    config_file = infrakit_dir / "config.yaml"
+    if not config_file.exists():
+        config_file.write_text(
+            yaml.dump(config_data, sort_keys=False), encoding="utf-8"
+        )
+
+    # Copy assets (context.md, coding-style.md) from the IaC asset templates.
+    assets_dir = iac_templates_dir / "assets"
+    if assets_dir.is_dir():
+        for asset_file in assets_dir.iterdir():
+            if asset_file.is_file():
+                dest = infrakit_dir / asset_file.name.replace(
+                    "context_template", "context"
+                ).replace("default_coding_style", "coding-style").replace(
+                    "coding-style-template", "coding-style"
+                )
+                if not dest.exists():
+                    shutil.copy2(asset_file, dest)
+
+    # tracks.md — master resource registry.
+    tracks_md = infrakit_tracks_dir / "tracks.md"
+    if not tracks_md.exists():
+        tracks_md.write_text(
+            "# Infrastructure Resource Registry\n\n"
+            "Track all infrastructure compositions and their current status.\n\n"
+            "## Status Reference\n\n"
+            "| Symbol | Meaning |\n"
+            "|--------|---------|\n"
+            "| 🔵 `initializing` | Track created, spec in progress |\n"
+            "| 📝 `spec-generated` | Spec confirmed, ready for plan |\n"
+            "| 📋 `planned` | Plan generated, ready for implementation |\n"
+            "| ⚙️ `in-progress` | Implementation underway |\n"
+            "| ✅ `done` | Implementation complete and reviewed |\n"
+            "| ❌ `blocked` | Blocked, needs attention |\n\n"
+            "---\n\n"
+            "## Tracks\n\n"
+            "| Track | Type | Directory | Status | Created |\n"
+            "|-------|------|-----------|--------|---------|\n"
+            "| (none yet) | — | — | — | — |\n",
+            encoding="utf-8",
+        )
+
+    # tagging-standard.md — shared, IaC-agnostic tagging standard.
+    tagging_std_md = infrakit_dir / "tagging-standard.md"
+    if not tagging_std_md.exists():
+        shared_tagging_template = tpl_root / "tagging-standard-template.md"
+        if shared_tagging_template.is_file():
+            shutil.copy2(shared_tagging_template, tagging_std_md)
+        else:
+            tagging_std_md.write_text(
+                "# Tagging Standard\n\n"
+                "> Run `/infrakit:setup` to configure your project-specific tagging requirements.\n",
+                encoding="utf-8",
+            )
+
+    # mcp-use.md — installed MCP server index.
+    mcp_use_md = infrakit_dir / "mcp-use.md"
+    if not mcp_use_md.exists():
+        mcp_use_md.write_text(
+            "# Installed MCP Servers\n\n"
+            "MCP servers configured for this project.\n"
+            "Run `infrakit mcp` to add more.\n\n"
+            "| MCP | Description | Tools | Usage |\n"
+            "|-----|-------------|-------|-------|\n"
+            "| — | — | — | — |\n",
+            encoding="utf-8",
+        )
+
+    # Memory directory (for project context).
+    (infrakit_dir / "memory").mkdir(parents=True, exist_ok=True)
+
+    # Per-track working directories live under .infrakit_tracks/tracks/.
+    tracks_dir = infrakit_tracks_dir / "tracks"
+    tracks_dir.mkdir(parents=True, exist_ok=True)
+
+    if tracker:
+        tracker.complete("iac-config", f"{iac_tool} ({iac_cfg.get('name', '')})")
+
+    # --- 2. Render commands + personas via the runtime renderer ---------
+
+    if tracker:
+        tracker.start("iac-commands")
+
+    counts = materialize_project(
+        project_path,
+        ai_assistant=ai_assistant,
+        iac_tool=iac_tool,
+        overwrite=False,
+    )
+
+    agent_cfg = AGENT_CONFIG.get(ai_assistant, {})
+    agent_folder = (agent_cfg.get("folder") or "").rstrip("/")
+    commands_subdir = agent_cfg.get("commands_subdir", "commands")
+    cmds_dest = (
+        project_path / agent_folder / commands_subdir
+        if agent_folder
+        else project_path / commands_subdir
+    )
+
+    if tracker:
+        tracker.complete(
+            "iac-commands",
+            f"{counts['generic_commands']} generic + {counts['iac_commands']} IaC commands → {cmds_dest.relative_to(project_path)}",
+        )