lnp-devopscli 1.0.0__tar.gz → 1.0.3__tar.gz

{lnp_devopscli-1.0.0 → lnp_devopscli-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lnp-devopscli
-Version: 1.0.0
+Version: 1.0.3
 Summary: DevOps CLI — workspace sync (GDrive, git, GPG, systemd timers) + GitLab/Bitwarden tooling
 Author-email: Lucas Neves Pires <npires.lucas@gmail.com>
 License: MIT
@@ -24,6 +24,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: click>=8.0
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: requests>=2.28
+Requires-Dist: rich>=13.0
 
 # lnp-devopscli
 

{lnp_devopscli-1.0.0 → lnp_devopscli-1.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lnp-devopscli"
-version = "1.0.0"
+version = "1.0.3"
 description = "DevOps CLI — workspace sync (GDrive, git, GPG, systemd timers) + GitLab/Bitwarden tooling"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -30,6 +30,7 @@ dependencies = [
     "click>=8.0",
     "pyyaml>=6.0",
     "requests>=2.28",
+    "rich>=13.0",
 ]
 
 [project.urls]

lnp_devopscli-1.0.3/scripts/dc_cli/groups/test.py

@@ -0,0 +1,581 @@
+"""Grupo: dc test - Testing infrastructure (Multipass VM)."""
+
+from __future__ import annotations
+
+import shutil
+import subprocess
+import sys
+import time
+from pathlib import Path
+
+import click
+
+
+DEFAULT_VM_NAME = "dev-test"
+DEFAULT_IMAGE = "24.04"
+SNAPPED_SNAPSHOT_NAME = "fresh"
+BOOTSTRAP_URL = (
+    "https://gitlab.com/-/snippets/6003334/raw/main/bootstrap.sh"
+)
+
+
+def _require_multipass() -> None:
+    if not shutil.which("multipass"):
+        click.echo(
+            "[ERRO] multipass nao instalado.\n"
+            "       Instale com: sudo snap install multipass",
+            err=True,
+        )
+        sys.exit(1)
+
+
+def _vm_exists(name: str) -> bool:
+    result = subprocess.run(
+        ["multipass", "info", name],
+        capture_output=True,
+        text=True,
+    )
+    return result.returncode == 0
+
+
+def _vm_state(name: str) -> str:
+    """Return: running | stopped | not-found."""
+    result = subprocess.run(
+        ["multipass", "info", name, "--format", "csv"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        return "not-found"
+    for line in result.stdout.splitlines():
+        if line.startswith(name + ","):
+            parts = line.split(",")
+            if len(parts) > 1:
+                return parts[1].lower().strip()
+    return "unknown"
+
+
+def _snapshot_exists(name: str, snapshot: str) -> bool:
+    result = subprocess.run(
+        ["multipass", "list", "--snapshots", "--format", "csv"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        return False
+    target = f"{name}.{snapshot}"
+    for line in result.stdout.splitlines():
+        if target in line:
+            return True
+    return False
+
+
+@click.group(name="test")
+def test_group():
+    """Testing infrastructure (Multipass VM, etc)."""
+
+
+@test_group.group(name="vm")
+def vm_group():
+    """Multipass VM management for isolated testing."""
+
+
+# ─── vm create ──────────────────────────────────────────────────
+
+
+@vm_group.command(name="create")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option("--cpus", default=2, show_default=True, type=int)
+@click.option("--memory", default="4G", show_default=True)
+@click.option("--disk", default="20G", show_default=True)
+@click.option("--image", default=DEFAULT_IMAGE, show_default=True)
+@click.option(
+    "--gui/--no-gui",
+    default=False,
+    help="Pre-install X11 deps (xauth, firefox) for GUI via SSH -X",
+)
+@click.option(
+    "--snapshot/--no-snapshot",
+    default=True,
+    help="Take snapshot after baseline setup",
+)
+def vm_create_cmd(
+    name: str,
+    cpus: int,
+    memory: str,
+    disk: str,
+    image: str,
+    gui: bool,
+    snapshot: bool,
+) -> None:
+    """Create a baseline Ubuntu VM with apt deps pre-installed."""
+    _require_multipass()
+
+    if _vm_exists(name):
+        click.echo(f"[INFO] VM '{name}' ja existe. Use 'vm destroy' primeiro.")
+        sys.exit(1)
+
+    click.echo(f"▶ Launching VM '{name}' ({cpus} CPUs, {memory} RAM, {disk} disk)...")
+    subprocess.run(
+        [
+            "multipass",
+            "launch",
+            "--name",
+            name,
+            "--cpus",
+            str(cpus),
+            "--memory",
+            memory,
+            "--disk",
+            disk,
+            image,
+        ],
+        check=True,
+    )
+
+    click.echo("\n▶ Pre-installing apt deps...")
+    base_deps = [
+        "git",
+        "curl",
+        "wget",
+        "ca-certificates",
+        "python3",
+        "python3-pip",
+        "pipx",
+        "rclone",
+        "git-crypt",
+        "gnupg",
+        "libnotify-bin",
+        "whiptail",
+        "jq",
+        "sudo",
+    ]
+    if gui:
+        base_deps += ["xauth", "x11-apps", "firefox"]
+        click.echo("  (GUI mode: incluindo xauth + firefox)")
+
+    apt_cmd = "sudo apt-get update -qq && sudo apt-get install -y " + " ".join(base_deps)
+    subprocess.run(["multipass", "exec", name, "--", "bash", "-c", apt_cmd], check=True)
+
+    click.echo("\n▶ Setup pipx PATH...")
+    subprocess.run(
+        ["multipass", "exec", name, "--", "bash", "-c", "pipx ensurepath"],
+        check=False,
+    )
+
+    if snapshot:
+        click.echo(f"\n▶ Stopping VM for snapshot...")
+        subprocess.run(["multipass", "stop", name], check=True)
+
+        click.echo(f"▶ Taking snapshot '{SNAPPED_SNAPSHOT_NAME}'...")
+        subprocess.run(
+            ["multipass", "snapshot", name, "--name", SNAPPED_SNAPSHOT_NAME],
+            check=True,
+        )
+
+        click.echo(f"▶ Starting VM again...")
+        subprocess.run(["multipass", "start", name], check=True)
+
+    click.echo("\n✓ VM pronta!")
+    click.echo(f"  Nome:     {name}")
+    click.echo(f"  Snapshot: {SNAPPED_SNAPSHOT_NAME}")
+    click.echo("\nProximos passos:")
+    click.echo(f"  devopscli test vm bootstrap --name {name}    # roda o bootstrap snippet")
+    click.echo(f"  devopscli test vm shell --name {name}        # abre shell")
+    click.echo(f"  devopscli test vm reset --name {name}        # restaura snapshot")
+
+
+# ─── vm reset ───────────────────────────────────────────────────
+
+
+@vm_group.command(name="reset")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option(
+    "--snapshot",
+    default=SNAPPED_SNAPSHOT_NAME,
+    show_default=True,
+    help="Snapshot name to restore",
+)
+def vm_reset_cmd(name: str, snapshot: str) -> None:
+    """Restore VM to a snapshot (default: fresh baseline)."""
+    _require_multipass()
+
+    if not _vm_exists(name):
+        click.echo(f"[ERRO] VM '{name}' nao existe", err=True)
+        sys.exit(1)
+
+    if not _snapshot_exists(name, snapshot):
+        click.echo(
+            f"[ERRO] Snapshot '{name}.{snapshot}' nao existe", err=True
+        )
+        sys.exit(1)
+
+    click.echo(f"▶ Stopping VM '{name}'...")
+    subprocess.run(["multipass", "stop", name], check=False)
+
+    click.echo(f"▶ Restoring to snapshot '{snapshot}'...")
+    subprocess.run(
+        ["multipass", "restore", "--destructive", f"{name}.{snapshot}"],
+        check=True,
+    )
+
+    click.echo(f"▶ Starting VM...")
+    subprocess.run(["multipass", "start", name], check=True)
+
+    click.echo(f"✓ VM '{name}' restaurada para snapshot '{snapshot}'")
+
+
+# ─── vm shell ───────────────────────────────────────────────────
+
+
+@vm_group.command(name="shell")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+def vm_shell_cmd(name: str) -> None:
+    """Open interactive shell in the VM."""
+    _require_multipass()
+
+    if _vm_state(name) != "running":
+        click.echo(f"▶ Starting VM '{name}'...")
+        subprocess.run(["multipass", "start", name], check=False)
+
+    subprocess.run(["multipass", "shell", name])
+
+
+# ─── vm exec ────────────────────────────────────────────────────
+
+
+@vm_group.command(name="exec")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.argument("command", nargs=-1, required=True)
+def vm_exec_cmd(name: str, command: tuple[str, ...]) -> None:
+    """Execute a command in the VM."""
+    _require_multipass()
+
+    if _vm_state(name) != "running":
+        subprocess.run(["multipass", "start", name], check=False)
+
+    cmd_str = " ".join(command)
+    subprocess.run(["multipass", "exec", name, "--", "bash", "-c", cmd_str])
+
+
+# ─── vm bootstrap ───────────────────────────────────────────────
+
+
+@vm_group.command(name="bootstrap")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option(
+    "--url",
+    default=BOOTSTRAP_URL,
+    show_default=True,
+    help="URL of bootstrap script to execute",
+)
+def vm_bootstrap_cmd(name: str, url: str) -> None:
+    """Run the bootstrap snippet inside the VM (full setup)."""
+    _require_multipass()
+
+    if _vm_state(name) != "running":
+        subprocess.run(["multipass", "start", name], check=False)
+
+    click.echo(f"▶ Executing bootstrap in VM '{name}'...")
+    click.echo(f"  URL: {url}")
+    click.echo()
+
+    subprocess.run(
+        [
+            "multipass",
+            "exec",
+            name,
+            "--",
+            "bash",
+            "-c",
+            f"bash <(curl -fsSL {url})",
+        ]
+    )
+
+
+# ─── vm mount ───────────────────────────────────────────────────
+
+
+@vm_group.command(name="mount")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option(
+    "--source",
+    default=str(Path.home() / "bin" / "devops-cli"),
+    show_default=True,
+    help="Local path to mount",
+)
+@click.option(
+    "--target",
+    default="/home/ubuntu/devops-cli",
+    show_default=True,
+    help="Path inside VM",
+)
+def vm_mount_cmd(name: str, source: str, target: str) -> None:
+    """Mount a local directory inside the VM (for development)."""
+    _require_multipass()
+
+    src_path = Path(source).expanduser().resolve()
+    if not src_path.exists():
+        click.echo(f"[ERRO] Source path nao existe: {src_path}", err=True)
+        sys.exit(1)
+
+    click.echo(f"▶ Mounting {src_path} -> {name}:{target}")
+    subprocess.run(
+        ["multipass", "mount", str(src_path), f"{name}:{target}"],
+        check=True,
+    )
+    click.echo(f"✓ Mounted. Dentro da VM, use:")
+    click.echo(f"    cd {target} && pipx install -e .")
+    click.echo(f"    # edita no host, testa na VM, sem ciclo de push")
+
+
+# ─── vm info ────────────────────────────────────────────────────
+
+
+@vm_group.command(name="info")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+def vm_info_cmd(name: str) -> None:
+    """Show VM status, IP, snapshots."""
+    _require_multipass()
+    subprocess.run(["multipass", "info", name])
+    click.echo()
+    click.echo("Snapshots:")
+    subprocess.run(
+        ["multipass", "list", "--snapshots", "--filter", name],
+        check=False,
+    )
+
+
+# ─── vm destroy ─────────────────────────────────────────────────
+
+
+@vm_group.command(name="destroy")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.confirmation_option(prompt="Tem certeza? VM e todos snapshots serao deletados.")
+def vm_destroy_cmd(name: str) -> None:
+    """Stop and permanently delete VM + snapshots."""
+    _require_multipass()
+
+    subprocess.run(["multipass", "stop", name], check=False)
+    subprocess.run(["multipass", "delete", name, "--purge"], check=False)
+    click.echo(f"✓ VM '{name}' destroyed.")
+
+
+# ─── vm ssh ─────────────────────────────────────────────────────
+
+
+def _get_vm_ip(name: str) -> str | None:
+    result = subprocess.run(
+        ["multipass", "info", name, "--format", "csv"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        return None
+    for line in result.stdout.splitlines():
+        if line.startswith(name + ","):
+            parts = line.split(",")
+            # CSV: Name,State,Ipv4,Image,Release,Cpu(s),Load,Disk usage,Memory usage,Mounts
+            if len(parts) > 2 and parts[2].strip():
+                return parts[2].strip()
+    return None
+
+
+def _ensure_ssh_key_in_vm(name: str) -> None:
+    """Copia ~/.ssh/id_*.pub do host pra ~/.ssh/authorized_keys da VM."""
+    host_pubkeys = list(Path.home().joinpath(".ssh").glob("id_*.pub"))
+    if not host_pubkeys:
+        click.echo(
+            "[WARN] Nenhuma chave publica em ~/.ssh/id_*.pub — gera com ssh-keygen",
+            err=True,
+        )
+        return
+    pubkey_content = "\n".join(p.read_text().strip() for p in host_pubkeys)
+    subprocess.run(
+        ["multipass", "exec", name, "--", "bash", "-c", "mkdir -p ~/.ssh && chmod 700 ~/.ssh"],
+        check=False,
+    )
+    subprocess.run(
+        [
+            "multipass",
+            "exec",
+            name,
+            "--",
+            "bash",
+            "-c",
+            f"echo '{pubkey_content}' >> ~/.ssh/authorized_keys && chmod 600 ~/.ssh/authorized_keys",
+        ],
+        check=False,
+    )
+
+
+@vm_group.command(name="ssh")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option(
+    "--x11/--no-x11",
+    default=True,
+    help="Enable X11 forwarding (-X) for GUI apps inside VM",
+)
+@click.option(
+    "--cmd",
+    default=None,
+    help="Run a single command via SSH and exit (instead of interactive shell)",
+)
+def vm_ssh_cmd(name: str, x11: bool, cmd: str | None) -> None:
+    """SSH to VM with X11 forwarding (for GUI apps like browser, rclone OAuth)."""
+    _require_multipass()
+
+    if _vm_state(name) != "running":
+        click.echo(f"▶ Starting VM '{name}'...")
+        subprocess.run(["multipass", "start", name], check=False)
+        time.sleep(2)
+
+    ip = _get_vm_ip(name)
+    if not ip:
+        click.echo(f"[ERRO] Nao consegui pegar IP da VM '{name}'", err=True)
+        sys.exit(1)
+
+    click.echo(f"▶ Setting up SSH key in VM...")
+    _ensure_ssh_key_in_vm(name)
+
+    ssh_args = ["ssh"]
+    if x11:
+        ssh_args.append("-X")
+        click.echo(f"  X11 forwarding: ENABLED (apps GUI abrem no host)")
+    ssh_args += [
+        "-o",
+        "StrictHostKeyChecking=no",
+        "-o",
+        "UserKnownHostsFile=/dev/null",
+        "-o",
+        "LogLevel=ERROR",
+        f"ubuntu@{ip}",
+    ]
+    if cmd:
+        ssh_args.append(cmd)
+
+    click.echo(f"  Connecting to ubuntu@{ip}...")
+    click.echo()
+    subprocess.run(ssh_args)
+
+
+# ─── vm rclone-setup ────────────────────────────────────────────
+
+
+@vm_group.command(name="rclone-setup")
+@click.option("--name", default=DEFAULT_VM_NAME, show_default=True)
+@click.option(
+    "--remote",
+    default="gdrive",
+    show_default=True,
+    help="Name of rclone remote to create",
+)
+def vm_rclone_setup_cmd(name: str, remote: str) -> None:
+    """Setup rclone OAuth in VM without browser (uses host browser for auth)."""
+    _require_multipass()
+
+    if _vm_state(name) != "running":
+        subprocess.run(["multipass", "start", name], check=False)
+        time.sleep(2)
+
+    click.echo("""
+┌──────────────────────────────────────────────────────────────┐
+│   Rclone OAuth setup (token vem do browser do host)          │
+└──────────────────────────────────────────────────────────────┘
+
+Vou abrir o browser do HOST agora pra gerar o token Google Drive.
+""")
+
+    click.echo("▶ Executando 'rclone authorize drive' no host...")
+    click.echo("  (cole o token retornado quando solicitado)")
+    click.echo()
+
+    # rclone authorize abre browser no host e imprime token JSON
+    result = subprocess.run(
+        ["rclone", "authorize", "drive"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        click.echo(f"[ERRO] rclone authorize falhou: {result.stderr}", err=True)
+        sys.exit(1)
+
+    # Extrair o token JSON da saida
+    token = None
+    for line in result.stdout.splitlines():
+        line = line.strip()
+        if line.startswith("{") and "access_token" in line:
+            token = line
+            break
+
+    if not token:
+        click.echo("[ERRO] Nao consegui extrair token da saida do rclone", err=True)
+        click.echo("Saida:")
+        click.echo(result.stdout)
+        sys.exit(1)
+
+    click.echo(f"✓ Token gerado.")
+    click.echo(f"▶ Importando para VM '{name}' como remote '{remote}'...")
+
+    rclone_cmd = (
+        f"rclone config create {remote} drive "
+        f"config_is_local=false "
+        f"scope=drive "
+        f"token='{token}'"
+    )
+    subprocess.run(
+        ["multipass", "exec", name, "--", "bash", "-c", rclone_cmd],
+        check=True,
+    )
+
+    click.echo(f"\n▶ Validando...")
+    subprocess.run(
+        ["multipass", "exec", name, "--", "rclone", "lsf", f"{remote}:", "--max-depth", "1"],
+        check=False,
+    )
+
+    click.echo(f"\n✓ Rclone configurado na VM. Remote: {remote}:")
+
+
+# ─── vm gui-help ────────────────────────────────────────────────
+
+
+@vm_group.command(name="gui-help")
+def vm_gui_help_cmd() -> None:
+    """Show how to access GUI (X11 forwarding, browser for rclone OAuth)."""
+    click.echo("""
+┌──────────────────────────────────────────────────────────────┐
+│              Acesso a GUI dentro da VM Multipass             │
+└──────────────────────────────────────────────────────────────┘
+
+Opcao 1 — X11 forwarding via SSH (recomendado)
+───────────────────────────────────────────────
+  Pre-requisito: VM criada com --gui (instala xauth + firefox)
+
+  # 1. Pegar IP da VM
+  devopscli test vm info | grep IPv4
+
+  # 2. SSH com -X
+  ssh -X ubuntu@<IP-da-VM>
+
+  # 3. Rodar app GUI dentro da VM, janela abre no host
+  firefox &
+  rclone config              # abre browser pra OAuth
+
+Opcao 2 — rclone OAuth sem GUI (truque)
+────────────────────────────────────────
+  # No host (sua maquina), gera token:
+  rclone authorize "drive"
+  # Cola codigo no browser, retorna JSON token
+
+  # Na VM, importa direto:
+  rclone config create gdrive drive \\
+    config_is_local=false \\
+    token='<json colado>'
+
+Opcao 3 — Desktop completo (VNC)
+─────────────────────────────────
+  Mais complexo, raramente necessario. Veja:
+  https://multipass.run/docs
+
+""")