@rebasepro/studio 0.1.2 → 0.2.3

package/src/components/RLSEditor/PolicyEditor.tsx

@@ -1,7 +1,24 @@
 
 import React, { useState, useEffect } from "react";
-import { Button, Paper, Typography, TextField, Select, SelectItem, MultiSelect, MultiSelectItem, cls, defaultBorderMixin, IconButton, Dialog, DialogTitle, DialogContent, DialogActions , iconSize } from "@rebasepro/ui";
-import { HelpCircleIcon } from "lucide-react";
+import {
+    Button,
+    cls,
+    defaultBorderMixin,
+    Dialog,
+    DialogActions,
+    DialogContent,
+    DialogTitle,
+    HelpCircleIcon,
+    IconButton,
+    iconSize,
+    MultiSelect,
+    MultiSelectItem,
+    Paper,
+    Select,
+    SelectItem,
+    TextField,
+    Typography
+} from "@rebasepro/ui";
 import { useTranslation } from "@rebasepro/core";
 import { MonacoEditor } from "../SQLEditor/MonacoEditor";
 import { PostgresPolicy } from "./RLSEditor";

package/src/components/RLSEditor/RLSEditor.tsx

@@ -1,8 +1,28 @@
 
 import { useStudioCollectionRegistry } from "@rebasepro/core";
 import React, { useState, useEffect, useCallback, useMemo } from "react";
-import { Paper, Typography, CircularProgress, cls, Alert, defaultBorderMixin, Card, Chip, Button, Tooltip, ResizablePanels, IconButton, Tabs, Tab , iconSize } from "@rebasepro/ui";
-import { ShieldIcon, RefreshCwIcon, AlertTriangleIcon, KeyIcon, Trash2Icon } from "lucide-react";
+import {
+    Alert,
+    AlertTriangleIcon,
+    Button,
+    Card,
+    Chip,
+    CircularProgress,
+    cls,
+    defaultBorderMixin,
+    IconButton,
+    iconSize,
+    KeyIcon,
+    Paper,
+    RefreshCwIcon,
+    ResizablePanels,
+    ShieldIcon,
+    Tab,
+    Tabs,
+    Tooltip,
+    Trash2Icon,
+    Typography
+} from "@rebasepro/ui";
 import { useRebaseContext, useSnackbarController, ErrorView, useTranslation } from "@rebasepro/core";
 import { isPostgresCollection } from "@rebasepro/types";
 import { PolicyEditor } from "./PolicyEditor";

package/src/components/SQLEditor/SQLEditor.tsx

@@ -3,16 +3,45 @@ import { IconForView } from "@rebasepro/core";
 import { useStudioCollectionRegistry, useStudioSideEntityController } from "@rebasepro/core";
 import React, { useState, useEffect, useCallback, useRef, useMemo } from "react";
 import { createPortal } from "react-dom";
-import { Button, Paper, Typography, CircularProgress, cls, IconButton, InputLabel, Dialog, DialogContent, DialogActions, DialogTitle, TextField, Tooltip, Alert, Tabs, Tab, defaultBorderMixin, Select, SelectItem, Menu, MenuItem, ResizablePanels, Chip, VirtualTable, VirtualTableColumn , iconSize, Checkbox, TextareaAutosize } from "@rebasepro/ui";
-import { DatabaseIcon, TerminalIcon, XIcon, PlusIcon, PencilIcon, MoreVerticalIcon, MenuIcon, PlayIcon } from "lucide-react";
-// VirtualTableInput is conditionally loaded from CMS when available
-let VirtualTableInput: React.ComponentType<any> | null = null;
-try {
-    // @ts-ignore — optional peer dependency
-    // eslint-disable-next-line @typescript-eslint/no-require-imports
-    const cms = require("@rebasepro/admin");
-    VirtualTableInput = cms.VirtualTableInput;
-} catch { /* CMS not available */ }
+import {
+    Alert,
+    Button,
+    Checkbox,
+    Chip,
+    CircularProgress,
+    cls,
+    DatabaseIcon,
+    defaultBorderMixin,
+    Dialog,
+    DialogActions,
+    DialogContent,
+    DialogTitle,
+    IconButton,
+    iconSize,
+    InputLabel,
+    Menu,
+    MenuIcon,
+    MenuItem,
+    MoreVerticalIcon,
+    Paper,
+    PencilIcon,
+    PlayIcon,
+    PlusIcon,
+    ResizablePanels,
+    Select,
+    SelectItem,
+    Tab,
+    Tabs,
+    TerminalIcon,
+    TextareaAutosize,
+    TextField,
+    Tooltip,
+    Typography,
+    VirtualTable,
+    VirtualTableColumn,
+    XIcon
+} from "@rebasepro/ui";
+
 import { useRebaseContext, useSnackbarController, ConfirmationDialog, ErrorView, useTranslation } from "@rebasepro/core";
 import { MonacoEditor } from "./MonacoEditor";
 import { SQLEditorSidebar, Snippet } from "./SQLEditorSidebar";
@@ -152,14 +181,21 @@ const FixedEditorOverlay = ({
     );
 };
 
+const getStoragePrefix = (baseUrl?: string) => {
+    if (!baseUrl) return "default";
+    return baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_");
+};
+
 export const SQLEditor = () => {
-    const { databaseAdmin } = useRebaseContext();
+    const { databaseAdmin, client } = useRebaseContext();
     const sideEntityController = useStudioSideEntityController();
     const snackbarController = useSnackbarController();
     const collectionRegistry = useStudioCollectionRegistry();
 
     const { t } = useTranslation();
 
+    const projectPrefix = useMemo(() => getStoragePrefix(client?.baseUrl), [client?.baseUrl]);
+
     // Schema state
     const [schemas, setSchemas] = useState<Record<string, TableInfo[]>>({});
     const [isSchemaLoading, setIsSchemaLoading] = useState(true);
@@ -168,10 +204,12 @@ export const SQLEditor = () => {
 
     // Connection state
     const [selectedDatabase, setSelectedDatabase] = useState<string | undefined>(() => {
-        return localStorage.getItem("rebase_sql_selected_db") || undefined;
+        const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+        return localStorage.getItem(`rebase_sql_selected_db_${projectPrefixSync}`) || undefined;
     });
     const [selectedRole, setSelectedRole] = useState<string | undefined>(() => {
-        return localStorage.getItem("rebase_sql_selected_role") || undefined;
+        const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+        return localStorage.getItem(`rebase_sql_selected_role_${projectPrefixSync}`) || undefined;
     });
 
     const [availableDatabases, setAvailableDatabases] = useState<string[]>([]);
@@ -192,7 +230,8 @@ export const SQLEditor = () => {
         execTime: number | null,
         lastExecutedSql: string | null
     }>>(() => {
-        const saved = localStorage.getItem(STORAGE_KEY_TABS);
+        const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+        const saved = localStorage.getItem(`rebase_sql_tabs_${projectPrefixSync}`);
         if (saved) {
             const parsed = JSON.parse(saved);
             return parsed.map((t: Record<string, unknown>) => ({
@@ -208,8 +247,8 @@ export const SQLEditor = () => {
             id: "1",
             name: "Query 1",
             sql: "SELECT * FROM ",
-            database: localStorage.getItem("rebase_sql_selected_db") || undefined,
-            role: localStorage.getItem("rebase_sql_selected_role") || undefined,
+            database: localStorage.getItem(`rebase_sql_selected_db_${projectPrefixSync}`) || undefined,
+            role: localStorage.getItem(`rebase_sql_selected_role_${projectPrefixSync}`) || undefined,
             results: null,
             loading: false,
             error: null,
@@ -218,7 +257,8 @@ export const SQLEditor = () => {
         }];
     });
     const [activeTabId, setActiveTabId] = useState<string>(() => {
-        return localStorage.getItem(STORAGE_KEY_ACTIVE_TAB) || "1";
+        const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+        return localStorage.getItem(`rebase_sql_active_tab_${projectPrefixSync}`) || "1";
     });
 
     const activeTab = tabs.find(t => t.id === activeTabId) || tabs[0];
@@ -243,30 +283,31 @@ export const SQLEditor = () => {
     useEffect(() => {
         let mounted = true;
         const fetchConnectionConfig = async () => {
-            if (!databaseAdmin?.fetchAvailableDatabases || !databaseAdmin?.fetchAvailableRoles) {
+            if (!databaseAdmin?.fetchAvailableDatabases || !databaseAdmin?.fetchAvailableRoles || !databaseAdmin?.executeSql) {
                 setConnectionConfigError(t("studio_sql_sql_not_supported"));
                 setIsLoadingConfig(false);
                 return;
             }
 
             try {
-                const [dbs, roles, currentDbFromApi] = await Promise.all([
+                const [dbs, roles, currentDbFromApi, currentUserResult] = await Promise.all([
                     databaseAdmin.fetchAvailableDatabases(),
                     databaseAdmin.fetchAvailableRoles(),
-                    typeof databaseAdmin?.fetchCurrentDatabase === "function" ? databaseAdmin.fetchCurrentDatabase() : Promise.resolve(undefined)
+                    typeof databaseAdmin?.fetchCurrentDatabase === "function" ? databaseAdmin.fetchCurrentDatabase() : Promise.resolve(undefined),
+                    databaseAdmin.executeSql("SELECT current_user AS role").catch(() => [])
                 ]);
 
                 if (mounted) {
                     setAvailableDatabases(dbs);
                     setAvailableRoles(roles);
 
-                    const loadedDb = localStorage.getItem("rebase_sql_selected_db") || undefined;
-                    const loadedRole = localStorage.getItem("rebase_sql_selected_role") || undefined;
+                    const loadedDb = localStorage.getItem(`rebase_sql_selected_db_${projectPrefix}`) || undefined;
+                    const loadedRole = localStorage.getItem(`rebase_sql_selected_role_${projectPrefix}`) || undefined;
 
-                    const initialActiveTabId = localStorage.getItem(STORAGE_KEY_ACTIVE_TAB) || "1";
-                    let initialTabs: any[] = [];
+                    const initialActiveTabId = localStorage.getItem(`rebase_sql_active_tab_${projectPrefix}`) || "1";
+                    let initialTabs: Array<{ id?: string; database?: string; role?: string }> = [];
                     try {
-                        const savedTabs = localStorage.getItem(STORAGE_KEY_TABS);
+                        const savedTabs = localStorage.getItem(`rebase_sql_tabs_${projectPrefix}`);
                         if (savedTabs) initialTabs = JSON.parse(savedTabs);
                     } catch (e) { /* ignore */ }
                     const currentActiveTab = initialTabs.find(t => t.id === initialActiveTabId);
@@ -279,22 +320,26 @@ export const SQLEditor = () => {
 
                     if (actualDb) {
                         setSelectedDatabase(actualDb);
-                        localStorage.setItem("rebase_sql_selected_db", actualDb);
-                        setTabs(prev => prev.map(t => t.id === initialActiveTabId && !t.database ? { ...t,
-database: actualDb } : t));
+                        localStorage.setItem(`rebase_sql_selected_db_${projectPrefix}`, actualDb);
+                        setTabs(prev => prev.map(t => t.id === initialActiveTabId && (!t.database || !dbs.includes(t.database)) ? { ...t, database: actualDb } : t));
                     }
 
+                    const currentUser = (currentUserResult?.[0] as Record<string, unknown> | undefined)?.role as string | undefined;
                     let actualRole = currentActiveTab?.role || loadedRole;
+
                     if (actualRole && !roles.includes(actualRole)) actualRole = undefined;
                     if (!actualRole && roles.length > 0) {
-                        actualRole = roles.includes("postgres") ? "postgres" : roles[0];
+                        if (currentUser && roles.includes(currentUser)) {
+                            actualRole = currentUser;
+                        } else {
+                            actualRole = roles.includes("postgres") ? "postgres" : roles[0];
+                        }
                     }
 
                     if (actualRole) {
                         setSelectedRole(actualRole);
-                        localStorage.setItem("rebase_sql_selected_role", actualRole);
-                        setTabs(prev => prev.map(t => t.id === initialActiveTabId && !t.role ? { ...t,
-role: actualRole } : t));
+                        localStorage.setItem(`rebase_sql_selected_role_${projectPrefix}`, actualRole);
+                        setTabs(prev => prev.map(t => t.id === initialActiveTabId && (!t.role || !roles.includes(t.role)) ? { ...t, role: actualRole } : t));
                     }
                 }
             } catch (err: unknown) {
@@ -313,22 +358,20 @@ role: actualRole } : t));
         fetchConnectionConfig();
 
         return () => { mounted = false; };
-    }, [databaseAdmin]);
+    }, [databaseAdmin, projectPrefix]);
 
     const handleDatabaseChange = (db: string, tabId?: string) => {
         setSelectedDatabase(db);
-        localStorage.setItem("rebase_sql_selected_db", db);
-        setTabs(prev => prev.map(t => t.id === (tabId || activeTabId) ? { ...t,
-database: db } : t));
+        localStorage.setItem(`rebase_sql_selected_db_${projectPrefix}`, db);
+        setTabs(prev => prev.map(t => t.id === (tabId || activeTabId) ? { ...t, database: db } : t));
         // Reset so the schema will be re-fetched for the new database
         schemaFetchedRef.current = false;
     };
 
     const handleRoleChange = (role: string, tabId?: string) => {
         setSelectedRole(role);
-        localStorage.setItem("rebase_sql_selected_role", role);
-        setTabs(prev => prev.map(t => t.id === (tabId || activeTabId) ? { ...t,
-role } : t));
+        localStorage.setItem(`rebase_sql_selected_role_${projectPrefix}`, role);
+        setTabs(prev => prev.map(t => t.id === (tabId || activeTabId) ? { ...t, role } : t));
     };
 
     const handleTabChange = useCallback((newTabId: string) => {
@@ -337,22 +380,20 @@ role } : t));
         if (newTab) {
             if (newTab.database && newTab.database !== selectedDatabase) {
                 setSelectedDatabase(newTab.database);
-                localStorage.setItem("rebase_sql_selected_db", newTab.database);
+                localStorage.setItem(`rebase_sql_selected_db_${projectPrefix}`, newTab.database);
                 schemaFetchedRef.current = false;
             } else if (!newTab.database && selectedDatabase) {
-                setTabs(prev => prev.map(t => t.id === newTabId ? { ...t,
-database: selectedDatabase } : t));
+                setTabs(prev => prev.map(t => t.id === newTabId ? { ...t, database: selectedDatabase } : t));
             }
 
             if (newTab.role && newTab.role !== selectedRole) {
                 setSelectedRole(newTab.role);
-                localStorage.setItem("rebase_sql_selected_role", newTab.role);
+                localStorage.setItem(`rebase_sql_selected_role_${projectPrefix}`, newTab.role);
             } else if (!newTab.role && selectedRole) {
-                setTabs(prev => prev.map(t => t.id === newTabId ? { ...t,
-role: selectedRole } : t));
+                setTabs(prev => prev.map(t => t.id === newTabId ? { ...t, role: selectedRole } : t));
             }
         }
-    }, [tabs, selectedDatabase, selectedRole]);
+    }, [tabs, selectedDatabase, selectedRole, projectPrefix]);
 
     const fetchSchema = useCallback(async () => {
         if (!databaseAdmin?.executeSql) {
@@ -566,7 +607,8 @@ role: selectedRole });
     }, [editingCell, schemas, activeTab.lastExecutedSql, activeTab.results, databaseAdmin, updateActiveTab, snackbarController, selectedDatabase, selectedRole]);
 
     const [columnWidths, setColumnWidths] = useState<Record<string, Record<string, number>>>(() => {
-        const saved = localStorage.getItem("rebase_sql_column_widths");
+        const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+        const saved = localStorage.getItem(`rebase_sql_column_widths_${projectPrefixSync}`);
         return saved ? JSON.parse(saved) : {};
     });
     const [snippets, setSnippets] = useState<Snippet[]>([]);
@@ -576,12 +618,20 @@ role: selectedRole });
 
     // Load from local storage
     useEffect(() => {
-        const savedSnippets = localStorage.getItem("rebase_sql_snippets");
-        if (savedSnippets) setSnippets(JSON.parse(savedSnippets));
+        const savedSnippets = localStorage.getItem(`rebase_sql_snippets_${projectPrefix}`);
+        if (savedSnippets) {
+            setSnippets(JSON.parse(savedSnippets));
+        } else {
+            setSnippets([]);
+        }
 
-        const savedHistory = localStorage.getItem("rebase_sql_history");
-        if (savedHistory) setHistory(JSON.parse(savedHistory));
-    }, []);
+        const savedHistory = localStorage.getItem(`rebase_sql_history_${projectPrefix}`);
+        if (savedHistory) {
+            setHistory(JSON.parse(savedHistory));
+        } else {
+            setHistory([]);
+        }
+    }, [projectPrefix]);
 
     // Save tabs and active tab to local storage
     useEffect(() => {
@@ -592,21 +642,21 @@ role: selectedRole });
             database: t.database,
             role: t.role
         }));
-        localStorage.setItem(STORAGE_KEY_TABS, JSON.stringify(sanitizedTabs));
-    }, [tabs]);
+        localStorage.setItem(`rebase_sql_tabs_${projectPrefix}`, JSON.stringify(sanitizedTabs));
+    }, [tabs, projectPrefix]);
 
     useEffect(() => {
-        localStorage.setItem(STORAGE_KEY_ACTIVE_TAB, activeTabId);
-    }, [activeTabId]);
+        localStorage.setItem(`rebase_sql_active_tab_${projectPrefix}`, activeTabId);
+    }, [activeTabId, projectPrefix]);
 
     const saveSnippets = (newSnippets: Snippet[]) => {
         setSnippets(newSnippets);
-        localStorage.setItem("rebase_sql_snippets", JSON.stringify(newSnippets));
+        localStorage.setItem(`rebase_sql_snippets_${projectPrefix}`, JSON.stringify(newSnippets));
     };
 
     const saveHistory = (newHistory: string[]) => {
         setHistory(newHistory);
-        localStorage.setItem("rebase_sql_history", JSON.stringify(newHistory.slice(-50)));
+        localStorage.setItem(`rebase_sql_history_${projectPrefix}`, JSON.stringify(newHistory.slice(-50)));
     };
 
     const handleDeleteSnippet = (id: string) => {
@@ -667,10 +717,10 @@ role: selectedRole });
                     [key]: width
                 }
             };
-            localStorage.setItem("rebase_sql_column_widths", JSON.stringify(newWidths));
+            localStorage.setItem(`rebase_sql_column_widths_${projectPrefix}`, JSON.stringify(newWidths));
             return newWidths;
         });
-    }, [activeTab.sql]);
+    }, [activeTab.sql, projectPrefix]);
 
     const handlePrettify = () => {
         // Simple formatting for now
@@ -743,7 +793,7 @@ role: selectedRole });
         } finally {
             updateActiveTab({ loading: false });
         }
-    }, [activeTab.sql, autoLimit, databaseAdmin, history, updateActiveTab]);
+    }, [activeTab.sql, autoLimit, databaseAdmin, history, updateActiveTab, selectedDatabase, selectedRole]);
 
     const handleRun = useCallback(async (selectedText?: string) => {
         const sqlTarget = selectedText || activeTab.sql;
@@ -1142,7 +1192,8 @@ id: String(ra.entityId) })}
 
     const [sidebarSize, setSidebarSize] = useState(() => {
         try {
-            const saved = localStorage.getItem("rebase_sql_editor_sidebar_size");
+            const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+            const saved = localStorage.getItem(`rebase_sql_editor_sidebar_size_${projectPrefixSync}`);
             return saved !== null ? parseFloat(saved) : 20;
         } catch (e) {
             return 20;
@@ -1150,7 +1201,8 @@ id: String(ra.entityId) })}
     });
     const [editorHeight, setEditorHeight] = useState(() => {
         try {
-            const saved = localStorage.getItem("rebase_sql_editor_height");
+            const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+            const saved = localStorage.getItem(`rebase_sql_editor_height_${projectPrefixSync}`);
             return saved !== null ? parseFloat(saved) : 50;
         } catch (e) {
             return 50;
@@ -1159,15 +1211,15 @@ id: String(ra.entityId) })}
 
     useEffect(() => {
         try {
-            localStorage.setItem("rebase_sql_editor_sidebar_size", sidebarSize.toString());
+            localStorage.setItem(`rebase_sql_editor_sidebar_size_${projectPrefix}`, sidebarSize.toString());
         } catch (e) { /* ignore */ }
-    }, [sidebarSize]);
+    }, [sidebarSize, projectPrefix]);
 
     useEffect(() => {
         try {
-            localStorage.setItem("rebase_sql_editor_height", editorHeight.toString());
+            localStorage.setItem(`rebase_sql_editor_height_${projectPrefix}`, editorHeight.toString());
         } catch (e) { /* ignore */ }
-    }, [editorHeight]);
+    }, [editorHeight, projectPrefix]);
 
     const activeSnippet = snippets.find(s => s.sql === activeTab.sql);
     const isFavorite = activeSnippet?.isFavorite || false;
@@ -1296,7 +1348,11 @@ isFavorite: !s.isFavorite } : s));
                                             className="text-text-secondary dark:text-text-secondary-dark font-medium mr-2"
                                         >
                                             <DatabaseIcon size={iconSize.small} className="mr-1.5 text-text-disabled dark:text-text-disabled-dark"/>
-                                            <span className="max-w-[80px] truncate">{isLoadingConfig ? "..." : (selectedDatabase || t("studio_sql_select_db"))}</span>
+                                            <span className="max-w-[160px] truncate">
+                                                {isLoadingConfig 
+                                                    ? "..." 
+                                                    : `${selectedDatabase || t("studio_sql_select_db")}${selectedRole ? ` (${selectedRole})` : ""}`}
+                                            </span>
                                         </Button>
                                     }
                                 >

package/src/components/SQLEditor/SQLEditorSidebar.tsx

@@ -1,7 +1,6 @@
 
 import React, { useState } from "react";
-import { Typography, cls, defaultBorderMixin, Tabs, Tab, IconButton , iconSize } from "@rebasepro/ui";
-import { Trash2Icon } from "lucide-react";
+import { cls, defaultBorderMixin, IconButton, iconSize, Tab, Tabs, Trash2Icon, Typography } from "@rebasepro/ui";
 import { useTranslation } from "@rebasepro/core";
 import { SchemaBrowser } from "./SchemaBrowser";
 import { TableInfo } from "./SQLEditor";

package/src/components/SQLEditor/SchemaBrowser.tsx

@@ -1,7 +1,19 @@
 
 import React, { useState } from "react";
-import { Typography, CircularProgress, cls, IconButton, defaultBorderMixin, Menu, MenuItem, Button , iconSize } from "@rebasepro/ui";
-import { CopyIcon, MoreVerticalIcon, RefreshCwIcon } from "lucide-react";
+import {
+    Button,
+    CircularProgress,
+    cls,
+    CopyIcon,
+    defaultBorderMixin,
+    IconButton,
+    iconSize,
+    Menu,
+    MenuItem,
+    MoreVerticalIcon,
+    RefreshCwIcon,
+    Typography
+} from "@rebasepro/ui";
 import { TableInfo } from "./SQLEditor";
 import { ErrorView, useTranslation } from "@rebasepro/core";
 

package/src/components/StorageView/StorageView.tsx

@@ -1,7 +1,44 @@
 
 import React, { useState, useEffect, useCallback, useMemo, useRef } from "react";
-import { Typography, cls, defaultBorderMixin, Button, IconButton, Tooltip, CircularProgress, ResizablePanels, Chip, Dialog, DialogTitle, DialogContent, DialogActions, FileUpload, iconSize, Checkbox, LoadingButton, TextField } from "@rebasepro/ui";
-import { VideoIcon, Music2Icon, RefreshCwIcon, Trash2Icon, XIcon, PlusIcon, DownloadIcon, UploadCloudIcon, FolderIcon, FolderPlusIcon, FileTextIcon, ImageIcon, ArrowLeftIcon, FileIcon, HomeIcon, LayoutGridIcon, ListIcon, CopyIcon, CheckIcon } from "lucide-react";
+import {
+    ArrowLeftIcon,
+    Button,
+    Checkbox,
+    CheckIcon,
+    Chip,
+    CircularProgress,
+    cls,
+    CopyIcon,
+    defaultBorderMixin,
+    Dialog,
+    DialogActions,
+    DialogContent,
+    DialogTitle,
+    DownloadIcon,
+    FileIcon,
+    FileTextIcon,
+    FileUpload,
+    FolderIcon,
+    FolderPlusIcon,
+    HomeIcon,
+    IconButton,
+    iconSize,
+    ImageIcon,
+    LayoutGridIcon,
+    ListIcon,
+    LoadingButton,
+    Music2Icon,
+    PlusIcon,
+    RefreshCwIcon,
+    ResizablePanels,
+    TextField,
+    Tooltip,
+    Trash2Icon,
+    Typography,
+    UploadCloudIcon,
+    VideoIcon,
+    XIcon
+} from "@rebasepro/ui";
 import { useStorageSource, useSnackbarController, ErrorView, useApiConfig } from "@rebasepro/core";
 import type { StorageListResult } from "@rebasepro/types";
 import { useSearchParams } from "react-router-dom";

package/src/components/StudioHomePage.tsx

@@ -1,8 +1,7 @@
 import type { HomePageSection, PluginGenericProps } from "@rebasepro/types";
 import React, { useEffect } from "react";
 import { Card, cls, Container, Typography } from "@rebasepro/ui";
-import { ArrowRightIcon } from "lucide-react";
-import { iconSize } from "@rebasepro/ui";
+import { ArrowRightIcon, iconSize } from "@rebasepro/ui";
 import { IconForView, useRebaseContext, useRestoreScroll, useSlot } from "@rebasepro/core";
 import { useNavigate } from "react-router-dom";
 import { useStudioBreadcrumbs, BootstrapAdminBanner } from "@rebasepro/core";

package/src/utils/sql_utils.ts

@@ -105,7 +105,7 @@ export function resolveQueryCollections(
     for (const table of tables) {
         // Match table name against collection table or slug->snake_case
         const matched = collections.find(c => {
-            const tableName = (c as any).table || toSnakeCase(c.slug);
+            const tableName = (c as EntityCollection & { table?: string }).table || toSnakeCase(c.slug);
             return tableName === table.name;
         });