@selesai/code 0.3.0 → 0.3.1

package/dist/extensions/caveman/test/extension.test.js

@@ -0,0 +1,185 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import cavemanExtension from "../index.js";
+
+function createPiHarness() {
+  const events = new Map();
+  const commands = new Map();
+  const appendedEntries = [];
+  const sentUserMessages = [];
+
+  const pi = {
+    on(eventName, handler) { events.set(eventName, handler); },
+    registerCommand(name, options) { commands.set(name, options); },
+    appendEntry(customType, data) { appendedEntries.push({ customType, data }); },
+    sendUserMessage(text, options) { sentUserMessages.push({ text, options }); },
+  };
+
+  cavemanExtension(pi);
+  return { events, commands, appendedEntries, sentUserMessages };
+}
+
+function createCommandContext(overrides = {}) {
+  return {
+    isIdle: () => true,
+    sessionManager: { getEntries: () => [] },
+    ui: { notify() {} },
+    ...overrides,
+  };
+}
+
+test("extension registers the /caveman command", () => {
+  const { commands } = createPiHarness();
+  assert.deepEqual([...commands.keys()], ["caveman"]);
+});
+
+test("/caveman on/off appends a caveman-mode entry", async () => {
+  const { commands, events, appendedEntries } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await commands.get("caveman").handler("off", ctx);
+
+  assert.deepEqual(appendedEntries.at(-1), { customType: "caveman-mode", data: { active: false } });
+
+  await commands.get("caveman").handler("on", ctx);
+  assert.deepEqual(appendedEntries.at(-1), { customType: "caveman-mode", data: { active: true } });
+});
+
+test("bare /caveman toggles off then back on", async () => {
+  const { commands, events, appendedEntries } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx); // starts on
+  await commands.get("caveman").handler("", ctx);
+  assert.equal(appendedEntries.at(-1).data.active, false);
+  await commands.get("caveman").handler("", ctx);
+  assert.equal(appendedEntries.at(-1).data.active, true);
+});
+
+test("before_agent_start injects instructions while active", async () => {
+  const { events } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
+
+  assert.ok(result.systemPrompt.startsWith("BASE"));
+  assert.ok(result.systemPrompt.includes("CAVEMAN MODE ACTIVE"));
+});
+
+test("disabled caveman skips instruction injection", async () => {
+  const { commands, events } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await commands.get("caveman").handler("off", ctx);
+  const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
+
+  assert.equal(result, undefined);
+});
+
+test("session_start restores latest persisted active flag", async () => {
+  const { events } = createPiHarness();
+  const ctx = createCommandContext({
+    sessionManager: {
+      getEntries: () => [
+        { type: "custom", customType: "caveman-mode", data: { active: false } },
+      ],
+    },
+  });
+
+  await events.get("session_start")({ reason: "resume" }, ctx);
+  const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
+
+  assert.equal(result, undefined); // persisted false -> no injection
+});
+
+test("'stop caveman' as a whole message deactivates mid-session", async () => {
+  const { events } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await events.get("input")({ text: "stop caveman", source: "interactive" }, ctx);
+
+  const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
+  assert.equal(result, undefined);
+});
+
+test("a request merely mentioning 'stop caveman' stays active", async () => {
+  const { events } = createPiHarness();
+  const ctx = createCommandContext();
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await events.get("input")({ text: "how do I stop caveman re-rendering?", source: "interactive" }, ctx);
+
+  const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
+  assert.ok(result.systemPrompt.includes("CAVEMAN MODE ACTIVE"));
+});
+
+test("status bar renders ON and flips active on agent_start", async () => {
+  const { events } = createPiHarness();
+  const statusWrites = [];
+  const ctx = createCommandContext({
+    sessionManager: { getEntries: () => [] },
+    ui: { notify() {}, setStatus: (key, text) => statusWrites.push({ key, text }), theme: { fg: (_c, t) => t } },
+  });
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await events.get("agent_start")({}, ctx);
+
+  assert.equal(statusWrites.at(-2).key, "caveman");
+  assert.match(statusWrites.at(-2).text, /○.*ON/);
+  assert.match(statusWrites.at(-1).text, /●.*ON/);
+});
+
+test("status bar clears when disabled", async () => {
+  const { events, commands } = createPiHarness();
+  const statusWrites = [];
+  const ctx = createCommandContext({
+    ui: { notify() {}, setStatus: (key, text) => statusWrites.push({ key, text }), theme: { fg: (_c, t) => t } },
+  });
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await commands.get("caveman").handler("off", ctx);
+
+  assert.deepEqual(statusWrites.at(-1), { key: "caveman", text: "" });
+});
+
+test("status bar stays silent when ui lacks a theme", async () => {
+  const { events } = createPiHarness();
+  const calls = [];
+  const ctx = createCommandContext({
+    sessionManager: { getEntries: () => [] },
+    ui: { notify() {}, setStatus: (_key, text) => calls.push(text) }, // no theme
+  });
+
+  await events.get("session_start")({ reason: "startup" }, ctx);
+  await events.get("agent_start")({}, ctx);
+
+  assert.deepEqual(calls, []);
+});
+
+test("invalid subcommand notifies warning and does not append", async () => {
+  const { commands, appendedEntries } = createPiHarness();
+  let notified = null;
+  const ctx = createCommandContext({ ui: { notify: (_m, kind) => { notified = kind; } } });
+
+  await commands.get("caveman").handler("banana", ctx);
+
+  assert.equal(notified, "warning");
+  assert.equal(appendedEntries.length, 0);
+});
+
+test("status subcommand reports without changing state", async () => {
+  const { commands, events, appendedEntries } = createPiHarness();
+  let reported = null;
+  const ctx = createCommandContext({ ui: { notify: (m) => { reported = m; } } });
+
+  await events.get("session_start")({ reason: "startup" }, ctx); // starts on
+  await commands.get("caveman").handler("status", ctx);
+
+  assert.match(reported, /ON/);
+  assert.equal(appendedEntries.length, 0);
+});

package/dist/extensions/caveman/test/helpers.test.js

@@ -0,0 +1,67 @@
+import assert from "node:assert/strict";
+import { readFileSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+import test from "node:test";
+
+import {
+  getCavemanInstructions,
+  isDeactivationCommand,
+  parseCavemanCommand,
+  resolveSessionActive,
+} from "../index.js";
+
+test("parseCavemanCommand: bare toggles", () => {
+  assert.deepEqual(parseCavemanCommand(""), { type: "toggle" });
+  assert.deepEqual(parseCavemanCommand("   "), { type: "toggle" });
+});
+
+test("parseCavemanCommand: on/off/status", () => {
+  assert.deepEqual(parseCavemanCommand("on"), { type: "set", active: true });
+  assert.deepEqual(parseCavemanCommand("ENABLE"), { type: "set", active: true });
+  assert.deepEqual(parseCavemanCommand("off"), { type: "set", active: false });
+  assert.deepEqual(parseCavemanCommand("disable"), { type: "set", active: false });
+  assert.deepEqual(parseCavemanCommand("status"), { type: "status" });
+});
+
+test("parseCavemanCommand: unknown arg is invalid", () => {
+  assert.deepEqual(parseCavemanCommand("banana"), { type: "invalid", reason: "unknown-subcommand", arg: "banana" });
+});
+
+test("resolveSessionActive: latest persisted entry wins", () => {
+  const entries = [
+    { type: "custom", customType: "caveman-mode", data: { active: false } },
+    { type: "custom", customType: "caveman-mode", data: { active: true } },
+  ];
+  assert.equal(resolveSessionActive(entries, true), true);
+});
+
+test("resolveSessionActive: non-array entries fall back", () => {
+  assert.equal(resolveSessionActive(null, true), true);
+  assert.equal(resolveSessionActive(undefined, false), false);
+  assert.equal(resolveSessionActive({}, true), true);
+  assert.equal(resolveSessionActive("nope"), true); // DEFAULT fallback
+});
+
+test("isDeactivationCommand: only the whole message, case-insensitive, trailing punct ok", () => {
+  assert.equal(isDeactivationCommand("stop caveman"), true);
+  assert.equal(isDeactivationCommand("STOP CAVEMAN!"), true);
+  assert.equal(isDeactivationCommand("normal mode"), true);
+  assert.equal(isDeactivationCommand("Normal Mode."), true);
+  assert.equal(isDeactivationCommand("add a normal mode toggle"), false);
+  assert.equal(isDeactivationCommand("how do I stop caveman re-rendering"), false);
+});
+
+test("getCavemanInstructions reads SKILL.md and strips frontmatter", () => {
+  const instructions = getCavemanInstructions();
+  assert.ok(instructions.startsWith("CAVEMAN MODE ACTIVE"));
+  assert.ok(!instructions.includes("name: caveman")); // frontmatter gone
+  assert.ok(instructions.includes("Respond terse like smart caveman"));
+});
+
+test("getCavemanInstructions matches src/skills/caveman/SKILL.md body", () => {
+  const skillPath = fileURLToPath(new URL("../../../skills/caveman/SKILL.md", import.meta.url));
+  const raw = readFileSync(skillPath, "utf8");
+  const body = raw.replace(/^---[\s\S]*?---\s*/, "");
+
+  assert.ok(getCavemanInstructions().endsWith(body));
+});

package/dist/extensions/pi-intercom/broker/broker.ts

@@ -1,13 +1,12 @@
 import net from "net";
 import { writeFileSync, unlinkSync, mkdirSync } from "fs";
 import { join } from "path";
-import { homedir } from "os";
 import { randomUUID } from "crypto";
 import { writeMessage, createMessageReader } from "./framing.js";
-import { getBrokerSocketPath } from "./paths.js";
+import { getBrokerSocketPath, getIntercomDir } from "./paths.js";
 import type { SessionInfo, Message, Attachment, BrokerMessage } from "../types.js";
 
-const INTERCOM_DIR = join(homedir(), ".pi/agent/intercom");
+const INTERCOM_DIR = getIntercomDir();
 const SOCKET_PATH = getBrokerSocketPath();
 const PID_PATH = join(INTERCOM_DIR, "broker.pid");
 

package/dist/extensions/pi-intercom/broker/paths.ts

@@ -1,5 +1,5 @@
 import { join } from "path";
-import { homedir } from "os";
+import { getAgentDir } from "@selesai/code";
 
 function sanitizePipeSegment(value: string): string {
   return value
@@ -8,13 +8,18 @@ function sanitizePipeSegment(value: string): string {
     .toLowerCase() || "default";
 }
 
+/** selesai config dir (~/.selesai/agent on the selesai fork, ~/.pi/agent upstream). */
+export function getIntercomDir(): string {
+  return join(getAgentDir(), "intercom");
+}
+
 export function getBrokerSocketPath(
   platform: NodeJS.Platform = process.platform,
-  homeDir: string = homedir(),
+  intercomDir: string = getIntercomDir(),
 ): string {
   if (platform === "win32") {
-    return `\\\\.\\pipe\\pi-intercom-${sanitizePipeSegment(homeDir)}`;
+    return `\\\\.\\pipe\\pi-intercom-${sanitizePipeSegment(intercomDir)}`;
   }
 
-  return join(homeDir, ".pi/agent/intercom/broker.sock");
-}
+  return join(intercomDir, "broker.sock");
+}

package/dist/extensions/pi-intercom/broker/spawn.ts

@@ -2,11 +2,10 @@ import { spawn } from "child_process";
 import { existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync } from "fs";
 import { join, dirname } from "path";
 import { fileURLToPath } from "url";
-import { homedir } from "os";
 import net from "net";
-import { getBrokerSocketPath } from "./paths.js";
+import { getBrokerSocketPath, getIntercomDir } from "./paths.js";
 
-const INTERCOM_DIR = join(homedir(), ".pi/agent/intercom");
+const INTERCOM_DIR = getIntercomDir();
 const EXTENSION_DIR = join(dirname(fileURLToPath(import.meta.url)), "..");
 const BROKER_SOCKET = getBrokerSocketPath();
 const BROKER_PID = join(INTERCOM_DIR, "broker.pid");

package/dist/extensions/pi-intercom/config.ts

@@ -1,6 +1,6 @@
 import { existsSync, readFileSync } from "fs";
 import { join } from "path";
-import { homedir } from "os";
+import { getIntercomDir } from "./broker/paths.js";
 
 export interface IntercomConfig {
   /** Broker command used to spawn the broker process (e.g. "npx" or "bun") */
@@ -22,7 +22,7 @@ export interface IntercomConfig {
   replyHint: boolean;
 }
 
-const CONFIG_PATH = join(homedir(), ".pi/agent/intercom/config.json");
+const CONFIG_PATH = join(getIntercomDir(), "config.json");
 
 const defaults: IntercomConfig = {
   brokerCommand: "npx",

package/dist/extensions/pi-intercom/index.ts

@@ -1,7 +1,7 @@
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@selesai/code";
 import { randomUUID } from "crypto";
 import { Type } from "typebox";
-import { Text } from "@mariozechner/pi-tui";
+import { Text } from "@earendil-works/pi-tui";
 import { IntercomClient } from "./broker/client.ts";
 import { spawnBrokerIfNeeded } from "./broker/spawn.ts";
 import { SessionListOverlay } from "./ui/session-list.ts";

package/dist/extensions/pi-intercom/package.json

@@ -28,8 +28,8 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@selesai/code": "*",
+    "@earendil-works/pi-tui": "*"
   },
   "dependencies": {
     "tsx": "^4.20.0",

package/dist/extensions/pi-intercom/ui/compose.ts

@@ -1,6 +1,6 @@
-import type { Component, TUI } from "@mariozechner/pi-tui";
-import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
-import type { KeybindingsManager, Theme } from "@mariozechner/pi-coding-agent";
+import type { Component, TUI } from "@earendil-works/pi-tui";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import type { KeybindingsManager, Theme } from "@selesai/code";
 import type { IntercomClient } from "../broker/client.js";
 import type { SessionInfo } from "../types.js";
 

package/dist/extensions/pi-intercom/ui/inline-message.ts

@@ -1,6 +1,6 @@
-import type { Component } from "@mariozechner/pi-tui";
-import { truncateToWidth, visibleWidth, wrapTextWithAnsi } from "@mariozechner/pi-tui";
-import type { Theme } from "@mariozechner/pi-coding-agent";
+import type { Component } from "@earendil-works/pi-tui";
+import { truncateToWidth, visibleWidth, wrapTextWithAnsi } from "@earendil-works/pi-tui";
+import type { Theme } from "@selesai/code";
 import type { SessionInfo, Message } from "../types.js";
 
 export class InlineMessageComponent implements Component {

package/dist/extensions/pi-intercom/ui/session-list.ts

@@ -1,6 +1,6 @@
-import type { Component } from "@mariozechner/pi-tui";
-import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
-import type { KeybindingsManager, Theme } from "@mariozechner/pi-coding-agent";
+import type { Component } from "@earendil-works/pi-tui";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import type { KeybindingsManager, Theme } from "@selesai/code";
 import type { SessionInfo } from "../types.js";
 
 function middleTruncate(text: string, maxWidth: number): string {

package/dist/extensions/pi-powerline-footer/bash-mode/history.ts

@@ -1,6 +1,7 @@
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { dirname, join } from "node:path";
 import { homedir } from "node:os";
+import { getAgentDir } from "@selesai/code";
 
 interface PersistedHistoryEntry {
   command: string;
@@ -13,7 +14,7 @@ function getHomeDir(): string {
 }
 
 function getHistoryDir(): string {
-  return join(getHomeDir(), ".pi", "agent", "powerline-footer", "bash-history");
+  return join(getAgentDir(), "powerline-footer", "bash-history");
 }
 
 function projectKey(cwd: string): string {

package/dist/extensions/pi-powerline-footer/index.ts

@@ -1,5 +1,8 @@
 import {
   copyToClipboard,
+  CONFIG_DIR_NAME,
+  getAgentDir,
+  getSettingsPath,
   type ExtensionAPI,
   type ReadonlyFooterDataProvider,
   type Theme,
@@ -8,7 +11,6 @@ import type { AssistantMessage } from "@earendil-works/pi-ai";
 import { isKeyRelease, matchesKey, type AutocompleteProvider, type SelectItem, SelectList, truncateToWidth, TUI_KEYBINDINGS, visibleWidth } from "@earendil-works/pi-tui";
 import { readFileSync, writeFileSync, existsSync, mkdirSync, readdirSync } from "node:fs";
 import { join, dirname } from "node:path";
-import { homedir } from "node:os";
 
 import type { ColorScheme, SegmentContext, StatusLinePreset, StatusLineSegmentId } from "./types.ts";
 import type { PowerlineConfig } from "./powerline-config.ts";
@@ -324,23 +326,16 @@ function trackPromptHistory(editor: any): void {
   snapshotPromptHistory(editor);
 }
 
-function getSettingsPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "settings.json");
-}
-
 function getProjectSettingsPath(cwd: string): string {
-  return join(cwd, ".pi", "settings.json");
+  return join(cwd, CONFIG_DIR_NAME, "settings.json");
 }
 
 function getGlobalCompactionPolicyPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "compaction-policy.json");
+  return join(getAgentDir(), "compaction-policy.json");
 }
 
 function getCustomCompactionExtensionPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "extensions", "pi-custom-compaction");
+  return join(getAgentDir(), "extensions", "pi-custom-compaction");
 }
 
 function mergeSettings(base: Record<string, unknown>, override: Record<string, unknown>): Record<string, unknown> {
@@ -413,7 +408,7 @@ function readCompactionPolicyEnabled(configPath: string): boolean | undefined {
 function detectCustomCompactionEnabled(cwd: string): boolean {
   if (!existsSync(getCustomCompactionExtensionPath())) return false;
 
-  const projectSetting = readCompactionPolicyEnabled(join(cwd, ".pi", "compaction-policy.json"));
+  const projectSetting = readCompactionPolicyEnabled(join(cwd, CONFIG_DIR_NAME, "compaction-policy.json"));
   if (projectSetting !== undefined) return projectSetting;
 
   return readCompactionPolicyEnabled(getGlobalCompactionPolicyPath()) ?? false;
@@ -424,13 +419,11 @@ function isRecord(value: unknown): value is Record<string, unknown> {
 }
 
 function getStashHistoryPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "powerline-footer", "stash-history.json");
+  return join(getAgentDir(), "powerline-footer", "stash-history.json");
 }
 
 function getSessionsPath(): string {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
-  return join(homeDir, ".pi", "agent", "sessions");
+  return join(getAgentDir(), "sessions");
 }
 
 function getProjectSessionsPath(cwd: string): string {

package/dist/extensions/pi-powerline-footer/welcome.ts

@@ -1,6 +1,7 @@
 import { readdirSync, existsSync, statSync, readFileSync } from "node:fs";
 import { join, basename } from "node:path";
 import { homedir as osHomedir } from "node:os";
+import { CONFIG_DIR_NAME, getAgentDir, getSettingsPath } from "@selesai/code";
 import type { Component } from "@earendil-works/pi-tui";
 import { truncateToWidth as tuiTruncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 import { ansi, fgOnly, getFgAnsiCode } from "./colors.ts";
@@ -330,11 +331,16 @@ function logDiscoveryError(scope: string, error: unknown): void {
   console.debug(`[powerline-welcome] ${scope}:`, error);
 }
 
+// ponytail: only .claude paths still need raw home; everything else routes via getAgentDir.
+function homedirSafe(): string {
+  return process.env.HOME || process.env.USERPROFILE || osHomedir();
+}
+
 /**
  * Discover loaded counts by scanning filesystem.
  */
 export function discoverLoadedCounts(): LoadedCounts {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || osHomedir();
+  const homeDir = getAgentDir();
   const cwd = process.cwd();
 
   let contextFiles = 0;
@@ -343,10 +349,10 @@ export function discoverLoadedCounts(): LoadedCounts {
   let promptTemplates = 0;
 
   const agentsMdPaths = [
-    join(homeDir, ".pi", "agent", "AGENTS.md"),
-    join(homeDir, ".claude", "AGENTS.md"),
+    join(homeDir, "AGENTS.md"),
+    join(homedirSafe(), ".claude", "AGENTS.md"),
     join(cwd, "AGENTS.md"),
-    join(cwd, ".pi", "AGENTS.md"),
+    join(cwd, CONFIG_DIR_NAME, "AGENTS.md"),
     join(cwd, ".claude", "AGENTS.md"),
   ];
 
@@ -355,16 +361,16 @@ export function discoverLoadedCounts(): LoadedCounts {
   }
 
   const extensionDirs = [
-    join(homeDir, ".pi", "agent", "extensions"),
+    join(homeDir, "extensions"),
     join(cwd, "extensions"),
-    join(cwd, ".pi", "extensions"),
+    join(cwd, CONFIG_DIR_NAME, "extensions"),
   ];
 
   const countedExtensions = new Set<string>();
 
   const settingsPaths = [
-    join(homeDir, ".pi", "agent", "settings.json"),
-    join(cwd, ".pi", "settings.json"),
+    getSettingsPath(),
+    join(cwd, CONFIG_DIR_NAME, "settings.json"),
   ];
 
   for (const settingsPath of settingsPaths) {
@@ -460,8 +466,8 @@ export function discoverLoadedCounts(): LoadedCounts {
   }
 
   const skillDirs = [
-    join(homeDir, ".pi", "agent", "skills"),
-    join(cwd, ".pi", "skills"),
+    join(homeDir, "skills"),
+    join(cwd, CONFIG_DIR_NAME, "skills"),
     join(cwd, "skills"),
   ];
 
@@ -493,9 +499,9 @@ export function discoverLoadedCounts(): LoadedCounts {
   }
 
   const templateDirs = [
-    join(homeDir, ".pi", "agent", "commands"),
-    join(homeDir, ".claude", "commands"),
-    join(cwd, ".pi", "commands"),
+    join(homeDir, "commands"),
+    join(homedirSafe(), ".claude", "commands"),
+    join(cwd, CONFIG_DIR_NAME, "commands"),
     join(cwd, ".claude", "commands"),
   ];
 
@@ -538,11 +544,12 @@ export function discoverLoadedCounts(): LoadedCounts {
  * Get recent sessions from the sessions directory.
  */
 export function getRecentSessions(maxCount: number = 3): RecentSession[] {
-  const homeDir = process.env.HOME || process.env.USERPROFILE || osHomedir();
+  const homeDir = getAgentDir();
+  const home = homedirSafe();
 
   const sessionsDirs = [
-    join(homeDir, ".pi", "agent", "sessions"),
-    join(homeDir, ".pi", "sessions"),
+    join(homeDir, "sessions"),
+    join(home, CONFIG_DIR_NAME, "sessions"),
   ];
 
   const sessions: { name: string; mtime: number }[] = [];

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 "@selesai/code";
+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 {

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": "*"
   }
 }