feat: compact selection from array (#1067)

Author: BrianHung

packages/core/src/internal/data-grid/data-grid-types.ts

@@ -567,6 +567,13 @@ export class CompactSelection {
         return CompactSelection.empty().add(selection);
     };
 
+    static fromArray = (items: readonly number[]): CompactSelection => {
+        if (items.length === 0) return CompactSelection.empty();
+        const slices = items.map(s => [s, s + 1] as Slice);
+        const newItems = mergeRanges(slices);
+        return new CompactSelection(newItems);
+    };
+
     public offset(amount: number): CompactSelection {
         if (amount === 0) return this;
         const newItems = this.items.map(x => [x[0] + amount, x[1] + amount] as Slice);
@@ -575,9 +582,7 @@ export class CompactSelection {
 
     public add(selection: number | Slice): CompactSelection {
         const slice: Slice = typeof selection === "number" ? [selection, selection + 1] : selection;
-
         const newItems = mergeRanges([...this.items, slice]);
-
         return new CompactSelection(newItems);
     }
 

packages/core/test/data-grid-types.test.ts

@@ -79,6 +79,17 @@ describe("data-grid-types", () => {
         expect(sel.toArray()).toEqual([3, 4]);
     });
 
+    test("Compact selection fromArray", () => {
+        const sel = CompactSelection.fromArray([3, 4, 6]);
+        expect([...sel]).toEqual([3, 4, 6]);
+    });
+
+    test("Compact selection array roundtrip", () => {
+        const sel = CompactSelection.fromSingleSelection([2, 7]).add(8);
+        const altSel = CompactSelection.fromArray(sel.toArray());
+        expect(altSel.equals(sel)).toBe(true);
+    });
+
     test("Smoke test compact selection remove", () => {
         const sel = CompactSelection.fromSingleSelection([3, 8]);