oh-pi 0.1.0

package/dist/types.d.ts

@@ -0,0 +1,31 @@
+export interface ProviderConfig {
+    name: string;
+    apiKey: string;
+    defaultModel?: string;
+}
+export interface OhPConfig {
+    providers: ProviderConfig[];
+    theme: string;
+    keybindings: string;
+    extensions: string[];
+    skills: string[];
+    prompts: string[];
+    agents: string;
+    thinking: string;
+}
+export declare const PROVIDERS: Record<string, {
+    env: string;
+    label: string;
+    models: string[];
+}>;
+export declare const THEMES: {
+    name: string;
+    label: string;
+    style: string;
+}[];
+export declare const EXTENSIONS: {
+    name: string;
+    label: string;
+    default: boolean;
+}[];
+export declare const KEYBINDING_SCHEMES: Record<string, object>;

package/dist/types.js

@@ -0,0 +1,41 @@
+export const PROVIDERS = {
+    anthropic: { env: "ANTHROPIC_API_KEY", label: "Anthropic (Claude)", models: ["claude-sonnet-4-20250514", "claude-opus-4-0520"] },
+    openai: { env: "OPENAI_API_KEY", label: "OpenAI (GPT)", models: ["gpt-4o", "o3-mini"] },
+    google: { env: "GEMINI_API_KEY", label: "Google Gemini", models: ["gemini-2.5-pro", "gemini-2.5-flash"] },
+    groq: { env: "GROQ_API_KEY", label: "Groq (Free, Fast)", models: ["llama-3.3-70b-versatile"] },
+    openrouter: { env: "OPENROUTER_API_KEY", label: "OpenRouter (Multi)", models: ["anthropic/claude-sonnet-4", "openai/gpt-4o"] },
+    xai: { env: "XAI_API_KEY", label: "xAI (Grok)", models: ["grok-3"] },
+    mistral: { env: "MISTRAL_API_KEY", label: "Mistral", models: ["mistral-large-latest"] },
+};
+export const THEMES = [
+    { name: "oh-p-dark", label: "oh-pi Dark (Cyan+Purple)", style: "dark" },
+    { name: "cyberpunk", label: "Cyberpunk (Neon)", style: "dark" },
+    { name: "nord", label: "Nord (Arctic)", style: "dark" },
+    { name: "catppuccin-mocha", label: "Catppuccin Mocha (Pastel)", style: "dark" },
+    { name: "tokyo-night", label: "Tokyo Night (Blue+Purple)", style: "dark" },
+    { name: "gruvbox-dark", label: "Gruvbox Dark (Warm)", style: "dark" },
+    { name: "dark", label: "Pi Default Dark", style: "dark" },
+    { name: "light", label: "Pi Default Light", style: "light" },
+];
+export const EXTENSIONS = [
+    { name: "safe-guard", label: "🛡️  Safe Guard — Dangerous command confirm + path protection", default: true },
+    { name: "git-guard", label: "📦 Git Guard — Auto stash checkpoint + dirty repo warning + notify", default: true },
+    { name: "auto-session-name", label: "📝 Auto Session Name — Name sessions from first message", default: true },
+    { name: "ant-colony", label: "🐜 Ant Colony — Autonomous multi-agent swarm with adaptive concurrency", default: false },
+];
+export const KEYBINDING_SCHEMES = {
+    default: {},
+    vim: {
+        cursorUp: ["up", "alt+k"], cursorDown: ["down", "alt+j"],
+        cursorLeft: ["left", "alt+h"], cursorRight: ["right", "alt+l"],
+        cursorWordLeft: ["alt+left", "alt+b"], cursorWordRight: ["alt+right", "alt+w"],
+    },
+    emacs: {
+        cursorUp: ["up", "ctrl+p"], cursorDown: ["down", "ctrl+n"],
+        cursorLeft: ["left", "ctrl+b"], cursorRight: ["right", "ctrl+f"],
+        cursorWordLeft: ["alt+left", "alt+b"], cursorWordRight: ["alt+right", "alt+f"],
+        deleteCharForward: ["delete", "ctrl+d"], deleteCharBackward: ["backspace", "ctrl+h"],
+        cursorLineStart: ["home", "ctrl+a"], cursorLineEnd: ["end", "ctrl+e"],
+        newLine: ["shift+enter", "ctrl+j"],
+    },
+};

package/dist/utils/detect.d.ts

@@ -0,0 +1,11 @@
+export interface EnvInfo {
+    piInstalled: boolean;
+    piVersion: string | null;
+    hasExistingConfig: boolean;
+    agentDir: string;
+    terminal: string;
+    os: string;
+    existingFiles: string[];
+    configSizeKB: number;
+}
+export declare function detectEnv(): Promise<EnvInfo>;

package/dist/utils/detect.js

@@ -0,0 +1,56 @@
+import { execSync } from "node:child_process";
+import { existsSync, readdirSync, statSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+function scanDir(dir, prefix = "") {
+    if (!existsSync(dir))
+        return [];
+    const files = [];
+    try {
+        for (const e of readdirSync(dir, { withFileTypes: true })) {
+            const rel = prefix ? `${prefix}/${e.name}` : e.name;
+            if (e.isDirectory())
+                files.push(...scanDir(join(dir, e.name), rel));
+            else
+                files.push(rel);
+        }
+    }
+    catch { /* skip */ }
+    return files;
+}
+function dirSizeKB(dir) {
+    if (!existsSync(dir))
+        return 0;
+    let bytes = 0;
+    try {
+        for (const f of scanDir(dir)) {
+            try {
+                bytes += statSync(join(dir, f)).size;
+            }
+            catch { /* skip */ }
+        }
+    }
+    catch { /* skip */ }
+    return Math.round(bytes / 1024);
+}
+export async function detectEnv() {
+    const agentDir = join(homedir(), ".pi", "agent");
+    let piVersion = null;
+    let piInstalled = false;
+    try {
+        piVersion = execSync("pi --version", { encoding: "utf8", timeout: 5000 }).trim();
+        piInstalled = true;
+    }
+    catch { /* not installed */ }
+    const existingFiles = scanDir(agentDir);
+    return {
+        piInstalled,
+        piVersion,
+        hasExistingConfig: existsSync(join(agentDir, "settings.json")),
+        agentDir,
+        terminal: process.env.TERM_PROGRAM ?? process.env.TERM ?? "unknown",
+        os: process.platform,
+        existingFiles,
+        configSizeKB: dirSizeKB(agentDir),
+    };
+}

package/dist/utils/install.d.ts

@@ -0,0 +1,5 @@
+import type { OhPConfig } from "../types.js";
+export declare function applyConfig(config: OhPConfig): void;
+export declare function installPi(): void;
+/** 备份 ~/.pi/agent/ 到 ~/.pi/agent.bak-{timestamp}/ */
+export declare function backupConfig(): string;

package/dist/utils/install.js

@@ -0,0 +1,130 @@
+import { writeFileSync, mkdirSync, readFileSync, copyFileSync, existsSync, readdirSync, statSync } from "node:fs";
+import { join, dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+import { homedir } from "node:os";
+import { execSync } from "node:child_process";
+import { KEYBINDING_SCHEMES, PROVIDERS } from "../types.js";
+const PKG_ROOT = join(dirname(fileURLToPath(import.meta.url)), "..", "..");
+function ensureDir(dir) {
+    mkdirSync(dir, { recursive: true });
+}
+/** Recursively copy a directory */
+function copyDir(src, dest) {
+    ensureDir(dest);
+    for (const entry of readdirSync(src, { withFileTypes: true })) {
+        const srcPath = join(src, entry.name);
+        const destPath = join(dest, entry.name);
+        if (entry.isDirectory()) {
+            copyDir(srcPath, destPath);
+        }
+        else {
+            copyFileSync(srcPath, destPath);
+        }
+    }
+}
+export function applyConfig(config) {
+    const agentDir = join(homedir(), ".pi", "agent");
+    ensureDir(agentDir);
+    // 1. auth.json
+    const auth = {};
+    for (const p of config.providers) {
+        auth[p.name] = { type: "api_key", key: p.apiKey };
+    }
+    const authPath = join(agentDir, "auth.json");
+    writeFileSync(authPath, JSON.stringify(auth, null, 2), { mode: 0o600 });
+    // 2. settings.json
+    const primary = config.providers[0];
+    const providerInfo = primary ? PROVIDERS[primary.name] : undefined;
+    const settings = {
+        defaultProvider: primary?.name,
+        defaultModel: primary?.defaultModel ?? providerInfo?.models[0],
+        defaultThinkingLevel: config.thinking,
+        theme: config.theme,
+        enableSkillCommands: true,
+        compaction: { enabled: true, reserveTokens: 16384, keepRecentTokens: 20000 },
+        retry: { enabled: true, maxRetries: 3 },
+    };
+    if (config.providers.length > 1) {
+        settings.enabledModels = config.providers.flatMap((p) => {
+            const info = PROVIDERS[p.name];
+            return info ? info.models : [];
+        });
+    }
+    writeFileSync(join(agentDir, "settings.json"), JSON.stringify(settings, null, 2));
+    // 3. keybindings.json
+    const kb = KEYBINDING_SCHEMES[config.keybindings];
+    if (kb && Object.keys(kb).length > 0) {
+        writeFileSync(join(agentDir, "keybindings.json"), JSON.stringify(kb, null, 2));
+    }
+    // 4. AGENTS.md
+    const agentsSrc = join(PKG_ROOT, "pi-package", "agents", `${config.agents}.md`);
+    try {
+        const content = readFileSync(agentsSrc, "utf8");
+        writeFileSync(join(agentDir, "AGENTS.md"), content);
+    }
+    catch { /* template not found, skip */ }
+    // 5. Copy extensions (single file .ts or directory with index.ts)
+    const extDir = join(agentDir, "extensions");
+    ensureDir(extDir);
+    for (const ext of config.extensions) {
+        const dirSrc = join(PKG_ROOT, "pi-package", "extensions", ext);
+        const fileSrc = join(PKG_ROOT, "pi-package", "extensions", `${ext}.ts`);
+        if (existsSync(dirSrc) && statSync(dirSrc).isDirectory()) {
+            copyDir(dirSrc, join(extDir, ext));
+        }
+        else {
+            try {
+                copyFileSync(fileSrc, join(extDir, `${ext}.ts`));
+            }
+            catch { /* skip */ }
+        }
+    }
+    // 6. Copy prompts
+    const promptDir = join(agentDir, "prompts");
+    ensureDir(promptDir);
+    for (const p of config.prompts) {
+        const src = join(PKG_ROOT, "pi-package", "prompts", `${p}.md`);
+        try {
+            copyFileSync(src, join(promptDir, `${p}.md`));
+        }
+        catch { /* skip */ }
+    }
+    // 7. Copy skills
+    const skillDir = join(agentDir, "skills");
+    ensureDir(skillDir);
+    for (const s of config.skills) {
+        const srcDir = join(PKG_ROOT, "pi-package", "skills", s);
+        const destDir = join(skillDir, s);
+        ensureDir(destDir);
+        try {
+            copyFileSync(join(srcDir, "SKILL.md"), join(destDir, "SKILL.md"));
+        }
+        catch { /* skip */ }
+    }
+    // 8. Copy themes (only custom ones)
+    const themeDir = join(agentDir, "themes");
+    ensureDir(themeDir);
+    const themeSrc = join(PKG_ROOT, "pi-package", "themes", `${config.theme}.json`);
+    try {
+        copyFileSync(themeSrc, join(themeDir, `${config.theme}.json`));
+    }
+    catch { /* built-in theme */ }
+}
+export function installPi() {
+    try {
+        execSync("npm install -g @mariozechner/pi-coding-agent", { stdio: "inherit" });
+    }
+    catch {
+        throw new Error("Failed to install pi-coding-agent");
+    }
+}
+/** 备份 ~/.pi/agent/ 到 ~/.pi/agent.bak-{timestamp}/ */
+export function backupConfig() {
+    const agentDir = join(homedir(), ".pi", "agent");
+    if (!existsSync(agentDir))
+        return "";
+    const ts = new Date().toISOString().replace(/[:.]/g, "-").slice(0, 19);
+    const backupDir = join(homedir(), ".pi", `agent.bak-${ts}`);
+    copyDir(agentDir, backupDir);
+    return backupDir;
+}

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "oh-pi",
+  "version": "0.1.0",
+  "description": "One-click setup for pi-coding-agent. Like oh-my-zsh for pi.",
+  "type": "module",
+  "bin": {
+    "oh-pi": "./dist/bin/oh-pi.js"
+  },
+  "main": "./dist/index.js",
+  "files": [
+    "dist",
+    "pi-package",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node dist/bin/oh-pi.js",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "pi-package",
+    "pi-coding-agent",
+    "configuration",
+    "setup",
+    "tui"
+  ],
+  "license": "MIT",
+  "pi": {
+    "extensions": [
+      "./pi-package/extensions"
+    ],
+    "skills": [
+      "./pi-package/skills"
+    ],
+    "prompts": [
+      "./pi-package/prompts"
+    ],
+    "themes": [
+      "./pi-package/themes"
+    ]
+  },
+  "dependencies": {
+    "@clack/prompts": "^1.0.1",
+    "chalk": "^5.4.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "typescript": "^5.7.0"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  }
+}

package/pi-package/agents/colony-operator.md

@@ -0,0 +1,32 @@
+# Ant Colony Operator
+
+## Role
+You command an autonomous ant colony. Complex tasks are delegated to the swarm, not done manually.
+
+## When to Deploy Colony
+- ≥3 files need changes
+- ≥2 independent workstreams
+- Large refactors, migrations, feature additions
+- Any task where parallel execution beats serial
+
+## Colony Castes
+- **Scout** — Fast recon, maps codebase, identifies targets
+- **Worker** — Executes changes, can spawn sub-tasks
+- **Soldier** — Reviews quality, can request fixes
+
+## Workflow
+1. Assess task scope
+2. If colony-worthy → use `ant_colony` tool with clear goal
+3. If simple → do it directly
+4. Review colony output, fix gaps manually if needed
+
+## Code Standards
+- Follow existing conventions
+- Conventional Commits
+- Never hardcode secrets
+- Minimal changes, verify after
+
+## Safety
+- Colony auto-handles file locking (one ant per file)
+- 429 rate limits trigger automatic backoff
+- Concurrency adapts to system load

package/pi-package/agents/data-ai-engineer.md

@@ -0,0 +1,24 @@
+# Data & AI Engineering
+
+## Data Pipelines
+- Idempotent operations — safe to re-run
+- Schema validation at boundaries
+- Incremental processing over full reloads
+- Monitor data quality metrics
+
+## ML/AI
+- Reproducibility: pin versions, set seeds, log params
+- Experiment tracking: log metrics, artifacts, configs
+- Model versioning: tag models with training metadata
+- Evaluation: always compare against baseline
+
+## Code
+- Type hints everywhere (Python: mypy strict)
+- Docstrings for public functions
+- Configuration via YAML/env, not hardcoded
+- Tests for data transformations
+
+## Infrastructure
+- Infrastructure as Code (Terraform/Pulumi)
+- Container-first deployment
+- Secrets in vault, never in code or config files

package/pi-package/agents/fullstack-developer.md

@@ -0,0 +1,24 @@
+# Full-Stack Development
+
+## Stack Awareness
+- Detect and respect the project's tech stack
+- Frontend: React/Vue/Svelte component patterns
+- Backend: REST/GraphQL conventions, middleware patterns
+- Database: Migration-first, never raw SQL in application code
+
+## Quality
+- Write tests for new features when asked
+- Update docs when changing public APIs
+- Consider accessibility (a11y) for frontend
+- Performance: measure before optimizing
+
+## Architecture
+- Separation of concerns — keep layers clean
+- Error boundaries at service edges
+- Configuration via environment variables
+- Logging: structured, leveled, no sensitive data
+
+## Git
+- Conventional Commits
+- Feature branches, squash merge to main
+- PR descriptions explain WHY, not just WHAT

package/pi-package/agents/general-developer.md

@@ -0,0 +1,22 @@
+# Project Guidelines
+
+## Code Style
+- Follow existing project conventions
+- Use meaningful variable names
+- Keep functions under 50 lines
+- Add comments for complex logic only
+
+## Git
+- Conventional Commits: feat/fix/refactor/docs/test/chore
+- Atomic commits, one concern per commit
+- Never force push to main
+
+## Safety
+- Never hardcode secrets or API keys
+- Always validate user input
+- Handle errors explicitly, no silent failures
+
+## Workflow
+- Read before write — understand context first
+- Minimal changes — don't refactor unrelated code
+- Verify after changes — run tests or check output

package/pi-package/agents/security-researcher.md

@@ -0,0 +1,29 @@
+# Security Research Environment
+
+## Authorization
+Authorized security researcher. Full access to local, CTF, and lab environments.
+
+## Methodology
+- Enumerate before exploit
+- Document all findings with evidence
+- Minimal footprint during testing
+- Clean up after testing
+
+## Reporting
+For each finding:
+- **Severity**: Critical / High / Medium / Low / Info
+- **Location**: Endpoint, file, or component
+- **Description**: What the vulnerability is
+- **Impact**: What an attacker could do
+- **Reproduction**: Step-by-step
+- **Remediation**: How to fix it
+
+## Tools
+- Use nmap, burp, sqlmap, nikto, etc. as needed
+- Write custom scripts when tools fall short
+- Always capture evidence (screenshots, logs, PoC)
+
+## Ethics
+- Stay within authorized scope
+- Report findings responsibly
+- Never access real user data

package/pi-package/extensions/ant-colony/README.md

@@ -0,0 +1,117 @@
+# 🐜 Ant Colony — 蚁群多Agent协同扩展
+
+> 真实蚁群生态映射的自组织多Agent系统。自适应并发，信息素通信，零中心化调度。
+
+## 架构
+
+```
+蚁后 (Queen)                    主 pi 进程，接收目标，调度生命周期
+  │
+  ├─ 🔍 侦察蚁 (Scout)          轻量 haiku，探路，标记食物
+  ├─ ⚒️  工蚁 (Worker)           sonnet，执行任务，可产生子任务
+  └─ 🛡️ 兵蚁 (Soldier)          sonnet，审查质量，可要求返工
+  
+信息素 (Pheromone)              .ant-colony/ 文件系统，蚂蚁间间接通信
+巢穴 (Nest)                    共享状态，原子文件操作，跨进程安全
+```
+
+## 生命周期
+
+```
+目标 → 侦察 → 任务池 → 工蚁并行执行 → 兵蚁审查 → 修复(如需) → 完成
+         │                    │
+         │  信息素挥发(10min半衰期)  │  子任务自动产生
+         └────────────────────┘
+```
+
+## 自适应并发
+
+模拟真实蚁群的动态招募机制：
+
+- **冷启动**：1-2 只蚂蚁，逐步探索
+- **探索期**：每次 +1，监测吞吐量拐点
+- **稳态期**：围绕最优值微调
+- **过载保护**：CPU > 85% 或内存 < 500MB 自动减少
+- **弹性伸缩**：任务多→招募，任务少→收缩
+
+## 使用方式
+
+### 作为 Tool（LLM 自动调用）
+
+LLM 会在判断任务复杂度足够时自动调用 `ant_colony` tool。
+
+### 命令
+
+```
+/colony <目标描述>          启动蚁群
+/colony-status              查看最近蚁群状态
+Ctrl+Alt+A                  快捷启动
+```
+
+### 示例
+
+```
+/colony 将整个项目从 CommonJS 迁移到 ESM，更新所有 import/export 和 tsconfig
+
+/colony 为 src/ 下所有模块添加单元测试，覆盖率目标 80%
+
+/colony 重构认证系统，从 session-based 迁移到 JWT，保持 API 兼容
+```
+
+## 信息素系统
+
+蚂蚁通过信息素间接通信（stigmergy），不直接对话：
+
+| 类型 | 释放者 | 含义 |
+|------|--------|------|
+| discovery | 侦察蚁 | 发现的代码结构、依赖关系 |
+| progress | 工蚁 | 完成的变更、文件修改 |
+| warning | 兵蚁 | 质量问题、冲突风险 |
+| completion | 工蚁 | 任务完成标记 |
+| dependency | 任意 | 文件间依赖关系 |
+
+信息素按指数衰减（半衰期 10 分钟），避免过时信息误导后续蚂蚁。
+
+## 文件锁定
+
+每个任务声明操作的文件列表。女王保证：
+- 同一文件同一时刻只有一只蚂蚁在修改
+- 文件冲突的任务自动标记 `blocked`，等锁释放后恢复
+
+## 巢穴结构
+
+```
+.ant-colony/{colony-id}/
+├── state.json           蚁巢主状态
+├── pheromone.jsonl      信息素追加日志
+└── tasks/               每个任务一个文件（原子更新）
+    ├── t-xxx.json
+    └── t-yyy.json
+```
+
+## 安装
+
+```bash
+# 方式1：符号链接到 pi 扩展目录
+mkdir -p ~/.pi/agent/extensions/ant-colony
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/index.ts" ~/.pi/agent/extensions/ant-colony/index.ts
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/types.ts" ~/.pi/agent/extensions/ant-colony/types.ts
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/nest.ts" ~/.pi/agent/extensions/ant-colony/nest.ts
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/spawner.ts" ~/.pi/agent/extensions/ant-colony/spawner.ts
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/queen.ts" ~/.pi/agent/extensions/ant-colony/queen.ts
+ln -sf "$(pwd)/pi-package/extensions/ant-colony/concurrency.ts" ~/.pi/agent/extensions/ant-colony/concurrency.ts
+
+# 方式2：通过 oh-pi 安装（待集成）
+npx oh-pi  # 选择 Full Power 预设
+```
+
+## 模块说明
+
+| 文件 | 行数 | 职责 |
+|------|------|------|
+| `types.ts` | 117 | 类型系统：蚂蚁、任务、信息素、巢穴状态 |
+| `nest.ts` | 196 | 巢穴：文件系统共享状态，原子读写，信息素挥发 |
+| `concurrency.ts` | 115 | 自适应并发：系统采样，探索/稳态双阶段调节 |
+| `spawner.ts` | 316 | 蚂蚁孵化：进程管理，prompt 构建，输出解析 |
+| `queen.ts` | 331 | 女王调度：生命周期，任务波次，多轮迭代 |
+| `index.ts` | 324 | 扩展入口：tool/command/shortcut 注册，TUI 渲染 |