indusagi-coding-agent 0.1.28 → 0.1.30

package/examples/extensions/antigravity-image-gen.ts

@@ -1,414 +0,0 @@
-/**
- * Antigravity Image Generation
- *
- * Generates images via Google Antigravity's image models (gemini-3-pro-image, imagen-3).
- * Returns images as tool result attachments for inline terminal rendering.
- * Requires OAuth login via /login for google-antigravity.
- *
- * Usage:
- *   "Generate an image of a sunset over mountains"
- *   "Create a 16:9 wallpaper of a cyberpunk city"
- *
- * Save modes (tool param, env var, or config file):
- *   save=none     - Don't save to disk (default)
- *   save=project  - Save to <repo>/.indusagi/generated-images/
- *   save=global   - Save to ~/.indusagi/agent/generated-images/
- *   save=custom   - Save to saveDir param or INDUSAGI_IMAGE_SAVE_DIR
- *
- * Environment variables:
- *   INDUSAGI_IMAGE_SAVE_MODE  - Default save mode (none|project|global|custom)
- *   INDUSAGI_IMAGE_SAVE_DIR   - Directory for custom save mode
- *
- * Config files (project overrides global):
- *   ~/.indusagi/agent/extensions/antigravity-image-gen.json
- *   <repo>/.indusagi/extensions/antigravity-image-gen.json
- *   Example: { "save": "global" }
- */
-
-import { randomUUID } from "node:crypto";
-import { existsSync, readFileSync } from "node:fs";
-import { mkdir, writeFile } from "node:fs/promises";
-import { homedir } from "node:os";
-import { join } from "node:path";
-import { StringEnum } from "indusagi/ai";
-import type { ExtensionAPI } from "indusagi-coding-agent";
-import { type Static, Type } from "@sinclair/typebox";
-
-const PROVIDER = "google-antigravity";
-
-const ASPECT_RATIOS = ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9"] as const;
-
-type AspectRatio = (typeof ASPECT_RATIOS)[number];
-
-const DEFAULT_MODEL = "gemini-3-pro-image";
-const DEFAULT_ASPECT_RATIO: AspectRatio = "1:1";
-const DEFAULT_SAVE_MODE = "none";
-
-const SAVE_MODES = ["none", "project", "global", "custom"] as const;
-type SaveMode = (typeof SAVE_MODES)[number];
-
-const ANTIGRAVITY_ENDPOINT = "https://daily-cloudcode-pa.sandbox.googleapis.com";
-
-const ANTIGRAVITY_HEADERS = {
-	"User-Agent": "antigravity/1.11.5 darwin/arm64",
-	"X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-	"Client-Metadata": JSON.stringify({
-		ideType: "IDE_UNSPECIFIED",
-		platform: "PLATFORM_UNSPECIFIED",
-		pluginType: "GEMINI",
-	}),
-};
-
-const IMAGE_SYSTEM_INSTRUCTION =
-	"You are an AI image generator. Generate images based on user descriptions. Focus on creating high-quality, visually appealing images that match the user's request.";
-
-const TOOL_PARAMS = Type.Object({
-	prompt: Type.String({ description: "Image description." }),
-	model: Type.Optional(
-		Type.String({
-			description: "Image model id (e.g., gemini-3-pro-image, imagen-3). Default: gemini-3-pro-image.",
-		}),
-	),
-	aspectRatio: Type.Optional(StringEnum(ASPECT_RATIOS)),
-	save: Type.Optional(StringEnum(SAVE_MODES)),
-	saveDir: Type.Optional(
-		Type.String({
-			description: "Directory to save image when save=custom. Defaults to INDUSAGI_IMAGE_SAVE_DIR if set.",
-		}),
-	),
-});
-
-type ToolParams = Static<typeof TOOL_PARAMS>;
-
-interface CloudCodeAssistRequest {
-	project: string;
-	model: string;
-	request: {
-		contents: Content[];
-		sessionId?: string;
-		systemInstruction?: { role?: string; parts: { text: string }[] };
-		generationConfig?: {
-			maxOutputTokens?: number;
-			temperature?: number;
-			imageConfig?: { aspectRatio?: string };
-			candidateCount?: number;
-		};
-		safetySettings?: Array<{ category: string; threshold: string }>;
-	};
-	requestType?: string;
-	userAgent?: string;
-	requestId?: string;
-}
-
-interface CloudCodeAssistResponseChunk {
-	response?: {
-		candidates?: Array<{
-			content?: {
-				role: string;
-				parts?: Array<{
-					text?: string;
-					inlineData?: {
-						mimeType?: string;
-						data?: string;
-					};
-				}>;
-			};
-		}>;
-		usageMetadata?: {
-			promptTokenCount?: number;
-			candidatesTokenCount?: number;
-			thoughtsTokenCount?: number;
-			totalTokenCount?: number;
-			cachedContentTokenCount?: number;
-		};
-		modelVersion?: string;
-		responseId?: string;
-	};
-	traceId?: string;
-}
-
-interface Content {
-	role: "user" | "model";
-	parts: Part[];
-}
-
-interface Part {
-	text?: string;
-	inlineData?: {
-		mimeType?: string;
-		data?: string;
-	};
-}
-
-interface ParsedCredentials {
-	accessToken: string;
-	projectId: string;
-}
-
-interface ExtensionConfig {
-	save?: SaveMode;
-	saveDir?: string;
-}
-
-interface SaveConfig {
-	mode: SaveMode;
-	outputDir?: string;
-}
-
-function parseOAuthCredentials(raw: string): ParsedCredentials {
-	let parsed: { token?: string; projectId?: string };
-	try {
-		parsed = JSON.parse(raw) as { token?: string; projectId?: string };
-	} catch {
-		throw new Error("Invalid Google OAuth credentials. Run /login to re-authenticate.");
-	}
-	if (!parsed.token || !parsed.projectId) {
-		throw new Error("Missing token or projectId in Google OAuth credentials. Run /login.");
-	}
-	return { accessToken: parsed.token, projectId: parsed.projectId };
-}
-
-function readConfigFile(path: string): ExtensionConfig {
-	if (!existsSync(path)) {
-		return {};
-	}
-	try {
-		const content = readFileSync(path, "utf-8");
-		const parsed = JSON.parse(content) as ExtensionConfig;
-		return parsed ?? {};
-	} catch {
-		return {};
-	}
-}
-
-function loadConfig(cwd: string): ExtensionConfig {
-	const globalConfig = readConfigFile(join(homedir(), ".indusagi", "agent", "extensions", "antigravity-image-gen.json"));
-	const projectConfig = readConfigFile(join(cwd, ".indusagi", "extensions", "antigravity-image-gen.json"));
-	return { ...globalConfig, ...projectConfig };
-}
-
-function resolveSaveConfig(params: ToolParams, cwd: string): SaveConfig {
-	const config = loadConfig(cwd);
-	const envMode = (process.env.INDUSAGI_IMAGE_SAVE_MODE || "").toLowerCase();
-	const paramMode = params.save;
-	const mode = (paramMode || envMode || config.save || DEFAULT_SAVE_MODE) as SaveMode;
-
-	if (!SAVE_MODES.includes(mode)) {
-		return { mode: DEFAULT_SAVE_MODE as SaveMode };
-	}
-
-	if (mode === "project") {
-		return { mode, outputDir: join(cwd, ".indusagi", "generated-images") };
-	}
-
-	if (mode === "global") {
-		return { mode, outputDir: join(homedir(), ".indusagi", "agent", "generated-images") };
-	}
-
-	if (mode === "custom") {
-		const dir = params.saveDir || process.env.INDUSAGI_IMAGE_SAVE_DIR || config.saveDir;
-		if (!dir || !dir.trim()) {
-			throw new Error("save=custom requires saveDir or INDUSAGI_IMAGE_SAVE_DIR.");
-		}
-		return { mode, outputDir: dir };
-	}
-
-	return { mode };
-}
-
-function imageExtension(mimeType: string): string {
-	const lower = mimeType.toLowerCase();
-	if (lower.includes("jpeg") || lower.includes("jpg")) return "jpg";
-	if (lower.includes("gif")) return "gif";
-	if (lower.includes("webp")) return "webp";
-	return "png";
-}
-
-async function saveImage(base64Data: string, mimeType: string, outputDir: string): Promise<string> {
-	await mkdir(outputDir, { recursive: true });
-	const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
-	const ext = imageExtension(mimeType);
-	const filename = `image-${timestamp}-${randomUUID().slice(0, 8)}.${ext}`;
-	const filePath = join(outputDir, filename);
-	await writeFile(filePath, Buffer.from(base64Data, "base64"));
-	return filePath;
-}
-
-function buildRequest(prompt: string, model: string, projectId: string, aspectRatio: string): CloudCodeAssistRequest {
-	return {
-		project: projectId,
-		model,
-		request: {
-			contents: [
-				{
-					role: "user",
-					parts: [{ text: prompt }],
-				},
-			],
-			systemInstruction: {
-				parts: [{ text: IMAGE_SYSTEM_INSTRUCTION }],
-			},
-			generationConfig: {
-				imageConfig: { aspectRatio },
-				candidateCount: 1,
-			},
-			safetySettings: [
-				{ category: "HARM_CATEGORY_HARASSMENT", threshold: "BLOCK_ONLY_HIGH" },
-				{ category: "HARM_CATEGORY_HATE_SPEECH", threshold: "BLOCK_ONLY_HIGH" },
-				{ category: "HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold: "BLOCK_ONLY_HIGH" },
-				{ category: "HARM_CATEGORY_DANGEROUS_CONTENT", threshold: "BLOCK_ONLY_HIGH" },
-				{ category: "HARM_CATEGORY_CIVIC_INTEGRITY", threshold: "BLOCK_ONLY_HIGH" },
-			],
-		},
-		requestType: "agent",
-		requestId: `agent-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
-		userAgent: "antigravity",
-	};
-}
-
-async function parseSseForImage(
-	response: Response,
-	signal?: AbortSignal,
-): Promise<{ image: { data: string; mimeType: string }; text: string[] }> {
-	if (!response.body) {
-		throw new Error("No response body");
-	}
-
-	const reader = response.body.getReader();
-	const decoder = new TextDecoder();
-	let buffer = "";
-	const textParts: string[] = [];
-
-	try {
-		while (true) {
-			if (signal?.aborted) {
-				throw new Error("Request was aborted");
-			}
-
-			const { done, value } = await reader.read();
-			if (done) break;
-
-			buffer += decoder.decode(value, { stream: true });
-			const lines = buffer.split("\n");
-			buffer = lines.pop() || "";
-
-			for (const line of lines) {
-				if (!line.startsWith("data:")) continue;
-				const jsonStr = line.slice(5).trim();
-				if (!jsonStr) continue;
-
-				let chunk: CloudCodeAssistResponseChunk;
-				try {
-					chunk = JSON.parse(jsonStr) as CloudCodeAssistResponseChunk;
-				} catch {
-					continue;
-				}
-
-				const responseData = chunk.response;
-				if (!responseData?.candidates) continue;
-
-				for (const candidate of responseData.candidates) {
-					const parts = candidate.content?.parts;
-					if (!parts) continue;
-					for (const part of parts) {
-						if (part.text) {
-							textParts.push(part.text);
-						}
-						if (part.inlineData?.data) {
-							await reader.cancel();
-							return {
-								image: {
-									data: part.inlineData.data,
-									mimeType: part.inlineData.mimeType || "image/png",
-								},
-								text: textParts,
-							};
-						}
-					}
-				}
-			}
-		}
-	} finally {
-		reader.releaseLock();
-	}
-
-	throw new Error("No image data returned by the model");
-}
-
-async function getCredentials(ctx: {
-	modelRegistry: { getApiKeyForProvider: (provider: string) => Promise<string | undefined> };
-}): Promise<ParsedCredentials> {
-	const apiKey = await ctx.modelRegistry.getApiKeyForProvider(PROVIDER);
-	if (!apiKey) {
-		throw new Error("Missing Google Antigravity OAuth credentials. Run /login for google-antigravity.");
-	}
-	return parseOAuthCredentials(apiKey);
-}
-
-export default function antigravityImageGen(indusagi: ExtensionAPI) {
-	indusagi.registerTool({
-		name: "generate_image",
-		label: "Generate image",
-		description:
-			"Generate an image via Google Antigravity image models. Returns the image as a tool result attachment. Optional saving via save=project|global|custom|none, or INDUSAGI_IMAGE_SAVE_MODE/INDUSAGI_IMAGE_SAVE_DIR.",
-		parameters: TOOL_PARAMS,
-		async execute(_toolCallId, params: ToolParams, onUpdate, ctx, signal) {
-			const { accessToken, projectId } = await getCredentials(ctx);
-			const model = params.model || DEFAULT_MODEL;
-			const aspectRatio = params.aspectRatio || DEFAULT_ASPECT_RATIO;
-
-			const requestBody = buildRequest(params.prompt, model, projectId, aspectRatio);
-			onUpdate?.({
-				content: [{ type: "text", text: `Requesting image from ${PROVIDER}/${model}...` }],
-				details: { provider: PROVIDER, model, aspectRatio },
-			});
-
-			const response = await fetch(`${ANTIGRAVITY_ENDPOINT}/v1internal:streamGenerateContent?alt=sse`, {
-				method: "POST",
-				headers: {
-					Authorization: `Bearer ${accessToken}`,
-					"Content-Type": "application/json",
-					Accept: "text/event-stream",
-					...ANTIGRAVITY_HEADERS,
-				},
-				body: JSON.stringify(requestBody),
-				signal,
-			});
-
-			if (!response.ok) {
-				const errorText = await response.text();
-				throw new Error(`Image request failed (${response.status}): ${errorText}`);
-			}
-
-			const parsed = await parseSseForImage(response, signal);
-			const saveConfig = resolveSaveConfig(params, ctx.cwd);
-			let savedPath: string | undefined;
-			let saveError: string | undefined;
-			if (saveConfig.mode !== "none" && saveConfig.outputDir) {
-				try {
-					savedPath = await saveImage(parsed.image.data, parsed.image.mimeType, saveConfig.outputDir);
-				} catch (error) {
-					saveError = error instanceof Error ? error.message : String(error);
-				}
-			}
-			const summaryParts = [`Generated image via ${PROVIDER}/${model}.`, `Aspect ratio: ${aspectRatio}.`];
-			if (savedPath) {
-				summaryParts.push(`Saved image to: ${savedPath}`);
-			} else if (saveError) {
-				summaryParts.push(`Failed to save image: ${saveError}`);
-			}
-			if (parsed.text.length > 0) {
-				summaryParts.push(`Model notes: ${parsed.text.join(" ")}`);
-			}
-
-			return {
-				content: [
-					{ type: "text", text: summaryParts.join(" ") },
-					{ type: "image", data: parsed.image.data, mimeType: parsed.image.mimeType },
-				],
-				details: { provider: PROVIDER, model, aspectRatio, savedPath, saveMode: saveConfig.mode },
-			};
-		},
-	});
-}
-

package/examples/extensions/auto-commit-on-exit.ts

@@ -1,49 +0,0 @@
-/**
- * Auto-Commit on Exit Extension
- *
- * Automatically commits changes when the agent exits.
- * Uses the last assistant message to generate a commit message.
- */
-
-import type { ExtensionAPI } from "indusagi-coding-agent";
-
-export default function (indusagi: ExtensionAPI) {
-	indusagi.on("session_shutdown", async (_event, ctx) => {
-		// Check for uncommitted changes
-		const { stdout: status, code } = await indusagi.exec("git", ["status", "--porcelain"]);
-
-		if (code !== 0 || status.trim().length === 0) {
-			// Not a git repo or no changes
-			return;
-		}
-
-		// Find the last assistant message for commit context
-		const entries = ctx.sessionManager.getEntries();
-		let lastAssistantText = "";
-		for (let i = entries.length - 1; i >= 0; i--) {
-			const entry = entries[i];
-			if (entry.type === "message" && entry.message.role === "assistant") {
-				const content = entry.message.content;
-				if (Array.isArray(content)) {
-					lastAssistantText = content
-						.filter((c): c is { type: "text"; text: string } => c.type === "text")
-						.map((c) => c.text)
-						.join("\n");
-				}
-				break;
-			}
-		}
-
-		// Generate a simple commit message
-		const firstLine = lastAssistantText.split("\n")[0] || "Work in progress";
-		const commitMessage = `[indusagi] ${firstLine.slice(0, 50)}${firstLine.length > 50 ? "..." : ""}`;
-
-		// Stage and commit
-		await indusagi.exec("git", ["add", "-A"]);
-		const { code: commitCode } = await indusagi.exec("git", ["commit", "-m", commitMessage]);
-
-		if (commitCode === 0 && ctx.hasUI) {
-			ctx.ui.notify(`Auto-committed: ${commitMessage}`, "info");
-		}
-	});
-}

package/examples/extensions/bookmark.ts

@@ -1,50 +0,0 @@
-/**
- * Entry bookmarking example.
- *
- * Shows setLabel to mark entries with labels for easy navigation in /tree.
- * Labels appear in the tree view and help you find important points.
- *
- * Usage: /bookmark [label] - bookmark the last assistant message
- */
-
-import type { ExtensionAPI } from "indusagi-coding-agent";
-
-export default function (indusagi: ExtensionAPI) {
-	indusagi.registerCommand("bookmark", {
-		description: "Bookmark last message (usage: /bookmark [label])",
-		handler: async (args, ctx) => {
-			const label = args.trim() || `bookmark-${Date.now()}`;
-
-			// Find the last assistant message entry
-			const entries = ctx.sessionManager.getEntries();
-			for (let i = entries.length - 1; i >= 0; i--) {
-				const entry = entries[i];
-				if (entry.type === "message" && entry.message.role === "assistant") {
-					indusagi.setLabel(entry.id, label);
-					ctx.ui.notify(`Bookmarked as: ${label}`, "info");
-					return;
-				}
-			}
-
-			ctx.ui.notify("No assistant message to bookmark", "warning");
-		},
-	});
-
-	// Remove bookmark
-	indusagi.registerCommand("unbookmark", {
-		description: "Remove bookmark from last labeled entry",
-		handler: async (_args, ctx) => {
-			const entries = ctx.sessionManager.getEntries();
-			for (let i = entries.length - 1; i >= 0; i--) {
-				const entry = entries[i];
-				const label = ctx.sessionManager.getLabel(entry.id);
-				if (label) {
-					indusagi.setLabel(entry.id, undefined);
-					ctx.ui.notify(`Removed bookmark: ${label}`, "info");
-					return;
-				}
-			}
-			ctx.ui.notify("No bookmarked entry found", "warning");
-		},
-	});
-}

package/examples/extensions/claude-rules.ts

@@ -1,86 +0,0 @@
-/**
- * Claude Rules Extension
- *
- * Scans the project's .claude/rules/ folder for rule files and lists them
- * in the system prompt. The agent can then use the read tool to load
- * specific rules when needed.
- *
- * Best practices for .claude/rules/:
- * - Keep rules focused: Each file should cover one topic (e.g., testing.md, api-design.md)
- * - Use descriptive filenames: The filename should indicate what the rules cover
- * - Use conditional rules sparingly: Only add paths frontmatter when rules truly apply to specific file types
- * - Organize with subdirectories: Group related rules (e.g., frontend/, backend/)
- *
- * Usage:
- * 1. Copy this file to ~/.indusagi/agent/extensions/ or your project's .indusagi/extensions/
- * 2. Create .claude/rules/ folder in your project root
- * 3. Add .md files with your rules
- */
-
-import * as fs from "node:fs";
-import * as path from "node:path";
-import type { ExtensionAPI } from "indusagi-coding-agent";
-
-/**
- * Recursively find all .md files in a directory
- */
-function findMarkdownFiles(dir: string, basePath: string = ""): string[] {
-	const results: string[] = [];
-
-	if (!fs.existsSync(dir)) {
-		return results;
-	}
-
-	const entries = fs.readdirSync(dir, { withFileTypes: true });
-
-	for (const entry of entries) {
-		const relativePath = basePath ? `${basePath}/${entry.name}` : entry.name;
-
-		if (entry.isDirectory()) {
-			results.push(...findMarkdownFiles(path.join(dir, entry.name), relativePath));
-		} else if (entry.isFile() && entry.name.endsWith(".md")) {
-			results.push(relativePath);
-		}
-	}
-
-	return results;
-}
-
-export default function claudeRulesExtension(indusagi: ExtensionAPI) {
-	let ruleFiles: string[] = [];
-	let rulesDir: string = "";
-
-	// Scan for rules on session start
-	indusagi.on("session_start", async (_event, ctx) => {
-		rulesDir = path.join(ctx.cwd, ".claude", "rules");
-		ruleFiles = findMarkdownFiles(rulesDir);
-
-		if (ruleFiles.length > 0) {
-			ctx.ui.notify(`Found ${ruleFiles.length} rule(s) in .claude/rules/`, "info");
-		}
-	});
-
-	// Append available rules to system prompt
-	indusagi.on("before_agent_start", async (event) => {
-		if (ruleFiles.length === 0) {
-			return;
-		}
-
-		const rulesList = ruleFiles.map((f) => `- .claude/rules/${f}`).join("\n");
-
-		return {
-			systemPrompt:
-				event.systemPrompt +
-				`
-
-## Project Rules
-
-The following project rules are available in .claude/rules/:
-
-${rulesList}
-
-When working on tasks related to these rules, use the read tool to load the relevant rule files for guidance.
-`,
-		};
-	});
-}

package/examples/extensions/confirm-destructive.ts

@@ -1,59 +0,0 @@
-/**
- * Confirm Destructive Actions Extension
- *
- * Prompts for confirmation before destructive session actions (clear, switch, branch).
- * Demonstrates how to cancel session events using the before_* events.
- */
-
-import type { ExtensionAPI, SessionBeforeSwitchEvent, SessionMessageEntry } from "indusagi-coding-agent";
-
-export default function (indusagi: ExtensionAPI) {
-	indusagi.on("session_before_switch", async (event: SessionBeforeSwitchEvent, ctx) => {
-		if (!ctx.hasUI) return;
-
-		if (event.reason === "new") {
-			const confirmed = await ctx.ui.confirm(
-				"Clear session?",
-				"This will delete all messages in the current session.",
-			);
-
-			if (!confirmed) {
-				ctx.ui.notify("Clear cancelled", "info");
-				return { cancel: true };
-			}
-			return;
-		}
-
-		// reason === "resume" - check if there are unsaved changes (messages since last assistant response)
-		const entries = ctx.sessionManager.getEntries();
-		const hasUnsavedWork = entries.some(
-			(e): e is SessionMessageEntry => e.type === "message" && e.message.role === "user",
-		);
-
-		if (hasUnsavedWork) {
-			const confirmed = await ctx.ui.confirm(
-				"Switch session?",
-				"You have messages in the current session. Switch anyway?",
-			);
-
-			if (!confirmed) {
-				ctx.ui.notify("Switch cancelled", "info");
-				return { cancel: true };
-			}
-		}
-	});
-
-	indusagi.on("session_before_fork", async (event, ctx) => {
-		if (!ctx.hasUI) return;
-
-		const choice = await ctx.ui.select(`Fork from entry ${event.entryId.slice(0, 8)}?`, [
-			"Yes, create fork",
-			"No, stay in current session",
-		]);
-
-		if (choice !== "Yes, create fork") {
-			ctx.ui.notify("Fork cancelled", "info");
-			return { cancel: true };
-		}
-	});
-}