@marimo-team/frontend 0.23.7-dev6 → 0.23.7-dev60

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

@@ -9,7 +9,7 @@ import {
   TextIcon,
   XIcon,
 } from "lucide-react";
-import { useRef, useState } from "react";
+import { useState } from "react";
 import { useLocale } from "react-aria";
 import {
   DropdownMenu,
@@ -26,26 +26,37 @@ 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 { capitalize } from "@/utils/strings";
 import { Button } from "../ui/button";
 import { DraggablePopover } from "../ui/draggable-popover";
 import { Input } from "../ui/input";
+import { RegexInput } from "./regex-input";
 import { NumberField } from "../ui/number-field";
 import { PopoverClose } from "../ui/popover";
 import {
   Select,
   SelectContent,
   SelectItem,
-  SelectSeparator,
   SelectTrigger,
   SelectValue,
 } from "../ui/select";
 import { FilterByValuesList } from "./filter-by-values-picker";
+import { OPERATOR_LABELS } from "./operator-labels";
 import {
   type ColumnFilterForType,
   type ColumnFilterValue,
+  DATETIME_OPS,
   Filter,
+  isDatetimeComparisonOp,
+  isNumberComparisonOp,
+  isTextScalarOp,
+  NUMBER_OPS,
+  TEXT_OPS,
 } from "./filters";
+import {
+  type DateLikeFilterType,
+  DateLikeInput,
+  DateLikeRangeInput,
+} from "./date-filter-inputs";
 import {
   ClearFilterMenuItem,
   FilterButtons,
@@ -148,7 +159,7 @@ export const DataTableColumnHeader = <TData, TValue>({
               {renderColumnWrapping(column)}
               {renderFormatOptions(column, locale)}
               <DropdownMenuSeparator />
-              {renderMenuItemFilter(column)}
+              {renderMenuItemFilter(column, calculateTopKRows)}
               {renderFilterByValues(column, setIsFilterValueOpen)}
               {hasFilter && <ClearFilterMenuItem column={column} />}
             </DropdownMenuContent>
@@ -211,6 +222,7 @@ const SortButton = <TData, TValue>({
 
 export function renderMenuItemFilter<TData, TValue>(
   column: Column<TData, TValue>,
+  calculateTopKRows?: CalculateTopKRows,
 ) {
   const canFilter = column.getCanFilter();
   if (!canFilter) {
@@ -248,7 +260,10 @@ export function renderMenuItemFilter<TData, TValue>(
         {filterMenuItem}
         <DropdownMenuPortal>
           <DropdownMenuSubContent>
-            <TextFilter column={column} />
+            <TextFilterMenu
+              column={column}
+              calculateTopKRows={calculateTopKRows}
+            />
           </DropdownMenuSubContent>
         </DropdownMenuPortal>
       </DropdownMenuSub>
@@ -261,7 +276,7 @@ export function renderMenuItemFilter<TData, TValue>(
         {filterMenuItem}
         <DropdownMenuPortal>
           <DropdownMenuSubContent>
-            <NumberRangeFilter column={column} />
+            <NumberFilterMenu column={column} />
           </DropdownMenuSubContent>
         </DropdownMenuPortal>
       </DropdownMenuSub>
@@ -273,77 +288,49 @@ export function renderMenuItemFilter<TData, TValue>(
     return null;
   }
 
-  if (filterType === "time") {
-    // Not implemented
-    return null;
-  }
-
-  if (filterType === "datetime") {
-    // Not implemented
-    return null;
-  }
-
-  if (filterType === "date") {
-    // Not implemented
-    return null;
+  if (
+    filterType === "date" ||
+    filterType === "datetime" ||
+    filterType === "time"
+  ) {
+    return (
+      <DropdownMenuSub>
+        {filterMenuItem}
+        <DropdownMenuPortal>
+          <DropdownMenuSubContent>
+            <DateFilterMenu column={column} filterType={filterType} />
+          </DropdownMenuSubContent>
+        </DropdownMenuPortal>
+      </DropdownMenuSub>
+    );
   }
 
   logNever(filterType);
   return null;
 }
 
-// Type-safe constants for null filter operators
-const NULL_FILTER_OPERATORS = {
-  is_null: "is_null",
-  is_not_null: "is_not_null",
-} satisfies Record<string, OperatorType>;
-
-const NullFilter = <TData, TValue>({
-  column,
-  defaultItem,
+const OperatorSelect = ({
   operator,
-  setOperator,
+  options,
+  onChange,
 }: {
-  column: Column<TData, TValue>;
-  defaultItem?: OperatorType | "between";
-  operator: OperatorType | "between";
-  setOperator: (operator: OperatorType) => void;
-}) => {
-  const handleValueChange = (value: OperatorType) => {
-    setOperator(value);
-    if (value === "is_null" || value === "is_not_null") {
-      column.setFilterValue(Filter.text({ operator: value }));
-    }
-  };
-
-  const isNullOrNotNull = operator === "is_null" || operator === "is_not_null";
-
-  return (
-    <Select
-      value={operator}
-      onValueChange={(value) => handleValueChange(value as OperatorType)}
-    >
-      <SelectTrigger
-        className={cn(
-          "border-border shadow-none! ring-0! w-full mb-0.5",
-          isNullOrNotNull && "mb-2",
-        )}
-      >
-        <SelectValue defaultValue={operator} />
-      </SelectTrigger>
-      <SelectContent>
-        {defaultItem && (
-          <SelectItem value={defaultItem}>{capitalize(defaultItem)}</SelectItem>
-        )}
-        <SelectSeparator />
-        <SelectItem value={NULL_FILTER_OPERATORS.is_null}>Is null</SelectItem>
-        <SelectItem value={NULL_FILTER_OPERATORS.is_not_null}>
-          Is not null
+  operator: OperatorType;
+  options: readonly OperatorType[];
+  onChange: (next: OperatorType) => void;
+}) => (
+  <Select value={operator} onValueChange={(v) => onChange(v as OperatorType)}>
+    <SelectTrigger className="border-border shadow-none! ring-0! w-full mb-0.5">
+      <SelectValue />
+    </SelectTrigger>
+    <SelectContent>
+      {options.map((op) => (
+        <SelectItem key={op} value={op}>
+          {OPERATOR_LABELS[op]}
         </SelectItem>
-      </SelectContent>
-    </Select>
-  );
-};
+      ))}
+    </SelectContent>
+  </Select>
+);
 
 const BooleanFilter = <TData, TValue>({
   column,
@@ -389,7 +376,15 @@ const BooleanFilter = <TData, TValue>({
   );
 };
 
-const NumberRangeFilter = <TData, TValue>({
+type NumberComparisonFilter = Extract<
+  ColumnFilterForType<"number">,
+  { value: number }
+>;
+const isNumberComparisonFilter = (
+  filter: ColumnFilterForType<"number">,
+): filter is NumberComparisonFilter => isNumberComparisonOp(filter.operator);
+
+export const NumberFilterMenu = <TData, TValue>({
   column,
 }: {
   column: Column<TData, TValue>;
@@ -399,149 +394,333 @@ const NumberRangeFilter = <TData, TValue>({
     | undefined;
   const hasFilter = currentFilter !== undefined;
 
-  const [operator, setOperator] = useState<OperatorType | "between">(
+  const [operator, setOperator] = useState<OperatorType>(
     currentFilter?.operator ?? "between",
   );
-  const [min, setMin] = useState<number | undefined>(currentFilter?.min);
-  const [max, setMax] = useState<number | undefined>(currentFilter?.max);
-  const minRef = useRef<HTMLInputElement>(null);
-  const maxRef = useRef<HTMLInputElement>(null);
+  const [min, setMin] = useState<number | undefined>(
+    currentFilter?.operator === "between" ? currentFilter.min : undefined,
+  );
+  const [max, setMax] = useState<number | undefined>(
+    currentFilter?.operator === "between" ? currentFilter.max : undefined,
+  );
+  const [value, setValue] = useState<number | undefined>(
+    currentFilter !== undefined && isNumberComparisonFilter(currentFilter)
+      ? currentFilter.value
+      : undefined,
+  );
 
-  const handleApply = (opts: { min?: number; max?: number } = {}) => {
-    column.setFilterValue(
-      Filter.number({
-        min: opts.min ?? min,
-        max: opts.max ?? max,
-        operator: operator === "between" ? undefined : operator,
-      }),
-    );
+  const isComparison = isNumberComparisonOp(operator);
+  const isNullish = operator === "is_null" || operator === "is_not_null";
+
+  const applyDisabled =
+    (operator === "between" && (min === undefined || max === undefined)) ||
+    (isComparison && value === undefined);
+
+  const handleApply = () => {
+    if (isNullish) {
+      column.setFilterValue(Filter.number({ operator }));
+      return;
+    }
+    if (operator === "between" && min !== undefined && max !== undefined) {
+      column.setFilterValue(Filter.number({ operator: "between", min, max }));
+      return;
+    }
+    if (isComparison && value !== undefined) {
+      column.setFilterValue(Filter.number({ operator, value }));
+    }
+  };
+
+  const handleClear = () => {
+    setMin(undefined);
+    setMax(undefined);
+    setValue(undefined);
+    column.setFilterValue(undefined);
+  };
+
+  const handleOperatorChange = (next: OperatorType) => {
+    setOperator(next);
   };
 
   return (
     <div className="flex flex-col gap-1 pt-3 px-2">
-      <NullFilter
-        column={column}
-        defaultItem="between"
+      <OperatorSelect
         operator={operator}
-        setOperator={setOperator}
+        options={NUMBER_OPS}
+        onChange={handleOperatorChange}
       />
       {operator === "between" && (
-        <>
-          <div className="flex gap-1 items-center">
-            <NumberField
-              ref={minRef}
-              value={min}
-              onChange={(value) => setMin(value)}
-              aria-label="min"
-              placeholder="min"
-              onKeyDown={(e) => {
-                if (e.key === "Enter") {
-                  handleApply({
-                    min: Number.parseFloat(e.currentTarget.value),
-                  });
-                }
-                if (e.key === "Tab") {
-                  maxRef.current?.focus();
-                }
-              }}
-              className="shadow-none! border-border hover:shadow-none!"
-            />
-            <MinusIcon className="h-5 w-5 text-muted-foreground" />
-            <NumberField
-              ref={maxRef}
-              value={max}
-              onChange={(value) => setMax(value)}
-              aria-label="max"
-              onKeyDown={(e) => {
-                if (e.key === "Enter") {
-                  handleApply({
-                    max: Number.parseFloat(e.currentTarget.value),
-                  });
-                }
-                if (e.key === "Tab") {
-                  minRef.current?.focus();
-                }
-              }}
-              placeholder="max"
-              className="shadow-none! border-border hover:shadow-none!"
-            />
-          </div>
-          <FilterButtons
-            onApply={handleApply}
-            onClear={() => {
-              setMin(undefined);
-              setMax(undefined);
-              column.setFilterValue(undefined);
-            }}
-            clearButtonDisabled={!hasFilter}
+        <div className="flex gap-1 items-center">
+          <NumberField
+            value={min}
+            onChange={setMin}
+            aria-label="min"
+            placeholder="min"
+            className="shadow-none! border-border hover:shadow-none!"
           />
-        </>
+          <MinusIcon className="h-5 w-5 text-muted-foreground" />
+          <NumberField
+            value={max}
+            onChange={setMax}
+            aria-label="max"
+            placeholder="max"
+            className="shadow-none! border-border hover:shadow-none!"
+          />
+        </div>
+      )}
+      {isComparison && (
+        <NumberField
+          value={value}
+          onChange={setValue}
+          aria-label="value"
+          placeholder="value"
+          className="shadow-none! border-border hover:shadow-none!"
+        />
       )}
+      <FilterButtons
+        onApply={handleApply}
+        onClear={handleClear}
+        clearButtonDisabled={!hasFilter}
+        applyButtonDisabled={applyDisabled}
+      />
     </div>
   );
 };
 
-const TextFilter = <TData, TValue>({
+type DateComparisonFilter = Extract<
+  ColumnFilterForType<DateLikeFilterType>,
+  { value: Date }
+>;
+const isDateComparisonFilter = (
+  filter: ColumnFilterForType<DateLikeFilterType>,
+): filter is DateComparisonFilter => isDatetimeComparisonOp(filter.operator);
+
+export const DateFilterMenu = <TData, TValue>({
   column,
+  filterType,
 }: {
   column: Column<TData, TValue>;
+  filterType: DateLikeFilterType;
+}) => {
+  const currentFilter = column.getFilterValue() as
+    | ColumnFilterForType<DateLikeFilterType>
+    | undefined;
+  const hasFilter = currentFilter !== undefined;
+
+  const [operator, setOperator] = useState<OperatorType>(
+    currentFilter?.operator ?? "between",
+  );
+  const [min, setMin] = useState<Date | undefined>(
+    currentFilter?.operator === "between" ? currentFilter.min : undefined,
+  );
+  const [max, setMax] = useState<Date | undefined>(
+    currentFilter?.operator === "between" ? currentFilter.max : undefined,
+  );
+  const [value, setValue] = useState<Date | undefined>(
+    currentFilter !== undefined && isDateComparisonFilter(currentFilter)
+      ? currentFilter.value
+      : undefined,
+  );
+
+  const isComparison = isDatetimeComparisonOp(operator);
+  const isNullish = operator === "is_null" || operator === "is_not_null";
+
+  const applyDisabled =
+    (operator === "between" && (min === undefined || max === undefined)) ||
+    (isComparison && value === undefined);
+
+  const buildFilter = (
+    opts: Parameters<typeof Filter.date>[0],
+  ): ColumnFilterForType<DateLikeFilterType> => {
+    switch (filterType) {
+      case "date":
+        return Filter.date(opts);
+      case "datetime":
+        return Filter.datetime(opts);
+      case "time":
+        return Filter.time(opts);
+    }
+  };
+
+  const handleApply = () => {
+    if (isNullish) {
+      column.setFilterValue(buildFilter({ operator }));
+      return;
+    }
+    if (operator === "between" && min !== undefined && max !== undefined) {
+      column.setFilterValue(buildFilter({ operator: "between", min, max }));
+      return;
+    }
+    if (isComparison && value !== undefined) {
+      column.setFilterValue(buildFilter({ operator, value }));
+    }
+  };
+
+  const [resetKey, setResetKey] = useState(0);
+  const handleClear = () => {
+    setMin(undefined);
+    setMax(undefined);
+    setValue(undefined);
+    setResetKey((k) => k + 1);
+    column.setFilterValue(undefined);
+  };
+
+  const handleOperatorChange = (next: OperatorType) => {
+    setOperator(next);
+  };
+
+  return (
+    <div
+      className="flex flex-col gap-1 pt-3 px-2"
+      onKeyDownCapture={(e) => {
+        if (e.key === "Tab") {
+          e.stopPropagation();
+        }
+      }}
+    >
+      <OperatorSelect
+        operator={operator}
+        options={DATETIME_OPS}
+        onChange={handleOperatorChange}
+      />
+      {operator === "between" && (
+        <DateLikeRangeInput
+          key={`${filterType}-${resetKey}`}
+          filterType={filterType}
+          min={min}
+          max={max}
+          onRangeChange={(nextMin, nextMax) => {
+            setMin(nextMin);
+            setMax(nextMax);
+          }}
+          className="shadow-none! border-border hover:shadow-none!"
+        />
+      )}
+      {isComparison && (
+        <DateLikeInput
+          key={`${filterType}-${resetKey}`}
+          filterType={filterType}
+          value={value}
+          onChange={setValue}
+          aria-label="value"
+          className="shadow-none! border-border hover:shadow-none!"
+        />
+      )}
+      <FilterButtons
+        onApply={handleApply}
+        onClear={handleClear}
+        clearButtonDisabled={!hasFilter}
+        applyButtonDisabled={applyDisabled}
+      />
+    </div>
+  );
+};
+
+export const TextFilterMenu = <TData, TValue>({
+  column,
+  calculateTopKRows,
+}: {
+  column: Column<TData, TValue>;
+  calculateTopKRows?: CalculateTopKRows;
 }) => {
   const currentFilter = column.getFilterValue() as
     | ColumnFilterForType<"text">
     | undefined;
   const hasFilter = currentFilter !== undefined;
-  const [value, setValue] = useState<string>(currentFilter?.text ?? "");
+
   const [operator, setOperator] = useState<OperatorType>(
     currentFilter?.operator ?? "contains",
   );
+  const [text, setText] = useState<string>(
+    currentFilter && "text" in currentFilter ? currentFilter.text : "",
+  );
+  const [values, setValues] = useState<string[]>(
+    currentFilter && "values" in currentFilter ? [...currentFilter.values] : [],
+  );
+
+  const isScalar = isTextScalarOp(operator);
+  const isMulti = operator === "in" || operator === "not_in";
+  const isNullish =
+    operator === "is_null" ||
+    operator === "is_not_null" ||
+    operator === "is_empty";
+
+  const applyDisabled =
+    (isScalar && text === "") || (isMulti && values.length === 0);
 
   const handleApply = () => {
-    if (operator !== "contains") {
+    if (isNullish) {
       column.setFilterValue(Filter.text({ operator }));
       return;
     }
-
-    if (value === "") {
-      column.setFilterValue(undefined);
+    if (isScalar && text !== "") {
+      column.setFilterValue(Filter.text({ operator, text }));
       return;
     }
+    if (isMulti && values.length > 0) {
+      column.setFilterValue(Filter.text({ operator, values }));
+    }
+  };
+
+  const handleClear = () => {
+    setText("");
+    setValues([]);
+    column.setFilterValue(undefined);
+  };
 
-    column.setFilterValue(Filter.text({ text: value, operator }));
+  const handleOperatorChange = (next: OperatorType) => {
+    setOperator(next);
   };
 
   return (
     <div className="flex flex-col gap-1 pt-3 px-2">
-      <NullFilter
-        column={column}
-        defaultItem="contains"
+      <OperatorSelect
         operator={operator}
-        setOperator={setOperator}
+        options={TEXT_OPS}
+        onChange={handleOperatorChange}
       />
-      {operator === "contains" && (
-        <>
-          <Input
-            type="text"
-            icon={<TextIcon className="h-3 w-3 text-muted-foreground mb-1" />}
-            value={value ?? ""}
-            onChange={(e) => setValue(e.target.value)}
-            placeholder="Text..."
-            onKeyDown={(e) => {
-              if (e.key === "Enter") {
-                handleApply();
-              }
-            }}
-            className="shadow-none! border-border hover:shadow-none!"
-          />
-          <FilterButtons
-            onApply={handleApply}
-            onClear={() => {
-              setValue("");
-              column.setFilterValue(undefined);
-            }}
-            clearButtonDisabled={!hasFilter}
-          />
-        </>
+      {isScalar && operator === "regex" && (
+        <RegexInput
+          value={text}
+          onChange={setText}
+          onKeyDown={(e) => {
+            e.stopPropagation();
+            if (e.key === "Enter") {
+              handleApply();
+            }
+          }}
+        />
       )}
+      {isScalar && operator !== "regex" && (
+        <Input
+          type="text"
+          icon={<TextIcon className="h-3 w-3 text-muted-foreground mb-1" />}
+          value={text}
+          onChange={(e) => setText(e.target.value)}
+          placeholder="Text..."
+          onKeyDown={(e) => {
+            e.stopPropagation();
+            if (e.key === "Enter") {
+              handleApply();
+            }
+          }}
+          className="shadow-none! border-border hover:shadow-none!"
+        />
+      )}
+      {isMulti && (
+        <FilterByValuesList
+          column={column}
+          calculateTopKRows={calculateTopKRows}
+          chosenValues={new Set(values)}
+          onChange={(next) => setValues(next.map(String))}
+          creatable={true}
+        />
+      )}
+      <FilterButtons
+        onApply={handleApply}
+        onClear={handleClear}
+        clearButtonDisabled={!hasFilter}
+        applyButtonDisabled={applyDisabled}
+      />
     </div>
   );
 };

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

@@ -70,6 +70,9 @@ interface DataTableProps<TData> extends Partial<ExportActionProps> {
   setSorting?: OnChangeFn<SortingState>; // controlled sorting
   // Pagination
   totalRows: number | TooManyRows;
+  // JSON-serialized size of the currently-rendered data. Forwarded to
+  // ExportMenu so hosts can size-gate the Export button via downloadSizeLimitAtom.
+  sizeBytes?: number | null;
   totalColumns: number;
   pagination?: boolean;
   manualPagination?: boolean; // server-side pagination
@@ -121,6 +124,7 @@ const DataTableInternal = <TData,>({
   selection,
   totalColumns,
   totalRows,
+  sizeBytes,
   manualSorting = false,
   sorting,
   setSorting,
@@ -309,6 +313,7 @@ const DataTableInternal = <TData,>({
             togglePanel={togglePanel}
             isAnyPanelOpen={isAnyPanelOpen}
             downloadAs={downloadAs}
+            sizeBytes={sizeBytes}
           />
           <Table
             className={cn(