@weng-lab/genomebrowser-ui 0.3.6 → 0.4.0-beta.0

package/test/main.tsx

@@ -1,15 +1,10 @@
 /// <reference types="vite/client" />
 
 // react
-import { useCallback, useEffect, useMemo, useState } from "react";
+import { useState } from "react";
 import { createRoot } from "react-dom/client";
 
-// license
-import { LicenseInfo } from "@mui/x-license";
-const muiLicenseKey = import.meta.env.VITE_MUI_X_LICENSE_KEY;
-if (muiLicenseKey) {
-  LicenseInfo.setLicenseKey(muiLicenseKey);
-}
+import "../src/muiLicense";
 
 // mui
 import EditIcon from "@mui/icons-material/Edit";
@@ -17,76 +12,27 @@ import { Button } from "@mui/material";
 
 // weng lab
 import {
-  BigBedConfig,
-  BigWigConfig,
   Browser,
   createBrowserStoreMemo,
   createTrackStoreMemo,
-  DisplayMode,
-  Domain,
   GQLWrapper,
-  MethylCConfig,
-  Rect,
-  Track,
-  TrackType,
-  TranscriptConfig,
 } from "@weng-lab/genomebrowser";
 
 // local
-import { foldersByAssembly, TrackSelect } from "../src/lib";
-import type { BiosampleRowInfo } from "../src/TrackSelect/Folders/biosamples/shared/types";
-import type { GeneRowInfo } from "../src/TrackSelect/Folders/genes/shared/types";
-import type { OtherTrackInfo } from "../src/TrackSelect/Folders/other-tracks/shared/types";
-import { Exon } from "@weng-lab/genomebrowser/dist/components/tracks/transcript/types";
-import { tfPeaksTrack } from "../src/TrackSelect/Custom/TfPeaks";
-
-interface Transcript {
-  id: string;
-  name: string;
-  coordinates: Domain;
-  strand: string;
-  exons?: Exon[];
-  color?: string;
-}
+import {
+  foldersByAssembly,
+  type InitialSelectedIdsByAssembly,
+  TrackSelect,
+  type TrackSelectTrackContext,
+} from "../src/lib";
 
 type Assembly = "GRCh38" | "mm10";
 
-// Callback types for track interactions (using any to avoid type conflicts with library types)
-interface TrackCallbacks {
-  onHover: (item: any) => void;
-  onLeave: () => void;
-  onCCREClick: (item: any) => void;
-  onGeneClick: (item: any) => void;
-}
-
-// Helper to inject callbacks based on track type
-function injectCallbacks(track: Track, callbacks: TrackCallbacks): Track {
-  if (track.trackType === TrackType.Transcript) {
-    return {
-      ...track,
-      onHover: callbacks.onHover,
-      onLeave: callbacks.onLeave,
-      onClick: callbacks.onGeneClick,
-    };
-  }
-  if (track.trackType === TrackType.BigBed) {
-    return {
-      ...track,
-      onHover: callbacks.onHover,
-      onLeave: callbacks.onLeave,
-      onClick: callbacks.onCCREClick,
-    };
-  }
-  return track;
-}
-
 function Main() {
   const [open, setOpen] = useState(false);
   const currentAssembly: Assembly = "GRCh38";
 
   const browserStore = createBrowserStoreMemo({
-    // chr7:19,695,494-19,699,803
-    // chr1:11103779-11262792
     domain: { chromosome: "chr12", start: 53380108, end: 53416378 },
     marginWidth: 100,
     trackWidth: 1400,
@@ -95,123 +41,18 @@ function Main() {
 
   const domain = browserStore((s) => s.domain);
 
-  const addHighlight = browserStore((s) => s.addHighlight);
-  const removeHighlight = browserStore((s) => s.removeHighlight);
-  const onHover = useCallback(
-    (item: Rect | Transcript) => {
-      const domain =
-        "start" in item
-          ? { start: item.start, end: item.end }
-          : { start: item.coordinates.start, end: item.coordinates.end };
+  const folders = foldersByAssembly[currentAssembly];
 
-      addHighlight({
-        id: "hover-highlight",
-        domain,
-        color: item.color || "blue",
-      });
+  const trackStore = createTrackStoreMemo([], [currentAssembly]);
+  const trackContext: TrackSelectTrackContext = {
+    onGeneClick: ({ trackId, transcript }) => {
+      console.log("Gene clicked", trackId, transcript.name);
     },
-    [addHighlight],
-  );
-  const onLeave = useCallback(() => {
-    removeHighlight("hover-highlight");
-  }, [removeHighlight]);
-
-  const onCCREClick = useCallback((item: Rect) => {
-    console.log(item);
-  }, []);
-  const onGeneClick = useCallback((item: Transcript) => {
-    console.log(item);
-  }, []);
-
-  // Bundle callbacks for track injection
-  const callbacks = useMemo<TrackCallbacks>(
-    () => ({
-      onHover,
-      onLeave,
-      onCCREClick,
-      onGeneClick,
-    }),
-    [onHover, onLeave, onCCREClick, onGeneClick],
-  );
-
-  const trackStore = useLocalTracks(currentAssembly, callbacks);
-
-  const tracks = trackStore((s) => s.tracks);
-  const insertTrack = trackStore((s) => s.insertTrack);
-  const removeTrack = trackStore((s) => s.removeTrack);
-
-  const folders = useMemo(
-    () => foldersByAssembly[currentAssembly],
-    [currentAssembly],
-  );
-
-  const storageKey = `${currentAssembly}-selected-tracks`;
-
-  const initialSelection = useMemo(
-    () =>
-      (currentAssembly as Assembly) === "GRCh38"
-        ? defaultHumanSelections
-        : defaultMouseSelections,
-    [currentAssembly],
-  );
-
-  // sync tracks to browser and save to localStorage
-  const handleSubmit = useCallback(
-    (selectedByFolder: Map<string, Set<string>>) => {
-      const currentIds = new Set(tracks.map((t) => t.id));
-      const selectedIds = new Set<string>();
-      const tracksToAdd: Array<{ row: unknown; folderId: string }> = [];
-
-      for (const folder of folders) {
-        const folderSelection =
-          selectedByFolder.get(folder.id) ?? new Set<string>();
-        folderSelection.forEach((id) => {
-          selectedIds.add(id);
-          if (currentIds.has(id)) {
-            return;
-          }
-          const row = folder.rowById.get(id);
-          if (row) {
-            tracksToAdd.push({ row, folderId: folder.id });
-          }
-        });
-      }
-
-      const tracksToRemove = tracks.filter((t) => !selectedIds.has(t.id));
-      for (const t of tracksToRemove) {
-        removeTrack(t.id);
-      }
-
-      for (const { row, folderId } of tracksToAdd) {
-        const track = generateTrack(
-          row as BiosampleRowInfo | GeneRowInfo,
-          folderId,
-          currentAssembly,
-          callbacks,
-        );
-        if (track === null) continue;
-        insertTrack(track);
-      }
+    onBiosampleFeatureHover: ({ trackId, rect }) => {
+      console.log("Biosample feature hovered", trackId, rect.name ?? "unknown");
     },
-    [tracks, removeTrack, insertTrack, callbacks, folders, currentAssembly],
-  );
-
-  // clear selections and remove all tracks
-  const handleClear = () => {
-    for (const t of tracks) {
-      removeTrack(t.id);
-    }
   };
 
-  // On first load, if no stored selection exists, apply initial selection
-  useEffect(() => {
-    const stored = sessionStorage.getItem(storageKey);
-    if (!stored) {
-      handleSubmit(initialSelection);
-    }
-    // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, []);
-
   return (
     <>
       <div>
@@ -226,15 +67,16 @@ function Main() {
         Select Tracks
       </Button>
       <TrackSelect
+        assembly={currentAssembly}
         folders={folders}
-        storageKey={storageKey}
-        initialSelection={initialSelection}
-        onSubmit={handleSubmit}
-        onClear={handleClear}
+        initialSelectedIds={defaultSelections}
+        sessionStorageKey={`track-select:${currentAssembly}`}
+        trackContext={trackContext}
+        trackStore={trackStore}
         maxTracks={30}
         open={open}
         onClose={() => setOpen(false)}
-        title="Biosample Tracks"
+        title="Select Tracks"
       />
       <GQLWrapper>
         <Browser browserStore={browserStore} trackStore={trackStore} />
@@ -245,178 +87,20 @@ function Main() {
 
 createRoot(document.getElementById("root")!).render(<Main />);
 
-const ASSAY_COLORS: Record<string, string> = {
-  dnase: "#06da93",
-  h3k4me3: "#ff0000",
-  h3k27ac: "#ffcd00",
-  ctcf: "#00b0d0",
-  atac: "#02c7b9",
-  rnaseq: "#00aa00",
-  chromhmm: "#00ff00",
-  ccre: "#000000",
-  wgbs: "#648bd8",
-};
-
-function generateTrack(
-  row: BiosampleRowInfo | GeneRowInfo | OtherTrackInfo,
-  folderId: string,
-  assembly: Assembly,
-  callbacks?: TrackCallbacks,
-): Track | null {
-  // Handle gene folders
-  if (folderId.includes("genes")) {
-    const geneRow = row as GeneRowInfo;
-    const track: Track = {
-      ...defaultTranscript,
-      id: geneRow.id,
-      assembly,
-      version: geneRow.versions[geneRow.versions.length - 1], // latest version
-    };
-    return callbacks ? injectCallbacks(track, callbacks) : track;
-  }
-
-  // Handle other-tracks folder
-  if (folderId.includes("other-tracks")) {
-    if (row.id === "tf-peaks") {
-      return { ...tfPeaksTrack };
-    }
-    return null;
-  }
-
-  // Handle biosample folders
-  const sel = row as BiosampleRowInfo;
-  const color = ASSAY_COLORS[sel.assay.toLowerCase()] || "#000000";
-  let track: Track;
-
-  switch (sel.assay.toLowerCase()) {
-    case "chromhmm":
-    case "ccre":
-      track = {
-        ...defaultBigBed,
-        id: sel.id,
-        url: sel.url ?? "",
-        title: sel.displayName,
-        color,
-      };
-      break;
-    case "wgbs":
-      track = {
-        ...defaultMethylC,
-        id: sel.id,
-        title: sel.displayName,
-        urls: {
-          plusStrand: {
-            cpg: { url: sel.cpgPlus ?? "" },
-            chg: { url: "" },
-            chh: { url: "" },
-            depth: { url: sel.coverage ?? "" },
-          },
-          minusStrand: {
-            cpg: { url: sel.cpgMinus ?? "" },
-            chg: { url: "" },
-            chh: { url: "" },
-            depth: { url: sel.coverage ?? "" },
-          },
-        },
-      };
-      break;
-    default:
-      track = {
-        ...defaultBigWig,
-        id: sel.id,
-        url: sel.url ?? "",
-        title: sel.displayName,
-        color,
-      };
-  }
-
-  return callbacks ? injectCallbacks(track, callbacks) : track;
-}
-
-export const defaultBigWig: Omit<BigWigConfig, "id" | "title" | "url"> = {
-  trackType: TrackType.BigWig,
-  height: 50,
-  displayMode: DisplayMode.Full,
-  titleSize: 12,
-};
-
-export const defaultBigBed: Omit<BigBedConfig, "id" | "title" | "url"> = {
-  trackType: TrackType.BigBed,
-  height: 20,
-  displayMode: DisplayMode.Dense,
-  titleSize: 12,
-};
-
-export const defaultMethylC: Omit<MethylCConfig, "id" | "title" | "urls"> = {
-  trackType: TrackType.MethylC,
-  height: 100,
-  displayMode: DisplayMode.Split,
-  titleSize: 12,
-  color: "#648bd8",
-  colors: {
-    cpg: "#648bd8",
-    chg: "#ff944d",
-    chh: "#ff00ff",
-    depth: "#525252",
+// Default selections for TrackSelect UI (uses folder row IDs)
+const defaultSelections: InitialSelectedIdsByAssembly = {
+  GRCh38: {
+    "human-genes": ["human-genes/gencode-basic"],
+    "human-biosamples": [
+      "human-biosamples/ccre-aggregate",
+      "human-biosamples/dnase-aggregate",
+    ],
+  },
+  mm10: {
+    "mouse-genes": ["mouse-genes/gencode-basic"],
+    "mouse-biosamples": [
+      "mouse-biosamples/ccre-aggregate",
+      "mouse-biosamples/dnase-aggregate",
+    ],
   },
 };
-
-export const defaultTranscript: Omit<
-  TranscriptConfig,
-  "id" | "assembly" | "version"
-> = {
-  title: "GENCODE Genes",
-  trackType: TrackType.Transcript,
-  displayMode: DisplayMode.Squish,
-  height: 100,
-  color: "#0c184a", // screen theme default
-  canonicalColor: "#100e98", // screen theme light
-  highlightColor: "#3c69e8", // bright blue
-  titleSize: 12,
-};
-
-export function useLocalTracks(assembly: string, callbacks?: TrackCallbacks) {
-  const localTracks = getLocalTracks(assembly);
-
-  // Start empty if no stored tracks - TrackSelect will populate defaults
-  let initialTracks: Track[] = localTracks || [];
-
-  // Inject callbacks if provided (callbacks are lost on JSON serialization)
-  if (callbacks) {
-    initialTracks = initialTracks.map((t) => injectCallbacks(t, callbacks));
-  }
-
-  const trackStore = createTrackStoreMemo(initialTracks, []);
-  const tracks = trackStore((state) => state.tracks);
-
-  // any time the track list changes, update local storage
-  useEffect(() => {
-    setLocalTracks(tracks, assembly);
-  }, [tracks, assembly]);
-
-  return trackStore;
-}
-
-export function getLocalTracks(assembly: string): Track[] | null {
-  if (typeof window === "undefined" || !window.sessionStorage) return null;
-
-  const localTracks = sessionStorage.getItem(assembly + "-tracks");
-  if (!localTracks) return null;
-  const localTracksJson = JSON.parse(localTracks) as Track[];
-  return localTracksJson;
-}
-
-export function setLocalTracks(tracks: Track[], assembly: string) {
-  sessionStorage.setItem(assembly + "-tracks", JSON.stringify(tracks));
-}
-
-// Default selections for TrackSelect UI (uses folder row IDs)
-const defaultHumanSelections = new Map<string, Set<string>>([
-  ["human-genes", new Set(["gencode-basic"])],
-  ["human-biosamples", new Set(["ccre-aggregate", "dnase-aggregate"])],
-]);
-
-const defaultMouseSelections = new Map<string, Set<string>>([
-  ["mouse-genes", new Set(["gencode-basic"])],
-  ["mouse-biosamples", new Set(["ccre-aggregate", "dnase-aggregate"])],
-]);

package/test/mocks/logo-test.tsx

@@ -0,0 +1,11 @@
+import { ReactNode } from "react";
+
+function Stub({ children }: { children?: ReactNode }) {
+  return <>{children ?? null}</>;
+}
+
+export const A = Stub;
+export const C = Stub;
+export const G = Stub;
+export const T = Stub;
+export const DNALogo = Stub;

package/test/mohdDisplay.test.tsx

@@ -0,0 +1,45 @@
+import { renderToStaticMarkup } from "react-dom/server";
+import { describe, expect, it, vi } from "vitest";
+import { buildSelectedTree } from "../src/TrackSelect/buildSelectedTree";
+vi.mock("../src/TrackSelect/Folders/mohd/shared/MohdGroupingCell", () => ({
+  MohdGroupingCell: () => null,
+}));
+vi.mock("../src/TrackSelect/Folders/mohd/shared/MohdTreeItem", () => ({
+  MohdTreeItem: () => null,
+}));
+vi.mock("../src/TrackSelect/Folders/mohd/shared/MohdViewSelector", () => ({
+  MohdViewSelector: () => null,
+}));
+import { humanMohdFolder } from "../src/TrackSelect/Folders/mohd/human";
+import { MohdOmeIcon } from "../src/TrackSelect/Folders/mohd/shared/config";
+
+describe("MOHD display helpers", () => {
+  it("renders an ome color icon", () => {
+    const html = renderToStaticMarkup(<MohdOmeIcon type="ATAC" />);
+
+    expect(html).toContain("mohd-ome-icon-atac");
+  });
+
+  it("wires MOHD-specific grouping and tree components onto the folder", () => {
+    expect(humanMohdFolder.GroupingCellComponent).toBeTruthy();
+    expect(humanMohdFolder.TreeItemComponent).toBeTruthy();
+  });
+
+  it("marks ome nodes in the selected tree for icon rendering", () => {
+    const selectedRows = humanMohdFolder.rows.filter(
+      (row) => row.sampleId === "MOHD_EA100001",
+    );
+
+    const tree = buildSelectedTree({
+      folderId: humanMohdFolder.id,
+      rootLabel: humanMohdFolder.label,
+      selectedRows,
+      groupingModel: ["ome", "site", "sampleId"],
+      leafField: "description",
+    });
+
+    expect(tree[0]?.children?.[0]?.label).toBe("ATAC");
+    expect(tree[0]?.children?.[0]?.highlightName).toBe("ATAC");
+    expect(tree[0]?.children?.[0]?.isHighlightedItem).toBe(true);
+  });
+});