Initial import

etc/servers/work-server/scripts/backfill-chat-resource-urls.ts

@@ -0,0 +1,76 @@
+import { db } from '../src/db/client.js';
+import {
+  CHAT_CONVERSATION_MESSAGE_TABLE,
+  CHAT_CONVERSATION_REQUEST_TABLE,
+} from '../src/services/chat-room-service.js';
+
+const LEGACY_CHAT_RESOURCE_PREFIX = '/.codex_chat/';
+const API_CHAT_RESOURCE_PREFIX = '/api/chat/resources/.codex_chat/';
+const requestedSessionId = process.argv[2]?.trim() || null;
+
+function rewriteLegacyChatResourceUrls(text: string) {
+  const normalized = String(text ?? '').replaceAll(LEGACY_CHAT_RESOURCE_PREFIX, API_CHAT_RESOURCE_PREFIX);
+
+  return normalized.replace(
+    /\((?:\/[^)\s]*?)?(\/api\/chat\/resources\/\.codex_chat\/[^)\s]*?)(?:\/api\/chat\/resources\/\.codex_chat\/[^)\s]*)?\)/g,
+    (_match, resourcePath) => `(${resourcePath})`,
+  );
+}
+
+async function backfillTable(
+  tableName: string,
+  textColumn: string,
+) {
+  const rows = await db(tableName)
+    .modify((query) => {
+      if (requestedSessionId) {
+        query.where('session_id', requestedSessionId);
+      }
+    })
+    .where(textColumn, 'like', `%${LEGACY_CHAT_RESOURCE_PREFIX}%`)
+    .select('id', 'session_id', textColumn);
+
+  let updatedCount = 0;
+  const touchedSessionIds = new Set<string>();
+
+  for (const row of rows) {
+    const currentText = String(row[textColumn] ?? '');
+    const nextText = rewriteLegacyChatResourceUrls(currentText);
+
+    if (nextText === currentText) {
+      continue;
+    }
+
+    await db(tableName)
+      .where('id', row.id)
+      .update({
+        [textColumn]: nextText,
+      });
+
+    updatedCount += 1;
+    touchedSessionIds.add(String(row.session_id ?? ''));
+  }
+
+  return {
+    tableName,
+    textColumn,
+    updatedCount,
+    touchedSessionIds: Array.from(touchedSessionIds).filter(Boolean),
+  };
+}
+
+try {
+  const messageResult = await backfillTable(CHAT_CONVERSATION_MESSAGE_TABLE, 'text');
+  const requestResult = await backfillTable(CHAT_CONVERSATION_REQUEST_TABLE, 'response_text');
+
+  console.log(JSON.stringify({
+    requestedSessionId,
+    updatedRowCount: messageResult.updatedCount + requestResult.updatedCount,
+    tables: [messageResult, requestResult],
+  }, null, 2));
+} catch (error) {
+  console.error(error);
+  process.exitCode = 1;
+} finally {
+  await db.destroy();
+}