smith.delivery.cli

  1"""CLI entry point for the smith command-line tool."""
  2
  3import argparse
  4import sys
  5from pathlib import Path
  6
  7from smith.application.connect import ConnectUseCase
  8from smith.application.disconnect import DisconnectUseCase
  9from smith.application.status import StatusUseCase
 10from smith.application.update import UpdateUseCase
 11from smith.domain.ports import TemplateSourceError
 12from smith.domain.value_objects import TemplateSource
 13
 14EXIT_SUCCESS = 0
 15EXIT_ERROR = 1
 16
 17
 18def build_parser() -> argparse.ArgumentParser:
 19    """Build and return the argument parser for the smith CLI."""
 20    parser = argparse.ArgumentParser(
 21        prog="smith", description="Connect AI agent configurations to any project"
 22    )
 23    parser.add_argument("--version", action="version", version=_get_version())
 24
 25    subparsers = parser.add_subparsers(dest="command", help="Available commands")
 26
 27    connect_parser = subparsers.add_parser(
 28        "connect", help="Connect agentic files to a project"
 29    )
 30    connect_parser.add_argument(
 31        "--from",
 32        dest="source",
 33        help="Template source (bundled:agents-smith, local path, or URL)",
 34    )
 35    connect_parser.add_argument(
 36        "--overwrite",
 37        action="store_true",
 38        dest="overwrite",
 39        help="Replace existing agentic files without prompting",
 40    )
 41    connect_parser.set_defaults(func=handle_connect)
 42
 43    disconnect_parser = subparsers.add_parser(
 44        "disconnect", help="Disconnect agentic files from a project"
 45    )
 46    disconnect_parser.set_defaults(func=handle_disconnect)
 47
 48    update_parser = subparsers.add_parser("update", help="Update agentic files")
 49    update_parser.add_argument(
 50        "--from",
 51        dest="source",
 52        help="Template source (bundled:agents-smith, local path, or URL)",
 53    )
 54    update_parser.set_defaults(func=handle_update)
 55
 56    status_parser = subparsers.add_parser("status", help="Show connection status")
 57    status_parser.add_argument("--json", action="store_true", help="Output as JSON")
 58    status_parser.set_defaults(func=handle_status)
 59
 60    return parser
 61
 62
 63def _get_version() -> str:
 64    """Return the current smith version string."""
 65    try:
 66        from importlib.metadata import metadata
 67
 68        meta = metadata("agents-smith")
 69        return f"smith {meta['Version']}"
 70    except Exception:
 71        return "smith 0.1.0"
 72
 73
 74def _parse_source(source_arg: str | None) -> TemplateSource:
 75    """Parse a source argument string into a TemplateSource value object."""
 76    if source_arg is None:
 77        return TemplateSource(kind="bundled", location="agents-smith")
 78    if source_arg.startswith("/"):
 79        return TemplateSource(kind="local", location=source_arg)
 80    if source_arg.startswith("http://") or source_arg.startswith("https://"):
 81        return TemplateSource(kind="url", location=source_arg)
 82    if ":" in source_arg:
 83        kind, _, location = source_arg.partition(":")
 84        return TemplateSource(kind=kind, location=location)  # type: ignore[arg-type]
 85    return TemplateSource(kind="local", location=source_arg)
 86
 87
 88def handle_connect(args: argparse.Namespace) -> int:
 89    """Handle the ``connect`` sub-command."""
 90    project_dir = Path(getattr(args, "project_dir", ".")).resolve()
 91    source = _parse_source(getattr(args, "source", None))
 92    overwrite = getattr(args, "overwrite", False)
 93    try:
 94        ConnectUseCase(project_dir=project_dir).execute(
 95            source=source, overwrite=overwrite
 96        )
 97        return EXIT_SUCCESS
 98    except TemplateSourceError as e:
 99        sys.stderr.write(f"Error: {e}\n")
100        return EXIT_ERROR
101    except Exception as e:
102        sys.stderr.write(f"Error: {e}\n")
103        return EXIT_ERROR
104
105
106def handle_disconnect(args: argparse.Namespace) -> int:
107    """Handle the ``disconnect`` sub-command."""
108    project_dir = Path(getattr(args, "project_dir", ".")).resolve()
109    try:
110        DisconnectUseCase(project_dir=project_dir).execute()
111        return EXIT_SUCCESS
112    except Exception as e:
113        sys.stderr.write(f"Error: {e}\n")
114        return EXIT_ERROR
115
116
117def handle_update(args: argparse.Namespace) -> int:
118    """Handle the ``update`` sub-command."""
119    project_dir = Path(getattr(args, "project_dir", ".")).resolve()
120    source = _parse_source(getattr(args, "source", None))
121    try:
122        UpdateUseCase(project_dir=project_dir).execute(
123            source=source if args.source else None
124        )
125        return EXIT_SUCCESS
126    except TemplateSourceError as e:
127        sys.stderr.write(f"Error: {e}\n")
128        return EXIT_ERROR
129    except Exception as e:
130        sys.stderr.write(f"Error: {e}\n")
131        return EXIT_ERROR
132
133
134def handle_status(args: argparse.Namespace) -> int:
135    """Handle the ``status`` sub-command."""
136    project_dir = Path(getattr(args, "project_dir", ".")).resolve()
137    try:
138        status = StatusUseCase(project_dir=project_dir).execute()
139        if getattr(args, "json", False):
140            import json
141
142            sys.stdout.write(json.dumps(status.to_dict(), indent=2) + "\n")
143        else:
144            sys.stdout.write(f"State: {status.state.value}\n")
145            if status.source:
146                sys.stdout.write(
147                    f"Source: {status.source.kind}:{status.source.location}\n"
148                )
149            if status.present_files:
150                sys.stdout.write("Present files:\n")
151                for f in status.present_files:
152                    sys.stdout.write(f"  {f}\n")
153            if status.missing_files:
154                sys.stdout.write("Missing files:\n")
155                for f in status.missing_files:
156                    sys.stdout.write(f"  {f}\n")
157        return EXIT_SUCCESS
158    except Exception as e:
159        sys.stderr.write(f"Error: {e}\n")
160        return EXIT_ERROR
161
162
163def main(argv: list[str] | None = None) -> int:
164    """Run the smith CLI and return an exit code."""
165    parser = build_parser()
166    args = parser.parse_args(argv)
167    if not hasattr(args, "func"):
168        parser.print_help()
169        return EXIT_ERROR
170    return args.func(args)