@marimo-team/frontend 0.25.2-dev56 → 0.25.2-dev62

package/src/components/data-table/__tests__/schemas.test.ts

@@ -0,0 +1,57 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { describe, expect, it } from "vitest";
+import {
+  DownloadAsSchema,
+  DownloadGeoJSONSchema,
+  GetExportMetadataSchema,
+} from "../schemas";
+
+describe("geometry export contracts", () => {
+  it("keeps the existing download_as formats unchanged", () => {
+    for (const format of ["csv", "tsv", "json", "parquet"]) {
+      expect(DownloadAsSchema.input.parse({ format })).toEqual({ format });
+    }
+    expect(
+      DownloadAsSchema.input.safeParse({ format: "geojson" }).success,
+    ).toBe(false);
+  });
+  it.each([undefined, "geom_b", ""])(
+    "accepts a GeoJSON request with geometry %s",
+    (geometry_column) => {
+      const request = { format: "geojson", geometry_column };
+      expect(DownloadGeoJSONSchema.input.parse(request)).toEqual(request);
+    },
+  );
+
+  it("accepts a missing CRS response for an unnamed geometry", () => {
+    const response = {
+      url: "",
+      filename: "",
+      code: "missing_crs",
+      column: "",
+      error: "Declare a CRS.",
+    };
+    expect(DownloadGeoJSONSchema.output.parse(response)).toEqual(response);
+  });
+
+  it("accepts GeoJSON support when the default geometry has no CRS", () => {
+    const metadata = {
+      geometry_columns: [
+        { name: "", encoding: "objects", crs: null },
+        { name: "alternate", encoding: "objects", crs: "EPSG:4326" },
+      ],
+      primary_geometry_column: "",
+      default_geometry_column: "",
+      formats: {
+        geojson: { available: true, reason: null, missing_packages: [] },
+        parquet: {
+          available: false,
+          reason: "GeoParquet export requires pyarrow.",
+          missing_packages: ["pyarrow"],
+        },
+      },
+    };
+    expect(GetExportMetadataSchema.output.parse(metadata)).toEqual(metadata);
+  });
+});

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

@@ -155,6 +155,7 @@ const DataTableInternal = <TData,>({
   paginationState,
   setPaginationState,
   downloadAs,
+  getExportMetadata,
   manualPagination = false,
   pagination = false,
   onRowSelectionChange,
@@ -376,6 +377,8 @@ const DataTableInternal = <TData,>({
                 togglePanel={togglePanel}
                 isAnyPanelOpen={isAnyPanelOpen}
                 downloadAs={downloadAs}
+                getExportMetadata={getExportMetadata}
+                metadataSource={data}
                 sizeBytes={sizeBytes}
                 sizeBytesIsLoading={sizeBytesIsLoading}
               />