dftech add command

Fixes #25

Author: spoorcc

dfetch/__main__.py

@@ -9,6 +9,7 @@
 
 from rich.console import Console
 
+import dfetch.commands.add
 import dfetch.commands.check
 import dfetch.commands.diff
 import dfetch.commands.environment
@@ -43,6 +44,7 @@ def create_parser() -> argparse.ArgumentParser:
     parser.set_defaults(func=_help)
     subparsers = parser.add_subparsers(help="commands")
 
+    dfetch.commands.add.Add.create_menu(subparsers)
     dfetch.commands.check.Check.create_menu(subparsers)
     dfetch.commands.diff.Diff.create_menu(subparsers)
     dfetch.commands.environment.Environment.create_menu(subparsers)

dfetch/commands/add.py

@@ -0,0 +1,111 @@
+"""*Dfetch* can add projects through the cli to the manifest.
+
+Sometimes you want to add a project to your manifest, but you don't want to
+edit the manifest by hand. With ``dfetch add`` you can add a project to your manifest
+through the command line. This will add the project to your manifest and fetch it
+to your disk. You can also specify a version to add, or it will be added with the
+latest version available.
+
+"""
+
+import argparse
+from pathlib import Path
+
+import yaml
+from rich.console import Console
+from rich.prompt import Prompt
+from rich.table import Table
+
+import dfetch.commands.command
+import dfetch.manifest.project
+import dfetch.project
+from dfetch.log import get_logger
+from dfetch.manifest.manifest import ManifestDumper
+from dfetch.manifest.project import ProjectEntry, ProjectEntryDict
+from dfetch.project import create_sub_project, create_super_project
+from dfetch.util.util import in_directory
+
+console = Console()
+
+logger = get_logger(__name__)
+
+
+class Add(dfetch.commands.command.Command):
+    """Add a new project to the manifest.
+
+    Add a new project to the manifest.
+    """
+
+    @staticmethod
+    def create_menu(subparsers: dfetch.commands.command.SubparserActionType) -> None:
+        """Add the parser menu for this action."""
+        parser = dfetch.commands.command.Command.parser(subparsers, Add)
+
+        parser.add_argument(
+            "remote_url",
+            metavar="<remote_url>",
+            type=str,
+            nargs=1,
+            help="Remote URL of project to add",
+        )
+
+    def __call__(self, args: argparse.Namespace) -> None:
+        """Perform the add."""
+        superproject = create_super_project()
+
+        with in_directory(superproject.root_directory):
+
+            project_entry = ProjectEntry(
+                ProjectEntryDict(name="project-to-be-added", url=args.remote_url[0])
+            )
+
+            subproject = create_sub_project(project_entry)
+
+            manifest = superproject.manifest
+
+            if project_entry.name in [p.name for p in manifest.projects]:
+                raise RuntimeError(
+                    f"Project with name {project_entry.name} already exists in manifest!"
+                )
+
+            project_entry = ProjectEntry(
+                ProjectEntryDict(
+                    name=project_entry.name,
+                    url=project_entry.remote_url,
+                    branch=subproject.get_default_branch(),
+                )
+            )
+
+            if not confirm(project_entry):
+                console.print("[yellow]Aborted[/yellow]")
+                return
+
+            with Path(manifest.path).open("a", encoding="utf-8") as manifest_file:
+
+                new_entry = yaml.dump(
+                    [project_entry.as_yaml()],
+                    Dumper=ManifestDumper,
+                    sort_keys=False,
+                    # line_break=os.linesep,
+                    indent=2,
+                )
+                manifest_file.write("\n")
+                for line in new_entry.splitlines():
+                    manifest_file.write(f"  {line}\n")
+
+            console.print("[bold green]✔ Project added successfully[/bold green]")
+
+
+def confirm(project: ProjectEntry) -> bool:
+    """Show a confirmation prompt to the user before adding the project."""
+    table = Table(title="Confirm dfetch add", show_lines=True)
+    table.add_column("Field", style="bold")
+    table.add_column("Value", style="cyan")
+
+    for k, v in project.as_yaml().items():
+        table.add_row(k, str(v))
+
+    console.clear()
+    console.print(table)
+
+    return Prompt.ask("Proceed?", choices=["y", "n"], default="y") == "y"