opencode-codex-memory 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-codex-memory",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Persistent memory plugin for opencode — ports codex's two-phase memory system (extraction → consolidation → injection → citation feedback)",
   "type": "module",
   "main": "src/index.ts",

package/src/capture.ts

@@ -43,7 +43,6 @@ export interface TranscriptMessage {
   type: string
   role?: string
   text?: string
-  seq: number
 }
 
 export function loadTranscript(sessionId: string): TranscriptMessage[] {
@@ -59,7 +58,7 @@ export function loadTranscript(sessionId: string): TranscriptMessage[] {
          ORDER BY p.time_created ASC`,
       )
       .all(sessionId) as { data: string; msg_data: string }[]
-    return rows.map((r, i) => {
+    return rows.map((r) => {
       let parsed: any = {}
       try {
         parsed = JSON.parse(r.data)
@@ -72,7 +71,6 @@ export function loadTranscript(sessionId: string): TranscriptMessage[] {
       } catch {
       }
       return {
-        seq: i,
         type: parsed.type ?? "unknown",
         role,
         text: extractText(parsed),

package/src/db.ts

@@ -1,7 +1,7 @@
 import { Database } from "bun:sqlite"
 import { memoryDbPath } from "./paths.js"
 
-export const SCHEMA_V1 = [
+const SCHEMA_V1 = [
   `CREATE TABLE IF NOT EXISTS memory_stage1_outputs (
     session_id TEXT PRIMARY KEY,
     source_updated_at INTEGER NOT NULL,
@@ -77,8 +77,4 @@ export function closeDb(): void {
     dbInstance.close()
     dbInstance = null
   }
-}
-
-export function resetDbForTest(): void {
-  closeDb()
 }

package/src/index.ts

@@ -3,14 +3,13 @@ import { stripCitations, extractCitedSessionIds } from "./citation.js"
 import { memory_read, memory_search, memory_list, memory_add_note } from "../tools/memory.js"
 import { memory_reset, memory_inspect, memory_mode } from "../tools/control.js"
 import { MemoryStore } from "./store.js"
-import { runPhase1, DEFAULT_PHASE1_OPTIONS } from "./phase1.js"
-import { runPhase2, DEFAULT_PHASE2_OPTIONS } from "./phase2.js"
+import { runPhase1 } from "./phase1.js"
+import { runPhase2 } from "./phase2.js"
 import { setPluginInput, cleanupOldSubSessions, isMemorySubSession } from "./llm.js"
 import type { PluginInput, PluginOptions } from "@opencode-ai/plugin"
 import fs from "fs"
 import path from "path"
 
-let store: MemoryStore | null = null
 let phase1InFlight = false
 let pluginClient: PluginInput["client"] | null = null
 // Configured MCP server names, fetched lazily; null until first successful fetch.
@@ -43,9 +42,10 @@ let pluginOptions: {
   min_rollout_idle_hours: 6,
 }
 
+// Deliberately uncached: openDb() is already a singleton, and caching a store
+// here would hold a stale handle across closeDb() (e.g. after memory_reset).
 function getStore(): MemoryStore {
-  if (!store) store = new MemoryStore()
-  return store
+  return new MemoryStore()
 }
 
 // Citation blocks arrive via message.part.updated once per streaming delta,

package/src/llm.ts

@@ -14,7 +14,7 @@ export function setPluginInput(input: PluginInput): void {
   inputRef = input
 }
 
-export function getPluginInput(): PluginInput | null {
+function getPluginInput(): PluginInput | null {
   return inputRef
 }
 
@@ -167,7 +167,10 @@ export async function consolidateViaSubagent(memoryRoot: string, diffFileName: s
   }
 }
 
-export async function cleanupOldSubSessions(maxAgeMinutes = 30): Promise<void> {
+// Must exceed the longest legitimate sub-session lifetime (consolidation may
+// run up to CONSOLIDATION_TIMEOUT_MS = 60min), or a second opencode instance /
+// plugin reload would delete a working sub-session mid-run.
+export async function cleanupOldSubSessions(maxAgeMinutes = 90): Promise<void> {
   const input = getPluginInput()
   if (!input) return
   try {
@@ -207,7 +210,7 @@ async function deleteSession(id: string): Promise<void> {
 export function fillTemplate(tmpl: string, vars: Record<string, string>): string {
   let out = tmpl
   for (const [key, value] of Object.entries(vars)) {
-    out = out.replace(`{{ ${key} }}`, () => value)
+    out = out.replaceAll(`{{ ${key} }}`, () => value)
   }
   return out
 }
@@ -221,12 +224,10 @@ function buildExtractionInput(sessionId: string, cwd: string, transcript: string
 }
 
 function buildConsolidationPrompt(memoryRoot: string, diffFileName: string): string {
-  const tmpl = readTemplate("consolidation.md")
-  let out = tmpl
-  // These placeholders appear many times in the template; replace all occurrences.
-  out = out.split("{{ memory_root }}").join(memoryRoot)
-  out = out.split("{{ phase2_workspace_diff_file }}").join(diffFileName)
-  return out
+  return fillTemplate(readTemplate("consolidation.md"), {
+    memory_root: memoryRoot,
+    phase2_workspace_diff_file: diffFileName,
+  })
 }
 
 function readTemplate(name: string): string {

package/src/phase2.ts

@@ -1,4 +1,4 @@
-import { MemoryStore, PHASE2_COOLDOWN_MS } from "./store.js"
+import { MemoryStore } from "./store.js"
 import { ensureLayout, rebuildRawMemories, writeRolloutSummaries, pruneExtensionResources, writeWorkspaceDiff } from "./workspace.js"
 import { ensureBaseline, captureWorkspaceDiff, resetBaseline, DIFF_ARTIFACT } from "./git-baseline.js"
 import { consolidateViaSubagent } from "./llm.js"
@@ -97,5 +97,3 @@ export async function runPhase2(store: MemoryStore, opts: Phase2Options = DEFAUL
     phase2InFlight = false
   }
 }
-
-export const PHASE2_COOLDOWN = PHASE2_COOLDOWN_MS

package/src/source.ts

@@ -55,8 +55,5 @@ export function buildMemorySystemPrompt(): string | null {
 }
 
 export function ensureMemoryLayout(): void {
-  const root = memoryRoot()
-  if (!fs.existsSync(root)) {
-    fs.mkdirSync(root, { recursive: true })
-  }
+  fs.mkdirSync(memoryRoot(), { recursive: true })
 }

package/src/store.ts

@@ -26,10 +26,6 @@ export interface Stage1Output {
   last_usage: number | null
 }
 
-export type ClaimResult =
-  | { type: "claimed"; sessionId: string; workerId: string; ownershipToken: string }
-  | { type: "skipped" }
-
 export interface Stage1Claim {
   sessionId: string
   ownershipToken: string

package/src/workspace.ts

@@ -1,3 +1,4 @@
+import { createHash } from "crypto"
 import fs from "fs"
 import path from "path"
 import { memoryRoot } from "./paths.js"
@@ -65,17 +66,7 @@ export function rolloutSummaryFileStem(o: Pick<Stage1Output, "session_id" | "sou
   const ts = new Date(o.source_updated_at)
   const pad = (n: number) => String(n).padStart(2, "0")
   const timestamp = `${ts.getUTCFullYear()}-${pad(ts.getUTCMonth() + 1)}-${pad(ts.getUTCDate())}T${pad(ts.getUTCHours())}-${pad(ts.getUTCMinutes())}-${pad(ts.getUTCSeconds())}`
-  let h = 0
-  for (let i = 0; i < o.session_id.length; i++) {
-    h = (h * 31 + o.session_id.charCodeAt(i)) >>> 0
-  }
-  const alphabet = "0123456789abcdefghijklmnopqrstuvwxyz"
-  let hash = ""
-  let v = h % 36 ** 4
-  for (let i = 0; i < 4; i++) {
-    hash = alphabet[v % 36] + hash
-    v = Math.floor(v / 36)
-  }
+  const hash = createHash("sha1").update(o.session_id).digest("hex").slice(0, 4)
   const prefix = `${timestamp}-${hash}`
   const slug = (o.rollout_slug ?? "")
     .toLowerCase()

package/tools/control.ts

@@ -4,7 +4,6 @@ import { tool } from "@opencode-ai/plugin"
 import { memoryRoot, memorySummaryPath } from "@/paths"
 import { MemoryStore } from "@/store"
 import { invalidateCache } from "@/source"
-import { closeDb } from "@/db"
 import { estimateTokens } from "@/token"
 
 function isSymlinkedRoot(): boolean {
@@ -72,7 +71,8 @@ export const memory_reset = tool({
       const store = new MemoryStore()
       store.clearMemoryData()
       wipeMemoriesDir()
-      closeDb()
+      // codex keeps its state DB pool open across resets (clear_memory_roots_contents
+      // only wipes directories); closing here would strand cached handles elsewhere.
       invalidateCache()
       return { output: "Memory reset complete. Extracted memories and jobs cleared, memories directory (incl. git history) wiped, cache invalidated. Per-session memory modes were preserved." }
     } catch (err) {