coderfleet 0.1.0__py3-none-any.whl

coderfleet/docker_ops.py

@@ -0,0 +1,385 @@
+"""
+docker_ops.py — Docker / compose 生命周期命令
+
+build / apply / up / down / restart / status / logs / enter / check-proxy
+"""
+from __future__ import annotations
+
+import os
+import shutil
+import subprocess
+from pathlib import Path
+from typing import Optional
+
+import click
+
+from coderfleet.config import load_config, parse_conf
+from coderfleet.compose import write_compose
+
+
+# ── helpers ───────────────────────────────────────────────────
+
+
+def _dc_prefix() -> list[str]:
+    """Detect docker compose v2 or legacy docker-compose."""
+    r = subprocess.run(["docker", "compose", "version"], capture_output=True, text=True)
+    if r.returncode == 0:
+        return ["docker", "compose"]
+    if shutil.which("docker-compose"):
+        return ["docker-compose"]
+    raise click.ClickException("找不到 docker compose 或 docker-compose，请先安装 Docker")
+
+
+def _dc(ws: Path) -> list[str]:
+    """Full docker compose command prefix including -f flag."""
+    return _dc_prefix() + ["-f", str(ws / "docker-compose.yml")]
+
+
+def _container_name(project_name: str, acc_type: str) -> str:
+    return f"{acc_type}-{project_name}"
+
+
+def _service_name(project_name: str, acc_type: str) -> str:
+    return f"{acc_type}-project-{project_name}"
+
+
+def _is_running(container: str) -> bool:
+    r = subprocess.run(
+        ["docker", "inspect", container, "--format", "{{.State.Running}}"],
+        capture_output=True, text=True,
+    )
+    return r.stdout.strip() == "true"
+
+
+def _get_accounts(ws: Path) -> dict[str, dict[str, str]]:
+    return {r["NAME"]: r for r in parse_conf(ws / "accounts.conf") if "NAME" in r}
+
+
+def _get_projects(ws: Path) -> list[dict[str, str]]:
+    return [p for p in parse_conf(ws / "projects.conf") if "NAME" in p and "ACCOUNT" in p]
+
+
+# ── commands ──────────────────────────────────────────────────
+
+
+@click.command("build")
+@click.option("--no-cache", is_flag=True, help="不使用构建缓存（强制完整重新构建）")
+@click.option("--pull", is_flag=True, help="强制拉取最新基础镜像")
+@click.option("--platform", "platform_override", default=None, metavar="PLATFORM",
+              help="覆盖 config.conf 中的 BUILD_PLATFORM（如 linux/amd64）")
+@click.option("--tag", "tag_override", default=None, metavar="TAG",
+              help="覆盖 config.conf 中的 IMAGE_TAG")
+@click.pass_context
+def cmd_build(
+    ctx: click.Context,
+    no_cache: bool,
+    pull: bool,
+    platform_override: Optional[str],
+    tag_override: Optional[str],
+) -> None:
+    """Build the Docker image."""
+    ws: Path = ctx.obj["workspace"]
+    cfg = load_config(ws)
+
+    dockerfile = ws / "Dockerfile"
+    if not dockerfile.exists():
+        raise click.ClickException(
+            f"找不到 Dockerfile（{dockerfile}），请先执行 coderfleet init"
+        )
+
+    image_name = cfg.get("IMAGE_NAME", "coderfleet")
+    image_tag = tag_override or cfg.get("IMAGE_TAG", "latest")
+    image = f"{image_name}:{image_tag}"
+    platform = platform_override or cfg.get("BUILD_PLATFORM", "linux/amd64")
+
+    click.echo(f"构建镜像 {image}（平台：{platform}）...")
+    if no_cache:
+        click.secho("  --no-cache：跳过所有缓存层", fg="yellow")
+    if pull:
+        click.secho("  --pull：强制拉取最新基础镜像", fg="yellow")
+    click.secho("首次构建约需 5~10 分钟，请耐心等待", dim=True)
+    click.echo()
+
+    cmd = [
+        "docker", "build",
+        "--platform", platform,
+        "--tag", image,
+        "--file", str(dockerfile),
+    ]
+    if no_cache:
+        cmd.append("--no-cache")
+    if pull:
+        cmd.append("--pull")
+    cmd.append(str(ws))
+
+    result = subprocess.run(cmd)
+
+    if result.returncode != 0:
+        raise click.ClickException("镜像构建失败")
+
+    click.echo()
+    click.secho(f"✓ 镜像构建完成：{image}", fg="green")
+
+
+@click.command("apply")
+@click.pass_context
+def cmd_apply(ctx: click.Context) -> None:
+    """Regenerate docker-compose.yml and restart all containers."""
+    ws: Path = ctx.obj["workspace"]
+
+    click.echo("生成 docker-compose.yml...")
+    write_compose(ws)
+    click.secho("✓ docker-compose.yml 已生成", fg="green")
+    click.echo()
+
+    click.echo("重启容器以应用新配置...")
+    dc = _dc(ws)
+    subprocess.run(dc + ["down", "--remove-orphans"])
+    result = subprocess.run(dc + ["up", "-d"])
+
+    if result.returncode != 0:
+        raise click.ClickException("容器启动失败")
+
+    click.echo()
+    click.secho("✓ 完成！使用 coderfleet status 查看状态", fg="green")
+
+
+@click.command("up")
+@click.pass_context
+def cmd_up(ctx: click.Context) -> None:
+    """Start all containers."""
+    ws: Path = ctx.obj["workspace"]
+    compose_file = ws / "docker-compose.yml"
+
+    if not compose_file.exists():
+        click.secho("docker-compose.yml 不存在，先生成...", fg="yellow")
+        write_compose(ws)
+        click.echo()
+
+    click.echo("启动所有容器...")
+    result = subprocess.run(_dc(ws) + ["up", "-d"])
+    if result.returncode == 0:
+        click.secho("✓ 启动完成", fg="green")
+    else:
+        raise click.ClickException("启动失败")
+
+
+@click.command("down")
+@click.pass_context
+def cmd_down(ctx: click.Context) -> None:
+    """Stop all containers."""
+    ws: Path = ctx.obj["workspace"]
+    click.echo("停止所有容器...")
+    subprocess.run(_dc(ws) + ["down"])
+    click.secho("✓ 已停止", fg="green")
+
+
+@click.command("restart")
+@click.pass_context
+def cmd_restart(ctx: click.Context) -> None:
+    """Restart all containers."""
+    ws: Path = ctx.obj["workspace"]
+    click.echo("重启...")
+    dc = _dc(ws)
+    subprocess.run(dc + ["down"])
+    result = subprocess.run(dc + ["up", "-d"])
+    if result.returncode == 0:
+        click.secho("✓ 重启完成", fg="green")
+    else:
+        raise click.ClickException("重启失败")
+
+
+@click.command("status")
+@click.pass_context
+def cmd_status(ctx: click.Context) -> None:
+    """Show container and image status."""
+    ws: Path = ctx.obj["workspace"]
+    cfg = load_config(ws)
+    accounts = _get_accounts(ws)
+    projects = _get_projects(ws)
+
+    click.echo()
+    click.echo("  ── 项目与容器状态 " + "─" * 50)
+
+    for p in projects:
+        pname = p["NAME"]
+        paccount = p.get("ACCOUNT", "")
+        acc = accounts.get(paccount, {})
+        acc_type = acc.get("TYPE", "")
+        ctr = _container_name(pname, acc_type) if acc_type else pname
+
+        if _is_running(ctr):
+            status = click.style("● 运行中", fg="green")
+        else:
+            status = click.style("○ 已停止", fg="red")
+
+        click.echo(f"  {pname:<20} [{acc_type:<6}] {status}  账号：{paccount}")
+
+    click.echo()
+    click.echo("  ── 代理中继 " + "─" * 56)
+    r = subprocess.run(
+        ["docker", "inspect", "coderfleet-proxy-relay",
+         "--format", "{{.State.Health.Status}}"],
+        capture_output=True, text=True,
+    )
+    health = r.stdout.strip() if r.returncode == 0 else "未运行"
+    if health == "healthy":
+        click.echo(f"  coderfleet-proxy-relay: {click.style(health, fg='green')}")
+    elif health == "starting":
+        click.echo(f"  coderfleet-proxy-relay: {click.style(health + '（启动中）', fg='yellow')}")
+    else:
+        click.echo(f"  coderfleet-proxy-relay: {click.style(health, fg='red')}")
+
+    click.echo()
+    click.echo("  ── 镜像信息 " + "─" * 56)
+    image = f"{cfg.get('IMAGE_NAME', 'coderfleet')}:{cfg.get('IMAGE_TAG', 'latest')}"
+    ri = subprocess.run(
+        ["docker", "image", "inspect", image,
+         "--format", "{{.Created}}\t{{.Size}}"],
+        capture_output=True, text=True,
+    )
+    if ri.returncode == 0 and ri.stdout.strip():
+        created, size_str = ri.stdout.strip().split("\t", 1)
+        try:
+            size_gb = int(size_str) / 1024 / 1024 / 1024
+            size_label = f"{size_gb:.1f} GB"
+        except ValueError:
+            size_label = size_str
+        click.echo(
+            f"  {image}: {click.style('已构建', fg='green')}"
+            f"（创建于 {created[:10]}，大小 {size_label}）"
+        )
+    else:
+        click.echo(
+            f"  {image}: {click.style('未构建', fg='red')}，请执行 coderfleet build"
+        )
+    click.echo()
+
+
+@click.command("logs")
+@click.argument("project", required=False)
+@click.pass_context
+def cmd_logs(ctx: click.Context, project: Optional[str]) -> None:
+    """Stream container logs (all containers, or a specific project)."""
+    ws: Path = ctx.obj["workspace"]
+    dc = _dc(ws)
+
+    if not project:
+        subprocess.run(dc + ["logs", "-f"])
+        return
+
+    accounts = _get_accounts(ws)
+    for p in _get_projects(ws):
+        if p["NAME"] != project:
+            continue
+        paccount = p.get("ACCOUNT", "")
+        acc_type = accounts.get(paccount, {}).get("TYPE", "")
+        svc = _service_name(project, acc_type)
+        subprocess.run(dc + ["logs", "-f", svc])
+        return
+
+    raise click.ClickException(f"项目 '{project}' 不在 projects.conf 中")
+
+
+@click.command("enter")
+@click.argument("project")
+@click.pass_context
+def cmd_enter(ctx: click.Context, project: str) -> None:
+    """Enter a container shell (replaces current process)."""
+    ws: Path = ctx.obj["workspace"]
+    accounts = _get_accounts(ws)
+
+    for p in _get_projects(ws):
+        if p["NAME"] != project:
+            continue
+        paccount = p.get("ACCOUNT", "")
+        acc_type = accounts.get(paccount, {}).get("TYPE", "")
+        ctr = _container_name(project, acc_type)
+
+        if not _is_running(ctr):
+            raise click.ClickException(f"容器 {ctr} 未运行，请先执行：coderfleet up")
+
+        click.secho(f"进入 {ctr}（类型：{acc_type}）...", dim=True)
+        os.execvp("docker", ["docker", "exec", "-it", ctr, "bash"])
+
+    raise click.ClickException(f"项目 '{project}' 不在 projects.conf 中")
+
+
+@click.command("check-proxy")
+@click.pass_context
+def cmd_check_proxy(ctx: click.Context) -> None:
+    """Check proxy connectivity for all containers."""
+    ws: Path = ctx.obj["workspace"]
+    cfg = load_config(ws)
+    accounts = _get_accounts(ws)
+    projects = _get_projects(ws)
+
+    relay_ip = cfg.get("RELAY_IP", "172.21.0.2")
+    relay_port = cfg.get("RELAY_LISTEN_PORT", "7890")
+
+    click.echo()
+    click.echo("  ── 代理连通性（应全部通）" + "─" * 43)
+    for p in projects:
+        pname = p["NAME"]
+        paccount = p.get("ACCOUNT", "")
+        acc_type = accounts.get(paccount, {}).get("TYPE", "")
+        ctr = _container_name(pname, acc_type)
+
+        click.echo(f"  {ctr:<28} → proxy-relay: ", nl=False)
+        if not _is_running(ctr):
+            click.secho("容器未运行", fg="yellow")
+            continue
+        r = subprocess.run(
+            ["docker", "exec", ctr, "sh", "-c", f"nc -z {relay_ip} {relay_port}"],
+            capture_output=True,
+        )
+        if r.returncode == 0:
+            click.secho("通", fg="green")
+        else:
+            click.secho("不通", fg="red")
+
+    click.echo()
+    click.echo("  ── 直连公网封锁（应全部封锁）" + "─" * 40)
+    for p in projects:
+        pname = p["NAME"]
+        paccount = p.get("ACCOUNT", "")
+        acc_type = accounts.get(paccount, {}).get("TYPE", "")
+        ctr = _container_name(pname, acc_type)
+
+        click.echo(f"  {ctr:<28} → 8.8.8.8:443: ", nl=False)
+        if not _is_running(ctr):
+            click.secho("容器未运行", fg="yellow")
+            continue
+        r = subprocess.run(
+            ["docker", "exec", ctr, "sh", "-c", "timeout 3 nc -z 8.8.8.8 443 2>/dev/null"],
+            capture_output=True,
+        )
+        if r.returncode == 0:
+            click.secho("可直连 ← 隔离未生效！", fg="red")
+        else:
+            click.secho("已封锁", fg="green")
+
+    click.echo()
+    click.echo("  ── 代理出口 IP " + "─" * 52)
+    for p in projects:
+        pname = p["NAME"]
+        paccount = p.get("ACCOUNT", "")
+        acc_type = accounts.get(paccount, {}).get("TYPE", "")
+        ctr = _container_name(pname, acc_type)
+
+        click.echo(f"  {ctr:<28} 出口 IP: ", nl=False)
+        if not _is_running(ctr):
+            click.secho("容器未运行", fg="yellow")
+            continue
+        r = subprocess.run(
+            ["docker", "exec", ctr, "sh", "-c",
+             "curl -s --max-time 6 https://api.ipify.org 2>/dev/null"],
+            capture_output=True, text=True,
+        )
+        ip = r.stdout.strip()
+        if ip:
+            click.secho(ip, fg="green")
+        else:
+            click.secho("请求失败", fg="yellow")
+
+    click.echo()

coderfleet/init_wizard.py

@@ -0,0 +1,227 @@
+"""
+init_wizard.py — coderfleet init 交互式初始化向导
+
+向导流程：
+1. 确认工作区路径
+2. 创建目录结构
+3. 配置代理与镜像参数 → 写入 config.conf
+4. 拷贝 Dockerfile / entrypoint.sh 到工作区
+5. 创建空白 accounts.conf / projects.conf（如不存在）
+6. 打印后续步骤提示
+"""
+from __future__ import annotations
+
+import importlib.resources
+import shutil
+import sys
+from pathlib import Path
+
+import click
+
+from coderfleet.config import ensure_workspace, get_workspace, load_config
+
+# ── 工具函数 ──────────────────────────────────────────────
+
+
+def _data_file(name: str) -> Path:
+    """返回 coderfleet/data/ 中打包资源文件的路径。"""
+    pkg = importlib.resources.files("coderfleet.data")
+    return Path(str(pkg / name))
+
+
+def _write_config(ws: Path, values: dict[str, str]) -> None:
+    """将 key=value 对写入 config.conf（整体覆盖）。"""
+    conf_path = ws / "config.conf"
+    lines = [
+        "# ============================================================",
+        "#  config.conf — CoderFleet 全局配置",
+        "#  修改后执行 coderfleet apply 生效",
+        "# ============================================================",
+        "",
+        "# ── 镜像配置 ──────────────────────────────────────────────",
+        f"IMAGE_NAME={values['IMAGE_NAME']}",
+        f"IMAGE_TAG={values['IMAGE_TAG']}",
+        f"BUILD_PLATFORM={values['BUILD_PLATFORM']}",
+        "",
+        "# ── 代理配置 ──────────────────────────────────────────────",
+        "# 宿主机代理地址（容器通过 host.docker.internal 访问宿主机）",
+        f"PROXY_HOST={values['PROXY_HOST']}",
+        f"PROXY_HTTP_PORT={values['PROXY_HTTP_PORT']}",
+        f"PROXY_SOCKS5_PORT={values['PROXY_SOCKS5_PORT']}",
+        "",
+        "# ── 内部网络配置 ──────────────────────────────────────────",
+        f"INTERNAL_SUBNET={values['INTERNAL_SUBNET']}",
+        f"RELAY_IP={values['RELAY_IP']}",
+        f"RELAY_LISTEN_PORT={values['RELAY_LISTEN_PORT']}",
+        f"RELAY_IMAGE={values['RELAY_IMAGE']}",
+        "",
+    ]
+    conf_path.write_text("\n".join(lines), encoding="utf-8")
+
+
+def _create_empty_conf(path: Path, example_name: str) -> None:
+    """拷贝 example 文件作为初始 conf（保留注释，去掉示例数据行）。"""
+    if path.exists():
+        return
+    src = _data_file(example_name)
+    if src.exists():
+        # 复制 example，但去掉实际数据行（非注释/非空的行）
+        lines = src.read_text(encoding="utf-8").splitlines(keepends=True)
+        kept = []
+        for line in lines:
+            stripped = line.strip()
+            # 保留空行和注释行；去掉 example 末尾的示例数据行
+            if not stripped or stripped.startswith("#"):
+                kept.append(line)
+        path.write_text("".join(kept), encoding="utf-8")
+    else:
+        path.write_text("", encoding="utf-8")
+
+
+# ── 向导主体 ──────────────────────────────────────────────
+
+def run_init_wizard(ws: Path) -> None:
+    """交互式初始化工作区。"""
+    click.echo()
+    click.secho("CoderFleet 初始化向导", bold=True)
+    click.echo("=" * 40)
+
+    # ── 步骤 1：确认工作区 ──────────────────────────────
+    click.echo(f"\n工作区目录：{click.style(str(ws), fg='cyan')}")
+    if ws.exists() and (ws / "config.conf").exists():
+        if not click.confirm("config.conf 已存在，是否重新配置？", default=False):
+            click.echo("跳过配置，仅补全缺失目录和文件。")
+            ensure_workspace(ws)
+            _copy_runtime_files(ws)
+            _init_empty_confs(ws)
+            _print_next_steps(ws)
+            return
+
+    ensure_workspace(ws)
+    click.secho("✓ 工作区目录已就绪", fg="green")
+
+    # ── 步骤 2：代理配置 ────────────────────────────────
+    click.echo("\n── 代理配置 ──────────────────────────────────────")
+    click.echo("容器通过宿主机代理访问互联网（Clash / v2ray / sing-box 等）。")
+
+    proxy_host = click.prompt(
+        "代理地址",
+        default="host.docker.internal",
+    )
+    proxy_http_port = click.prompt(
+        "代理 HTTP 端口",
+        default="10808",
+    )
+    same_port = click.confirm(
+        f"SOCKS5 端口与 HTTP 端口相同（{proxy_http_port}）",
+        default=True,
+    )
+    proxy_socks5_port = proxy_http_port if same_port else click.prompt(
+        "代理 SOCKS5 端口",
+        default="10808",
+    )
+
+    # ── 步骤 3：镜像配置 ────────────────────────────────
+    click.echo("\n── Docker 镜像配置 ───────────────────────────────")
+
+    import platform as _platform
+    default_platform = (
+        "linux/arm64" if _platform.machine() in ("arm64", "aarch64") else "linux/amd64"
+    )
+
+    image_name = click.prompt("镜像名称", default="coderfleet")
+    image_tag = click.prompt("镜像标签", default="latest")
+    build_platform = click.prompt("构建平台", default=default_platform)
+
+    # ── 步骤 4：写入 config.conf ─────────────────────────
+    relay_listen_port = proxy_http_port   # relay 监听同一端口转发给宿主机
+    values = {
+        "IMAGE_NAME":       image_name,
+        "IMAGE_TAG":        image_tag,
+        "BUILD_PLATFORM":   build_platform,
+        "PROXY_HOST":       proxy_host,
+        "PROXY_HTTP_PORT":  proxy_http_port,
+        "PROXY_SOCKS5_PORT": proxy_socks5_port,
+        "INTERNAL_SUBNET":  "172.21.0.0/16",
+        "RELAY_IP":         "172.21.0.2",
+        "RELAY_LISTEN_PORT": relay_listen_port,
+        "RELAY_IMAGE":      "gogost/gost:3",
+    }
+    _write_config(ws, values)
+    click.secho("✓ config.conf 已生成", fg="green")
+
+    # ── 步骤 5：拷贝运行时文件 ───────────────────────────
+    _copy_runtime_files(ws)
+
+    # ── 步骤 6：初始化 conf 文件 ─────────────────────────
+    _init_empty_confs(ws)
+
+    _print_next_steps(ws)
+
+
+def _copy_runtime_files(ws: Path) -> None:
+    """将 Dockerfile、entrypoint.sh 和 scripts/ 拷贝到工作区（已存在则跳过）。"""
+    for name in ("Dockerfile", "entrypoint.sh"):
+        dst = ws / name
+        if dst.exists():
+            click.echo(f"  {name} 已存在，跳过")
+            continue
+        src = _data_file(name)
+        if src.exists():
+            shutil.copy2(str(src), str(dst))
+            if name.endswith(".sh"):
+                dst.chmod(0o755)
+            click.secho(f"✓ {name} 已拷贝", fg="green")
+        else:
+            click.secho(f"警告：找不到内置 {name}，请手动放置", fg="yellow")
+
+    # scripts/ — Dockerfile 构建时需要的辅助脚本
+    scripts_src = Path(str(_data_file("scripts")))
+    scripts_dst = ws / "scripts"
+    if scripts_src.is_dir():
+        scripts_dst.mkdir(exist_ok=True)
+        for f in scripts_src.iterdir():
+            dst = scripts_dst / f.name
+            if dst.exists():
+                continue
+            shutil.copy2(str(f), str(dst))
+        click.secho("✓ scripts/ 已拷贝", fg="green")
+
+
+def _init_empty_confs(ws: Path) -> None:
+    """创建空白的 accounts.conf 和 projects.conf（如不存在）。"""
+    for fname, example in (
+        ("accounts.conf", "accounts.conf.example"),
+        ("projects.conf", "projects.conf.example"),
+    ):
+        path = ws / fname
+        if not path.exists():
+            _create_empty_conf(path, example)
+            click.secho(f"✓ {fname} 已创建（空白，仅含注释）", fg="green")
+
+
+def _print_next_steps(ws: Path) -> None:
+    """打印后续操作提示。"""
+    click.echo()
+    click.secho("初始化完成！后续步骤：", bold=True)
+    click.echo()
+    click.echo(f"  1. 构建 Docker 镜像（首次约 5-10 分钟）：")
+    click.secho(f"       coderfleet build", fg="cyan")
+    click.echo(f"  2. 添加账号：")
+    click.secho(f"       coderfleet account add <名称> TYPE=claude", fg="cyan")
+    click.echo(f"  3. 添加项目：")
+    click.secho(f"       coderfleet project add <名称> <账号名> ~/projects/myproject", fg="cyan")
+    click.echo(f"  4. 应用配置并启动容器：")
+    click.secho(f"       coderfleet apply", fg="cyan")
+    click.echo(f"  5. 登录账号：")
+    click.secho(f"       coderfleet login <账号名>", fg="cyan")
+    click.echo(f"  6. 启动 Web UI：")
+    click.secho(f"       coderfleet server", fg="cyan")
+    click.echo()
+    click.echo(f"工作区位置：{ws}")
+    if ws != Path.home() / ".coderfleet":
+        click.echo(
+            f"提示：非默认工作区，运行 coderfleet 命令时需设置环境变量：\n"
+            f"  export CODERFLEET_WORKSPACE={ws}"
+        )
+    click.echo()