@tarquinen/opencode-dcp 3.2.5-beta0 → 3.2.6-beta0

package/lib/commands/decompress.ts

@@ -0,0 +1,275 @@
+import type { Logger } from "../logger"
+import type { CompressionBlock, PruneMessagesState, SessionState, WithParts } from "../state"
+import { syncCompressionBlocks } from "../messages"
+import { parseBlockRef } from "../message-ids"
+import { getCurrentParams } from "../token-utils"
+import { saveSessionState } from "../state/persistence"
+import { sendIgnoredMessage } from "../ui/notification"
+import { formatTokenCount } from "../ui/utils"
+import {
+    getActiveCompressionTargets,
+    resolveCompressionTarget,
+    type CompressionTarget,
+} from "./compression-targets"
+
+export interface DecompressCommandContext {
+    client: any
+    state: SessionState
+    logger: Logger
+    sessionId: string
+    messages: WithParts[]
+    args: string[]
+}
+
+function parseBlockIdArg(arg: string): number | null {
+    const normalized = arg.trim().toLowerCase()
+    const blockRef = parseBlockRef(normalized)
+    if (blockRef !== null) {
+        return blockRef
+    }
+
+    if (!/^[1-9]\d*$/.test(normalized)) {
+        return null
+    }
+
+    const parsed = Number.parseInt(normalized, 10)
+    return Number.isInteger(parsed) && parsed > 0 ? parsed : null
+}
+
+function findActiveParentBlockId(
+    messagesState: PruneMessagesState,
+    block: CompressionBlock,
+): number | null {
+    const queue = [...block.parentBlockIds]
+    const visited = new Set<number>()
+
+    while (queue.length > 0) {
+        const parentBlockId = queue.shift()
+        if (parentBlockId === undefined || visited.has(parentBlockId)) {
+            continue
+        }
+        visited.add(parentBlockId)
+
+        const parent = messagesState.blocksById.get(parentBlockId)
+        if (!parent) {
+            continue
+        }
+
+        if (parent.active) {
+            return parent.blockId
+        }
+
+        for (const ancestorId of parent.parentBlockIds) {
+            if (!visited.has(ancestorId)) {
+                queue.push(ancestorId)
+            }
+        }
+    }
+
+    return null
+}
+
+function findActiveAncestorBlockId(
+    messagesState: PruneMessagesState,
+    target: CompressionTarget,
+): number | null {
+    for (const block of target.blocks) {
+        const activeAncestorBlockId = findActiveParentBlockId(messagesState, block)
+        if (activeAncestorBlockId !== null) {
+            return activeAncestorBlockId
+        }
+    }
+
+    return null
+}
+
+function snapshotActiveMessages(messagesState: PruneMessagesState): Map<string, number> {
+    const activeMessages = new Map<string, number>()
+    for (const [messageId, entry] of messagesState.byMessageId) {
+        if (entry.activeBlockIds.length > 0) {
+            activeMessages.set(messageId, entry.tokenCount)
+        }
+    }
+    return activeMessages
+}
+
+function formatDecompressMessage(
+    target: CompressionTarget,
+    restoredMessageCount: number,
+    restoredTokens: number,
+    reactivatedBlockIds: number[],
+): string {
+    const lines: string[] = []
+
+    lines.push(`Restored compression ${target.displayId}.`)
+    if (target.runId !== target.displayId || target.grouped) {
+        lines.push(`Tool call label: Compression #${target.runId}.`)
+    }
+    if (reactivatedBlockIds.length > 0) {
+        const refs = reactivatedBlockIds.map((id) => String(id)).join(", ")
+        lines.push(`Also restored nested compression(s): ${refs}.`)
+    }
+
+    if (restoredMessageCount > 0) {
+        lines.push(
+            `Restored ${restoredMessageCount} message(s) (~${formatTokenCount(restoredTokens)}).`,
+        )
+    } else {
+        lines.push("No messages were restored.")
+    }
+
+    return lines.join("\n")
+}
+
+function formatAvailableBlocksMessage(availableTargets: CompressionTarget[]): string {
+    const lines: string[] = []
+
+    lines.push("Usage: /dcp decompress <n>")
+    lines.push("")
+
+    if (availableTargets.length === 0) {
+        lines.push("No compressions are available to restore.")
+        return lines.join("\n")
+    }
+
+    lines.push("Available compressions:")
+    const entries = availableTargets.map((target) => {
+        const topic = target.topic.replace(/\s+/g, " ").trim() || "(no topic)"
+        const label = `${target.displayId} (${formatTokenCount(target.compressedTokens)})`
+        const details = target.grouped
+            ? `Compression #${target.runId} - ${target.blocks.length} messages`
+            : `Compression #${target.runId}`
+        return { label, topic: `${details} - ${topic}` }
+    })
+
+    const labelWidth = Math.max(...entries.map((entry) => entry.label.length)) + 4
+    for (const entry of entries) {
+        lines.push(`  ${entry.label.padEnd(labelWidth)}${entry.topic}`)
+    }
+
+    return lines.join("\n")
+}
+
+export async function handleDecompressCommand(ctx: DecompressCommandContext): Promise<void> {
+    const { client, state, logger, sessionId, messages, args } = ctx
+
+    const params = getCurrentParams(state, messages, logger)
+    const targetArg = args[0]
+
+    if (args.length > 1) {
+        await sendIgnoredMessage(
+            client,
+            sessionId,
+            "Invalid arguments. Usage: /dcp decompress <n>",
+            params,
+            logger,
+        )
+        return
+    }
+
+    syncCompressionBlocks(state, logger, messages)
+    const messagesState = state.prune.messages
+
+    if (!targetArg) {
+        const availableTargets = getActiveCompressionTargets(messagesState)
+        const message = formatAvailableBlocksMessage(availableTargets)
+        await sendIgnoredMessage(client, sessionId, message, params, logger)
+        return
+    }
+
+    const targetBlockId = parseBlockIdArg(targetArg)
+    if (targetBlockId === null) {
+        await sendIgnoredMessage(
+            client,
+            sessionId,
+            `Please enter a compression number. Example: /dcp decompress 2`,
+            params,
+            logger,
+        )
+        return
+    }
+
+    const target = resolveCompressionTarget(messagesState, targetBlockId)
+    if (!target) {
+        await sendIgnoredMessage(
+            client,
+            sessionId,
+            `Compression ${targetBlockId} does not exist.`,
+            params,
+            logger,
+        )
+        return
+    }
+
+    const activeBlocks = target.blocks.filter((block) => block.active)
+    if (activeBlocks.length === 0) {
+        const activeAncestorBlockId = findActiveAncestorBlockId(messagesState, target)
+        if (activeAncestorBlockId !== null) {
+            await sendIgnoredMessage(
+                client,
+                sessionId,
+                `Compression ${target.displayId} is inside compression ${activeAncestorBlockId}. Restore compression ${activeAncestorBlockId} first.`,
+                params,
+                logger,
+            )
+            return
+        }
+
+        await sendIgnoredMessage(
+            client,
+            sessionId,
+            `Compression ${target.displayId} is not active.`,
+            params,
+            logger,
+        )
+        return
+    }
+
+    const activeMessagesBefore = snapshotActiveMessages(messagesState)
+    const activeBlockIdsBefore = new Set(messagesState.activeBlockIds)
+    const deactivatedAt = Date.now()
+
+    for (const block of target.blocks) {
+        block.active = false
+        block.deactivatedByUser = true
+        block.deactivatedAt = deactivatedAt
+        block.deactivatedByBlockId = undefined
+    }
+
+    syncCompressionBlocks(state, logger, messages)
+
+    let restoredMessageCount = 0
+    let restoredTokens = 0
+    for (const [messageId, tokenCount] of activeMessagesBefore) {
+        const entry = messagesState.byMessageId.get(messageId)
+        const isActiveNow = entry ? entry.activeBlockIds.length > 0 : false
+        if (!isActiveNow) {
+            restoredMessageCount++
+            restoredTokens += tokenCount
+        }
+    }
+
+    state.stats.totalPruneTokens = Math.max(0, state.stats.totalPruneTokens - restoredTokens)
+
+    const reactivatedBlockIds = Array.from(messagesState.activeBlockIds)
+        .filter((blockId) => !activeBlockIdsBefore.has(blockId))
+        .sort((a, b) => a - b)
+
+    await saveSessionState(state, logger)
+
+    const message = formatDecompressMessage(
+        target,
+        restoredMessageCount,
+        restoredTokens,
+        reactivatedBlockIds,
+    )
+    await sendIgnoredMessage(client, sessionId, message, params, logger)
+
+    logger.info("Decompress command completed", {
+        targetBlockId: target.displayId,
+        targetRunId: target.runId,
+        restoredMessageCount,
+        restoredTokens,
+        reactivatedBlockIds,
+    })
+}

package/lib/commands/help.ts

@@ -0,0 +1,76 @@
+/**
+ * DCP Help command handler.
+ * Shows available DCP commands and their descriptions.
+ */
+
+import type { Logger } from "../logger"
+import type { PluginConfig } from "../config"
+import type { SessionState, WithParts } from "../state"
+import { compressPermission } from "../compress-permission"
+import { sendIgnoredMessage } from "../ui/notification"
+import { getCurrentParams } from "../token-utils"
+
+export interface HelpCommandContext {
+    client: any
+    state: SessionState
+    config: PluginConfig
+    logger: Logger
+    sessionId: string
+    messages: WithParts[]
+}
+
+const BASE_COMMANDS: [string, string][] = [
+    ["/dcp context", "Show token usage breakdown for current session"],
+    ["/dcp stats", "Show DCP pruning statistics"],
+    ["/dcp sweep [n]", "Prune tools since last user message, or last n tools"],
+    ["/dcp manual [on|off]", "Toggle manual mode or set explicit state"],
+]
+
+const TOOL_COMMANDS: Record<string, [string, string]> = {
+    compress: ["/dcp compress [focus]", "Trigger manual compress tool execution"],
+    decompress: ["/dcp decompress <n>", "Restore selected compression"],
+    recompress: ["/dcp recompress <n>", "Re-apply a user-decompressed compression"],
+}
+
+function getVisibleCommands(state: SessionState, config: PluginConfig): [string, string][] {
+    const commands = [...BASE_COMMANDS]
+
+    if (compressPermission(state, config) !== "deny") {
+        commands.push(TOOL_COMMANDS.compress)
+        commands.push(TOOL_COMMANDS.decompress)
+        commands.push(TOOL_COMMANDS.recompress)
+    }
+
+    return commands
+}
+
+function formatHelpMessage(state: SessionState, config: PluginConfig): string {
+    const commands = getVisibleCommands(state, config)
+    const colWidth = Math.max(...commands.map(([cmd]) => cmd.length)) + 4
+    const lines: string[] = []
+
+    lines.push("╭─────────────────────────────────────────────────────────────────────────╮")
+    lines.push("│                              DCP Commands                               │")
+    lines.push("╰─────────────────────────────────────────────────────────────────────────╯")
+    lines.push("")
+    lines.push(`  ${"Manual mode:".padEnd(colWidth)}${state.manualMode ? "ON" : "OFF"}`)
+    lines.push("")
+    for (const [cmd, desc] of commands) {
+        lines.push(`  ${cmd.padEnd(colWidth)}${desc}`)
+    }
+    lines.push("")
+
+    return lines.join("\n")
+}
+
+export async function handleHelpCommand(ctx: HelpCommandContext): Promise<void> {
+    const { client, state, logger, sessionId, messages } = ctx
+
+    const { config } = ctx
+    const message = formatHelpMessage(state, config)
+
+    const params = getCurrentParams(state, messages, logger)
+    await sendIgnoredMessage(client, sessionId, message, params, logger)
+
+    logger.info("Help command executed")
+}

package/lib/commands/index.ts

@@ -0,0 +1,11 @@
+export { handleContextCommand } from "./context"
+export { handleDecompressCommand } from "./decompress"
+export { handleHelpCommand } from "./help"
+export {
+    applyPendingManualTrigger,
+    handleManualToggleCommand,
+    handleManualTriggerCommand,
+} from "./manual"
+export { handleRecompressCommand } from "./recompress"
+export { handleStatsCommand } from "./stats"
+export { handleSweepCommand } from "./sweep"

package/lib/commands/manual.ts

@@ -0,0 +1,125 @@
+/**
+ * DCP Manual mode command handler.
+ * Handles toggling manual mode and triggering individual tool executions.
+ *
+ * Usage:
+ *   /dcp manual [on|off]  - Toggle manual mode or set explicit state
+ *   /dcp compress [focus]  - Trigger manual compress execution
+ */
+
+import type { Logger } from "../logger"
+import type { SessionState, WithParts } from "../state"
+import type { PluginConfig } from "../config"
+import { sendIgnoredMessage } from "../ui/notification"
+import { getCurrentParams } from "../token-utils"
+import { buildCompressedBlockGuidance } from "../prompts/extensions/nudge"
+import { isIgnoredUserMessage } from "../messages/query"
+
+const MANUAL_MODE_ON = "Manual mode is now ON. Use /dcp compress to trigger context tools manually."
+
+const MANUAL_MODE_OFF = "Manual mode is now OFF."
+
+const COMPRESS_TRIGGER_PROMPT = [
+    "<compress triggered manually>",
+    "Manual mode trigger received. You must now use the compress tool.",
+    "Find the most significant completed conversation content that can be compressed into a high-fidelity technical summary.",
+    "Follow the active compress mode, preserve all critical implementation details, and choose safe targets.",
+    "Return after compress with a brief explanation of what content was compressed.",
+].join("\n\n")
+
+function getTriggerPrompt(
+    tool: "compress",
+    state: SessionState,
+    config: PluginConfig,
+    userFocus?: string,
+): string {
+    const base = COMPRESS_TRIGGER_PROMPT
+    const compressedBlockGuidance =
+        config.compress.mode === "message" ? "" : buildCompressedBlockGuidance(state)
+
+    const sections = [base, compressedBlockGuidance]
+    if (userFocus && userFocus.trim().length > 0) {
+        sections.push(`Additional user focus:\n${userFocus.trim()}`)
+    }
+
+    return sections.join("\n\n")
+}
+
+export interface ManualCommandContext {
+    client: any
+    state: SessionState
+    config: PluginConfig
+    logger: Logger
+    sessionId: string
+    messages: WithParts[]
+}
+
+export async function handleManualToggleCommand(
+    ctx: ManualCommandContext,
+    modeArg?: string,
+): Promise<void> {
+    const { client, state, logger, sessionId, messages } = ctx
+
+    if (modeArg === "on") {
+        state.manualMode = "active"
+    } else if (modeArg === "off") {
+        state.manualMode = false
+    } else {
+        state.manualMode = state.manualMode ? false : "active"
+    }
+
+    const params = getCurrentParams(state, messages, logger)
+    await sendIgnoredMessage(
+        client,
+        sessionId,
+        state.manualMode ? MANUAL_MODE_ON : MANUAL_MODE_OFF,
+        params,
+        logger,
+    )
+
+    logger.info("Manual mode toggled", { manualMode: state.manualMode })
+}
+
+export async function handleManualTriggerCommand(
+    ctx: ManualCommandContext,
+    tool: "compress",
+    userFocus?: string,
+): Promise<string | null> {
+    return getTriggerPrompt(tool, ctx.state, ctx.config, userFocus)
+}
+
+export function applyPendingManualTrigger(
+    state: SessionState,
+    messages: WithParts[],
+    logger: Logger,
+): void {
+    const pending = state.pendingManualTrigger
+    if (!pending) {
+        return
+    }
+
+    if (!state.sessionId || pending.sessionId !== state.sessionId) {
+        state.pendingManualTrigger = null
+        return
+    }
+
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const msg = messages[i]
+        if (msg.info.role !== "user" || isIgnoredUserMessage(msg)) {
+            continue
+        }
+
+        for (const part of msg.parts) {
+            if (part.type !== "text" || part.ignored || part.synthetic) {
+                continue
+            }
+
+            part.text = pending.prompt
+            state.pendingManualTrigger = null
+            logger.debug("Applied manual prompt", { sessionId: pending.sessionId })
+            return
+        }
+    }
+
+    state.pendingManualTrigger = null
+}