@marimo-team/islands 0.23.4-dev9 → 0.23.5-dev1

package/dist/{vega-component-CjMUUeEZ.js → vega-component-BnCQmtxw.js}

@@ -15,7 +15,7 @@ import { a as isValid, i as AlertTitle, n as Alert, t as arrow } from "./formats
 import { n as formats } from "./vega-loader.browser-3_z8GoFC.js";
 import { a as getContainerWidth, n as vegaLoadData, s as tooltipHandler } from "./loader-BvW0-YWZ.js";
 import { t as useAsyncData } from "./useAsyncData-CKYzhCis.js";
-import { t as j } from "./react-vega-C2Rtgjb4.js";
+import { t as j } from "./react-vega-k9ODWPlI.js";
 import "./defaultLocale-BpsHxBd7.js";
 import "./defaultLocale-DoeErsX2.js";
 import { t as useDeepCompareMemoize } from "./useDeepCompareMemoize-je76AJS_.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.23.4-dev9",
+  "version": "0.23.5-dev1",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",
@@ -155,7 +155,7 @@
     "typescript-memoize": "^1.1.1",
     "use-acp": "0.2.6",
     "use-resize-observer": "^9.1.0",
-    "vega-lite": "6.3.1",
+    "vega-lite": "6.4.2",
     "vega-loader": "^5.1.0",
     "vega-parser": "^7.1.0",
     "vega-tooltip": "^1.1.0",

package/src/components/data-table/__tests__/column-header.test.ts

@@ -0,0 +1,63 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { describe, expect, it } from "vitest";
+import { seedFromFilter } from "../column-header";
+import { Filter } from "../filters";
+
+describe("seedFromFilter", () => {
+  it("returns empty defaults when there is no filter", () => {
+    expect(seedFromFilter(undefined)).toEqual({
+      values: [],
+      operator: "in",
+    });
+  });
+
+  it("seeds values from an `in` select filter", () => {
+    const filter = Filter.select({
+      options: ["Flying", "Ground"],
+      operator: "in",
+    });
+    expect(seedFromFilter(filter)).toEqual({
+      values: ["Flying", "Ground"],
+      operator: "in",
+    });
+  });
+
+  it("preserves `not_in` so reapplying does not silently flip to `in`", () => {
+    const filter = Filter.select({
+      options: ["Fire"],
+      operator: "not_in",
+    });
+    expect(seedFromFilter(filter)).toEqual({
+      values: ["Fire"],
+      operator: "not_in",
+    });
+  });
+
+  it("returns a fresh array (callers may mutate without affecting the filter)", () => {
+    const options = ["a", "b"];
+    const filter = Filter.select({ options, operator: "in" });
+    const seeded = seedFromFilter(filter);
+    seeded.values.push("c");
+    expect(options).toEqual(["a", "b"]);
+  });
+
+  it("ignores non-select filters and falls back to defaults", () => {
+    expect(
+      seedFromFilter(Filter.text({ text: "abc", operator: "equals" })),
+    ).toEqual({
+      values: [],
+      operator: "in",
+    });
+    expect(seedFromFilter(Filter.number({ min: 0, max: 10 }))).toEqual({
+      values: [],
+      operator: "in",
+    });
+    expect(
+      seedFromFilter(Filter.boolean({ value: true, operator: "is_true" })),
+    ).toEqual({
+      values: [],
+      operator: "in",
+    });
+  });
+});

package/src/components/data-table/__tests__/columns.test.tsx

@@ -312,7 +312,7 @@ describe("generateColumns", () => {
     expect(cell?.props.className).toContain("center");
   });
 
-  it("should always left-align column headers regardless of text justification", () => {
+  it("should align column headers to match textJustifyColumns", () => {
     const columns = generateColumns({
       rowHeaders: [],
       selection: null,
@@ -330,7 +330,8 @@ describe("generateColumns", () => {
       columnDef: { meta: col.meta },
     });
 
-    // Even with right justification, header is left-aligned with sort + menu buttons
+    // Right-justified column: outer summary wrapper aligns to end, and the
+    // header row uses flex-row-reverse so the title sits at the right edge.
     const { container: rightContainer } = render(
       <TooltipProvider>
         {/* oxlint-disable-next-line typescript/no-explicit-any */}
@@ -345,12 +346,11 @@ describe("generateColumns", () => {
         "[data-testid='data-table-column-menu-button']",
       ),
     ).toBeTruthy();
-    // No flex-row-reverse or items-end on header
-    const rightWrapper = rightContainer.firstElementChild;
-    expect(rightWrapper?.className).not.toContain("items-end");
-    expect(rightWrapper?.className).not.toContain("flex-row-reverse");
+    expect(rightContainer.firstElementChild?.className).toContain("items-end");
+    expect(rightContainer.querySelector(".flex-row-reverse")).toBeTruthy();
 
-    // Same for center-justified column
+    // Center-justified column: outer summary wrapper centers; header row
+    // keeps natural order.
     const { container: centerContainer } = render(
       <TooltipProvider>
         {/* oxlint-disable-next-line typescript/no-explicit-any */}
@@ -365,9 +365,39 @@ describe("generateColumns", () => {
         "[data-testid='data-table-column-menu-button']",
       ),
     ).toBeTruthy();
-    const centerWrapper = centerContainer.firstElementChild;
-    expect(centerWrapper?.className).not.toContain("items-center");
-    expect(centerWrapper?.className).not.toContain("flex-row-reverse");
+    expect(centerContainer.firstElementChild?.className).toContain(
+      "items-center",
+    );
+    expect(centerContainer.querySelector(".flex-row-reverse")).toBeNull();
+  });
+
+  it("should not auto-align numeric column headers without explicit override", () => {
+    const columns = generateColumns({
+      rowHeaders: [],
+      selection: null,
+      fieldTypes,
+    });
+
+    const mockColumn = (col: (typeof columns)[number]) => ({
+      id: col.id,
+      getCanSort: () => true,
+      getCanFilter: () => false,
+      getIsSorted: () => false,
+      getSortIndex: () => -1,
+      getFilterValue: () => undefined,
+      columnDef: { meta: col.meta },
+    });
+
+    // "age" is numeric: cells auto right-align, but the header stays
+    // left-aligned unless the user explicitly opts in via text_justify_columns.
+    const { container } = render(
+      <TooltipProvider>
+        {/* oxlint-disable-next-line typescript/no-explicit-any */}
+        {(columns[1].header as any)({ column: mockColumn(columns[1]) })}
+      </TooltipProvider>,
+    );
+    expect(container.firstElementChild?.className).not.toContain("items-end");
+    expect(container.querySelector(".flex-row-reverse")).toBeNull();
   });
 
   it("should cycle sort button through asc, desc, and clear on clicks", () => {

package/src/components/data-table/column-header.tsx

@@ -9,7 +9,7 @@ import {
   TextIcon,
   XIcon,
 } from "lucide-react";
-import { useMemo, useRef, useState } from "react";
+import { useRef, useState } from "react";
 import { useLocale } from "react-aria";
 import {
   DropdownMenu,
@@ -22,24 +22,12 @@ import {
   DropdownMenuSubTrigger,
   DropdownMenuTrigger,
 } from "@/components/ui/dropdown-menu";
-import { useAsyncData } from "@/hooks/useAsyncData";
-import { ErrorBanner } from "@/plugins/impl/common/error-banner";
 import type { CalculateTopKRows } from "@/plugins/impl/DataTablePlugin";
 import type { OperatorType } from "@/plugins/impl/data-frames/utils/operators";
 import { logNever } from "@/utils/assertNever";
 import { cn } from "@/utils/cn";
-import { Logger } from "@/utils/Logger";
 import { capitalize } from "@/utils/strings";
-import { Spinner } from "../icons/spinner";
 import { Button } from "../ui/button";
-import { Checkbox } from "../ui/checkbox";
-import {
-  Command,
-  CommandEmpty,
-  CommandInput,
-  CommandItem,
-  CommandList,
-} from "../ui/command";
 import { DraggablePopover } from "../ui/draggable-popover";
 import { Input } from "../ui/input";
 import { NumberField } from "../ui/number-field";
@@ -52,8 +40,12 @@ import {
   SelectTrigger,
   SelectValue,
 } from "../ui/select";
-import { type ColumnFilterForType, Filter } from "./filters";
-import { SentinelCell } from "./sentinel-cell";
+import { FilterByValuesList } from "./filter-by-values-picker";
+import {
+  type ColumnFilterForType,
+  type ColumnFilterValue,
+  Filter,
+} from "./filters";
 import {
   ClearFilterMenuItem,
   FilterButtons,
@@ -66,9 +58,6 @@ import {
   renderSortFilterIcon,
   renderSorts,
 } from "./header-items";
-import { detectSentinel, stringifyUnknownValue } from "./utils";
-
-const TOP_K_ROWS = 30;
 
 interface DataTableColumnHeaderProps<
   TData,
@@ -77,6 +66,7 @@ interface DataTableColumnHeaderProps<
   column: Column<TData, TValue>;
   header: React.ReactNode;
   subheader?: React.ReactNode;
+  justify?: "left" | "center" | "right";
   calculateTopKRows?: CalculateTopKRows;
   table?: Table<TData>;
 }
@@ -85,6 +75,7 @@ export const DataTableColumnHeader = <TData, TValue>({
   column,
   header,
   subheader,
+  justify,
   className,
   calculateTopKRows,
   table,
@@ -100,7 +91,13 @@ export const DataTableColumnHeader = <TData, TValue>({
   // No sorting or filtering
   if (!column.getCanSort() && !column.getCanFilter()) {
     return (
-      <div className={cn(className)}>
+      <div
+        className={cn(
+          justify === "center" && "text-center",
+          justify === "right" && "text-right",
+          className,
+        )}
+      >
         {header}
         {subheader}
       </div>
@@ -114,9 +111,24 @@ export const DataTableColumnHeader = <TData, TValue>({
       <div
         className={cn("group flex flex-col my-1 w-full select-none", className)}
       >
-        <div className="flex items-center gap-1">
-          <span>{header}</span>
-          {column.getCanSort() && <SortButton column={column} />}
+        <div
+          className={cn(
+            "flex items-center gap-1",
+            justify === "right" && "flex-row-reverse",
+            justify === "center" && "mx-auto",
+          )}
+        >
+          {justify === "center" ? (
+            <>
+              {column.getCanSort() && <SortButton column={column} />}
+              <span>{header}</span>
+            </>
+          ) : (
+            <>
+              <span>{header}</span>
+              {column.getCanSort() && <SortButton column={column} />}
+            </>
+          )}
           <DropdownMenu modal={false}>
             <DropdownMenuTrigger asChild={true}>
               <button
@@ -534,6 +546,26 @@ const TextFilter = <TData, TValue>({
   );
 };
 
+/**
+ * Seed the filter-by-values picker from a column's existing filter value.
+ *
+ * Reopening the picker should reflect what's already applied. Only `select`
+ * filters carry checkbox-style values; other filter shapes (number, text,
+ * etc.) seed an empty list.
+ */
+export function seedFromFilter(value: ColumnFilterValue | undefined): {
+  values: unknown[];
+  operator: Extract<OperatorType, "in" | "not_in">;
+} {
+  if (value && "type" in value && value.type === "select") {
+    return {
+      values: [...value.options],
+      operator: value.operator === "not_in" ? "not_in" : "in",
+    };
+  }
+  return { values: [], operator: "in" };
+}
+
 const PopoverFilterByValues = <TData, TValue>({
   setIsFilterValueOpen,
   calculateTopKRows,
@@ -543,69 +575,13 @@ const PopoverFilterByValues = <TData, TValue>({
   calculateTopKRows?: CalculateTopKRows;
   column: Column<TData, TValue>;
 }) => {
-  const [chosenValues, setChosenValues] = useState<Set<unknown>>(new Set());
-  const [query, setQuery] = useState<string>("");
-
-  const { data, isPending, error } = useAsyncData(async () => {
-    if (!calculateTopKRows) {
-      return null;
-    }
-    const res = await calculateTopKRows({ column: column.id, k: TOP_K_ROWS });
-    return res.data;
-  }, []);
-
-  const filteredData = useMemo(() => {
-    if (!data) {
-      return [];
-    }
-
-    try {
-      return data.filter(([value, _count]) => {
-        // Check if value exists and can be converted to string
-        // Keep null values for filtering
-        return value === undefined
-          ? false
-          : String(value).toLowerCase().includes(query.toLowerCase());
-      });
-    } catch (error_) {
-      Logger.error("Error filtering data", error_);
-      return [];
-    }
-  }, [data, query]);
-
-  let dataTable: React.ReactNode;
-
-  if (isPending) {
-    dataTable = <Spinner size="medium" className="mx-auto mt-12 mb-10" />;
-  }
-
-  if (error) {
-    dataTable = <ErrorBanner error={error} className="my-10 mx-4" />;
-  }
-
-  const handleToggle = (value: unknown) => {
-    setChosenValues((prev) => {
-      const checked = prev.has(value);
-      const newSet = new Set(prev);
-      if (checked) {
-        newSet.delete(value);
-      } else {
-        newSet.add(value);
-      }
-      return newSet;
-    });
-  };
+  const seed = seedFromFilter(
+    column.getFilterValue() as ColumnFilterValue | undefined,
+  );
 
-  const handleToggleAll = (checked: boolean) => {
-    if (!data) {
-      return;
-    }
-    if (checked) {
-      setChosenValues(new Set(filteredData.map(([value]) => value)));
-    } else {
-      setChosenValues(new Set());
-    }
-  };
+  const [chosenValues, setChosenValues] = useState<Set<unknown>>(
+    () => new Set(seed.values),
+  );
 
   const handleApply = () => {
     if (chosenValues.size === 0) {
@@ -613,87 +589,13 @@ const PopoverFilterByValues = <TData, TValue>({
       return;
     }
     column.setFilterValue(
-      Filter.select({ options: [...chosenValues], operator: "in" }),
+      Filter.select({
+        options: [...chosenValues],
+        operator: seed.operator,
+      }),
     );
   };
 
-  if (data) {
-    const allChecked = chosenValues.size === filteredData.length;
-
-    dataTable = (
-      <>
-        <Command className="text-sm outline-hidden" shouldFilter={false}>
-          <CommandInput
-            placeholder={`Search among the top ${data.length} values`}
-            autoFocus={true}
-            onValueChange={(value) => setQuery(value.trim())}
-          />
-          <CommandEmpty>No results found.</CommandEmpty>
-          <CommandList className="border-b">
-            {filteredData.length > 0 && (
-              <CommandItem
-                value="__select-all__"
-                className="border-b rounded-none px-3"
-                onSelect={() => handleToggleAll(!allChecked)}
-              >
-                <Checkbox
-                  checked={chosenValues.size === filteredData.length}
-                  aria-label="Select all"
-                  className="mr-3 h-3.5 w-3.5"
-                />
-                <span className="font-bold flex-1">{column.id}</span>
-                <span className="font-bold">Count</span>
-              </CommandItem>
-            )}
-            {filteredData.map(([value, count], rowIndex) => {
-              const isSelected = chosenValues.has(value);
-              const valueString = stringifyUnknownValue({ value });
-              const sentinel = detectSentinel(
-                value,
-                column.columnDef.meta?.dataType,
-              );
-
-              return (
-                <CommandItem
-                  key={rowIndex}
-                  value={valueString}
-                  className="not-last:border-b rounded-none px-3"
-                  onSelect={() => handleToggle(value)}
-                >
-                  <Checkbox
-                    checked={isSelected}
-                    aria-label="Select row"
-                    className="mr-3 h-3.5 w-3.5"
-                  />
-                  <span className="flex-1 overflow-hidden max-h-20 line-clamp-3">
-                    {sentinel ? (
-                      <SentinelCell sentinel={sentinel} />
-                    ) : (
-                      valueString
-                    )}
-                  </span>
-                  <span className="ml-3">{count}</span>
-                </CommandItem>
-              );
-            })}
-          </CommandList>
-          {filteredData.length === TOP_K_ROWS && (
-            <span className="text-xs text-muted-foreground mt-1.5 text-center">
-              Only showing the top {TOP_K_ROWS} values
-            </span>
-          )}
-        </Command>
-        <FilterButtons
-          onApply={handleApply}
-          onClear={() => {
-            setChosenValues(new Set());
-          }}
-          clearButtonDisabled={chosenValues.size === 0}
-        />
-      </>
-    );
-  }
-
   return (
     <DraggablePopover
       open={true}
@@ -709,7 +611,19 @@ const PopoverFilterByValues = <TData, TValue>({
           <XIcon className="h-4 w-4" />
         </Button>
       </PopoverClose>
-      <div className="flex flex-col gap-1.5 py-2">{dataTable}</div>
+      <div className="flex flex-col gap-1.5 py-2">
+        <FilterByValuesList
+          column={column}
+          calculateTopKRows={calculateTopKRows}
+          chosenValues={chosenValues}
+          onChange={(values) => setChosenValues(new Set(values))}
+        />
+        <FilterButtons
+          onApply={handleApply}
+          onClear={() => setChosenValues(new Set())}
+          clearButtonDisabled={chosenValues.size === 0}
+        />
+      </div>
     </DraggablePopover>
   );
 };

package/src/components/data-table/columns.tsx

@@ -197,9 +197,17 @@ export function generateColumns<T>({
         const stats = chartSpecModel?.getColumnStats(key);
         const dtype = column.columnDef.meta?.dtype;
         const headerTitle = headerTooltip?.[key];
+        const headerJustify = textJustifyColumns?.[key];
+
         const dtypeHeader =
           showDataTypes && dtype ? (
-            <div className="flex flex-row gap-1">
+            <div
+              className={cn(
+                "flex flex-row gap-1",
+                headerJustify === "center" && "justify-center",
+                headerJustify === "right" && "justify-end",
+              )}
+            >
               <span className="text-xs text-muted-foreground">{dtype}</span>
               {stats && typeof stats.nulls === "number" && stats.nulls > 0 && (
                 <span className="text-xs text-muted-foreground">
@@ -233,6 +241,7 @@ export function generateColumns<T>({
             header={headerWithTooltip}
             subheader={dtypeHeader}
             column={column}
+            justify={headerJustify}
             calculateTopKRows={calculateTopKRows}
             table={table}
           />
@@ -247,6 +256,8 @@ export function generateColumns<T>({
           <div
             className={cn(
               "flex flex-col h-full pt-0.5 pb-3 justify-between items-start",
+              headerJustify === "center" && "items-center",
+              headerJustify === "right" && "items-end",
             )}
           >
             {dataTableColumnHeader}

package/src/components/data-table/filter-by-values-picker.tsx

@@ -74,7 +74,7 @@ export const FilterByValuesPicker = <TData, TValue>({
         </Button>
       </PopoverTrigger>
       <PopoverContent className="w-80 p-0">
-        <PickerBody
+        <FilterByValuesList
           column={column}
           calculateTopKRows={calculateTopKRows}
           chosenValues={chosenValuesSet}
@@ -85,19 +85,22 @@ export const FilterByValuesPicker = <TData, TValue>({
   );
 };
 
-interface PickerBodyProps<TData, TValue> {
+interface FilterByValuesListProps<TData, TValue> {
   column: Column<TData, TValue>;
   calculateTopKRows?: CalculateTopKRows;
   chosenValues: Set<unknown>;
   onChange: (values: unknown[]) => void;
 }
 
-const PickerBody = <TData, TValue>({
+/**
+ * Search + checkbox list that powers the "filter by values" picker.
+ */
+export const FilterByValuesList = <TData, TValue>({
   column,
   calculateTopKRows,
   chosenValues,
   onChange,
-}: PickerBodyProps<TData, TValue>) => {
+}: FilterByValuesListProps<TData, TValue>) => {
   const [query, setQuery] = useState<string>("");
 
   const { data, isPending, error } = useAsyncData(async () => {