@pi-unipi/core 0.1.16 → 2.0.0

package/constants.ts

@@ -38,6 +38,7 @@ export const MODULES = {
   FOOTER: "@pi-unipi/footer",
   UPDATER: "@pi-unipi/updater",
   INPUT_SHORTCUTS: "@pi-unipi/input-shortcuts",
+  COCOINDEX: "@pi-unipi/cocoindex",
 } as const;
 
 /** Workflow command names */
@@ -215,6 +216,11 @@ export const MCP_DEFAULTS = {
   TOOL_NAME_SEPARATOR: "__",
 } as const;
 
+/** Compactor sentinel — when passed as customInstructions to ctx.compact(),
+ *  the compactor extension recognizes it and uses its zero-LLM pipeline.
+ *  Other extensions can use this to trigger compactor-aware compaction. */
+export const COMPACTOR_INSTRUCTION = "__compactor__" as const;
+
 /** Compactor tool names */
 export const COMPACTOR_TOOLS = {
   COMPACT: "compact",
@@ -222,24 +228,19 @@ export const COMPACTOR_TOOLS = {
   CTX_EXECUTE: "ctx_execute",
   CTX_EXECUTE_FILE: "ctx_execute_file",
   CTX_BATCH_EXECUTE: "ctx_batch_execute",
-  CTX_INDEX: "ctx_index",
-  CTX_SEARCH: "ctx_search",
-  CTX_FETCH_AND_INDEX: "ctx_fetch_and_index",
   CTX_STATS: "ctx_stats",
   CTX_DOCTOR: "ctx_doctor",
 } as const;
 
 /** Compactor command names */
 export const COMPACTOR_COMMANDS = {
+  LOSSLESS_COMPACT: "lossless-compact",
   COMPACT: "compact",
   COMPACT_RECALL: "compact-recall",
   COMPACT_STATS: "compact-stats",
   COMPACT_DOCTOR: "compact-doctor",
   COMPACT_SETTINGS: "compact-settings",
   COMPACT_PRESET: "compact-preset",
-  COMPACT_INDEX: "compact-index",
-  COMPACT_SEARCH: "compact-search",
-  COMPACT_PURGE: "compact-purge",
 } as const;
 
 /** Compactor directory paths */
@@ -334,5 +335,25 @@ export const COMPACTOR_DEFAULTS = {
   DEFAULT_TIMEOUT_MS: 30000,
   SESSION_TTL_DAYS: 7,
   CACHE_TTL_HOURS: 24,
-  FTS5_CHUNK_SIZE: 4096,
+} as const;
+
+/** CocoIndex minimum supported CLI version. */
+export const COCOINDEX_MIN_VERSION = "1.0" as const;
+
+/** CocoIndex package spec installed by the interactive installer. */
+export const COCOINDEX_PACKAGE_SPEC = `cocoindex[lancedb]>=${COCOINDEX_MIN_VERSION}` as const;
+
+/** CocoIndex tool names */
+export const COCOINDEX_TOOLS = {
+  SEARCH: "cocoindex_search",
+  STATUS: "cocoindex_status",
+} as const;
+
+/** CocoIndex command names */
+export const COCOINDEX_COMMANDS = {
+  UPDATE: "cocoindex-update",
+  STATUS: "cocoindex-status",
+  INIT: "cocoindex-init",
+  SETTINGS: "cocoindex-settings",
+  SEARCH: "cocoindex-search",
 } as const;

package/events.ts

@@ -90,6 +90,13 @@ export const UNIPI_EVENTS = {
   UPDATE_APPLIED: "unipi:update:applied",
   /** Update error */
   UPDATE_ERROR: "unipi:update:error",
+
+  /** CocoIndex: update started */
+  COCOINDEX_UPDATE_STARTED: "unipi:cocoindex:update:started",
+  /** CocoIndex: update completed */
+  COCOINDEX_UPDATE_COMPLETED: "unipi:cocoindex:update:completed",
+  /** CocoIndex: search performed */
+  COCOINDEX_SEARCH_PERFORMED: "unipi:cocoindex:search:performed",
 } as const;
 
 /** Payload for MODULE_READY / MODULE_GONE */
@@ -102,6 +109,8 @@ export interface UnipiModuleEvent {
   commands: string[];
   /** Tools registered by this module */
   tools: string[];
+  /** Load time in milliseconds (optional) */
+  loadTimeMs?: number;
 }
 
 /** Payload for WORKFLOW_START / WORKFLOW_END */
@@ -268,8 +277,6 @@ export interface UnipiCompactorStatsEvent {
   compactions: number;
   /** Tokens saved total */
   tokensSaved: number;
-  /** Indexed documents count */
-  indexedDocs: number;
   /** Sandbox executions count */
   sandboxRuns: number;
   /** Search queries count */
@@ -368,6 +375,32 @@ export interface UnipiUpdateErrorEvent {
   phase: "check" | "install";
 }
 
+/** Payload for COCOINDEX_UPDATE_STARTED */
+export interface UnipiCocoindexUpdateStartedEvent {
+  /** Project directory being indexed */
+  projectDir: string;
+}
+
+/** Payload for COCOINDEX_UPDATE_COMPLETED */
+export interface UnipiCocoindexUpdateCompletedEvent {
+  /** Whether the update succeeded */
+  success: boolean;
+  /** Number of chunks processed */
+  chunksProcessed: number;
+  /** Duration in ms */
+  durationMs: number;
+  /** Error message if failed */
+  error?: string;
+}
+
+/** Payload for COCOINDEX_SEARCH_PERFORMED */
+export interface UnipiCocoindexSearchPerformedEvent {
+  /** Search query */
+  query: string;
+  /** Number of results */
+  resultCount: number;
+}
+
 /** Payload for NOTIFICATION_SENT */
 export interface UnipiNotificationSentEvent {
   /** Event type that triggered notification */
@@ -408,4 +441,7 @@ export type UnipiEventPayload =
   | UnipiUpdateCheckEvent
   | UnipiUpdateAvailableEvent
   | UnipiUpdateAppliedEvent
-  | UnipiUpdateErrorEvent;
+  | UnipiUpdateErrorEvent
+  | UnipiCocoindexUpdateStartedEvent
+  | UnipiCocoindexUpdateCompletedEvent
+  | UnipiCocoindexSearchPerformedEvent;

package/global-types.ts

@@ -0,0 +1,46 @@
+/**
+ * @unipi/core — Minimal interfaces for global registry objects
+ *
+ * These interfaces cover the methods that external packages call.
+ * The actual implementations live in their respective packages
+ * (info-screen, footer, mcp) and may have richer APIs.
+ */
+
+/** Stat data shape returned by info data providers */
+export interface StatDataLike {
+  value: string;
+  detail?: string;
+}
+
+/** Minimal InfoRegistry interface for external consumers */
+export interface InfoRegistryLike {
+  registerGroup(group: {
+    id: string;
+    name: string;
+    icon: string;
+    priority: number;
+    config: {
+      showByDefault: boolean;
+      stats: Array<{ id: string; label: string; show: boolean }>;
+    };
+    dataProvider: () => Promise<Record<string, unknown>>;
+  }): void;
+  getGroupData(groupId: string): Promise<Record<string, unknown>>;
+  getCachedData(groupId: string): Record<string, unknown> | null;
+  invalidateCache(groupId: string): void;
+  refreshGroup(groupId: string): Record<string, unknown> | null;
+  refreshAll(): void;
+}
+
+/** Minimal FooterRegistry interface for external consumers */
+export interface FooterRegistryLike {
+  registerGroup(group: unknown): void;
+}
+
+/** MCP stats shape exposed via globalThis */
+export interface McpStatsLike {
+  serversTotal?: number;
+  serversActive?: number;
+  serversFailed?: number;
+  toolsTotal?: number;
+}

package/global.d.ts

@@ -0,0 +1,23 @@
+/**
+ * @unipi/core — Global type augmentations
+ *
+ * Declares `__unipi_*` properties on `globalThis` so packages
+ * can access shared registries without `as any` casts.
+ */
+
+declare global {
+  // eslint-disable-next-line no-var
+  var __unipi_info_registry: import("./global-types.js").InfoRegistryLike | undefined;
+
+  // eslint-disable-next-line no-var
+  var __unipi_footer_registry: import("./global-types.js").FooterRegistryLike | undefined;
+
+  // eslint-disable-next-line no-var
+  var __unipi_kanboard_registry: unknown;
+
+  // eslint-disable-next-line no-var
+  var __unipi_mcp_stats: import("./global-types.js").McpStatsLike | undefined;
+}
+
+// Force this to be treated as a module
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-unipi/core",
-  "version": "0.1.16",
+  "version": "2.0.0",
   "description": "Shared utilities, event types, and constants for Unipi extension suite",
   "type": "module",
   "license": "MIT",

package/sandbox.ts

@@ -10,20 +10,40 @@ import { WORKFLOW_COMMANDS } from "./constants.js";
 /** Sandbox levels */
 export type SandboxLevel = "read_only" | "brainstorm" | "write_unipi" | "review" | "full";
 
-/** Tool sets per sandbox level */
-const SANDBOX_TOOLS: Record<SandboxLevel, readonly string[]> = {
-  /** Only read-only tools — no bash, no write, no edit */
+/**
+ * Built-in workflow tools used when no active tool list is supplied.
+ *
+ * Workflow commands should normally call filterToolsForLevel() with the current
+ * active tool list. That keeps safe extension tools (memory, ask-user, web,
+ * notify, etc.) available while removing only tools that violate the sandbox.
+ */
+const FALLBACK_TOOLS: Record<SandboxLevel, readonly string[]> = {
+  /** Only read-only file tools — no bash, no write, no edit */
   read_only: ["read", "grep", "find", "ls"],
-  /** Read + constrained write + bash (setup only) — only to .unipi/docs/specs/ */
+  /** Read + constrained write + bash (setup only) — only to .unipi/docs/specs/ by instruction */
   brainstorm: ["read", "grep", "find", "ls", "write", "bash"],
   /** Read + write/edit + file discovery — no bash */
   write_unipi: ["read", "write", "edit", "grep", "find", "ls"],
-  /** Read + write + bash for git operations — no code editing outside .unipi */
+  /** Read + write/edit + bash for git/checks — no code editing by instruction */
   review: ["read", "write", "edit", "grep", "find", "ls", "bash"],
-  /** All tools */
+  /** Full workflow access */
   full: ["read", "write", "edit", "bash"],
 };
 
+/** Tools that are removed from the current active tool set at each level. */
+const BLOCKED_TOOLS: Record<SandboxLevel, readonly string[]> = {
+  /** Read-only workflows must not mutate files or run shell commands. */
+  read_only: ["write", "edit", "bash"],
+  /** Brainstorm may write specs and run limited setup commands, but should not edit existing files. */
+  brainstorm: ["edit"],
+  /** Planning/docs workflows may write .unipi docs, but should not run shell commands. */
+  write_unipi: ["bash"],
+  /** Review workflows rely on command instructions to constrain writes to reviewer remarks. */
+  review: [],
+  /** Full workflows do not filter active tools. */
+  full: [],
+};
+
 /** Command to sandbox level mapping */
 const COMMAND_SANDBOX: Record<string, SandboxLevel> = {
   [WORKFLOW_COMMANDS.BRAINSTORM]: "brainstorm",
@@ -56,25 +76,53 @@ export function getSandboxLevel(commandName: string): SandboxLevel {
 }
 
 /**
- * Get allowed tools for a sandbox level.
+ * Get fallback tools for a sandbox level.
+ *
+ * Prefer filterToolsForLevel(level, activeTools) when applying a sandbox so
+ * extension tools are preserved unless explicitly blocked.
  */
 export function getToolsForLevel(level: SandboxLevel): readonly string[] {
-  return SANDBOX_TOOLS[level];
+  return FALLBACK_TOOLS[level];
+}
+
+/**
+ * Get tool names explicitly blocked by a sandbox level.
+ */
+export function getBlockedToolsForLevel(level: SandboxLevel): readonly string[] {
+  return BLOCKED_TOOLS[level];
+}
+
+/**
+ * Filter the current active tool set for a sandbox level.
+ *
+ * This preserves safe extension tools (memory_search, memory_store, ask_user,
+ * web tools, notifications, etc.) instead of replacing the tool list with a
+ * small built-in allow-list. Only tools that violate the level are removed.
+ */
+export function filterToolsForLevel(
+  level: SandboxLevel,
+  activeTools: readonly string[],
+): readonly string[] {
+  const blocked = new Set(BLOCKED_TOOLS[level]);
+  return activeTools.filter((tool) => !blocked.has(tool));
 }
 
 /**
  * Get allowed tools for a command.
  */
-export function getToolsForCommand(commandName: string): readonly string[] {
+export function getToolsForCommand(
+  commandName: string,
+  activeTools?: readonly string[],
+): readonly string[] {
   const level = getSandboxLevel(commandName);
-  return getToolsForLevel(level);
+  return activeTools ? filterToolsForLevel(level, activeTools) : getToolsForLevel(level);
 }
 
 /**
  * Check if a tool is allowed at a sandbox level.
  */
 export function isToolAllowed(level: SandboxLevel, toolName: string): boolean {
-  return SANDBOX_TOOLS[level].includes(toolName);
+  return !BLOCKED_TOOLS[level].includes(toolName);
 }
 
 /**
@@ -82,7 +130,7 @@ export function isToolAllowed(level: SandboxLevel, toolName: string): boolean {
  */
 export function hasWriteAccess(commandName: string): boolean {
   const level = getSandboxLevel(commandName);
-  return level === "write_unipi" || level === "full";
+  return !BLOCKED_TOOLS[level].includes("write");
 }
 
 /**
@@ -90,5 +138,5 @@ export function hasWriteAccess(commandName: string): boolean {
  */
 export function hasBashAccess(commandName: string): boolean {
   const level = getSandboxLevel(commandName);
-  return level === "full" || level === "brainstorm";
+  return !BLOCKED_TOOLS[level].includes("bash");
 }