chore: move installation scripts from docs repo to here

scripts/install/maixcam/install_picoclaw.py

@@ -0,0 +1,247 @@
+import argparse
+import os
+import platform
+import signal
+import shutil
+import subprocess
+import tarfile
+import urllib.request
+from pathlib import Path
+
+
+DEFAULT_ACTION = "install"
+START_PICOCLAW = True
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description="Install or uninstall picoclaw.")
+    parser.add_argument(
+        "--action",
+        choices=["install", "uninstall"],
+        default=DEFAULT_ACTION,
+        help=f"Action to perform, defaults to '{DEFAULT_ACTION}'.",
+    )
+    return parser.parse_args()
+
+
+def detect_arch() -> str:
+    machine = platform.machine().lower()
+
+    if machine in {"aarch64", "arm64"}:
+        return "arm64"
+    if machine in {"riscv64"}:
+        return "riscv64"
+
+    return f"unknown ({machine})"
+
+
+def ensure_picoclaw_dir() -> Path:
+    target_dir = Path("/root/picoclaw")
+    target_dir.mkdir(parents=True, exist_ok=True)
+    return target_dir
+
+
+def get_download_url(arch: str) -> str:
+    if arch == "riscv64":
+        return "https://picoclaw-downloads.tos-cn-beijing.volces.com/latest/picoclaw_Linux_riscv64.tar.gz"
+    if arch == "arm64":
+        return "https://picoclaw-downloads.tos-cn-beijing.volces.com/latest/picoclaw_aarch64.deb"
+
+    raise ValueError(f"Unsupported architecture: {arch}")
+
+
+def download_package(url: str, target_dir: Path) -> Path:
+    file_name = Path(url).name
+    target_path = target_dir / file_name
+    urllib.request.urlretrieve(url, target_path)
+    return target_path
+
+
+def install_arm64(deb_file: Path) -> None:
+    subprocess.run(["dpkg", "-i", str(deb_file)], check=True)
+
+
+def find_file_recursive(root: Path, file_name: str) -> Path:
+    matches = list(root.rglob(file_name))
+    if not matches:
+        raise FileNotFoundError(f"Required file not found: {file_name}")
+    return matches[0]
+
+
+def install_riscv64(tar_file: Path, work_dir: Path) -> None:
+    install_dir = Path("/opt/picoclaw")
+    if install_dir.exists():
+        shutil.rmtree(install_dir)
+    install_dir.mkdir(parents=True, exist_ok=True)
+
+    with tarfile.open(tar_file, "r:gz") as tar:
+        tar.extractall(path=install_dir)
+
+    for binary_name in ["picoclaw", "picoclaw-launcher", "picoclaw-launcher-tui"]:
+        installed_binary = find_file_recursive(install_dir, binary_name)
+        installed_binary.chmod(0o755)
+
+        link_path = Path("/usr/bin") / binary_name
+        if link_path.exists() or link_path.is_symlink():
+            link_path.unlink()
+        link_path.symlink_to(installed_binary)
+
+
+def cleanup_dir(target_dir: Path) -> None:
+    if target_dir.exists():
+        shutil.rmtree(target_dir)
+
+
+def stop_picoclaw() -> None:
+    process_names = {"picoclaw", "picoclaw-launcher", "picoclaw-launcher-tui"}
+    stopped_any = False
+    self_pid = os.getpid()
+
+    try:
+        ps_result = subprocess.run(["ps", "-eo", "pid=,args="], capture_output=True, text=True, check=False)
+    except FileNotFoundError:
+        print("ps command not found, skip stopping picoclaw processes.")
+        return
+
+    if ps_result.returncode != 0:
+        print("Failed to list processes with ps, skip stopping picoclaw processes.")
+        return
+
+    for line in ps_result.stdout.splitlines():
+        line = line.strip()
+        if not line:
+            continue
+
+        parts = line.split(maxsplit=1)
+        if not parts:
+            continue
+
+        try:
+            pid = int(parts[0])
+        except ValueError:
+            continue
+
+        if pid == self_pid:
+            continue
+
+        cmdline = parts[1] if len(parts) > 1 else ""
+        executable = os.path.basename(cmdline.split(maxsplit=1)[0]) if cmdline else ""
+        if executable not in process_names and not any(f"/{name}" in cmdline for name in process_names):
+            continue
+
+        try:
+            os.kill(pid, signal.SIGTERM)
+            stopped_any = True
+        except (ProcessLookupError, PermissionError):
+            continue
+
+    if stopped_any:
+        print("Stopped picoclaw related processes.")
+    else:
+        print("No running picoclaw related process found.")
+
+
+def uninstall_arm64() -> None:
+    result = subprocess.run(["dpkg", "-l"], capture_output=True, text=True, check=True)
+    package_names = []
+
+    for line in result.stdout.splitlines():
+        if line.startswith("ii") and "picoclaw" in line:
+            package_names.append(line.split()[1])
+
+    if not package_names:
+        print("No installed picoclaw package found for arm64.")
+        return
+
+    for package_name in package_names:
+        print(f"Removing package: {package_name}")
+        subprocess.run(["dpkg", "-r", package_name], check=True)
+
+
+def uninstall_riscv64() -> None:
+    for binary_name in ["picoclaw", "picoclaw-launcher", "picoclaw-launcher-tui"]:
+        link_path = Path("/usr/bin") / binary_name
+        if link_path.exists() or link_path.is_symlink():
+            link_path.unlink()
+            print(f"Removed link: {link_path}")
+
+    install_dir = Path("/opt/picoclaw")
+    if install_dir.exists():
+        shutil.rmtree(install_dir)
+        print(f"Removed directory: {install_dir}")
+
+
+def remove_picoclaw_user_dir() -> None:
+    user_dir = Path("/root/.picoclaw")
+    if user_dir.exists():
+        shutil.rmtree(user_dir)
+        print(f"Removed directory: {user_dir}")
+
+
+def start_picoclaw_launcher() -> None:
+    launcher_path = shutil.which("picoclaw-launcher")
+    if not launcher_path:
+        print("picoclaw-launcher not found in PATH, skip start.")
+        return
+
+    env = dict(os.environ)
+    env["HOME"] = "/root"
+    env.setdefault("NO_COLOR", "1")
+
+    subprocess.Popen(
+        [launcher_path, "-no-browser", "-public"],
+        env=env,
+        stdin=subprocess.DEVNULL,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+        start_new_session=True,
+        close_fds=True,
+    )
+    print("Started picoclaw-launcher in background.")
+
+
+if __name__ == "__main__":
+    args = parse_args()
+    action = args.action
+
+    arch = detect_arch()
+    print(f"Arch: {arch}")
+    print(f"Action: {action}")
+
+    if action == "install":
+        if arch.startswith("unknown"):
+            print("Unsupported arch, exiting.")
+            raise SystemExit(1)
+
+        picoclaw_dir = ensure_picoclaw_dir()
+        print(f"Ensured directory exists: {picoclaw_dir}")
+
+        download_url = get_download_url(arch)
+        print(f"Downloading from: {download_url}")
+
+        downloaded_file = download_package(download_url, picoclaw_dir)
+        print(f"Downloaded file: {downloaded_file}")
+
+        if arch == "arm64":
+            print("Installing with dpkg...")
+            install_arm64(downloaded_file)
+        else:
+            print("Extracting all files to /opt/picoclaw and creating symlinks...")
+            install_riscv64(downloaded_file, picoclaw_dir)
+
+        if START_PICOCLAW:
+            start_picoclaw_launcher()
+
+        cleanup_dir(picoclaw_dir)
+        print(f"Cleaned up directory: {picoclaw_dir}")
+    else:
+        stop_picoclaw()
+
+        if arch == "arm64":
+            uninstall_arm64()
+        elif arch == "riscv64":
+            uninstall_riscv64()
+        else:
+            print("Unknown architecture, skipping uninstallation.")
+
+        remove_picoclaw_user_dir()