changes

Author: abidlabs

trackio/frontend/src/App.svelte

@@ -147,8 +147,13 @@
 
       if (JSON.stringify(runs) !== JSON.stringify(newRuns)) {
         const prevSelected = selectedRuns;
+        const prevOrdered = runs.map(runKey);
         runs = newRuns;
-        selectedRuns = reconcileSelectedRuns(prevSelected, newRuns.map(runKey));
+        selectedRuns = reconcileSelectedRuns(
+          prevSelected,
+          newRuns.map(runKey),
+          prevOrdered,
+        );
       }
     } catch (e) {
       console.error("Failed to load runs:", e);

trackio/frontend/src/lib/selection.js

@@ -3,15 +3,24 @@ export function latestOnlySelection(filteredRunIds) {
   return [filteredRunIds[0]];
 }
 
-export function reconcileSelectedRuns(prevSelected, newOrderedIds) {
+export function reconcileSelectedRuns(prevSelected, newOrderedIds, prevOrderedIds) {
   const prev = prevSelected ?? [];
   const ordered = newOrderedIds ?? [];
+  const prevOrdered = prevOrderedIds ?? [];
   const newIdSet = new Set(ordered);
   const kept = prev.filter((r) => newIdSet.has(r));
 
   if (prev.length === 0 || kept.length === 0) {
     return [...ordered];
   }
 
+  const allPrevSelected =
+    prevOrdered.length > 0 && prev.length === prevOrdered.length;
+  if (allPrevSelected) {
+    const keptSet = new Set(kept);
+    const additions = ordered.filter((r) => !keptSet.has(r));
+    return [...kept, ...additions];
+  }
+
   return kept;
 }

trackio/frontend/src/lib/selection.test.js

@@ -22,27 +22,37 @@ describe("reconcileSelectedRuns", () => {
     expect(reconcileSelectedRuns([], ["a", "b", "c"])).toEqual(["a", "b", "c"]);
   });
 
-  test("keeps the previously selected run without auto-selecting new runs", () => {
-    expect(reconcileSelectedRuns(["a"], ["a", "b", "c"])).toEqual(["a"]);
+  test("keeps a partial selection without auto-selecting new runs", () => {
+    expect(reconcileSelectedRuns(["a"], ["a", "b", "c"], ["a", "b"])).toEqual(["a"]);
+  });
+
+  test("auto-selects new runs when all previous runs were selected", () => {
+    expect(
+      reconcileSelectedRuns(["a", "b"], ["a", "b", "c"], ["a", "b"]),
+    ).toEqual(["a", "b", "c"]);
   });
 
   test("preserves the chosen runs when the run list is unchanged on refresh", () => {
     const prev = ["b"];
     const next = ["a", "b", "c"];
-    expect(reconcileSelectedRuns(prev, next)).toEqual(["b"]);
-    expect(reconcileSelectedRuns(["b", "a", "c"], next)).toEqual(["b", "a", "c"]);
+    expect(reconcileSelectedRuns(prev, next, next)).toEqual(["b"]);
+    expect(reconcileSelectedRuns(["b", "a", "c"], next, next)).toEqual([
+      "b",
+      "a",
+      "c",
+    ]);
   });
 
   test("drops runs that no longer exist on the server", () => {
-    expect(reconcileSelectedRuns(["a", "b", "c"], ["a", "c"])).toEqual(["a", "c"]);
+    expect(
+      reconcileSelectedRuns(["a", "b", "c"], ["a", "c"], ["a", "b", "c"]),
+    ).toEqual(["a", "c"]);
   });
 
   test("falls back to all runs when none of the previously selected runs exist anymore", () => {
-    expect(reconcileSelectedRuns(["a"], ["b"])).toEqual(["b"]);
-    expect(reconcileSelectedRuns(["x", "y"], ["a", "b", "c"])).toEqual([
-      "a",
-      "b",
-      "c",
-    ]);
+    expect(reconcileSelectedRuns(["a"], ["b"], ["a"])).toEqual(["b"]);
+    expect(
+      reconcileSelectedRuns(["x", "y"], ["a", "b", "c"], ["x", "y"]),
+    ).toEqual(["a", "b", "c"]);
   });
 });