copilotkit 4.28.0 → 4.30.0

package/exporters/claude-sdk/store-export.ts

@@ -0,0 +1,61 @@
+/** The read-only subset of the TypeScript SDK SessionStore contract. */
+export interface ReadonlySessionStore {
+  load(key: {
+    projectKey: string;
+    sessionId: string;
+    subpath?: string;
+  }): Promise<unknown[] | null>;
+  listSessions?(
+    projectKey: string,
+  ): Promise<{ sessionId: string; mtime: number }[]>;
+  listSubkeys?(key: {
+    projectKey: string;
+    sessionId: string;
+  }): Promise<string[]>;
+}
+export interface SessionStoreExport {
+  format: 'claude-session-store-v1';
+  language: 'typescript';
+  sdkVersion: string;
+  projectKey: string;
+  sessions: {
+    sessionId: string;
+    entries: unknown[];
+    subkeys: { subpath: string; entries: unknown[] }[];
+  }[];
+}
+/**
+ * Read a quiesced custom store without query(), hydration, append or delete.
+ * Caller supplies IDs when the store has no discovery method. Export while
+ * writers are stopped or through a backend's consistent read snapshot.
+ */
+export async function exportSessionStore(
+  store: ReadonlySessionStore,
+  config: { language: 'typescript'; sdkVersion: string; projectKey: string },
+  sessionIds?: string[],
+): Promise<SessionStoreExport> {
+  if (!sessionIds && !store.listSessions)
+    throw new Error(
+      'Supply session IDs when SessionStore does not implement listSessions',
+    );
+  const ids =
+    sessionIds ??
+    (store.listSessions
+      ? (await store.listSessions(config.projectKey)).map((s) => s.sessionId)
+      : []);
+  const sessions: SessionStoreExport['sessions'] = [];
+  for (const sessionId of [...new Set(ids)]) {
+    const key = { projectKey: config.projectKey, sessionId };
+    const entries = await store.load(key);
+    if (!entries)
+      throw new Error(`Session ${sessionId} is missing from the store`);
+    const subkeys: SessionStoreExport['sessions'][number]['subkeys'] = [];
+    for (const subpath of (await store.listSubkeys?.(key)) ?? []) {
+      const child = await store.load({ ...key, subpath });
+      if (!child) throw new Error(`Session subkey ${subpath} is missing`);
+      subkeys.push({ subpath, entries: child });
+    }
+    sessions.push({ sessionId, entries, subkeys });
+  }
+  return { format: 'claude-session-store-v1', ...config, sessions };
+}