pi-ui-extend 0.1.3 → 0.1.4

package/external/pi-tools-suite/src/{compress → dcp}/config.ts

@@ -220,48 +220,6 @@ function readDcpFromSuiteConfig(filePath: string): Record<string, unknown> {
   return dcp as Record<string, unknown>
 }
 
-/**
- * Walk up from `startDir` looking for `.pi/dcp.jsonc`.
- * Returns the path if found, otherwise null.
- */
-function findProjectConfig(startDir: string): string | null {
-  let dir = path.resolve(startDir)
-  const root = path.parse(dir).root
-
-  while (true) {
-    const candidate = path.join(dir, ".pi", "dcp.jsonc")
-    if (fs.existsSync(candidate)) return candidate
-    if (dir === root) return null
-    const parent = path.dirname(dir)
-    if (parent === dir) return null
-    dir = parent
-  }
-}
-
-/**
- * Walk up from `startDir` looking for `.pi/pi-tools-suite.jsonc`.
- * Returns the path if found, otherwise null.
- */
-function findProjectSuiteConfig(startDir: string): string | null {
-  let dir = path.resolve(startDir)
-  const root = path.parse(dir).root
-
-  while (true) {
-    const candidate = path.join(dir, ".pi", "pi-tools-suite.jsonc")
-    if (fs.existsSync(candidate)) return candidate
-    if (dir === root) return null
-    const parent = path.dirname(dir)
-    if (parent === dir) return null
-    dir = parent
-  }
-}
-
-function mergeConfigFile(config: DcpConfig, filePath: string): DcpConfig {
-  const raw = readJsoncFile(filePath)
-  if (Object.keys(raw).length === 0) return config
-  return deepMerge(config, raw as Partial<DcpConfig>)
-}
-
 function mergeSuiteDcpConfig(config: DcpConfig, filePath: string): DcpConfig {
   const raw = readDcpFromSuiteConfig(filePath)
   if (Object.keys(raw).length === 0) return config
@@ -272,39 +230,21 @@ function mergeSuiteDcpConfig(config: DcpConfig, filePath: string): DcpConfig {
 // Public API
 // ---------------------------------------------------------------------------
 
+export interface LoadConfigOptions {
+  homeDir?: string
+}
+
 /**
- * Load the DCP configuration by merging (in order):
- *  1. Built-in defaults
- *  2. legacy ~/.config/pi/dcp.jsonc, then dcp in ~/.config/pi/pi-tools-suite.jsonc
- *  3. legacy $PI_CONFIG_DIR/dcp.jsonc, then dcp in $PI_CONFIG_DIR/pi-tools-suite.jsonc
- *  4. legacy <project>/.pi/dcp.jsonc, then dcp in nearest <project>/.pi/pi-tools-suite.jsonc
- *
- * The shared pi-tools-suite config wins over legacy dcp.jsonc at each layer so
- * DCP settings can live beside the rest of the suite config while preserving
- * backwards compatibility with existing standalone files.
+ * Load the DCP configuration by merging built-in defaults with the `dcp` section
+ * from the user-level shared pi-tools-suite config only:
+ * `~/.config/pi/pi-tools-suite.jsonc`.
  */
-export function loadConfig(projectDir: string): DcpConfig {
+export function loadConfig(options: LoadConfigOptions = {}): DcpConfig {
   // Layer 1: defaults (deep clone so we never mutate the constant)
   let config: DcpConfig = deepMerge(DEFAULT_CONFIG, {})
 
-  // Layer 2: global config
-  const globalConfigPath = path.join(os.homedir(), ".config", "pi", "dcp.jsonc")
-  config = mergeConfigFile(config, globalConfigPath)
-  config = mergeSuiteDcpConfig(config, path.join(os.homedir(), ".config", "pi", "pi-tools-suite.jsonc"))
-
-  // Layer 3: $PI_CONFIG_DIR/dcp.jsonc
-  const piConfigDir = process.env["PI_CONFIG_DIR"]
-  if (piConfigDir) {
-    config = mergeConfigFile(config, path.join(piConfigDir, "dcp.jsonc"))
-    config = mergeSuiteDcpConfig(config, path.join(piConfigDir, "pi-tools-suite.jsonc"))
-  }
-
-  // Layer 4: project-local config (walk up from projectDir)
-  const projectConfigPath = findProjectConfig(projectDir)
-  if (projectConfigPath) config = mergeConfigFile(config, projectConfigPath)
-
-  const projectSuiteConfigPath = findProjectSuiteConfig(projectDir)
-  if (projectSuiteConfigPath) config = mergeSuiteDcpConfig(config, projectSuiteConfigPath)
+  const homeDir = options.homeDir ?? os.homedir()
+  config = mergeSuiteDcpConfig(config, path.join(homeDir, ".config", "pi", "pi-tools-suite.jsonc"))
 
   return config
 }

package/external/pi-tools-suite/src/{compress → dcp}/index.ts

@@ -37,11 +37,8 @@ import {
 import type { DcpNudgeType } from "./pruner-types.js"
 import { registerCompressTool } from "./compress-tool.js"
 import { DCP_STATS_MESSAGE_TYPE, registerCommands } from "./commands.js"
-import { DcpUiController, normalizeDcpContextUsage } from "./ui.js"
-import { registerTuiFilter } from "./dcp-tui-filter.js"
-import { ignoreStaleExtensionContextError, safeGetContextUsage } from "../context-usage.js"
-
-const DCP_CONTEXT_USAGE_EVENT = "pi-tools-suite:dcp-context-usage"
+import { normalizeDcpContextUsage } from "./ui.js"
+import { safeGetContextUsage } from "../context-usage.js"
 
 // ---------------------------------------------------------------------------
 // Helpers
@@ -101,31 +98,12 @@ function isUserVisibleOnlyMessage(message: any): boolean {
 
 export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 	// ── 1. Load config ────────────────────────────────────────────────────────
-	const config = loadConfig(process.cwd())
+	const config = loadConfig()
 
 	if (!config.enabled) return
 
 	// ── 2. Create state ───────────────────────────────────────────────────────
 	const state = createState()
-	const ui = new DcpUiController(state)
-	const updateUi = (ctx: ExtensionContext): void => {
-		try {
-			if (!ctx?.hasUI) return
-			ui.setUICtx(ctx.ui)
-			ui.update(ctx)
-		} catch (error) {
-			ignoreStaleExtensionContextError(error)
-		}
-	}
-	const emitContextUsage = (ctx: ExtensionContext): void => {
-		const usage = safeGetContextUsage(ctx)
-		if (!usage) return
-		try {
-			;(pi as { events?: { emit?: (name: string, data: unknown) => void } }).events?.emit?.(DCP_CONTEXT_USAGE_EVENT, usage)
-		} catch (error) {
-			ignoreStaleExtensionContextError(error)
-		}
-	}
 	const appendNudgeTelemetry = (
 		event: "emitted" | "upgraded",
 		type: DcpNudgeType,
@@ -158,20 +136,13 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 		state.manualMode = true
 	}
 
-	// ── 3. Register TUI filter (strip DCP tags from displayed messages) ───────
-	registerTuiFilter(pi)
-
-	// ── 4. Register compress tool ─────────────────────────────────────────────
+	// ── 3. Register compress tool ─────────────────────────────────────────────
 	registerCompressTool(pi, state, config)
 
-	// ── 5. Register /dcp commands ─────────────────────────────────────────────
-	registerCommands(pi, state, config, {
-		onStateChanged(ctx) {
-			updateUi(ctx)
-		},
-	})
+	// ── 4. Register /dcp commands ─────────────────────────────────────────────
+	registerCommands(pi, state, config)
 
-	// ── 6. session_start: restore state from session entries ──────────────────
+	// ── 5. session_start: restore state from session entries ──────────────────
 	pi.on("session_start", async (_event, ctx) => {
 		// Reset to a clean slate first.
 		resetState(state)
@@ -191,17 +162,15 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 
 		restoreState(state, latestDcpState)
 
-		// Show a rich status indicator + floating cleaned-context widget in the pi TUI.
-		updateUi(ctx)
+		// Headless by design: no extension status/footer/widgets are rendered.
 	})
 
-	// ── 7. session_shutdown: save state ───────────────────────────────────────
+	// ── 6. session_shutdown: save state ───────────────────────────────────────
 	pi.on("session_shutdown", async (_event, _ctx) => {
 		saveState(pi, state)
-		ui.dispose()
 	})
 
-	// ── 8. before_agent_start: inject system prompt ───────────────────────────
+	// ── 7. before_agent_start: inject system prompt ───────────────────────────
 	pi.on("before_agent_start", async (event, _ctx) => {
 		const promptAddition = state.manualMode
 			? MANUAL_MODE_SYSTEM_PROMPT
@@ -212,7 +181,7 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 		}
 	})
 
-	// ── 9. tool_call: record input args for dedup / purge fingerprinting ───────
+	// ── 8. tool_call: record input args for dedup / purge fingerprinting ───────
 	pi.on("tool_call", async (event, _ctx) => {
 		if (!state.toolCalls.has(event.toolCallId)) {
 			state.toolCalls.set(event.toolCallId, {
@@ -231,8 +200,8 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 		}
 	})
 
-	// ── 10. tool_result: finalise tool record with result info ─────────────────
-	pi.on("tool_result", async (event, ctx) => {
+	// ── 9. tool_result: finalise tool record with result info ─────────────────
+	pi.on("tool_result", async (event, _ctx) => {
 		const record = state.toolCalls.get(event.toolCallId)
 
 		const outputText = event.content
@@ -261,13 +230,9 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 			})
 		}
 
-		if (event.toolName === "compress" && ctx.hasUI) {
-			updateUi(ctx)
-			emitContextUsage(ctx)
-		}
 	})
 
-	// ── 11. context: apply pruning and inject nudges ──────────────────────────
+	// ── 10. context: apply pruning and inject nudges ──────────────────────────
 	pi.on("context", async (event, ctx) => {
 		const contextMessages = event.messages.filter((message: any) => !isUserVisibleOnlyMessage(message))
 		annotateMessagesWithBranchEntryIds(contextMessages, ctx)
@@ -295,8 +260,6 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 				applyAnchoredNudges(prunedMessages, state, (anchor) =>
 					appendConcreteNudgeGuidance(baseNudgeText(anchor.type), candidate, messageCandidates, state),
 				)
-				updateUi(ctx)
-				emitContextUsage(ctx)
 				return { messages: prunedMessages }
 			}
 
@@ -388,24 +351,11 @@ export default async function dcpModule(pi: ExtensionAPI): Promise<void> {
 			appendConcreteNudgeGuidance(baseNudgeText(anchor.type), candidate, messageCandidates, state),
 		)
 
-
-		// Update footer status and floating widget after each context event.
-		updateUi(ctx)
-		emitContextUsage(ctx)
-
 		return { messages: prunedMessages }
 	})
 
-	// ── 12. turn_end: refresh DCP status from Pi's final context percentage ───
-	pi.on("turn_end", async (_event, ctx) => {
-		updateUi(ctx)
-		emitContextUsage(ctx)
-	})
-
-	// ── 13. agent_end: persist state after each agent run ────────────────────
-	pi.on("agent_end", async (_event, ctx) => {
-		updateUi(ctx)
-		emitContextUsage(ctx)
+	// ── 11. agent_end: persist state after each agent run ────────────────────
+	pi.on("agent_end", async (_event, _ctx) => {
 		saveState(pi, state)
 	})
 }

package/external/pi-tools-suite/src/dcp/ui.ts

@@ -0,0 +1,45 @@
+export interface DcpCompressionVisualDetails {
+	topic: string
+	blockIds: number[]
+	ranges: number
+	messages: number
+	itemCount: number
+	totalSummaryTokens: number
+	activeBlocks: number
+	totalBlocks: number
+	prunedTools: number
+	tokensSaved: number
+	contextTokens?: number | null
+	contextWindow?: number
+	contextPercent?: number | null
+	skippedMessages?: number
+	skippedMessageIssues?: string[]
+}
+
+export interface DcpContextUsage {
+	tokens: number | null
+	contextWindow: number
+	percent?: number | null
+}
+
+type RawDcpContextUsage = {
+	tokens?: number | null
+	contextWindow?: number
+	percent?: number | null
+} | undefined
+
+export function normalizeDcpContextUsage(usage: RawDcpContextUsage): DcpContextUsage | undefined {
+	if (!usage || typeof usage.contextWindow !== "number" || usage.contextWindow <= 0) return undefined
+	const contextWindow = usage.contextWindow
+	const percent = typeof usage.percent === "number" && Number.isFinite(usage.percent)
+		? Math.max(0, usage.percent)
+		: usage.percent === null
+			? null
+			: undefined
+	const tokens = typeof percent === "number"
+		? Math.round((contextWindow * percent) / 100)
+		: typeof usage.tokens === "number"
+			? usage.tokens
+			: null
+	return { tokens, contextWindow, percent }
+}

package/external/pi-tools-suite/src/default-pi-tools-suite-config.ts

@@ -13,10 +13,11 @@ export const DEFAULT_PI_TOOLS_SUITE_CONFIG_JSONC = `{
     // "model-tools",
     // "usage",
     // "web-search",
-    // "compress"
+    // "dcp"
   ],
 
-  // Dynamic Context Pruning (DCP) / compress module config.
+  // Dynamic Context Pruning (DCP) module config.
+  // The DCP module owns the compress tool and /dcp commands.
   "dcp": {
     "enabled": true,
     "manualMode": {

package/external/pi-tools-suite/src/index.ts

@@ -20,7 +20,7 @@ const MODULES: Array<{ name: string; load: () => Promise<ExtensionModule> }> = [
 	{ name: "model-tools", load: () => import("./model-tools/index") },
 	{ name: "usage", load: () => import("./usage/index") },
 	{ name: "web-search", load: () => import("./web-search/index") },
-	{ name: "compress", load: () => import("./compress/index") },
+	{ name: "dcp", load: () => import("./dcp/index") },
 	{ name: "prompt-commands", load: () => import("./prompt-commands/index") },
 ];
 

package/external/pi-tools-suite/src/tool-descriptions.ts

@@ -1,4 +1,4 @@
-import { COMPRESS_RANGE_DESCRIPTION } from "./compress/prompts.js";
+import { COMPRESS_RANGE_DESCRIPTION } from "./dcp/prompts.js";
 
 export type ToolDescription = {
 	name: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "private": false,
   "type": "module",
   "bin": {