Add interactive navctl menu for install, uninstall, and update.
Provide navctl.py with a numbered menu, persist install path in config, and support curl bootstrap without pre-cloning the repo. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
dekun
@@ -0,0 +1,317 @@
|
||||
#!/usr/bin/env python3
|
||||
"""LocalNav 交互式部署管理：安装 / 卸载 / 更新。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import platform
|
||||
import shutil
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from nav_common import (
|
||||
APP_NAME,
|
||||
MIN_PYTHON,
|
||||
default_install_dir,
|
||||
default_repo,
|
||||
is_localnav_project,
|
||||
load_install_config,
|
||||
pm2_app_status,
|
||||
resolve_project_dir,
|
||||
save_install_config,
|
||||
service_status_text,
|
||||
stop_service,
|
||||
)
|
||||
|
||||
SCRIPT_DIR = Path(__file__).resolve().parent
|
||||
|
||||
|
||||
def _ok(msg: str) -> None:
|
||||
print(f"[OK] {msg}")
|
||||
|
||||
|
||||
def _warn(msg: str) -> None:
|
||||
print(f"[WARN] {msg}")
|
||||
|
||||
|
||||
def _fail(msg: str) -> None:
|
||||
print(f"[FAIL] {msg}", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def _pause() -> None:
|
||||
try:
|
||||
input("\n按回车返回菜单...")
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print()
|
||||
|
||||
|
||||
def prompt_value(label: str, default: str) -> str:
|
||||
try:
|
||||
raw = input(f"{label} [{default}]: ").strip()
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print()
|
||||
_fail("已取消")
|
||||
return raw or default
|
||||
|
||||
|
||||
def confirm(prompt: str, *, default_no: bool = True) -> bool:
|
||||
suffix = " [y/N]: " if default_no else " [Y/n]: "
|
||||
try:
|
||||
raw = input(prompt + suffix).strip().lower()
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print()
|
||||
return False
|
||||
if not raw:
|
||||
return not default_no
|
||||
return raw in ("y", "yes", "是")
|
||||
|
||||
|
||||
def check_python() -> str:
|
||||
ver = sys.version_info
|
||||
if ver < MIN_PYTHON:
|
||||
_fail(f"需要 Python {MIN_PYTHON[0]}.{MIN_PYTHON[1]}+，当前 {ver.major}.{ver.minor}")
|
||||
return sys.executable
|
||||
|
||||
|
||||
def check_git() -> str:
|
||||
git = shutil.which("git")
|
||||
if not git:
|
||||
_fail("未检测到 git，请先安装：sudo apt install -y git")
|
||||
return git
|
||||
|
||||
|
||||
def clone_project(git_exe: str, repo: str, dest: Path) -> None:
|
||||
if dest.exists() and not dest.is_dir():
|
||||
_fail(f"安装路径已存在且不是目录: {dest}")
|
||||
|
||||
if is_localnav_project(dest):
|
||||
_ok(f"目录已是 LocalNav 项目: {dest}")
|
||||
return
|
||||
|
||||
if dest.exists():
|
||||
if (dest / ".git").is_dir():
|
||||
_fail(f"目录已存在且不是 LocalNav 项目: {dest}")
|
||||
_fail(f"目录已存在: {dest}")
|
||||
|
||||
dest.parent.mkdir(parents=True, exist_ok=True)
|
||||
_ok(f"正在克隆 {repo} -> {dest}")
|
||||
subprocess.run([git_exe, "clone", repo, str(dest)], check=True)
|
||||
if not is_localnav_project(dest):
|
||||
_fail("克隆完成但未找到 app.py，请检查仓库地址")
|
||||
|
||||
|
||||
def git_pull(project_dir: Path) -> None:
|
||||
git = check_git()
|
||||
if not (project_dir / ".git").is_dir():
|
||||
_fail(f"目录不是 Git 仓库，无法更新: {project_dir}")
|
||||
_ok(f"正在拉取最新代码: {project_dir}")
|
||||
subprocess.run([git, "-C", str(project_dir), "pull", "--ff-only"], check=True)
|
||||
|
||||
|
||||
def run_deploy(project_dir: Path, py_exe: str) -> None:
|
||||
deploy_script = project_dir / "scripts" / "deploy.py"
|
||||
if not deploy_script.is_file():
|
||||
_fail(f"未找到部署脚本: {deploy_script}")
|
||||
subprocess.run([py_exe, str(deploy_script)], cwd=project_dir, check=True)
|
||||
|
||||
|
||||
def action_install(*, repo: str = "", dest: str = "", yes: bool = False) -> None:
|
||||
print()
|
||||
print("=" * 50)
|
||||
print(" 一键部署安装")
|
||||
print("=" * 50)
|
||||
|
||||
py_exe = check_python()
|
||||
git_exe = check_git()
|
||||
|
||||
repo = repo.strip() or default_repo()
|
||||
if yes:
|
||||
project_dir = Path(dest).expanduser() if dest.strip() else default_install_dir()
|
||||
else:
|
||||
repo = prompt_value("Git 仓库地址", repo)
|
||||
dest_str = dest.strip() or prompt_value("安装目录", str(default_install_dir()))
|
||||
project_dir = Path(dest_str).expanduser()
|
||||
|
||||
project_dir = project_dir.resolve()
|
||||
print()
|
||||
print(f" 仓库: {repo}")
|
||||
print(f" 目录: {project_dir}")
|
||||
print()
|
||||
|
||||
clone_project(git_exe, repo, project_dir)
|
||||
run_deploy(project_dir, py_exe)
|
||||
save_install_config(project_dir, repo)
|
||||
_ok(f"安装完成: {project_dir}")
|
||||
|
||||
|
||||
def action_update(*, dest: str = "") -> None:
|
||||
print()
|
||||
print("=" * 50)
|
||||
print(" 更新")
|
||||
print("=" * 50)
|
||||
|
||||
project_dir = resolve_project_dir(dest)
|
||||
if not project_dir:
|
||||
dest_str = dest.strip() or prompt_value("安装目录", str(default_install_dir()))
|
||||
project_dir = Path(dest_str).expanduser().resolve()
|
||||
if not is_localnav_project(project_dir):
|
||||
_fail(f"未找到 LocalNav 安装: {project_dir}")
|
||||
|
||||
py_exe = check_python()
|
||||
git_pull(project_dir)
|
||||
run_deploy(project_dir, py_exe)
|
||||
cfg = load_install_config()
|
||||
save_install_config(project_dir, cfg.get("repo") or default_repo())
|
||||
_ok(f"更新完成: {project_dir}")
|
||||
|
||||
|
||||
def action_uninstall(*, dest: str = "") -> None:
|
||||
print()
|
||||
print("=" * 50)
|
||||
print(" 一键卸载")
|
||||
print("=" * 50)
|
||||
|
||||
project_dir = resolve_project_dir(dest)
|
||||
if not project_dir:
|
||||
dest_str = dest.strip() or prompt_value("安装目录", str(default_install_dir()))
|
||||
candidate = Path(dest_str).expanduser().resolve()
|
||||
if candidate.exists() and is_localnav_project(candidate):
|
||||
project_dir = candidate
|
||||
elif not candidate.exists():
|
||||
_warn(f"目录不存在: {candidate}")
|
||||
project_dir = None
|
||||
else:
|
||||
_fail(f"目录不是 LocalNav 项目: {candidate}")
|
||||
|
||||
print()
|
||||
if project_dir:
|
||||
print(f" 安装目录: {project_dir}")
|
||||
print(f" 服务状态: {service_status_text(project_dir)}")
|
||||
else:
|
||||
print(" 未检测到已安装的 LocalNav 项目")
|
||||
print()
|
||||
|
||||
if not confirm("确认停止 nav-site 服务？", default_no=True):
|
||||
_warn("已取消卸载")
|
||||
return
|
||||
|
||||
actions = stop_service()
|
||||
if actions:
|
||||
for item in actions:
|
||||
_ok(item)
|
||||
else:
|
||||
_warn("未发现 pm2 / systemd 中的 nav-site 进程")
|
||||
|
||||
if not project_dir:
|
||||
_ok("卸载完成（仅停止服务）")
|
||||
return
|
||||
|
||||
if confirm("是否删除整个安装目录（含数据库，不可恢复）？", default_no=True):
|
||||
if not confirm("再次确认：删除 " + str(project_dir) + " ？", default_no=True):
|
||||
_warn("已保留安装目录")
|
||||
else:
|
||||
shutil.rmtree(project_dir)
|
||||
_ok(f"已删除目录: {project_dir}")
|
||||
cfg = load_install_config()
|
||||
if cfg.get("dir") == str(project_dir):
|
||||
from nav_common import config_path
|
||||
|
||||
try:
|
||||
config_path().unlink(missing_ok=True)
|
||||
except OSError:
|
||||
pass
|
||||
else:
|
||||
_warn(f"已保留安装目录: {project_dir}")
|
||||
|
||||
_ok("卸载完成")
|
||||
|
||||
|
||||
def render_menu(project_dir: Path | None) -> None:
|
||||
print()
|
||||
print("=" * 50)
|
||||
print(" LocalNav 部署管理")
|
||||
print("=" * 50)
|
||||
if project_dir:
|
||||
print(f" 安装目录: {project_dir}")
|
||||
else:
|
||||
print(f" 安装目录: 未安装（默认 {default_install_dir()}）")
|
||||
print(f" 服务状态: {service_status_text(project_dir)}")
|
||||
if pm2_app_status():
|
||||
print(f" PM2 进程: {APP_NAME}")
|
||||
print("-" * 50)
|
||||
print(" 1 一键部署安装")
|
||||
print(" 2 一键卸载")
|
||||
print(" 3 更新")
|
||||
print(" 0 退出")
|
||||
print("-" * 50)
|
||||
|
||||
|
||||
def interactive_menu() -> None:
|
||||
print(f"LocalNav 部署管理 · {platform.system()} {platform.release()}")
|
||||
while True:
|
||||
project_dir = resolve_project_dir()
|
||||
render_menu(project_dir)
|
||||
try:
|
||||
choice = input("请选择: ").strip()
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print("\n再见。")
|
||||
return
|
||||
|
||||
if choice == "1":
|
||||
try:
|
||||
action_install()
|
||||
except subprocess.CalledProcessError as exc:
|
||||
_fail(f"安装失败: {exc}")
|
||||
_pause()
|
||||
elif choice == "2":
|
||||
try:
|
||||
action_uninstall()
|
||||
except subprocess.CalledProcessError as exc:
|
||||
_fail(f"卸载失败: {exc}")
|
||||
_pause()
|
||||
elif choice == "3":
|
||||
try:
|
||||
action_update()
|
||||
except subprocess.CalledProcessError as exc:
|
||||
_fail(f"更新失败: {exc}")
|
||||
_pause()
|
||||
elif choice == "0":
|
||||
print("再见。")
|
||||
return
|
||||
else:
|
||||
_warn("无效选项，请输入 0-3")
|
||||
_pause()
|
||||
|
||||
|
||||
def parse_args() -> argparse.Namespace:
|
||||
parser = argparse.ArgumentParser(description="LocalNav 部署管理")
|
||||
parser.add_argument(
|
||||
"command",
|
||||
nargs="?",
|
||||
choices=("install", "uninstall", "update", "menu"),
|
||||
help="直接执行指定操作；省略则进入交互菜单",
|
||||
)
|
||||
parser.add_argument("--repo", default="", help="Git 仓库地址")
|
||||
parser.add_argument("--dir", default="", help="安装目录")
|
||||
parser.add_argument("-y", "--yes", action="store_true", help="安装时使用默认路径，减少提问")
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def main() -> None:
|
||||
args = parse_args()
|
||||
if args.command == "install":
|
||||
action_install(repo=args.repo, dest=args.dir, yes=args.yes)
|
||||
elif args.command == "uninstall":
|
||||
action_uninstall(dest=args.dir)
|
||||
elif args.command == "update":
|
||||
action_update(dest=args.dir)
|
||||
elif args.command == "menu":
|
||||
interactive_menu()
|
||||
else:
|
||||
interactive_menu()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user