my-pi-agent 0.1.0

package/src/my_coding_agent/paths.py

@@ -0,0 +1,119 @@
+"""AgentPaths: 统一管理全局用户目录与项目本地资源的路径调度中心。"""
+
+from __future__ import annotations
+
+import os
+import re
+from dataclasses import dataclass, field
+from hashlib import sha256
+from pathlib import Path
+
+__all__ = ["AgentPaths"]
+
+
+@dataclass(frozen=True, slots=True)
+class AgentPaths:
+    """集中解析与管理 my-pi-agent 的全局与项目级路径。"""
+
+    home: Path = field(
+        default_factory=lambda: Path(os.environ.get("MY_AGENT_HOME") or (Path.home() / ".my-pi-agent")).resolve()
+    )
+    agents_home: Path = field(default_factory=lambda: (Path.home() / ".agents").resolve())
+
+    # ── 全局资源路径 ──
+    @property
+    def auth_path(self) -> Path:
+        return self.home / "auth.json"
+
+    @property
+    def auth_lock_path(self) -> Path:
+        return self.home / "auth.json.lock"
+
+    @property
+    def settings_path(self) -> Path:
+        return self.home / "settings.json"
+
+    @property
+    def sessions_dir(self) -> Path:
+        return self.home / "sessions"
+
+    @property
+    def skills_dir(self) -> Path:
+        return self.home / "skills"
+
+    @property
+    def prompts_dir(self) -> Path:
+        return self.home / "prompts"
+
+    @property
+    def themes_dir(self) -> Path:
+        return self.home / "themes"
+
+    @property
+    def extensions_dir(self) -> Path:
+        return self.home / "extensions"
+
+    @property
+    def logs_dir(self) -> Path:
+        return self.home / "logs"
+
+    # ── 项目局部路径 ──
+    def project_agent_dir(self, cwd: Path) -> Path:
+        return cwd / ".my-pi-agent"
+
+    def project_settings_path(self, cwd: Path) -> Path:
+        return self.project_agent_dir(cwd) / "settings.json"
+
+    def project_skills_dir(self, cwd: Path) -> Path:
+        return self.project_agent_dir(cwd) / "skills"
+
+    def project_agents_skills_dir(self, cwd: Path) -> Path:
+        return cwd / ".agents" / "skills"
+
+    # ── 会话分区映射算法 (Tau Slug + Hash 优化版) ──
+    def project_session_dir(self, cwd: Path) -> Path:
+        """根据项目 cwd 计算全局唯一的 sessions/<slug>-<hash> 目录。"""
+        resolved = cwd.resolve()
+        digest = sha256(str(resolved).encode("utf-8")).hexdigest()[:6]
+        slug = self._slugify_path(resolved)
+        target = self.sessions_dir / f"{slug}-{digest}"
+        target.mkdir(parents=True, exist_ok=True)
+        return target
+
+    def default_session_path(self, cwd: Path) -> Path:
+        return self.project_session_dir(cwd) / "default.jsonl"
+
+    def ensure_directories(self) -> None:
+        """初次启动自动建巢，静默初始化目录骨架。"""
+        for d in (
+            self.home,
+            self.sessions_dir,
+            self.skills_dir,
+            self.prompts_dir,
+            self.themes_dir,
+            self.extensions_dir,
+            self.logs_dir,
+        ):
+            d.mkdir(parents=True, exist_ok=True)
+
+    @staticmethod
+    def _slugify_path(path: Path, max_length: int = 48) -> str:
+        parts = [p for p in path.parts if p not in (path.anchor, "")]
+        try:
+            rel = path.relative_to(Path.home())
+            parts = ["home", *rel.parts]
+        except ValueError:
+            pass
+        normalized = [clean for p in parts if (clean := re.sub(r"[^a-zA-Z0-9._-]+", "-", p).strip(".-_").lower())]
+        slug = "-".join(normalized)
+        if len(slug) <= max_length:
+            return slug or "project"
+
+        suffix_parts: list[str] = []
+        cur_len = 0
+        for p in reversed(normalized):
+            if cur_len + len(p) + 1 > max_length:
+                break
+            suffix_parts.append(p)
+            cur_len += len(p) + 1
+        return "-".join(reversed(suffix_parts)) or slug[-max_length:].strip("-") or "project"

package/src/my_coding_agent/permissions.py

@@ -0,0 +1,84 @@
+"""业务权限门禁（PermissionGate）：基于 ToolCallHook 实现的无侵入安全审批门禁。"""
+
+from __future__ import annotations
+
+import inspect
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass, field
+from typing import Any, Literal
+
+from my_agent_core.hooks import HookResult, ToolCallHook
+
+PermissionMode = Literal["review", "autonomous", "yolo", "strict"]
+READONLY_TOOLS = frozenset({"read", "grep", "find"})
+SAFE_BASH_PREFIXES = (
+    "git status",
+    "git diff",
+    "git log",
+    "pytest",
+    "python -m pytest",
+    "uv run",
+)
+
+
+@dataclass(frozen=True)
+class PermissionRequest:
+    """权限审批请求对象。"""
+
+    action: str
+    target: str
+    details: dict[str, Any] = field(default_factory=dict)
+    preview: str | None = None
+
+
+class PermissionGate:
+    """基于 ToolCallHook 实现的无侵入安全审批门禁。"""
+
+    def __init__(
+        self,
+        mode: PermissionMode = "review",
+        confirm_callback: Callable[[PermissionRequest], Awaitable[bool] | bool] | None = None,
+    ) -> None:
+        self.mode: PermissionMode = "autonomous" if mode == "yolo" else mode
+        self.confirm_callback = confirm_callback
+
+    async def __call__(self, hook: ToolCallHook) -> HookResult:
+        if self.mode == "autonomous":
+            return HookResult()
+
+        tool_name = hook.tool_name
+        args = hook.args or {}
+
+        # 1. 只读工具放行 (除非 strict 模式)
+        if self.mode != "strict" and tool_name in READONLY_TOOLS:
+            return HookResult()
+
+        # 2. 安全 Shell 命令放行
+        if tool_name == "bash" and self.mode == "review":
+            cmd = str(args.get("command", "")).strip()
+            if any(cmd.startswith(p) for p in SAFE_BASH_PREFIXES):
+                return HookResult()
+
+        # 3. 发起用户交互审查
+        if not self.confirm_callback:
+            # 无交互回调时默认放行
+            return HookResult()
+
+        target = str(args.get("path") or args.get("command") or tool_name)
+        preview = None
+        if tool_name in ("write", "edit"):
+            preview = args.get("content") or str(args.get("edits", ""))
+
+        req = PermissionRequest(
+            action=tool_name,
+            target=target,
+            details=args,
+            preview=preview,
+        )
+
+        res = self.confirm_callback(req)
+        approved = await res if inspect.isawaitable(res) else res
+        if not approved:
+            return HookResult(block=True, reason=f"用户拒绝了工具 [{tool_name}] 的执行请求。")
+
+        return HookResult()

package/src/my_coding_agent/prompt.py

@@ -0,0 +1,54 @@
+"""专业编码系统提示词与上下文自动注入。"""
+
+import logging
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+DEFAULT_CODING_INSTRUCTIONS = """You are an expert coding assistant operating inside the codebase.
+
+Core Engineering Principles:
+1. Read Before Edit: Always inspect file contents with 'read' before calling 'edit'. Never guess line numbers or code blocks.
+2. Surgical Edits: Keep edits concise and provide sufficient context in 'oldText' to ensure a unique match.
+3. Verification Before Completion: Run tests or linters using 'bash' to verify changes before declaring work complete.
+4. Minimal Changes: Do not add unsolicited refactorings, comments, or unnecessary abstractions.
+"""
+
+
+def build_default_coding_prompt(workspace: Path | str) -> str:
+    """构建专业编码系统提示词，自动扫描并注入项目指导文件。
+
+    Args:
+        workspace: 工作区目录路径（支持 Path 或 str）。
+
+    Returns:
+        包含工程规范、工作区路径以及项目上下文的完整系统提示词。
+    """
+    workspace_path = Path(workspace).resolve()
+    sections = [
+        DEFAULT_CODING_INSTRUCTIONS,
+        f"Workspace Directory: {workspace_path}",
+    ]
+
+    # 扫描并注入项目指导文件
+    context_files = ["AGENTS.md", "CLAUDE.md", "README.md"]
+    injected_contexts = []
+    for fname in context_files:
+        fpath = workspace_path / fname
+        if fpath.is_file():
+            try:
+                content = fpath.read_text(encoding="utf-8", errors="replace")
+                injected_contexts.append(
+                    f'<project_instructions path="{fpath}">\n{content}\n</project_instructions>'
+                )
+            except Exception:
+                logger.debug("Failed to read context file: %s", fpath, exc_info=True)
+
+    if injected_contexts:
+        sections.append(
+            "<project_context>\n"
+            + "\n\n".join(injected_contexts)
+            + "\n</project_context>"
+        )
+
+    return "\n\n".join(sections)