@selesai/code 0.2.1 → 0.3.1

package/dist/extensions/pi-powerline-footer/working-vibes.ts

@@ -3,10 +3,9 @@
 // Uses module-level state (matching powerline-footer pattern).
 
 import { complete, type Context } from "@earendil-works/pi-ai";
-import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { getAgentDir, getSettingsPath, type ExtensionContext } from "@selesai/code";
 import { existsSync, readFileSync, writeFileSync, mkdirSync } from "node:fs";
 import { join, dirname } from "node:path";
-import { homedir } from "node:os";
 
 type VibeMode = "generate" | "file";
 
@@ -76,11 +75,6 @@ let recentVibes: string[] = [];
 // Configuration Management
 // ═══════════════════════════════════════════════════════════════════════════
 
-function getSettingsPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "settings.json");
-}
-
 function isRecord(value: unknown): value is Record<string, unknown> {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
@@ -208,8 +202,7 @@ function saveModelConfig(): boolean {
 // ═══════════════════════════════════════════════════════════════════════════
 
 function getVibesDir(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "vibes");
+  return join(getAgentDir(), "vibes");
 }
 
 function toVibeFileSlug(theme: string): string {
@@ -231,7 +224,7 @@ function getVibeFilePath(theme: string): string {
 function loadVibesFromFile(theme: string): string[] {
   const filePath = getVibeFilePath(theme);
   if (!existsSync(filePath)) return [];
-  
+
   try {
     const content = readFileSync(filePath, "utf-8");
     return content
@@ -247,12 +240,12 @@ function loadVibesFromFile(theme: string): string[] {
 function saveVibesToFile(theme: string, vibes: string[]): void {
   const vibesDir = getVibesDir();
   const filePath = getVibeFilePath(theme);
-  
+
   // Ensure directory exists
   if (!existsSync(vibesDir)) {
     mkdirSync(vibesDir, { recursive: true });
   }
-  
+
   writeFileSync(filePath, vibes.join("\n"));
 }
 
@@ -269,26 +262,26 @@ function mulberry32(seed: number): () => number {
 // Get vibe at index using seeded shuffle (no-repeat until all used)
 function getVibeAtIndex(vibes: string[], index: number, seed: number): string {
   if (vibes.length === 0) return `${config.fallback}...`;
-  
+
   // For small lists or when we've cycled through, just use modulo
   const effectiveIndex = index % vibes.length;
-  
+
   // Create deterministic shuffle using seed
   const rng = mulberry32(seed);
   const indices = Array.from({ length: vibes.length }, (_, i) => i);
-  
+
   // Fisher-Yates shuffle with seeded RNG
   for (let i = indices.length - 1; i > 0; i--) {
     const j = Math.floor(rng() * (i + 1));
     [indices[i], indices[j]] = [indices[j], indices[i]];
   }
-  
+
   return vibes[indices[effectiveIndex]];
 }
 
 function getNextVibeFromFile(): string {
   if (!config.theme) return `${config.fallback}...`;
-  
+
   // Load/reload cache if theme changed
   if (vibeCacheTheme !== config.theme) {
     vibeCache = loadVibesFromFile(config.theme);
@@ -296,11 +289,11 @@ function getNextVibeFromFile(): string {
     vibeSeed = Date.now();  // New seed for new theme
     vibeIndex = 0;
   }
-  
+
   if (vibeCache.length === 0) {
     return `${config.fallback}...`;
   }
-  
+
   const vibe = getVibeAtIndex(vibeCache, vibeIndex, vibeSeed);
   vibeIndex++;
   return vibe;
@@ -313,12 +306,12 @@ function getNextVibeFromFile(): string {
 function buildVibePrompt(ctx: VibeGenContext): string {
   // Truncate user prompt to save tokens (most context in first 100 chars)
   const task = ctx.userPrompt.slice(0, 100);
-  
+
   // Build exclusion list from recent vibes
-  const exclude = recentVibes.length > 0 
+  const exclude = recentVibes.length > 0
     ? `Don't use: ${recentVibes.join(", ")}`
     : "";
-  
+
   // Use configured template with variable substitution
   return config.promptTemplate
     .replace(/\{theme\}/g, ctx.theme)
@@ -328,28 +321,28 @@ function buildVibePrompt(ctx: VibeGenContext): string {
 
 function parseVibeResponse(response: string, fallback: string): string {
   if (!response) return `${fallback}...`;
-  
+
   // Take only the first line (AI sometimes adds explanations)
   let vibe = response.trim().split('\n')[0].trim();
-  
+
   // Remove quotes if model wrapped the response
   vibe = vibe.replace(/^["']|["']$/g, "");
-  
+
   // Ensure ellipsis
   if (!vibe.endsWith("...")) {
     vibe = vibe.replace(/\.+$/, "") + "...";
   }
-  
+
   // Enforce length limit (configurable, default 65 chars)
   if (vibe.length > config.maxLength) {
     vibe = vibe.slice(0, config.maxLength - 3) + "...";
   }
-  
+
   // Final validation
   if (!vibe || vibe === "...") {
     return `${fallback}...`;
   }
-  
+
   return vibe;
 }
 
@@ -375,7 +368,7 @@ async function generateVibe(
   if (!extensionCtx) {
     return `${config.fallback}...`;
   }
-  
+
   // Parse model spec (provider/modelId format, where modelId may contain slashes)
   const slashIndex = config.modelSpec.indexOf("/");
   if (slashIndex === -1) {
@@ -386,23 +379,23 @@ async function generateVibe(
   if (!provider || !modelId) {
     return `${config.fallback}...`;
   }
-  
+
   // Resolve model from registry
   const model = extensionCtx.modelRegistry.find(provider, modelId);
   if (!model) {
     console.debug(`[working-vibes] Model not found: ${config.modelSpec}`);
     return `${config.fallback}...`;
   }
-  
+
   // Get auth
   const auth = await extensionCtx.modelRegistry.getApiKeyAndHeaders(model);
   if (!auth.ok) {
     console.debug(`[working-vibes] Auth failed for ${provider}: ${auth.error}`);
     return `${config.fallback}...`;
   }
-  
+
   const aiContext = buildAiContext(buildVibePrompt(ctx));
-  
+
   const response = await complete(model, aiContext, { apiKey: auth.apiKey, headers: auth.headers, signal });
 
   const textContent = response.content.find(c => c.type === "text");
@@ -415,7 +408,7 @@ async function generateVibe(
 function trackRecentVibe(vibe: string): void {
   // Don't track fallback messages
   if (vibe === `${config.fallback}...`) return;
-  
+
   // Add to front, remove duplicates
   recentVibes = [vibe, ...recentVibes.filter(v => v !== vibe)].slice(0, MAX_RECENT_VIBES);
 }
@@ -425,7 +418,7 @@ function updateVibeFromFile(setWorkingMessage: (msg?: string) => void): void {
 }
 
 async function generateAndUpdate(
-  prompt: string, 
+  prompt: string,
   setWorkingMessage: (msg?: string) => void,
 ): Promise<void> {
   // File mode: instant, no API call
@@ -433,27 +426,27 @@ async function generateAndUpdate(
     updateVibeFromFile(setWorkingMessage);
     return;
   }
-  
+
   // Generate mode: API call with abort handling
   // Cancel any in-flight generation and create new controller
   // Capture in local variable to avoid race condition with subsequent calls
   const controller = new AbortController();
   currentGeneration?.abort();
   currentGeneration = controller;
-  
+
   // Create timeout signal (3 seconds)
   const timeoutSignal = AbortSignal.timeout(config.timeout);
   const combinedSignal = AbortSignal.any([
     controller.signal,
     timeoutSignal,
   ]);
-  
+
   try {
     const vibe = await generateVibe(
       { theme: config.theme!, userPrompt: prompt },
       combinedSignal,
     );
-    
+
     // Only update if still streaming and THIS generation wasn't aborted
     if (isStreaming && !controller.signal.aborted) {
       trackRecentVibe(vibe);
@@ -499,19 +492,19 @@ export function setVibeModel(modelSpec: string): boolean {
 }
 
 export function onVibeBeforeAgentStart(
-  prompt: string, 
+  prompt: string,
   setWorkingMessage: (msg?: string) => void,
 ): void {
   // Skip if no theme configured or no extensionCtx
   if (!config.theme || !extensionCtx) return;
-  
+
   // Queue themed placeholder BEFORE agent_start creates the loader
   // This sets pendingWorkingMessage which is applied when loader is created
   setWorkingMessage(`Channeling ${config.theme}...`);
-  
+
   // Mark vibe generation time for rate limiting
   lastVibeTime = Date.now();
-  
+
   // Async: generate and update (fire-and-forget, don't await)
   generateAndUpdate(prompt, setWorkingMessage);
 }
@@ -528,11 +521,11 @@ export function onVibeToolCall(
 ): void {
   // Skip if no theme, not streaming, or no extensionCtx
   if (!config.theme || !extensionCtx || !isStreaming) return;
-  
+
   // Rate limit: skip if not enough time has passed
   const now = Date.now();
   if (now - lastVibeTime < config.refreshInterval) return;
-  
+
   // Prefer agent context if provided (richer, more contextual)
   // Fall back to tool-based hint
   let hint: string;
@@ -553,7 +546,7 @@ export function onVibeToolCall(
       hint = `running command: ${cmd}`;
     }
   }
-  
+
   // Update time and generate new vibe
   lastVibeTime = now;
   generateAndUpdate(hint, setWorkingMessage);
@@ -613,11 +606,11 @@ export async function generateVibesBatch(
 ): Promise<GenerateVibesResult> {
   const filePath = getVibeFilePath(theme);
   const safeCount = Number.isFinite(count) ? Math.min(Math.max(Math.floor(count), 1), 500) : 100;
-  
+
   if (!extensionCtx) {
     return { success: false, count: 0, filePath, error: "Extension not initialized" };
   }
-  
+
   // Parse model spec
   const slashIndex = config.modelSpec.indexOf("/");
   if (slashIndex === -1) {
@@ -625,31 +618,31 @@ export async function generateVibesBatch(
   }
   const provider = config.modelSpec.slice(0, slashIndex);
   const modelId = config.modelSpec.slice(slashIndex + 1);
-  
+
   // Resolve model
   const model = extensionCtx.modelRegistry.find(provider, modelId);
   if (!model) {
     return { success: false, count: 0, filePath, error: `Model not found: ${config.modelSpec}` };
   }
-  
+
   // Get auth
   const auth = await extensionCtx.modelRegistry.getApiKeyAndHeaders(model);
   if (!auth.ok) {
     return { success: false, count: 0, filePath, error: auth.error };
   }
-  
+
   // Build batch prompt
   const prompt = BATCH_PROMPT
     .replace(/\{theme\}/g, theme)
     .replace(/\{count\}/g, String(safeCount));
-  
+
   const aiContext = buildAiContext(prompt);
-  
+
   try {
     // Use longer timeout for batch generation (30 seconds)
     const signal = AbortSignal.timeout(30000);
     const response = await complete(model, aiContext, { apiKey: auth.apiKey, headers: auth.headers, signal });
-    
+
     const textContent = response.content.find(c => c.type === "text");
     if (!textContent?.text) {
       const error = response.stopReason === "error" && response.errorMessage
@@ -657,7 +650,7 @@ export async function generateVibesBatch(
         : "Empty response from model";
       return { success: false, count: 0, filePath, error };
     }
-    
+
     // Parse response: one vibe per line
     const vibes = textContent.text
       .split("\n")
@@ -673,20 +666,20 @@ export async function generateVibesBatch(
         return vibe;
       })
       .filter(vibe => vibe.length > 3 && vibe !== "...");  // Filter invalid
-    
+
     if (vibes.length === 0) {
       return { success: false, count: 0, filePath, error: "No valid vibes generated" };
     }
-    
+
     // Save to file
     saveVibesToFile(theme, vibes);
-    
+
     // Clear cache so next use loads fresh
     if (vibeCacheTheme === theme) {
       vibeCache = [];
       vibeCacheTheme = null;
     }
-    
+
     return { success: true, count: vibes.length, filePath };
   } catch (error) {
     const message = error instanceof Error ? error.message : "Unknown error";

package/dist/extensions/pi-rewind-hook/index.ts

@@ -1,4 +1,4 @@
-import { getAgentDir, type ExtensionAPI, type ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { getAgentDir, type ExtensionAPI, type ExtensionContext } from "@selesai/code";
 import { exec as execCb } from "child_process";
 import { existsSync, readFileSync, realpathSync } from "fs";
 import { mkdtemp, readdir, readFile, rm, stat } from "fs/promises";

package/dist/extensions/pi-rewind-hook/package.json

@@ -24,6 +24,6 @@
     "test": "npx --yes tsx --tsconfig tsconfig.test.json --test index.test.ts"
   },
   "devDependencies": {
-    "@mariozechner/pi-coding-agent": "^0.51.0"
+    "@selesai/code": "*"
   }
 }