@weng-lab/genomebrowser-ui 0.1.7 → 0.1.9

package/dist/lib.d.ts

@@ -1,4 +1,8 @@
 import { default as TrackSelect, TrackSelectProps } from './TrackSelect/TrackSelect';
 import { createSelectionStore, SelectionStoreInstance } from './TrackSelect/store.ts';
+import { rowById } from './TrackSelect/consts.ts';
+import { RowInfo } from './TrackSelect/types.ts';
 export { TrackSelect, TrackSelectProps };
 export { createSelectionStore, SelectionStoreInstance };
+export { rowById };
+export { RowInfo };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@weng-lab/genomebrowser-ui",
   "private": false,
-  "version": "0.1.7",
+  "version": "0.1.9",
   "license": "MIT",
   "type": "module",
   "publishConfig": {

package/src/TrackSelect/store.ts

@@ -1,5 +1,6 @@
 import { create, StoreApi, UseBoundStore } from "zustand";
-import { SelectionAction, SelectionState } from "./types";
+import { rowById } from "./consts";
+import { RowInfo, SelectionAction, SelectionState } from "./types";
 
 export type SelectionStoreInstance = UseBoundStore<
   StoreApi<SelectionState & SelectionAction>
@@ -18,6 +19,20 @@ export function createSelectionStore(initialIds?: Set<string>) {
       const all = get().selectedIds;
       return new Set([...all].filter((id) => !isAutoGeneratedId(id)));
     },
+    // Returns a Map of track IDs to RowInfo (no auto-generated IDs)
+    getTrackMap: () => {
+      const all = get().selectedIds;
+      const map = new Map<string, RowInfo>();
+      all.forEach((id) => {
+        if (!isAutoGeneratedId(id)) {
+          const row = rowById.get(id);
+          if (row) {
+            map.set(id, row);
+          }
+        }
+      });
+      return map;
+    },
     setSelected: (ids: Set<string>) =>
       set(() => ({
         selectedIds: new Set(ids),

package/src/TrackSelect/types.ts

@@ -105,6 +105,8 @@ export type SelectionState = {
 export type SelectionAction = {
   // Returns only real track IDs (filters out auto-generated group IDs)
   getTrackIds: () => Set<string>;
+  // Returns a Map of track IDs to RowInfo (no auto-generated IDs)
+  getTrackMap: () => Map<string, RowInfo>;
   setSelected: (ids: Set<string>) => void;
   removeIds: (removedIds: Set<string>) => void;
   clear: () => void;

package/src/lib.ts

@@ -6,3 +6,9 @@ import {
   type SelectionStoreInstance,
 } from "./TrackSelect/store.ts";
 export { createSelectionStore, SelectionStoreInstance };
+
+import { rowById } from "./TrackSelect/consts.ts";
+export { rowById };
+
+import type { RowInfo } from "./TrackSelect/types.ts";
+export { RowInfo };

package/test/main.tsx

@@ -1,10 +1,43 @@
+import { useEffect, useMemo } from "react";
 import { createSelectionStore } from "../src/TrackSelect/store";
 import TrackSelect from "../src/TrackSelect/TrackSelect";
 import { createRoot } from "react-dom/client";
 
+function getLocalStorage(assembly: string): Set<string> | null {
+  if (typeof window === "undefined" || !window.sessionStorage) return null;
+
+  const selectedIds = sessionStorage.getItem(assembly + "-selected-tracks");
+  if (!selectedIds) return null;
+  const idsArray = JSON.parse(selectedIds) as string[];
+  return new Set(idsArray);
+}
+
+function setLocalStorage(trackIds: Set<string>, assembly: string) {
+  sessionStorage.setItem(
+    assembly + "-selected-tracks",
+    JSON.stringify([...trackIds]),
+  );
+}
+
+const assembly = "grch38";
 function Main() {
-  const store = createSelectionStore();
-  return <TrackSelect store={store} />;
+  const selectionStore = useMemo(() => {
+    const localIds = getLocalStorage(assembly);
+    const ids = localIds != null ? localIds : new Set<string>();
+    return createSelectionStore(ids);
+  }, [assembly]);
+
+  const selectedIds = selectionStore((s) => s.selectedIds);
+  const getTrackIds = selectionStore((s) => s.getTrackIds);
+
+  // Get only real track IDs (no auto-generated group IDs)
+  const trackIds = useMemo(() => getTrackIds(), [selectedIds, getTrackIds]);
+
+  useEffect(() => {
+    setLocalStorage(trackIds, assembly);
+  }, [assembly, trackIds]);
+
+  return <TrackSelect store={selectionStore} />;
 }
 
 createRoot(document.getElementById("root")!).render(<Main />);