mosscode 0.1.0__py3-none-any.whl

mosscode/agents/tasks.py

@@ -0,0 +1,193 @@
+"""The sub-agents of one session, foreground or background."""
+
+from __future__ import annotations
+
+import asyncio
+import time
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Literal
+
+from ..messages import UsageTotals
+
+STATUS_LABELS = {"running": "运行中", "completed": "已完成", "failed": "失败", "cancelled": "已取消"}
+ORIGIN_LABELS = {"agent": "agent 工具", "skill": "Skill fork", "hook": "Hook"}
+CANCEL_WAIT = 5.0
+
+
+@dataclass
+class Task:
+    id: str
+    origin: Literal["agent", "skill", "hook"]
+    agent: Literal["fork", "subagent"]
+    agent_type: str
+    description: str
+    prompt: str
+    call_id: str = ""
+    status: Literal["running", "completed", "failed", "cancelled"] = "running"
+    background: bool = False
+    started: float = field(default_factory=time.monotonic)
+    ended: float | None = None
+    usage: UsageTotals = field(default_factory=UsageTotals)
+    tool_calls: int = 0
+    last_tool: str = ""
+    result: str | None = None
+    transcript: Path | None = None
+    notified: bool = False
+    stopped: bool = False  # Stopped by /tasks stop or task_stop: never notified.
+    job: asyncio.Task | None = field(default=None, repr=False)
+    detach: asyncio.Event = field(default_factory=asyncio.Event, repr=False)
+    policy: object = field(default=None, repr=False)  # The PolicyView, switched to non-interactive on detach.
+    isolated: bool = False  # Runs in a worktree (asked for and not skipped by plan mode).
+    root: Path | None = None  # The worktree directory it works in; None for the project.
+    # {"name", "path", "branch", "base", "outcome"}; outcome is 使用中 / 已保留 / 已删除 / 删除未完成.
+    worktree: dict | None = None
+
+    @property
+    def elapsed(self) -> float:
+        return (self.ended if self.ended is not None else time.monotonic()) - self.started
+
+    @property
+    def tokens(self) -> int:
+        return self.usage.input_tokens + self.usage.output_tokens
+
+    @property
+    def label(self) -> str:
+        return f"{self.agent_type} · {self.description}"
+
+    def to_background(self) -> None:
+        self.background = True
+        if self.policy is not None:
+            self.policy.to_background()
+
+
+def task_line(task: Task) -> str:
+    """The line shown when a background task starts or ends."""
+    state = "已启动" if task.status == "running" else STATUS_LABELS[task.status]
+    return f"后台任务 {task.id}（{task.label}）{state}"
+
+
+class Announcer:
+    """One line when a background task starts and one when it ends, however often on_change fires between.
+
+    Hook tasks are left out: the hook engine reports them.
+    """
+
+    def __init__(self, say: Callable[[str], None]):
+        self.say = say
+        self.started: set[str] = set()
+        self.ended: set[str] = set()
+
+    def __call__(self, task: Task) -> None:
+        if not task.background or task.origin == "hook":
+            return
+        if task.id not in self.started:
+            self.started.add(task.id)
+            if task.status == "running":
+                self.say(task_line(task))
+        if task.status != "running" and task.id not in self.ended:
+            self.ended.add(task.id)
+            self.say(task_line(task))
+
+
+class TaskManager:
+    def __init__(self, *, max_background: int = 4):
+        self.tasks: dict[str, Task] = {}
+        self.max_background = max_background
+        self._next = 0
+        self.on_finish: Callable[[Task], None] | None = None
+        self.on_change: Callable[[Task], None] | None = None  # The UI refreshes a tool row or the status bar.
+        self.finished = asyncio.Event()  # Set whenever a task ends; -p waits on it.
+
+    def new(self, **fields) -> Task:
+        self._next += 1
+        task = Task(f"t{self._next}", **fields)
+        self.tasks[task.id] = task
+        return task
+
+    def get(self, task_id: str) -> Task | None:
+        return self.tasks.get(task_id)
+
+    def running(self) -> list[Task]:
+        return [task for task in self.tasks.values() if task.status == "running"]
+
+    def background(self) -> list[Task]:
+        return [task for task in self.running() if task.background]
+
+    def full(self) -> bool:
+        return len(self.background()) >= self.max_background
+
+    def changed(self, task: Task) -> None:
+        if self.on_change is not None:
+            try:
+                self.on_change(task)
+            except Exception:
+                pass
+
+    def launch(self, task: Task, run: Callable[[], Awaitable[str]]) -> asyncio.Task:
+        task.job = asyncio.get_running_loop().create_task(self._run(task, run))
+        self.changed(task)
+        return task.job
+
+    async def _run(self, task: Task, run) -> str:
+        try:
+            text = await run()
+        except asyncio.CancelledError:
+            text = task.result or "[子 Agent 未完成：已取消]"
+            task.status = "cancelled"
+        except Exception as exc:  # run() turns provider errors into text; anything else is a bug.
+            text = f"[子 Agent 未完成：内部错误 {type(exc).__name__}]"
+            task.status = "failed"
+        self.settle(task, text)
+        return text
+
+    def settle(self, task: Task, text: str) -> None:
+        """The end of a task, also of one run outside launch (a Skill fork)."""
+        task.result = text
+        task.ended = time.monotonic()
+        if task.status == "running":
+            task.status = "completed"
+        self.finished.set()
+        self.changed(task)
+        if self.on_finish is not None:
+            try:
+                self.on_finish(task)
+            except Exception:
+                pass
+
+    def detach_foreground(self) -> int:
+        """ctrl+b moves every running foreground sub-agent of the agent tool to the background."""
+        count = 0
+        for task in self.running():
+            if not task.background and task.origin == "agent" and not task.detach.is_set():
+                task.detach.set()
+                count += 1
+        return count
+
+    async def stop(self, task_id: str) -> bool:
+        """True when it was running; its notification is never sent."""
+        task = self.tasks.get(task_id)
+        if task is None or task.status != "running" or task.job is None:
+            return False
+        task.stopped = task.notified = True
+        task.job.cancel()
+        await asyncio.gather(task.job, return_exceptions=True)
+        return True
+
+    async def cancel_all(self, wait: float = CANCEL_WAIT) -> int:
+        """On exit and when the user confirms: cancel, wait a little, never notify."""
+        jobs = []
+        for task in self.running():
+            task.stopped = task.notified = True
+            if task.job is not None:
+                task.job.cancel()
+                jobs.append(task.job)
+        if jobs:
+            await asyncio.wait(jobs, timeout=wait)
+        return len(jobs)
+
+    async def reset(self) -> None:
+        """/clear and resume start with an empty list."""
+        await self.cancel_all()
+        self.tasks.clear()

mosscode/appearance.py

@@ -0,0 +1,184 @@
+"""Terminal colors and the mascot of the startup banner."""
+
+from importlib.metadata import version
+from pathlib import Path
+
+from rich.style import Style
+from rich.table import Table
+from rich.text import Text
+from textual.theme import Theme
+
+TEXT_COLOR = "ansi_default"
+TOOL_SUCCESS_COLOR = "#8bd5a6"
+PARAMETER_COLOR = "#6fa8ff"
+OPERATION_COLOR = "#e3b95f"
+ERROR_COLOR = "#e87983"
+COMMAND_HIGHLIGHT_COLOR = "#3673ff"
+# User messages sit on a grey bar, as in Claude Code: a mid grey with near-white
+# text stands out on both dark and light terminal backgrounds.
+USER_MESSAGE_BACKGROUND = "#4a4a4a"
+USER_MESSAGE_COLOR = "#f2f2f2"
+
+# The mascot, a little ghost, as pixels: two pixel rows make one line of half blocks, so each pixel is
+# about square. "." is the terminal's own background; the other letters are MASCOT_COLORS.
+MASCOT_PIXELS = (
+    "....wwwwww....",
+    "..wwwwwwwwww..",
+    ".wwwwwwwwwwww.",
+    ".wwkhwwwwkhww.",
+    ".wwkkwwwwkkww.",
+    ".wrrwwkkwwrrw.",
+    ".wwwwwwwwwwww.",
+    ".wwwwwwwwwwww.",
+    ".ww.www.www.w.",
+    ".w...w...w....",
+)
+MASCOT_COLORS = {"w": "#eef1ff", "k": "#26212e", "h": "#ffffff", "r": "#ff9fb2"}
+
+
+# Transparent TCSS inherits the parent's painted background. ANSI default means
+# the terminal's own background, and must be preserved by ansi_color=True.
+TERMINAL_CSS = """
+App, Screen, * {
+    color: ansi_default !important;
+    text-opacity: 100% !important;
+    opacity: 1 !important;
+    background: ansi_default !important;
+    background-tint: transparent !important;
+    tint: transparent !important;
+    scrollbar-background: ansi_default;
+    scrollbar-background-hover: ansi_default;
+    scrollbar-background-active: ansi_default;
+    scrollbar-corner-color: ansi_default;
+}
+Input > .input--placeholder, Input > .input--suggestion {
+    color: ansi_default !important;
+    text-style: none !important;
+}
+Markdown MarkdownBlock > .code_inline {
+    background: ansi_default !important;
+}
+Input > .input--cursor {
+    color: $foreground;
+    background: ansi_default !important;
+    text-style: reverse;
+}
+Input > .input--selection {
+    color: $primary;
+    background: ansi_default !important;
+    text-style: bold underline;
+}
+OptionList:focus > .option-list--option-highlighted {
+    color: $primary;
+    background: ansi_default !important;
+    text-style: bold;
+}
+"""
+
+
+def moss_themes() -> tuple[Theme, ...]:
+    # Use the terminal's full-strength foreground; never dim supporting text.
+    variables = {
+        name: TEXT_COLOR
+        for name in (
+            "text",
+            "text-muted",
+            "text-disabled",
+            "foreground-muted",
+            "input-cursor-foreground",
+            "link-color",
+            "link-color-hover",
+        )
+    }
+    variables["input-cursor-background"] = "ansi_default"
+    return (
+        Theme(
+            name="moss-dark",
+            primary=TEXT_COLOR,
+            secondary=TEXT_COLOR,
+            accent=TEXT_COLOR,
+            foreground=TEXT_COLOR,
+            background="#0c0c0c",
+            surface="#0c0c0c",
+            panel="#0c0c0c",
+            success=TEXT_COLOR,
+            warning=TEXT_COLOR,
+            error=TEXT_COLOR,
+            dark=True,
+            variables=variables,
+        ),
+    )
+
+
+def pixel_art(pixels: tuple[str, ...], colors: dict[str, str]) -> Text:
+    """Each pair of pixel rows as one line: ▀ in the upper color over the lower one as background."""
+    art = Text(no_wrap=True)
+    for index, (upper, lower) in enumerate(zip(pixels[::2], pixels[1::2])):
+        if index:
+            art.append("\n")
+        for top, bottom in zip(upper, lower):
+            top, bottom = colors.get(top), colors.get(bottom)
+            if top and top == bottom:
+                art.append("█", Style(color=top))
+            elif top:
+                art.append("▀", Style(color=top, bgcolor=bottom))
+            elif bottom:
+                art.append("▄", Style(color=bottom))
+            else:
+                art.append(" ")
+    return art
+
+
+def startup_banner(
+    provider: str,
+) -> Table:
+    logo = pixel_art(MASCOT_PIXELS, MASCOT_COLORS)
+    info = Text(style="default", no_wrap=True, overflow="ellipsis")
+    info.append("MossCode", style=Style(bold=True))
+    info.append(f" v{version('mosscode')}\n")
+    info.append(provider)
+    info.append("\n" + str(Path.cwd()))
+    banner = Table.grid(padding=(0, 2), expand=True)
+    banner.add_column(width=len(MASCOT_PIXELS[0]), no_wrap=True)
+    banner.add_column(ratio=1, no_wrap=True, overflow="ellipsis", vertical="middle")
+    banner.add_row(logo, info)
+    return banner
+
+
+def compact_notice(event) -> str:
+    """One line for a finished or failed compaction, shared by the TUI and headless text output."""
+    if hasattr(event, "removed"):
+        return (
+            f"✓ 已压缩上下文 · {event.removed} 条消息 → 摘要 · 约 {event.summary_tokens} Token"
+            f" · 用时 {event.elapsed:.1f}s"
+        )
+    if event.code == "nothing":
+        return "└ " + event.reason
+    if event.code == "restore":
+        return "! " + event.reason
+    if event.manual:
+        return "! 压缩上下文失败：" + event.reason
+    notice = f"! 自动压缩上下文失败（连续 {event.consecutive} 次）：{event.reason}"
+    if event.paused:
+        notice += "\n  自动压缩已暂停，可以 /compact 手动重试或 /clear"
+    return notice
+
+
+# The key half of /help; the command half comes from the registry.
+KEY_HELP = """按键
+Enter                 发送消息或执行命令
+Shift+Enter / Ctrl+J  换行（也可在行尾输入 \\ 再按 Enter）
+↑ / ↓                 在命令提示中选择；首行/末行时浏览本次输入历史
+Tab                   补全命令或参数
+Shift+Tab             依次切换 default → acceptEdits → plan
+运行中 Enter          排队消息，本轮工具执行后发送给模型
+Esc                   关闭命令提示（保留输入）
+Esc / Ctrl+C          取消正在生成的回复；排队消息放回输入框
+授权面板              ↑ / ↓ 选择，Enter 确认，Esc 拒绝本次，Ctrl+C 取消任务
+Ctrl+B                把前台运行的子 Agent 转入后台
+Ctrl+T                展开团队成员面板：Enter 查看对话，x 停止成员
+成员视图              PgUp / PgDn 滚动，← / → 切换成员，Esc 返回
+Ctrl+Q                退出（有后台任务时先确认）
+鼠标滚轮              滚动对话记录
+鼠标拖选              使用终端原生文字选择
+Ctrl+V                粘贴（多行内容保持原样）"""