pi-mono/packages/coding-agent/examples/extensions

Extension Examples

Example extensions for pi-coding-agent.

Usage

# Load an extension with --extension flag
pi --extension examples/extensions/permission-gate.ts

# Or copy to extensions directory for auto-discovery
cp permission-gate.ts ~/.pi/agent/extensions/

Examples

Lifecycle & Safety

Extension	Description
permission-gate.ts	Prompts for confirmation before dangerous bash commands (rm -rf, sudo, etc.)
protected-paths.ts	Blocks writes to protected paths (.env, .git/, node_modules/)
confirm-destructive.ts	Confirms before destructive session actions (clear, switch, fork)
dirty-repo-guard.ts	Prevents session changes with uncommitted git changes
sandbox/	OS-level sandboxing using @anthropic-ai/sandbox-runtime with per-project config

Custom Tools

Extension	Description
todo.ts	Todo list tool + /todos command with custom rendering and state persistence
hello.ts	Minimal custom tool example
question.ts	Demonstrates ctx.ui.select() for asking the user questions with custom UI
questionnaire.ts	Multi-question input with tab bar navigation between questions
tool-override.ts	Override built-in tools (e.g., add logging/access control to read)
dynamic-tools.ts	Register tools after startup (session_start) and at runtime via command, with prompt snippets and tool-specific prompt guidelines
built-in-tool-renderer.ts	Custom compact rendering for built-in tools (read, bash, edit, write) while keeping original behavior
minimal-mode.ts	Override built-in tool rendering for minimal display (only tool calls, no output in collapsed mode)
truncated-tool.ts	Wraps ripgrep with proper output truncation (50KB/2000 lines)
antigravity-image-gen.ts	Generate images via Google Antigravity with optional save-to-disk modes
ssh.ts	Delegate all tools to a remote machine via SSH using pluggable operations
subagent/	Delegate tasks to specialized subagents with isolated context windows

Commands & UI

Extension	Description
preset.ts	Named presets for model, thinking level, tools, and instructions via --preset flag and /preset command
plan-mode/	Claude Code-style plan mode for read-only exploration with /plan command and step tracking
tools.ts	Interactive /tools command to enable/disable tools with session persistence
handoff.ts	Transfer context to a new focused session via /handoff <goal>
qna.ts	Extracts questions from last response into editor via ctx.ui.setEditorText()
status-line.ts	Shows turn progress in footer via ctx.ui.setStatus() with themed colors
widget-placement.ts	Shows widgets above and below the editor via ctx.ui.setWidget() placement
hidden-thinking-label.ts	Customizes the collapsed thinking label via ctx.ui.setHiddenThinkingLabel()
working-indicator.ts	Customizes the streaming working indicator via ctx.ui.setWorkingIndicator()
model-status.ts	Shows model changes in status bar via model_select hook
snake.ts	Snake game with custom UI, keyboard handling, and session persistence
tic-tac-toe.ts	Tic-tac-toe vs the agent with executionMode: "sequential" tools to prevent race conditions on shared cursor state
send-user-message.ts	Demonstrates pi.sendUserMessage() for sending user messages from extensions
timed-confirm.ts	Demonstrates AbortSignal for auto-dismissing ctx.ui.confirm() and ctx.ui.select() dialogs
rpc-demo.ts	Exercises all RPC-supported extension UI methods; pair with examples/rpc-extension-ui.ts
modal-editor.ts	Custom vim-like modal editor via ctx.ui.setEditorComponent()
rainbow-editor.ts	Animated rainbow text effect via custom editor
notify.ts	Desktop notifications via OSC 777 when agent finishes (Ghostty, iTerm2, WezTerm)
titlebar-spinner.ts	Braille spinner animation in terminal title while the agent is working
summarize.ts	Summarize conversation with GPT-5.2 and show in transient UI
custom-footer.ts	Custom footer with git branch and token stats via ctx.ui.setFooter()
custom-header.ts	Custom header via ctx.ui.setHeader()
overlay-test.ts	Test overlay compositing with inline text inputs and edge cases
overlay-qa-tests.ts	Comprehensive overlay QA tests: anchors, margins, stacking, overflow, animation
doom-overlay/	DOOM game running as an overlay at 35 FPS (demonstrates real-time game rendering)
shutdown-command.ts	Adds /quit command demonstrating ctx.shutdown()
reload-runtime.ts	Adds /reload-runtime and reload_runtime tool showing safe reload flow
interactive-shell.ts	Run interactive commands (vim, htop) with full terminal via user_bash hook
inline-bash.ts	Expands !{command} patterns in prompts via input event transformation

Git Integration

Extension	Description
git-checkpoint.ts	Creates git stash checkpoints at each turn for code restoration on fork
auto-commit-on-exit.ts	Auto-commits on exit using last assistant message for commit message

System Prompt & Compaction

Extension	Description
pirate.ts	Demonstrates systemPromptAppend to dynamically modify system prompt
claude-rules.ts	Scans .claude/rules/ folder and lists rules in system prompt
custom-compaction.ts	Custom compaction that summarizes entire conversation
trigger-compact.ts	Triggers compaction when context usage exceeds 100k tokens and adds /trigger-compact command

System Integration

Extension	Description
mac-system-theme.ts	Syncs pi theme with macOS dark/light mode

Resources

Extension	Description
dynamic-resources/	Loads skills, prompts, and themes using resources_discover

Messages & Communication

Extension	Description
message-renderer.ts	Custom message rendering with colors and expandable details via registerMessageRenderer
event-bus.ts	Inter-extension communication via pi.events

Session Metadata

Extension	Description
session-name.ts	Name sessions for the session selector via setSessionName
bookmark.ts	Bookmark entries with labels for /tree navigation via setLabel

Custom Providers

Extension	Description
custom-provider-anthropic/	Custom Anthropic provider with OAuth support and custom streaming implementation
custom-provider-gitlab-duo/	GitLab Duo provider using pi-ai's built-in Anthropic/OpenAI streaming via proxy
custom-provider-qwen-cli/	Qwen CLI provider with OAuth device flow and OpenAI-compatible models

External Dependencies

Extension	Description
with-deps/	Extension with its own package.json and dependencies (demonstrates jiti module resolution)
file-trigger.ts	Watches a trigger file and injects contents into conversation

Writing Extensions

See docs/extensions.md for full documentation.

import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { Type } from "@sinclair/typebox";

export default function (pi: ExtensionAPI) {
  // Subscribe to lifecycle events
  pi.on("tool_call", async (event, ctx) => {
    if (event.toolName === "bash" && event.input.command?.includes("rm -rf")) {
      const ok = await ctx.ui.confirm("Dangerous!", "Allow rm -rf?");
      if (!ok) return { block: true, reason: "Blocked by user" };
    }
  });

  // Register custom tools
  pi.registerTool({
    name: "greet",
    label: "Greeting",
    description: "Generate a greeting",
    parameters: Type.Object({
      name: Type.String({ description: "Name to greet" }),
    }),
    async execute(toolCallId, params, onUpdate, ctx, signal) {
      return {
        content: [{ type: "text", text: `Hello, ${params.name}!` }],
        details: {},
      };
    },
  });

  // Register commands
  pi.registerCommand("hello", {
    description: "Say hello",
    handler: async (args, ctx) => {
      ctx.ui.notify("Hello!", "info");
    },
  });
}

Key Patterns

Use StringEnum for string parameters (required for Google API compatibility):

import { StringEnum } from "@mariozechner/pi-ai";

// Good
action: StringEnum(["list", "add"] as const)

// Bad - doesn't work with Google
action: Type.Union([Type.Literal("list"), Type.Literal("add")])

State persistence via details:

// Store state in tool result details for proper forking support
return {
  content: [{ type: "text", text: "Done" }],
  details: { todos: [...todos], nextId },  // Persisted in session
};

// Reconstruct on session events
pi.on("session_start", async (_event, ctx) => {
  for (const entry of ctx.sessionManager.getBranch()) {
    if (entry.type === "message" && entry.message.toolName === "my_tool") {
      const details = entry.message.details;
      // Reconstruct state from details
    }
  }
});