@youtyan/code-viewer 0.5.3 → 0.5.4

package/dist/code-viewer.js

@@ -6016,10 +6016,13 @@ function sanitizeDbUiState(raw) {
     return emptyDbUiState();
   const prefs = sanitizeDbUiPrefs(raw.prefs);
   const expandedTables = sanitizeDbUiExpandedTables(raw.expandedTables);
+  const snapshotSelectedTables = sanitizeDbUiExpandedTables(raw.snapshotSelectedTables);
   if (!isRecord(raw.columnWidths)) {
     const out2 = emptyDbUiState();
     if (expandedTables)
       out2.expandedTables = expandedTables;
+    if (snapshotSelectedTables)
+      out2.snapshotSelectedTables = snapshotSelectedTables;
     if (prefs)
       out2.prefs = prefs;
     return out2;
@@ -6065,6 +6068,8 @@ function sanitizeDbUiState(raw) {
   const out = { version: 1, columnWidths };
   if (expandedTables)
     out.expandedTables = expandedTables;
+  if (snapshotSelectedTables)
+    out.snapshotSelectedTables = snapshotSelectedTables;
   if (prefs)
     out.prefs = prefs;
   return out;
@@ -6100,12 +6105,17 @@ function mergeDbUiState(current, patch) {
     return current;
   const mergedPrefs = "prefs" in patch ? mergeDbUiPrefs(current.prefs, patch.prefs) : current.prefs;
   const mergedExpandedTables = "expandedTables" in patch ? mergeDbUiExpandedTables(current.expandedTables, patch.expandedTables) : current.expandedTables;
+  const mergedSnapshotSelectedTables = "snapshotSelectedTables" in patch ? mergeDbUiExpandedTables(current.snapshotSelectedTables, patch.snapshotSelectedTables) : current.snapshotSelectedTables;
   if (!isRecord(patch.columnWidths)) {
     const merged = { ...current, version: 1 };
     if (mergedExpandedTables)
       merged.expandedTables = mergedExpandedTables;
     else
       delete merged.expandedTables;
+    if (mergedSnapshotSelectedTables)
+      merged.snapshotSelectedTables = mergedSnapshotSelectedTables;
+    else
+      delete merged.snapshotSelectedTables;
     if (mergedPrefs)
       merged.prefs = mergedPrefs;
     else
@@ -6146,6 +6156,7 @@ function mergeDbUiState(current, patch) {
     ...patch,
     columnWidths,
     expandedTables: mergedExpandedTables,
+    snapshotSelectedTables: mergedSnapshotSelectedTables,
     prefs: mergedPrefs,
     version: 1
   });
@@ -11220,17 +11231,20 @@ async function computeDiffTables(cwd, beforeId, afterId) {
         insertedCount: 0,
         updatedCount: 0,
         deletedCount: 0,
-        unchangedCount: b.row_count
+        unchangedCount: b.row_count,
+        coverage: "both"
       });
       continue;
     }
     if (!b) {
       results.push({
         tableName: table,
-        insertedCount: a ? a.row_count : 0,
+        insertedCount: 0,
         updatedCount: 0,
         deletedCount: 0,
-        unchangedCount: 0
+        unchangedCount: 0,
+        coverage: "after-only",
+        unsnapshottedRowCount: a ? a.row_count : 0
       });
       continue;
     }
@@ -11239,8 +11253,10 @@ async function computeDiffTables(cwd, beforeId, afterId) {
         tableName: table,
         insertedCount: 0,
         updatedCount: 0,
-        deletedCount: b.row_count,
-        unchangedCount: 0
+        deletedCount: 0,
+        unchangedCount: 0,
+        coverage: "before-only",
+        unsnapshottedRowCount: b.row_count
       });
       continue;
     }
@@ -11262,7 +11278,8 @@ async function computeDiffTables(cwd, beforeId, afterId) {
       insertedCount,
       updatedCount,
       deletedCount,
-      unchangedCount: Math.max(0, unchangedCount)
+      unchangedCount: Math.max(0, unchangedCount),
+      coverage: "both"
     });
   }
   results.sort((a, b) => {
@@ -11400,9 +11417,11 @@ async function runSnapshot(cwd, source, dbId, containers, note, onProgress, opti
   const snapshotId = await createSnapshot(cwd, dbId, snapshotSource.kind, containers, note, options.schema);
   try {
     options.onSnapshotId?.(snapshotId);
-    for (const container of containers) {
+    const total = containers.length;
+    for (let i = 0;i < containers.length; i++) {
+      const container = containers[i];
       throwIfAborted(options.signal, "snapshot cancelled");
-      onProgress?.(container, false);
+      onProgress?.({ container, done: false, index: i, total });
       const collected = [];
       let pkColumns = [];
       if (snapshotSource.model === "sql") {
@@ -11427,7 +11446,7 @@ async function runSnapshot(cwd, source, dbId, containers, note, onProgress, opti
       await addSnapshotTableData(cwd, snapshotId, container, pkColumns, collected);
     }
     await finalizeSnapshot(cwd, snapshotId);
-    onProgress?.("", true);
+    onProgress?.({ container: "", done: true, index: total, total });
     return snapshotId;
   } catch (err) {
     const msg = err instanceof Error ? err.message : String(err);
@@ -12465,13 +12484,16 @@ async function handleSnapshotCreate(cwd, req, sendSse, omitDirNames) {
   let activeSnapshotId;
   (async () => {
     try {
-      const snapshotId = await runSnapshot(cwd, source, snapshotDbId, snapshotContainers, note, (table, done) => {
+      const snapshotId = await runSnapshot(cwd, source, snapshotDbId, snapshotContainers, note, (progress) => {
         sendSse?.("db-snapshot", JSON.stringify({
           action: "progress",
           dbId: snapshotDbId,
           schema: body.schema,
-          table,
-          done
+          id: activeSnapshotId,
+          table: progress.container,
+          done: progress.done,
+          index: progress.index,
+          total: progress.total
         }));
       }, {
         signal: abortController.signal,
@@ -12484,7 +12506,8 @@ async function handleSnapshotCreate(cwd, req, sendSse, omitDirNames) {
             action: "started",
             dbId: snapshotDbId,
             schema: body.schema,
-            id
+            id,
+            total: snapshotContainers.length
           }));
         }
       });
@@ -12503,6 +12526,7 @@ async function handleSnapshotCreate(cwd, req, sendSse, omitDirNames) {
         action: aborted ? "aborted" : "error",
         dbId: snapshotDbId,
         schema: body.schema,
+        id: activeSnapshotId,
         ...aborted ? {} : { error: err instanceof Error ? err.message : String(err) }
       }));
     } finally {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@youtyan/code-viewer",
-  "version": "0.5.3",
+  "version": "0.5.4",
   "description": "Local browser-based code and git diff viewer",
   "type": "module",
   "bin": {