@marimo-team/frontend 0.20.5-dev0 → 0.20.5-dev3

package/src/components/editor/connections/storage/schemas.ts

@@ -21,7 +21,6 @@ export const S3StorageSchema = z
         FieldOptions.of({
           label: "Region",
           placeholder: "us-east-1",
-          optionRegex: ".*region.*",
         }),
       ),
     access_key_id: z
@@ -115,6 +114,50 @@ export const AzureStorageSchema = z
   })
   .describe(FieldOptions.of({ direction: "two-columns" }));
 
+export const CoreWeaveStorageSchema = z
+  .object({
+    type: z.literal("coreweave"),
+    bucket: z
+      .string()
+      .nonempty()
+      .describe(
+        FieldOptions.of({
+          label: "Bucket",
+          placeholder: "bucket-name",
+        }),
+      ),
+    region: z
+      .string()
+      .nonempty()
+      .describe(
+        FieldOptions.of({
+          label: "Region",
+          placeholder: "US-EAST-04A",
+        }),
+      ),
+    access_key_id: z
+      .string()
+      .optional()
+      .describe(
+        FieldOptions.of({
+          label: "Access Key ID",
+          inputType: "password",
+          optionRegex: ".*object_storage_key.*",
+        }),
+      ),
+    secret_access_key: z
+      .string()
+      .optional()
+      .describe(
+        FieldOptions.of({
+          label: "Secret Access Key",
+          inputType: "password",
+          optionRegex: ".*object_storage_secret.*",
+        }),
+      ),
+  })
+  .describe(FieldOptions.of({ direction: "two-columns" }));
+
 export const GoogleDriveStorageSchema = z
   .object({
     type: z.literal("gdrive"),
@@ -135,6 +178,7 @@ export const StorageConnectionSchema = z.discriminatedUnion("type", [
   S3StorageSchema,
   GCSStorageSchema,
   AzureStorageSchema,
+  CoreWeaveStorageSchema,
   GoogleDriveStorageSchema,
 ]);
 

package/src/components/storage/components.tsx

@@ -75,15 +75,19 @@ const PROTOCOL_ICONS: Record<KnownStorageProtocol, IconEntry> = {
 
 export const ProtocolIcon: React.FC<{
   protocol: KnownStorageProtocol | (string & {});
+  forceDark?: boolean;
   className?: string;
-}> = ({ protocol, className }) => {
+}> = ({ protocol, forceDark, className }) => {
   const { theme } = useTheme();
   const entry =
     PROTOCOL_ICONS[protocol.toLowerCase() as KnownStorageProtocol] ??
     HardDriveIcon;
 
   if ("src" in entry) {
-    const src = theme === "dark" && entry.dark ? entry.dark : entry.src;
+    let src = entry.src;
+    if (entry.dark && (forceDark || theme === "dark")) {
+      src = entry.dark;
+    }
     return (
       <img src={src} alt={protocol} className={cn("h-3.5 w-3.5", className)} />
     );