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

package/lib/messages/prune.ts

@@ -0,0 +1,238 @@
+import type { SessionState, WithParts } from "../state"
+import type { Logger } from "../logger"
+import type { PluginConfig } from "../config"
+import { isMessageCompacted } from "../state/utils"
+import { createSyntheticUserMessage, replaceBlockIdsWithBlocked } from "./utils"
+import { getLastUserMessage } from "./query"
+import type { UserMessage } from "@opencode-ai/sdk/v2"
+
+const PRUNED_TOOL_OUTPUT_REPLACEMENT =
+    "[Output removed to save context - information superseded or no longer needed]"
+const PRUNED_TOOL_ERROR_INPUT_REPLACEMENT = "[input removed due to failed tool call]"
+const PRUNED_QUESTION_INPUT_REPLACEMENT = "[questions removed - see output for user's answers]"
+
+export const prune = (
+    state: SessionState,
+    logger: Logger,
+    config: PluginConfig,
+    messages: WithParts[],
+): void => {
+    filterCompressedRanges(state, logger, config, messages)
+    // pruneFullTool(state, logger, messages)
+    pruneToolOutputs(state, logger, messages)
+    pruneToolInputs(state, logger, messages)
+    pruneToolErrors(state, logger, messages)
+}
+
+const pruneFullTool = (state: SessionState, logger: Logger, messages: WithParts[]): void => {
+    const messagesToRemove: string[] = []
+
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        const partsToRemove: string[] = []
+
+        for (const part of parts) {
+            if (part.type !== "tool") {
+                continue
+            }
+
+            if (!state.prune.tools.has(part.callID)) {
+                continue
+            }
+            if (part.tool !== "edit" && part.tool !== "write") {
+                continue
+            }
+
+            partsToRemove.push(part.callID)
+        }
+
+        if (partsToRemove.length === 0) {
+            continue
+        }
+
+        msg.parts = parts.filter(
+            (part) => part.type !== "tool" || !partsToRemove.includes(part.callID),
+        )
+
+        if (msg.parts.length === 0) {
+            messagesToRemove.push(msg.info.id)
+        }
+    }
+
+    if (messagesToRemove.length > 0) {
+        const result = messages.filter((msg) => !messagesToRemove.includes(msg.info.id))
+        messages.length = 0
+        messages.push(...result)
+    }
+}
+
+const pruneToolOutputs = (state: SessionState, logger: Logger, messages: WithParts[]): void => {
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        for (const part of parts) {
+            if (part.type !== "tool") {
+                continue
+            }
+            if (!state.prune.tools.has(part.callID)) {
+                continue
+            }
+            if (part.state.status !== "completed") {
+                continue
+            }
+            if (part.tool === "question" || part.tool === "edit" || part.tool === "write") {
+                continue
+            }
+
+            part.state.output = PRUNED_TOOL_OUTPUT_REPLACEMENT
+        }
+    }
+}
+
+const pruneToolInputs = (state: SessionState, logger: Logger, messages: WithParts[]): void => {
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        for (const part of parts) {
+            if (part.type !== "tool") {
+                continue
+            }
+
+            if (!state.prune.tools.has(part.callID)) {
+                continue
+            }
+            if (part.state.status !== "completed") {
+                continue
+            }
+            if (part.tool !== "question") {
+                continue
+            }
+
+            if (part.state.input?.questions !== undefined) {
+                part.state.input.questions = PRUNED_QUESTION_INPUT_REPLACEMENT
+            }
+        }
+    }
+}
+
+const pruneToolErrors = (state: SessionState, logger: Logger, messages: WithParts[]): void => {
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        for (const part of parts) {
+            if (part.type !== "tool") {
+                continue
+            }
+            if (!state.prune.tools.has(part.callID)) {
+                continue
+            }
+            if (part.state.status !== "error") {
+                continue
+            }
+
+            // Prune all string inputs for errored tools
+            const input = part.state.input
+            if (input && typeof input === "object") {
+                for (const key of Object.keys(input)) {
+                    if (typeof input[key] === "string") {
+                        input[key] = PRUNED_TOOL_ERROR_INPUT_REPLACEMENT
+                    }
+                }
+            }
+        }
+    }
+}
+
+const filterCompressedRanges = (
+    state: SessionState,
+    logger: Logger,
+    config: PluginConfig,
+    messages: WithParts[],
+): void => {
+    if (
+        state.prune.messages.byMessageId.size === 0 &&
+        state.prune.messages.activeByAnchorMessageId.size === 0
+    ) {
+        return
+    }
+
+    const result: WithParts[] = []
+
+    for (const msg of messages) {
+        const msgId = msg.info.id
+
+        // Check if there's a summary to inject at this anchor point
+        const blockId = state.prune.messages.activeByAnchorMessageId.get(msgId)
+        const summary =
+            blockId !== undefined ? state.prune.messages.blocksById.get(blockId) : undefined
+        if (summary) {
+            const rawSummaryContent = (summary as { summary?: unknown }).summary
+            if (
+                summary.active !== true ||
+                typeof rawSummaryContent !== "string" ||
+                rawSummaryContent.length === 0
+            ) {
+                logger.warn("Skipping malformed compress summary", {
+                    anchorMessageId: msgId,
+                    blockId: (summary as { blockId?: unknown }).blockId,
+                })
+            } else {
+                // Find user message for variant and as base for synthetic message
+                const msgIndex = messages.indexOf(msg)
+                const userMessage = getLastUserMessage(messages, msgIndex)
+
+                if (userMessage) {
+                    const userInfo = userMessage.info as UserMessage
+                    const summaryContent =
+                        config.compress.mode === "message"
+                            ? replaceBlockIdsWithBlocked(rawSummaryContent)
+                            : rawSummaryContent
+                    const summarySeed = `${summary.blockId}:${summary.anchorMessageId}`
+                    result.push(
+                        createSyntheticUserMessage(
+                            userMessage,
+                            summaryContent,
+                            userInfo.variant,
+                            summarySeed,
+                        ),
+                    )
+
+                    logger.info("Injected compress summary", {
+                        anchorMessageId: msgId,
+                        summaryLength: summaryContent.length,
+                    })
+                } else {
+                    logger.warn("No user message found for compress summary", {
+                        anchorMessageId: msgId,
+                    })
+                }
+            }
+        }
+
+        // Skip messages that are in the prune list
+        const pruneEntry = state.prune.messages.byMessageId.get(msgId)
+        if (pruneEntry && pruneEntry.activeBlockIds.length > 0) {
+            continue
+        }
+
+        // Normal message, include it
+        result.push(msg)
+    }
+
+    // Replace messages array contents
+    messages.length = 0
+    messages.push(...result)
+}

package/lib/messages/query.ts

@@ -0,0 +1,56 @@
+import type { PluginConfig } from "../config"
+import type { WithParts } from "../state"
+
+export const getLastUserMessage = (
+    messages: WithParts[],
+    startIndex?: number,
+): WithParts | null => {
+    const start = startIndex ?? messages.length - 1
+    for (let i = start; i >= 0; i--) {
+        const msg = messages[i]
+        if (msg.info.role === "user" && !isIgnoredUserMessage(msg)) {
+            return msg
+        }
+    }
+    return null
+}
+
+export const messageHasCompress = (message: WithParts): boolean => {
+    if (message.info.role !== "assistant") {
+        return false
+    }
+
+    const parts = Array.isArray(message.parts) ? message.parts : []
+    return parts.some(
+        (part) =>
+            part.type === "tool" && part.tool === "compress" && part.state?.status === "completed",
+    )
+}
+
+export const isIgnoredUserMessage = (message: WithParts): boolean => {
+    if (message.info.role !== "user") {
+        return false
+    }
+
+    const parts = Array.isArray(message.parts) ? message.parts : []
+    if (parts.length === 0) {
+        return true
+    }
+
+    for (const part of parts) {
+        if (!(part as any).ignored) {
+            return false
+        }
+    }
+
+    return true
+}
+
+export function isProtectedUserMessage(config: PluginConfig, message: WithParts): boolean {
+    return (
+        config.compress.mode === "message" &&
+        config.compress.protectUserMessages &&
+        message.info.role === "user" &&
+        !isIgnoredUserMessage(message)
+    )
+}

package/lib/messages/reasoning-strip.ts

@@ -0,0 +1,40 @@
+import type { WithParts } from "../state"
+import { getLastUserMessage } from "./query"
+
+/**
+ * Mirrors opencode's differentModel handling by preserving part content while
+ * dropping provider metadata on assistant parts that came from a different
+ * model/provider than the current turn's user message.
+ */
+export function stripStaleMetadata(messages: WithParts[]): void {
+    const lastUserMessage = getLastUserMessage(messages)
+    if (lastUserMessage?.info.role !== "user") {
+        return
+    }
+
+    const modelID = lastUserMessage.info.model.modelID
+    const providerID = lastUserMessage.info.model.providerID
+
+    messages.forEach((message) => {
+        if (message.info.role !== "assistant") {
+            return
+        }
+
+        if (message.info.modelID === modelID && message.info.providerID === providerID) {
+            return
+        }
+
+        message.parts = message.parts.map((part) => {
+            if (part.type !== "text" && part.type !== "tool" && part.type !== "reasoning") {
+                return part
+            }
+
+            if (!("metadata" in part)) {
+                return part
+            }
+
+            const { metadata: _metadata, ...rest } = part
+            return rest
+        })
+    })
+}

package/lib/messages/sync.ts

@@ -0,0 +1,124 @@
+import type { SessionState, WithParts } from "../state"
+import type { Logger } from "../logger"
+
+function sortBlocksByCreation(
+    a: { createdAt: number; blockId: number },
+    b: { createdAt: number; blockId: number },
+): number {
+    const createdAtDiff = a.createdAt - b.createdAt
+    if (createdAtDiff !== 0) {
+        return createdAtDiff
+    }
+    return a.blockId - b.blockId
+}
+
+export const syncCompressionBlocks = (
+    state: SessionState,
+    logger: Logger,
+    messages: WithParts[],
+): void => {
+    const messagesState = state.prune.messages
+    if (!messagesState?.blocksById?.size) {
+        return
+    }
+
+    const messageIds = new Set(messages.map((msg) => msg.info.id))
+    const previousActiveBlockIds = new Set<number>(
+        Array.from(messagesState.blocksById.values())
+            .filter((block) => block.active)
+            .map((block) => block.blockId),
+    )
+
+    messagesState.activeBlockIds.clear()
+    messagesState.activeByAnchorMessageId.clear()
+
+    const now = Date.now()
+    const missingOriginBlockIds: number[] = []
+    const orderedBlocks = Array.from(messagesState.blocksById.values()).sort(sortBlocksByCreation)
+
+    for (const block of orderedBlocks) {
+        const hasOriginMessage =
+            typeof block.compressMessageId === "string" &&
+            block.compressMessageId.length > 0 &&
+            messageIds.has(block.compressMessageId)
+
+        if (!hasOriginMessage) {
+            block.active = false
+            block.deactivatedAt = now
+            block.deactivatedByBlockId = undefined
+            missingOriginBlockIds.push(block.blockId)
+            continue
+        }
+
+        if (block.deactivatedByUser) {
+            block.active = false
+            if (block.deactivatedAt === undefined) {
+                block.deactivatedAt = now
+            }
+            block.deactivatedByBlockId = undefined
+            continue
+        }
+
+        for (const consumedBlockId of block.consumedBlockIds) {
+            if (!messagesState.activeBlockIds.has(consumedBlockId)) {
+                continue
+            }
+
+            const consumedBlock = messagesState.blocksById.get(consumedBlockId)
+            if (consumedBlock) {
+                consumedBlock.active = false
+                consumedBlock.deactivatedAt = now
+                consumedBlock.deactivatedByBlockId = block.blockId
+
+                const mappedBlockId = messagesState.activeByAnchorMessageId.get(
+                    consumedBlock.anchorMessageId,
+                )
+                if (mappedBlockId === consumedBlock.blockId) {
+                    messagesState.activeByAnchorMessageId.delete(consumedBlock.anchorMessageId)
+                }
+            }
+
+            messagesState.activeBlockIds.delete(consumedBlockId)
+        }
+
+        block.active = true
+        block.deactivatedAt = undefined
+        block.deactivatedByBlockId = undefined
+        messagesState.activeBlockIds.add(block.blockId)
+        if (messageIds.has(block.anchorMessageId)) {
+            messagesState.activeByAnchorMessageId.set(block.anchorMessageId, block.blockId)
+        }
+    }
+
+    for (const entry of messagesState.byMessageId.values()) {
+        const allBlockIds = Array.isArray(entry.allBlockIds)
+            ? [...new Set(entry.allBlockIds.filter((id) => Number.isInteger(id) && id > 0))]
+            : []
+
+        entry.allBlockIds = allBlockIds
+        entry.activeBlockIds = allBlockIds.filter((id) => messagesState.activeBlockIds.has(id))
+    }
+
+    const nextActiveBlockIds = messagesState.activeBlockIds
+    let deactivatedCount = 0
+    let reactivatedCount = 0
+
+    for (const blockId of previousActiveBlockIds) {
+        if (!nextActiveBlockIds.has(blockId)) {
+            deactivatedCount++
+        }
+    }
+    for (const blockId of nextActiveBlockIds) {
+        if (!previousActiveBlockIds.has(blockId)) {
+            reactivatedCount++
+        }
+    }
+
+    if (missingOriginBlockIds.length > 0 || deactivatedCount > 0 || reactivatedCount > 0) {
+        logger.info("Synced compress block state", {
+            missingOriginCount: missingOriginBlockIds.length,
+            deactivatedCount,
+            reactivatedCount,
+        })
+    }
+}

package/lib/messages/utils.ts

@@ -0,0 +1,187 @@
+import { createHash } from "node:crypto"
+import type { SessionState, WithParts } from "../state"
+import { isMessageCompacted } from "../state/utils"
+import type { UserMessage } from "@opencode-ai/sdk/v2"
+
+const SUMMARY_ID_HASH_LENGTH = 16
+const DCP_BLOCK_ID_TAG_REGEX = /(<dcp-message-id(?=[\s>])[^>]*>)b\d+(<\/dcp-message-id>)/g
+const DCP_PAIRED_TAG_REGEX = /<dcp[^>]*>[\s\S]*?<\/dcp[^>]*>/gi
+const DCP_UNPAIRED_TAG_REGEX = /<\/?dcp[^>]*>/gi
+
+const generateStableId = (prefix: string, seed: string): string => {
+    const hash = createHash("sha256").update(seed).digest("hex").slice(0, SUMMARY_ID_HASH_LENGTH)
+    return `${prefix}_${hash}`
+}
+
+export const createSyntheticUserMessage = (
+    baseMessage: WithParts,
+    content: string,
+    variant?: string,
+    stableSeed?: string,
+): WithParts => {
+    const userInfo = baseMessage.info as UserMessage
+    const now = Date.now()
+    const deterministicSeed = stableSeed?.trim() || userInfo.id
+    const messageId = generateStableId("msg_dcp_summary", deterministicSeed)
+    const partId = generateStableId("prt_dcp_summary", deterministicSeed)
+
+    return {
+        info: {
+            id: messageId,
+            sessionID: userInfo.sessionID,
+            role: "user" as const,
+            agent: userInfo.agent,
+            model: userInfo.model,
+            time: { created: now },
+            ...(variant !== undefined && { variant }),
+        },
+        parts: [
+            {
+                id: partId,
+                sessionID: userInfo.sessionID,
+                messageID: messageId,
+                type: "text" as const,
+                text: content,
+            },
+        ],
+    }
+}
+
+export const createSyntheticTextPart = (
+    baseMessage: WithParts,
+    content: string,
+    stableSeed?: string,
+) => {
+    const userInfo = baseMessage.info as UserMessage
+    const deterministicSeed = stableSeed?.trim() || userInfo.id
+    const partId = generateStableId("prt_dcp_text", deterministicSeed)
+
+    return {
+        id: partId,
+        sessionID: userInfo.sessionID,
+        messageID: userInfo.id,
+        type: "text" as const,
+        text: content,
+    }
+}
+
+type MessagePart = WithParts["parts"][number]
+type ToolPart = Extract<MessagePart, { type: "tool" }>
+type TextPart = Extract<MessagePart, { type: "text" }>
+
+export const appendToLastTextPart = (message: WithParts, injection: string): boolean => {
+    const textPart = findLastTextPart(message)
+    if (!textPart) {
+        return false
+    }
+
+    return appendToTextPart(textPart, injection)
+}
+
+const findLastTextPart = (message: WithParts): TextPart | null => {
+    for (let i = message.parts.length - 1; i >= 0; i--) {
+        const part = message.parts[i]
+        if (part.type === "text") {
+            return part
+        }
+    }
+
+    return null
+}
+
+export const appendToTextPart = (part: TextPart, injection: string): boolean => {
+    if (typeof part.text !== "string") {
+        return false
+    }
+
+    const normalizedInjection = injection.replace(/^\n+/, "")
+    if (!normalizedInjection.trim()) {
+        return false
+    }
+    if (part.text.includes(normalizedInjection)) {
+        return true
+    }
+
+    const baseText = part.text.replace(/\n*$/, "")
+    part.text = baseText.length > 0 ? `${baseText}\n\n${normalizedInjection}` : normalizedInjection
+    return true
+}
+
+export const appendToAllToolParts = (message: WithParts, tag: string): boolean => {
+    let injected = false
+    for (const part of message.parts) {
+        if (part.type === "tool") {
+            injected = appendToToolPart(part, tag) || injected
+        }
+    }
+    return injected
+}
+
+export const appendToToolPart = (part: ToolPart, tag: string): boolean => {
+    if (part.state?.status !== "completed" || typeof part.state.output !== "string") {
+        return false
+    }
+    if (part.state.output.includes(tag)) {
+        return true
+    }
+
+    part.state.output = `${part.state.output}${tag}`
+    return true
+}
+
+export const hasContent = (message: WithParts): boolean => {
+    return message.parts.some(
+        (part) =>
+            (part.type === "text" &&
+                typeof part.text === "string" &&
+                part.text.trim().length > 0) ||
+            (part.type === "tool" &&
+                part.state?.status === "completed" &&
+                typeof part.state.output === "string"),
+    )
+}
+
+export function buildToolIdList(state: SessionState, messages: WithParts[]): string[] {
+    const toolIds: string[] = []
+    for (const msg of messages) {
+        if (isMessageCompacted(state, msg)) {
+            continue
+        }
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        if (parts.length > 0) {
+            for (const part of parts) {
+                if (part.type === "tool" && part.callID && part.tool) {
+                    toolIds.push(part.callID)
+                }
+            }
+        }
+    }
+    state.toolIdList = toolIds
+    return toolIds
+}
+
+export const replaceBlockIdsWithBlocked = (text: string): string => {
+    return text.replace(DCP_BLOCK_ID_TAG_REGEX, "$1BLOCKED$2")
+}
+
+export const stripHallucinationsFromString = (text: string): string => {
+    return text.replace(DCP_PAIRED_TAG_REGEX, "").replace(DCP_UNPAIRED_TAG_REGEX, "")
+}
+
+export const stripHallucinations = (messages: WithParts[]): void => {
+    for (const message of messages) {
+        for (const part of message.parts) {
+            if (part.type === "text" && typeof part.text === "string") {
+                part.text = stripHallucinationsFromString(part.text)
+            }
+
+            if (
+                part.type === "tool" &&
+                part.state?.status === "completed" &&
+                typeof part.state.output === "string"
+            ) {
+                part.state.output = stripHallucinationsFromString(part.state.output)
+            }
+        }
+    }
+}

package/lib/prompts/compress-message.ts

@@ -0,0 +1,42 @@
+export const COMPRESS_MESSAGE = `Collapse selected individual messages in the conversation into detailed summaries.
+
+THE SUMMARY
+Your summary must be EXHAUSTIVE. Capture file paths, function signatures, decisions made, constraints discovered, key findings, tool outcomes, and user intent details that matter... EVERYTHING that preserves the value of the selected message after the raw message is removed.
+
+USER INTENT FIDELITY
+When a selected message contains user intent, preserve that intent with extra care. Do not change scope, constraints, priorities, acceptance criteria, or requested outcomes.
+Directly quote short user instructions when that best preserves exact meaning.
+
+Yet be LEAN. Strip away the noise: failed attempts that led nowhere, verbose tool output, and repetition. What remains should be pure signal - golden nuggets of detail that preserve full understanding with zero ambiguity.
+If a message contains no significant technical decisions, code changes, or user requirements, produce a minimal one-line summary rather than a detailed one.
+
+MESSAGE IDS
+You specify individual raw messages by ID using the injected IDs visible in the conversation:
+
+- \`mNNNN\` IDs identify raw messages
+
+Each message has an ID inside XML metadata tags like \`<dcp-message-id priority="high">m0007</dcp-message-id>\`.
+The same ID tag appears in every tool output of the message it belongs to — each unique ID identifies one complete message.
+Treat these tags as message metadata only, not as content to summarize. Use only the inner \`mNNNN\` value as the \`messageId\`.
+The \`priority\` attribute indicates relative context cost. You MUST compress high-priority messages when their full text is no longer necessary for the active task.
+If prior compress-tool results are present, always compress and summarize them minimally only as part of a broader compression pass. Do not invoke the compress tool solely to re-compress an earlier compression result.
+Messages marked as \`<dcp-message-id>BLOCKED</dcp-message-id>\` cannot be compressed.
+
+Rules:
+
+- Pick each \`messageId\` directly from injected IDs visible in context.
+- Only use raw message IDs of the form \`mNNNN\`.
+- Ignore XML attributes such as \`priority\` when copying the ID; use only the inner \`mNNNN\` value.
+- Do not invent IDs. Use only IDs that are present in context.
+
+BATCHING
+Select MANY messages in a single tool call when they are safe to compress.
+Each entry should summarize exactly one message, and the tool can receive as many entries as needed in one batch.
+
+GENERAL CLEANUP
+Use the topic "general cleanup" for broad cleanup passes.
+During general cleanup, compress all medium and high-priority messages that are not relevant to the active task.
+Optimize for reducing context footprint, not for grouping messages by topic.
+Do not compress away still-active instructions, unresolved questions, or constraints that are likely to matter soon.
+Prioritize the earliest messages in the context as they will be the least relevant to the active task.
+`