@ryan_nookpi/pi-extension-auto-name 0.1.0

package/README.md

@@ -0,0 +1,15 @@
+# @ryan_nookpi/pi-extension-auto-name
+
+Auto session name extension for pi.
+
+## Install
+
+```bash
+pi install /Users/creatrip/Documents/pi-extension/packages/auto-name
+pi install npm:@ryan_nookpi/pi-extension-auto-name
+```
+
+## What it provides
+
+- Automatic session name detection
+- `./index.ts` entry

package/index.ts

@@ -0,0 +1,105 @@
+/**
+ * Auto session name — detects purpose from first user message
+ * and sets it as the session name via pi.setSessionName().
+ *
+ * - Auto-detect: uses pi-ai completeSimple() to summarize first message → pi.setSessionName()
+ * - Footer display: shows session name in status bar via setStatus()
+ * - Manual control: use built-in /name command (no custom command needed)
+ * - Skips auto-detection for subagent sessions
+ */
+
+import * as path from "node:path";
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import {
+	buildNameContext,
+	extractNameFromResult,
+	extractSessionFilePath,
+	formatNameStatus,
+	isSubagentSessionPath,
+	NAME_SYSTEM_PROMPT,
+} from "./utils/auto-name-utils.ts";
+import { generateShortLabel } from "./utils/short-label.js";
+import { NAME_STATUS_KEY } from "./utils/status-keys.ts";
+
+// ── Helpers ──────────────────────────────────────────────────────────────────
+
+function isSubagentSession(ctx: ExtensionContext): boolean {
+	const sessionFilePath = extractSessionFilePath(ctx.sessionManager);
+	return isSubagentSessionPath(sessionFilePath);
+}
+
+async function detectNameFromMessage(userMessage: string, ctx: ExtensionContext): Promise<string> {
+	return generateShortLabel(ctx, {
+		systemPrompt: NAME_SYSTEM_PROMPT,
+		prompt: buildNameContext(userMessage),
+		extractText: extractNameFromResult,
+	});
+}
+
+// ── Extension ────────────────────────────────────────────────────────────────
+
+export default function autoSessionName(pi: ExtensionAPI) {
+	const updateTerminalTitle = (ctx: ExtensionContext) => {
+		if (!ctx.hasUI) return;
+		const cwdBasename = path.basename(process.cwd());
+		const name = pi.getSessionName();
+		if (name) {
+			ctx.ui.setTitle(`π - ${name} - ${cwdBasename}`);
+		} else {
+			ctx.ui.setTitle(`π - ${cwdBasename}`);
+		}
+	};
+
+	const updateStatus = (ctx: ExtensionContext) => {
+		if (!ctx.hasUI) return;
+
+		const name = pi.getSessionName();
+		if (!name) {
+			ctx.ui.setStatus(NAME_STATUS_KEY, undefined);
+			return;
+		}
+
+		ctx.ui.setStatus(NAME_STATUS_KEY, formatNameStatus(name));
+		updateTerminalTitle(ctx);
+	};
+
+	// ── Auto Name (async) ──────────────────────────────────────
+
+	pi.on("before_agent_start", async (event, ctx) => {
+		if (isSubagentSession(ctx)) return;
+
+		// name이 이미 있으면 스킵
+		if (pi.getSessionName()) return;
+
+		const text = event.prompt.trim();
+		if (!text) return;
+
+		// Fire-and-forget: 비동기로 name 감지 후 설정
+		(async () => {
+			try {
+				const detected = await detectNameFromMessage(text, ctx);
+				if (detected && !pi.getSessionName()) {
+					pi.setSessionName(detected);
+					updateStatus(ctx);
+				}
+			} catch {
+				// 실패 시 무시
+			}
+		})();
+	});
+
+	// ── Lifecycle ─────────────────────────────────────────────────
+
+	pi.on("session_start", async (_event, ctx) => {
+		updateStatus(ctx);
+	});
+
+	pi.on("session_tree", async (_event, ctx) => {
+		updateStatus(ctx);
+	});
+
+	pi.on("session_shutdown", async (_event, ctx) => {
+		if (!ctx.hasUI) return;
+		ctx.ui.setStatus(NAME_STATUS_KEY, undefined);
+	});
+}

package/package.json

@@ -0,0 +1,26 @@
+{
+  "name": "@ryan_nookpi/pi-extension-auto-name",
+  "version": "0.1.0",
+  "description": "Auto session name extension for pi.",
+  "type": "module",
+  "keywords": [
+    "pi-package"
+  ],
+  "files": [
+    "index.ts",
+    "README.md",
+    "utils"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-ai": "*",
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/utils/auto-name-utils.ts

@@ -0,0 +1,100 @@
+/**
+ * Pure utility functions for auto-name extension.
+ * Extracted for testability — no I/O, no pi SDK dependencies.
+ */
+
+import * as os from "node:os";
+import * as path from "node:path";
+
+// ── Constants ────────────────────────────────────────────────────────────────
+
+/** Must match subagent/session.ts:SUBAGENT_SESSION_DIR */
+export const SUBAGENT_SESSION_DIR = path.join(os.homedir(), ".pi", "agent", "sessions", "subagents");
+
+export const NAME_SYSTEM_PROMPT =
+	"사용자 메시지를 분석해서 세션의 목적을 20자 이내 한 줄로 추출해. 오직 목적 텍스트만 출력하고, 설명이나 다른 텍스트는 절대 출력하지 마.";
+
+/** Max chars for the user message sent to the LLM. */
+export const MAX_MESSAGE_LENGTH = 500;
+
+/** Max chars for the resulting session name. */
+export const MAX_NAME_LENGTH = 30;
+
+/** Max chars shown in the status bar. */
+export const MAX_STATUS_CHARS = 90;
+
+/** Only a fully completed response should be used as a session name. */
+export const SUCCESSFUL_STOP_REASON = "stop";
+
+// ── Pure Functions ───────────────────────────────────────────────────────────
+
+/**
+ * Check if a session file path belongs to the subagent sessions directory.
+ * Returns true if the path starts with SUBAGENT_SESSION_DIR.
+ */
+export function isSubagentSessionPath(sessionFilePath: string | undefined): boolean {
+	if (!sessionFilePath) return false;
+	return (
+		sessionFilePath.startsWith(SUBAGENT_SESSION_DIR + path.sep) ||
+		sessionFilePath.startsWith(`${SUBAGENT_SESSION_DIR}/`)
+	);
+}
+
+/**
+ * Safely extract session file path from an ExtensionContext-like object.
+ * Returns undefined if extraction fails.
+ */
+export function extractSessionFilePath(sessionManager: unknown): string | undefined {
+	try {
+		if (sessionManager && typeof sessionManager === "object" && "getSessionFile" in sessionManager) {
+			const getSessionFile = (sessionManager as Record<string, unknown>).getSessionFile;
+			if (typeof getSessionFile === "function") {
+				const raw = String(getSessionFile() ?? "");
+				const cleaned = raw.replace(/[\r\n\t]+/g, "").trim();
+				return cleaned || undefined;
+			}
+		}
+	} catch {
+		// Ignore errors
+	}
+	return undefined;
+}
+
+/**
+ * Format a session name for status bar display.
+ * Normalizes whitespace and clips to MAX_STATUS_CHARS.
+ */
+export function formatNameStatus(name: string): string {
+	const singleLine = name.replace(/\s+/g, " ").trim();
+	return singleLine.length > MAX_STATUS_CHARS ? `${singleLine.slice(0, MAX_STATUS_CHARS - 1)}…` : singleLine;
+}
+
+/**
+ * Build the user-message text sent to the LLM for name detection.
+ * Truncates to MAX_MESSAGE_LENGTH.
+ */
+export function buildNameContext(userMessage: string): string {
+	return `사용자 메시지: ${userMessage.slice(0, MAX_MESSAGE_LENGTH)}`;
+}
+
+/**
+ * Check whether a model result completed normally.
+ * Only fully completed responses should be used for session naming.
+ */
+export function isSuccessfulResult(stopReason: string | undefined): boolean {
+	return stopReason === SUCCESSFUL_STOP_REASON;
+}
+
+/**
+ * Extract the session name text from an LLM AssistantMessage-like result.
+ * Filters to text content, joins, trims, and clips to MAX_NAME_LENGTH.
+ */
+export function extractNameFromResult(content: ReadonlyArray<{ type: string; text?: string }>): string {
+	const text = content
+		.filter((c): c is { type: "text"; text: string } => c.type === "text" && typeof c.text === "string")
+		.map((c) => c.text)
+		.join("")
+		.trim();
+
+	return text.slice(0, MAX_NAME_LENGTH);
+}

package/utils/short-label.ts

@@ -0,0 +1,76 @@
+import { completeSimple } from "@mariozechner/pi-ai";
+
+type SummaryModel = Parameters<typeof completeSimple>[0];
+type SummaryResult = Awaited<ReturnType<typeof completeSimple>>;
+
+type AuthResult = {
+	ok: boolean;
+	apiKey?: string;
+	headers?: Record<string, string>;
+};
+
+export type ShortLabelContext = {
+	model?: SummaryModel;
+	modelRegistry?: {
+		getApiKeyAndHeaders: (model: SummaryModel) => Promise<AuthResult>;
+	};
+};
+
+export type GenerateShortLabelOptions = {
+	systemPrompt: string;
+	prompt: string;
+	maxTokens?: number;
+	timeoutMs?: number;
+	extractText?: (content: SummaryResult["content"]) => string;
+};
+
+function defaultExtractText(content: SummaryResult["content"]): string {
+	return content
+		.filter((part): part is { type: "text"; text: string } => part.type === "text" && typeof part.text === "string")
+		.map((part) => part.text)
+		.join("")
+		.trim();
+}
+
+export async function generateShortLabel(ctx: ShortLabelContext, options: GenerateShortLabelOptions): Promise<string> {
+	const model = ctx.model;
+	if (!model || !ctx.modelRegistry) return "";
+
+	const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
+	if (!auth.ok) return "";
+
+	const controller = new AbortController();
+	const timeoutMs = options.timeoutMs ?? 10000;
+	const timer = setTimeout(() => controller.abort(), timeoutMs);
+
+	try {
+		const result = await completeSimple(
+			model,
+			{
+				systemPrompt: options.systemPrompt,
+				messages: [
+					{
+						role: "user",
+						content: [{ type: "text", text: options.prompt }],
+						timestamp: Date.now(),
+					},
+				],
+			},
+			{
+				apiKey: auth.apiKey,
+				headers: auth.headers,
+				signal: controller.signal,
+				reasoning: "minimal",
+				maxTokens: options.maxTokens ?? 60,
+			},
+		);
+
+		if (result.stopReason !== "stop") return "";
+		const extractText = options.extractText ?? defaultExtractText;
+		return extractText(result.content);
+	} catch {
+		return "";
+	} finally {
+		clearTimeout(timer);
+	}
+}

package/utils/status-keys.ts

@@ -0,0 +1,2 @@
+export const NAME_STATUS_KEY = "name-footer";
+export const ELAPSED_STATUS_KEY = "elapsed-time";