iivo-sub 0.1.6 → 0.1.7

package/README.md

@@ -35,7 +35,9 @@ Backup configuration lets you choose Codex, Claude Code, Hermes, OpenClaw, or al
 
 Chat record backup lets you back up common local conversation paths for Codex, Claude Code, Hermes, and OpenClaw. Restore creates a safety backup before replacing existing chat record paths.
 
-For Codex, chat backup includes sessions, the recent-task index, history, attachments, shell snapshots, and local state files. SQLite state files are backed up with a consistent SQLite snapshot when `sqlite3` is available. Restore rebuilds the recent-task index from restored session files and also rebuilds the Codex SQLite `threads` task list when possible, because recent Codex builds read the task list from `~/.codex/state_*.sqlite`.
+For Codex, chat backup includes sessions, the recent-task index, history, attachments, shell snapshots, and local state files. SQLite state files are backed up with a consistent SQLite snapshot when `sqlite3` is available. Restore rebuilds the recent-task index from restored session files and also rebuilds the Codex SQLite `threads` task list plus `backfill_state` when possible, because recent Codex builds read the task list from `~/.codex/state_*.sqlite`.
+
+After restoring Codex chats in VS Code, run `Developer: Reload Window` or fully restart the VS Code/Codex window. The Codex extension keeps an in-process task-list cache, so the sidebar may continue to show the old small list until the window reloads.
 
 After you enter an API key once, it is cached in `~/.iivo-sub/cache.json`. Next time the prompt shows a masked key such as `sk-x...abcd`; press Enter to reuse it or type a new key to replace it.
 

package/bin/iivo-sub.js

@@ -6,7 +6,7 @@ import path from 'node:path'
 import readline from 'node:readline'
 import { spawnSync } from 'node:child_process'
 
-const VERSION = '0.1.6'
+const VERSION = '0.1.7'
 const APP_DIR = path.join(os.homedir(), '.iivo-sub')
 const CACHE_FILE = path.join(APP_DIR, 'cache.json')
 const SCRIPTED_INPUT = !process.stdin.isTTY ? fs.readFileSync(0, 'utf8').split(/\r?\n/) : null
@@ -663,6 +663,10 @@ function listCodexSessionFiles(dir) {
   return out
 }
 
+function codexRelativePath(filePath) {
+  return path.relative(path.join(os.homedir(), '.codex'), filePath).split(path.sep).join('/')
+}
+
 function rebuildCodexSessionIndex() {
   const sessionsDir = path.join(os.homedir(), '.codex', 'sessions')
   const indexPath = path.join(os.homedir(), '.codex', 'session_index.jsonl')
@@ -695,6 +699,25 @@ function sqlString(value) {
   return `'${String(value ?? '').replaceAll("'", "''")}'`
 }
 
+function updateCodexBackfillState(sqlite3, dbPath, watermark) {
+  if (!watermark) return false
+  const tableCheck = spawnSync(sqlite3, [dbPath, "select name from sqlite_master where type='table' and name='backfill_state';"], {
+    encoding: 'utf8',
+    timeout: 10000
+  })
+  if (tableCheck.status !== 0 || !tableCheck.stdout.includes('backfill_state')) return false
+
+  const completedAt = Math.trunc(Date.now() / 1000)
+  const script = [
+    'BEGIN;',
+    'DELETE FROM backfill_state WHERE id = 1;',
+    `INSERT INTO backfill_state (id, status, last_watermark, last_success_at, updated_at) VALUES (1, 'complete', ${sqlString(watermark)}, ${completedAt}, ${completedAt});`,
+    'COMMIT;'
+  ].join('\n')
+  const result = runSqliteScript(sqlite3, dbPath, script)
+  return result.status === 0
+}
+
 function rebuildCodexThreadsState() {
   const sqlite3 = findSqlite3()
   if (!sqlite3) return { rebuilt: 0, reason: 'sqlite3 not found' }
@@ -709,7 +732,14 @@ function rebuildCodexThreadsState() {
   const rows = files.map(codexSessionSummary).filter((row) => row.id)
   if (rows.length === 0) return { rebuilt: 0, reason: 'no session ids' }
 
+  const watermark = files
+    .map(codexRelativePath)
+    .filter((value) => value && !value.startsWith('..'))
+    .sort()
+    .at(-1) || ''
+
   let total = 0
+  let backfillUpdated = 0
   for (const dbPath of stateDbs) {
     const tableCheck = spawnSync(sqlite3, [dbPath, "select name from sqlite_master where type='table' and name='threads';"], {
       encoding: 'utf8',
@@ -744,9 +774,12 @@ ${createdMs}, ${updatedMs}, ${row.thread_source ? sqlString(row.thread_source) :
     statements.push('COMMIT;')
 
     const result = runSqliteScript(sqlite3, dbPath, statements.join('\n'))
-    if (result.status === 0) total += rows.length
+    if (result.status === 0) {
+      total += rows.length
+      if (updateCodexBackfillState(sqlite3, dbPath, watermark)) backfillUpdated += 1
+    }
   }
-  return { rebuilt: total, reason: total > 0 ? '' : 'threads table not rebuilt' }
+  return { rebuilt: total, backfillUpdated, reason: total > 0 ? '' : 'threads table not rebuilt' }
 }
 
 function loadBackupManifest(dir) {
@@ -1365,9 +1398,16 @@ async function restoreBackupByKind(kind, title, emptyMessage) {
   }
   if (rebuiltCodexThreads.rebuilt > 0) {
     console.log(c('cyan', `已重建 Codex sqlite 任务列表: ${rebuiltCodexThreads.rebuilt} 条`))
+    if (rebuiltCodexThreads.backfillUpdated > 0) {
+      console.log(c('cyan', `已同步 Codex sqlite 回填状态: ${rebuiltCodexThreads.backfillUpdated} 个数据库`))
+    }
   } else if (rebuiltCodexThreads.reason) {
     console.log(c('yellow', `未重建 Codex sqlite 任务列表: ${rebuiltCodexThreads.reason}`))
   }
+  if (kind === 'chat' && restoredCodexSessions) {
+    console.log()
+    console.log(c('yellow', '重要: Codex VS Code 插件会缓存任务列表。还原后请执行 "Developer: Reload Window"，或完全退出并重开 VS Code/Codex 窗口，否则侧边栏可能仍显示还原前的少量记录。'))
+  }
   console.log()
   console.log(c('yellow', `恢复前安全备份目录: ${safetyBackup}`))
   await pause()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "iivo-sub",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "IIVO AI gateway quick configuration CLI",
   "type": "module",
   "bin": {