keep for split

virtualstate · Jan 16, 2024 · 6a09d8f · 6a09d8f
1 parent 32f39b7
commit 6a09d8f
Showing 1 changed file with 19 additions and 8 deletions.
diff --git a/src/split/split.ts b/src/split/split.ts
@@ -220,17 +220,28 @@ export function split<T>(
       function* inner() {
         yield mainTarget?.push(snapshot);
 
-        for (const [index, target] of targets.entries()) {
-          if (index >= snapshot.length) {
-            continue;
+        if (options?.keep) {
+          for (let index = 0; index < snapshot.length; index += 1) {
+            const target = atTarget(index);
+            const value = snapshot.at(index);
+            assert(value);
+            target.push(value);
           }
-          if (index < 0 && Math.abs(index) > snapshot.length) {
-            continue;
+        } else {
+          for (const [index, target] of targets.entries()) {
+            if (index >= snapshot.length) {
+              continue;
+            }
+            if (index < 0 && Math.abs(index) > snapshot.length) {
+              continue;
+            }
+            const value = snapshot.at(index); // Using at here allows using -1 as an index
+            assert(value);
+            target.push(value);
           }
-          const value = snapshot.at(index); // Using at here allows using -1 as an index
-          assert(value);
-          target.push(value);
         }
+
+
       }
     }