coderfleet 0.1.0__py3-none-any.whl

coderfleet/server/terminal.py

@@ -0,0 +1,129 @@
+from __future__ import annotations
+
+import asyncio
+import fcntl
+import os
+import pty
+import signal
+import struct
+import termios
+from dataclasses import dataclass
+
+from coderfleet.server import docker_mgr
+from coderfleet.server.models import Account, Project
+from coderfleet.server.scheduler import Scheduler
+
+
+@dataclass(frozen=True)
+class TerminalTarget:
+    project: Project
+    account: Account
+    container_name: str
+    container_workdir: str
+    command: list[str]
+
+
+def build_terminal_command(container_name: str, container_workdir: str) -> list[str]:
+    return ["docker", "exec", "-it", "-w", container_workdir, container_name, "bash", "-l"]
+
+
+def resolve_terminal_target(scheduler: Scheduler, project_name: str) -> TerminalTarget:
+    project = scheduler.find_project_by_name(project_name)
+    if project is None:
+        raise ValueError(f"项目 '{project_name}' 不存在")
+
+    account = next((acc for acc in scheduler.get_accounts() if acc.name == project.account), None)
+    if account is None:
+        raise ValueError(f"账号 '{project.account}' 不存在")
+
+    container_name = project.container_name(account.type)
+    if not docker_mgr.is_container_running(container_name):
+        raise RuntimeError(f"容器 {container_name} 未运行")
+
+    container_workdir = scheduler.container_workdir_for_project(project, project.path)
+    command = build_terminal_command(container_name, container_workdir)
+    return TerminalTarget(
+        project=project,
+        account=account,
+        container_name=container_name,
+        container_workdir=container_workdir,
+        command=command,
+    )
+
+
+def resize_pty(fd: int, cols: int, rows: int) -> None:
+    if cols <= 0 or rows <= 0:
+        return
+    size = struct.pack("HHHH", rows, cols, 0, 0)
+    fcntl.ioctl(fd, termios.TIOCSWINSZ, size)
+
+
+class TerminalSession:
+    def __init__(self, command: list[str], project_name: str):
+        self.command = command
+        self.project_name = project_name
+        self.master_fd: int | None = None
+        self.child_pid: int | None = None
+
+    def start(self) -> None:
+        if self.master_fd is not None or self.child_pid is not None:
+            raise RuntimeError("terminal session already started")
+
+        child_pid, master_fd = pty.fork()
+        if child_pid == 0:
+            os.execvp(self.command[0], self.command)
+        self.child_pid = child_pid
+        self.master_fd = master_fd
+        os.set_blocking(master_fd, False)
+
+    async def read(self) -> bytes:
+        if self.master_fd is None:
+            return b""
+        loop = asyncio.get_running_loop()
+        return await loop.run_in_executor(None, self._read_once)
+
+    def _read_once(self) -> bytes:
+        if self.master_fd is None:
+            return b""
+        try:
+            return os.read(self.master_fd, 4096)
+        except BlockingIOError:
+            return b""
+        except OSError:
+            return b""
+
+    def write(self, data: str) -> None:
+        if self.master_fd is None or not data:
+            return
+        os.write(self.master_fd, data.encode("utf-8", errors="ignore"))
+
+    def resize(self, cols: int, rows: int) -> None:
+        if self.master_fd is None:
+            return
+        resize_pty(self.master_fd, cols, rows)
+
+    def close(self) -> None:
+        child_pid = self.child_pid
+        master_fd = self.master_fd
+        self.child_pid = None
+        self.master_fd = None
+
+        if child_pid is not None:
+            try:
+                os.kill(child_pid, signal.SIGHUP)
+            except ProcessLookupError:
+                pass
+            except OSError:
+                pass
+            try:
+                os.waitpid(child_pid, os.WNOHANG)
+            except ChildProcessError:
+                pass
+            except OSError:
+                pass
+
+        if master_fd is not None:
+            try:
+                os.close(master_fd)
+            except OSError:
+                pass

coderfleet/task_cmds.py

@@ -0,0 +1,311 @@
+"""
+task_cmds.py — coderfleet task 子命令
+
+通过 HTTP 调用 CoderFleet 调度服务 API，需先执行 coderfleet server 启动服务。
+CODERFLEET_API 环境变量可覆盖服务地址（默认 http://localhost:8765）。
+"""
+from __future__ import annotations
+
+import os
+import sys
+from pathlib import Path
+from typing import Optional
+
+import click
+import httpx
+
+
+# ── 工具函数 ──────────────────────────────────────────────
+
+def _api_base() -> str:
+    return os.environ.get("CODERFLEET_API", "http://localhost:8765").rstrip("/")
+
+
+def _require_server() -> str:
+    """检查服务是否在运行，返回 api base url。失败则 abort。"""
+    api = _api_base()
+    try:
+        httpx.get(f"{api}/api/health", timeout=3).raise_for_status()
+    except Exception:
+        raise click.ClickException(
+            f"CoderFleet 服务未启动，请先执行：coderfleet server\n"
+            f"（或设置 CODERFLEET_API=<地址> 指向远端服务）"
+        )
+    return api
+
+
+def _resolve_project_field(prefer_project: str) -> tuple[str, str]:
+    """
+    判断 --project 值是路径还是项目名：
+    - 绝对路径 / ~ 开头 / ./ 开头 → project 字段（路径匹配）
+    - 其他字符串              → project_name 字段（名称匹配）
+    返回 (field_name, value)
+    """
+    expanded = os.path.expanduser(prefer_project)
+    if (
+        os.path.isabs(expanded)
+        or prefer_project.startswith("~")
+        or prefer_project.startswith("./")
+    ):
+        return "project", str(Path(expanded).resolve())
+    return "project_name", prefer_project
+
+
+_STATUS_STYLE = {
+    "running":   ("green",  "● 运行中"),
+    "pending":   ("yellow", "○ 排队中"),
+    "scheduled": ("blue",   "◷ 定时中"),
+    "done":      ("cyan",   "✓ 完成  "),
+    "failed":    ("red",    "✗ 失败  "),
+    "killed":    ("yellow", "⊘ 已终止"),
+}
+
+
+# ── Click 命令组 ──────────────────────────────────────────
+
+@click.group("task")
+def task_group() -> None:
+    """Task management commands (requires server running)."""
+    pass
+
+
+# ── task run ──────────────────────────────────────────────
+
+@task_group.command("run")
+@click.argument("prompt")
+@click.option("--project", "prefer_project", default=None, help="项目名称或宿主机路径")
+@click.option("--account", default=None, help="指定账号名")
+@click.option("--type", "prefer_type", default=None,
+              type=click.Choice(["codex", "claude"]), help="指定账号类型")
+@click.option("--auto", is_flag=True, help="全自动模式（跳过权限确认）")
+@click.option("--conversation", "conversation_id", default=None, help="续接已有任务链 ID")
+@click.option("--new-chain", "conversation_name", default=None, help="新建任务链并命名")
+@click.option("--at", "execute_at", default=None, help="定时执行时间（ISO 8601）")
+def cmd_task_run(
+    prompt: str,
+    prefer_project: Optional[str],
+    account: Optional[str],
+    prefer_type: Optional[str],
+    auto: bool,
+    conversation_id: Optional[str],
+    conversation_name: Optional[str],
+    execute_at: Optional[str],
+) -> None:
+    """Submit a task for execution."""
+    api = _require_server()
+
+    body: dict = {"prompt": prompt, "auto": auto}
+
+    if prefer_project:
+        field, value = _resolve_project_field(prefer_project)
+        body[field] = value
+    if account:
+        body["account"] = account
+    if prefer_type:
+        body["type"] = prefer_type
+    if conversation_id:
+        body["conversation_id"] = conversation_id
+    if conversation_name:
+        body["conversation_name"] = conversation_name
+    if execute_at:
+        body["execute_at"] = execute_at
+
+    try:
+        r = httpx.post(f"{api}/api/tasks", json=body, timeout=10)
+    except httpx.RequestError as e:
+        raise click.ClickException(f"请求失败：{e}")
+
+    if r.status_code != 201:
+        detail = r.json().get("detail", r.text) if r.headers.get("content-type", "").startswith("application/json") else r.text
+        raise click.ClickException(f"提交失败（HTTP {r.status_code}）：{detail}")
+
+    d = r.json()
+    conv = d.get("conversation_id", "")
+    project_label = d["project"].split("/")[-1] if d.get("project") else d.get("project_name", "")
+
+    click.secho(f"✓ 任务已提交：{d['id']}", fg="green")
+    click.echo(f"  项目：{project_label}  账号：{d['account']} ({d['type']})")
+    if d.get("status") == "scheduled":
+        click.echo(f"  定时：{d.get('execute_at')}")
+    elif d.get("status") == "pending":
+        click.secho(f"  状态：排队中（账号忙碌）", fg="yellow")
+    if conv:
+        click.echo(f"  任务链：{conv}")
+    click.echo(f"  查看日志：coderfleet task logs {d['id']}")
+    click.echo(f"  实时跟踪：coderfleet task logs {d['id']} -f")
+
+
+# ── task list ─────────────────────────────────────────────
+
+@task_group.command("list")
+@click.option("--account", default=None, help="按账号名过滤")
+@click.option("--status", "status_filter", default=None,
+              type=click.Choice(["running", "pending", "scheduled", "done", "failed", "killed"]),
+              help="按状态过滤")
+@click.option("--limit", default=200, show_default=True, help="最多显示条数")
+@click.option("--all", "include_archived", is_flag=True, help="包含已归档的任务")
+def cmd_task_list(
+    account: Optional[str],
+    status_filter: Optional[str],
+    limit: int,
+    include_archived: bool,
+) -> None:
+    """List tasks."""
+    api = _require_server()
+
+    params: dict = {"limit": limit}
+    if account:
+        params["account"] = account
+    if status_filter:
+        params["status"] = status_filter
+    if include_archived:
+        params["include_archived"] = "true"
+
+    try:
+        r = httpx.get(f"{api}/api/tasks", params=params, timeout=10)
+    except httpx.RequestError as e:
+        raise click.ClickException(f"请求失败：{e}")
+
+    if r.status_code != 200:
+        raise click.ClickException(f"获取任务列表失败（HTTP {r.status_code}）")
+
+    tasks = r.json()
+    if not tasks:
+        click.secho("⚠ 暂无任务记录", fg="yellow")
+        return
+
+    click.echo()
+    click.echo("  ── 任务列表 " + "─" * 62)
+    click.echo(f"  {'任务 ID':<22} {'状态':<10} {'账号':<18} {'类型':<8}  任务描述")
+    click.echo("  " + "─" * 80)
+
+    for t in tasks:
+        color, label = _STATUS_STYLE.get(t["status"], ("white", t["status"]))
+        prompt = t["prompt"]
+        if len(prompt) > 38:
+            prompt = prompt[:38] + "…"
+        status_str = click.style(f"{label:<10}", fg=color)
+        click.echo(f"  {t['id']:<22} {status_str}  {t['account']:<18} {t['type']:<8}  {prompt}")
+
+    click.echo()
+
+
+# ── task status ───────────────────────────────────────────
+
+@task_group.command("status")
+@click.argument("task_id")
+def cmd_task_status(task_id: str) -> None:
+    """Show task details."""
+    api = _require_server()
+
+    try:
+        r = httpx.get(f"{api}/api/tasks/{task_id}", timeout=10)
+    except httpx.RequestError as e:
+        raise click.ClickException(f"请求失败：{e}")
+
+    if r.status_code == 404:
+        raise click.ClickException(f"任务 '{task_id}' 不存在")
+    if r.status_code != 200:
+        raise click.ClickException(f"查询失败（HTTP {r.status_code}）")
+
+    d = r.json()
+    color, label = _STATUS_STYLE.get(d["status"], ("white", d["status"]))
+
+    click.echo()
+    click.echo(f"  ID：      {d['id']}")
+    click.echo(f"  状态：    {click.style(label, fg=color)}")
+    click.echo(f"  账号：    {d['account']} ({d['type']})")
+    click.echo(f"  项目：    {d['project']}")
+    if d.get("project_name"):
+        click.echo(f"  项目名：  {d['project_name']}")
+    click.echo(f"  创建：    {d['created']}")
+    click.echo(f"  完成：    {d.get('finished') or '-'}")
+    if d.get("execute_at"):
+        click.echo(f"  定时：    {d['execute_at']}")
+    if d.get("conversation_id"):
+        click.echo(f"  任务链：  {d['conversation_id']}")
+    if d.get("native_session_id"):
+        click.echo(f"  会话ID：  {d['native_session_id']}")
+    click.echo(f"  任务描述：{d['prompt']}")
+    click.echo()
+
+
+# ── task logs ─────────────────────────────────────────────
+
+@task_group.command("logs")
+@click.argument("task_id")
+@click.option("-f", "--follow", is_flag=True, help="实时跟踪日志（流式输出）")
+def cmd_task_logs(task_id: str, follow: bool) -> None:
+    """Show task logs. Use -f to stream in real time."""
+    api = _require_server()
+
+    if follow:
+        click.secho("实时跟踪日志（Ctrl+C 退出）...", dim=True)
+        url = f"{api}/api/tasks/{task_id}/logs/stream?tail=80"
+        try:
+            with httpx.stream("GET", url, timeout=None) as r:
+                if r.status_code == 404:
+                    raise click.ClickException(f"任务 '{task_id}' 不存在")
+                for line in r.iter_lines():
+                    if line == "data: [DONE]":
+                        break
+                    if line.startswith("data: "):
+                        click.echo(line[6:])
+        except KeyboardInterrupt:
+            pass
+        except httpx.RequestError as e:
+            raise click.ClickException(f"请求失败：{e}")
+    else:
+        try:
+            r = httpx.get(f"{api}/api/tasks/{task_id}/logs", timeout=30)
+        except httpx.RequestError as e:
+            raise click.ClickException(f"请求失败：{e}")
+
+        if r.status_code == 404:
+            raise click.ClickException(f"任务 '{task_id}' 的日志不存在")
+        if r.status_code != 200:
+            raise click.ClickException(f"获取日志失败（HTTP {r.status_code}）")
+
+        click.echo(r.text, nl=False)
+
+
+# ── task kill ─────────────────────────────────────────────
+
+@task_group.command("kill")
+@click.argument("task_id")
+def cmd_task_kill(task_id: str) -> None:
+    """Kill a running or pending task."""
+    api = _require_server()
+
+    try:
+        r = httpx.delete(f"{api}/api/tasks/{task_id}", timeout=10)
+    except httpx.RequestError as e:
+        raise click.ClickException(f"请求失败：{e}")
+
+    if r.status_code == 200:
+        click.secho(f"✓ 已终止任务 {task_id}", fg="green")
+    elif r.status_code == 404:
+        raise click.ClickException(f"任务 '{task_id}' 不存在")
+    else:
+        detail = r.json().get("detail", r.text) if "json" in r.headers.get("content-type", "") else r.text
+        raise click.ClickException(f"终止失败（HTTP {r.status_code}）：{detail}")
+
+
+# ── task clean ────────────────────────────────────────────
+
+@task_group.command("clean")
+@click.argument("keep", default=30, required=False)
+def cmd_task_clean(keep: int) -> None:
+    """Clean old task records, keeping the N most recent (default: 30)."""
+    api = _require_server()
+
+    try:
+        r = httpx.post(f"{api}/api/tasks/clean", params={"keep": keep}, timeout=10)
+    except httpx.RequestError as e:
+        raise click.ClickException(f"请求失败：{e}")
+
+    if r.status_code != 200:
+        raise click.ClickException(f"清理失败（HTTP {r.status_code}）")
+
+    d = r.json()
+    click.secho(f"✓ 已清理 {d['cleaned']} 条历史记录（保留 {d['kept']} 条）", fg="green")