@rebasepro/studio 0.1.2 → 0.2.1

package/dist/{SQLEditor-Cr9Kg_Qg.js → SQLEditor-BELYJQRP.js}

@@ -1,10 +1,9 @@
 import { jsx, jsxs, Fragment } from "react/jsx-runtime";
 import { c } from "react-compiler-runtime";
 import { useTranslation, ErrorView, useRebaseContext, useStudioSideEntityController, useSnackbarController, useStudioCollectionRegistry, ConfirmationDialog, IconForView } from "@rebasepro/core";
-import { useState, useRef, useCallback, useEffect } from "react";
+import { useState, useMemo, useRef, useCallback, useEffect } from "react";
 import { createPortal } from "react-dom";
-import { CircularProgress, Typography, IconButton, iconSize, cls, defaultBorderMixin, Menu, MenuItem, Button, Tabs, Tab, Collapse, ResizablePanels, Tooltip, Checkbox, Dialog, DialogTitle, DialogContent, TextField, DialogActions, VirtualTable, TextareaAutosize } from "@rebasepro/ui";
-import { RefreshCwIcon, CopyIcon, MoreVerticalIcon, Trash2Icon, TerminalIcon, XIcon, PlusIcon, MenuIcon, DatabaseIcon, PlayIcon, PencilIcon } from "lucide-react";
+import { CircularProgress, Typography, IconButton, RefreshCwIcon, iconSize, cls, defaultBorderMixin, CopyIcon, Menu, MenuItem, MoreVerticalIcon, Button, Tabs, Tab, Trash2Icon, Collapse, ResizablePanels, TerminalIcon, XIcon, PlusIcon, Tooltip, MenuIcon, Checkbox, DatabaseIcon, PlayIcon, Dialog, DialogTitle, DialogContent, TextField, DialogActions, VirtualTable, PencilIcon, TextareaAutosize } from "@rebasepro/ui";
 import { M as MonacoEditor } from "./MonacoEditor-CMYEjiRf.js";
 import { parseFirst } from "pgsql-ast-parser";
 import { toSnakeCase } from "@rebasepro/utils";
@@ -698,14 +697,6 @@ const ExplainVisualizer = (t0) => {
 function _temp$1(childPlan, idx) {
   return /* @__PURE__ */ jsx(ExplainVisualizer, { plan: childPlan, isRoot: false }, idx);
 }
-let VirtualTableInput = null;
-try {
-  const cms = require("@rebasepro/admin");
-  VirtualTableInput = cms.VirtualTableInput;
-} catch {
-}
-const STORAGE_KEY_TABS = "rebase_sql_tabs";
-const STORAGE_KEY_ACTIVE_TAB = "rebase_sql_active_tab";
 const FixedEditorOverlay = (t0) => {
   const $ = c(26);
   const {
@@ -863,9 +854,14 @@ const FixedEditorOverlay = (t0) => {
   }
   return t13;
 };
+const getStoragePrefix = (baseUrl) => {
+  if (!baseUrl) return "default";
+  return baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_");
+};
 const SQLEditor = () => {
   const {
-    databaseAdmin
+    databaseAdmin,
+    client
   } = useRebaseContext();
   const sideEntityController = useStudioSideEntityController();
   const snackbarController = useSnackbarController();
@@ -873,22 +869,26 @@ const SQLEditor = () => {
   const {
     t
   } = useTranslation();
+  const projectPrefix = useMemo(() => getStoragePrefix(client?.baseUrl), [client?.baseUrl]);
   const [schemas, setSchemas] = useState({});
   const [isSchemaLoading, setIsSchemaLoading] = useState(true);
   const schemaFetchedRef = useRef(false);
   const [schemaError, setSchemaError] = useState(null);
   const [selectedDatabase, setSelectedDatabase] = useState(() => {
-    return localStorage.getItem("rebase_sql_selected_db") || void 0;
+    const projectPrefixSync = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+    return localStorage.getItem(`rebase_sql_selected_db_${projectPrefixSync}`) || void 0;
   });
   const [selectedRole, setSelectedRole] = useState(() => {
-    return localStorage.getItem("rebase_sql_selected_role") || void 0;
+    const projectPrefixSync_0 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+    return localStorage.getItem(`rebase_sql_selected_role_${projectPrefixSync_0}`) || void 0;
   });
   const [availableDatabases, setAvailableDatabases] = useState([]);
   const [availableRoles, setAvailableRoles] = useState([]);
   const [isLoadingConfig, setIsLoadingConfig] = useState(true);
   const [connectionConfigError, setConnectionConfigError] = useState(null);
   const [tabs, setTabs] = useState(() => {
-    const saved = localStorage.getItem(STORAGE_KEY_TABS);
+    const projectPrefixSync_1 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+    const saved = localStorage.getItem(`rebase_sql_tabs_${projectPrefixSync_1}`);
     if (saved) {
       const parsed = JSON.parse(saved);
       return parsed.map((t_0) => ({
@@ -904,8 +904,8 @@ const SQLEditor = () => {
       id: "1",
       name: "Query 1",
       sql: "SELECT * FROM ",
-      database: localStorage.getItem("rebase_sql_selected_db") || void 0,
-      role: localStorage.getItem("rebase_sql_selected_role") || void 0,
+      database: localStorage.getItem(`rebase_sql_selected_db_${projectPrefixSync_1}`) || void 0,
+      role: localStorage.getItem(`rebase_sql_selected_role_${projectPrefixSync_1}`) || void 0,
       results: null,
       loading: false,
       error: null,
@@ -914,7 +914,8 @@ const SQLEditor = () => {
     }];
   });
   const [activeTabId, setActiveTabId] = useState(() => {
-    return localStorage.getItem(STORAGE_KEY_ACTIVE_TAB) || "1";
+    const projectPrefixSync_2 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+    return localStorage.getItem(`rebase_sql_active_tab_${projectPrefixSync_2}`) || "1";
   });
   const activeTab = tabs.find((t_1) => t_1.id === activeTabId) || tabs[0];
   const updateActiveTab = useCallback((update) => {
@@ -934,22 +935,22 @@ 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([databaseAdmin.fetchAvailableDatabases(), databaseAdmin.fetchAvailableRoles(), typeof databaseAdmin?.fetchCurrentDatabase === "function" ? databaseAdmin.fetchCurrentDatabase() : Promise.resolve(void 0)]);
+        const [dbs, roles, currentDbFromApi, currentUserResult] = await Promise.all([databaseAdmin.fetchAvailableDatabases(), databaseAdmin.fetchAvailableRoles(), typeof databaseAdmin?.fetchCurrentDatabase === "function" ? databaseAdmin.fetchCurrentDatabase() : Promise.resolve(void 0), databaseAdmin.executeSql("SELECT current_user AS role").catch(() => [])]);
         if (mounted) {
           setAvailableDatabases(dbs);
           setAvailableRoles(roles);
-          const loadedDb = localStorage.getItem("rebase_sql_selected_db") || void 0;
-          const loadedRole = localStorage.getItem("rebase_sql_selected_role") || void 0;
-          const initialActiveTabId = localStorage.getItem(STORAGE_KEY_ACTIVE_TAB) || "1";
+          const loadedDb = localStorage.getItem(`rebase_sql_selected_db_${projectPrefix}`) || void 0;
+          const loadedRole = localStorage.getItem(`rebase_sql_selected_role_${projectPrefix}`) || void 0;
+          const initialActiveTabId = localStorage.getItem(`rebase_sql_active_tab_${projectPrefix}`) || "1";
           let initialTabs = [];
           try {
-            const savedTabs = localStorage.getItem(STORAGE_KEY_TABS);
+            const savedTabs = localStorage.getItem(`rebase_sql_tabs_${projectPrefix}`);
             if (savedTabs) initialTabs = JSON.parse(savedTabs);
           } catch (e) {
           }
@@ -961,21 +962,26 @@ const SQLEditor = () => {
           }
           if (actualDb) {
             setSelectedDatabase(actualDb);
-            localStorage.setItem("rebase_sql_selected_db", actualDb);
-            setTabs((prev_0) => prev_0.map((t_4) => t_4.id === initialActiveTabId && !t_4.database ? {
+            localStorage.setItem(`rebase_sql_selected_db_${projectPrefix}`, actualDb);
+            setTabs((prev_0) => prev_0.map((t_4) => t_4.id === initialActiveTabId && (!t_4.database || !dbs.includes(t_4.database)) ? {
               ...t_4,
               database: actualDb
             } : t_4));
           }
+          const currentUser = currentUserResult?.[0]?.role;
           let actualRole = currentActiveTab?.role || loadedRole;
           if (actualRole && !roles.includes(actualRole)) actualRole = void 0;
           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_1) => prev_1.map((t_5) => t_5.id === initialActiveTabId && !t_5.role ? {
+            localStorage.setItem(`rebase_sql_selected_role_${projectPrefix}`, actualRole);
+            setTabs((prev_1) => prev_1.map((t_5) => t_5.id === initialActiveTabId && (!t_5.role || !roles.includes(t_5.role)) ? {
               ...t_5,
               role: actualRole
             } : t_5));
@@ -999,10 +1005,10 @@ const SQLEditor = () => {
     return () => {
       mounted = false;
     };
-  }, [databaseAdmin]);
+  }, [databaseAdmin, projectPrefix]);
   const handleDatabaseChange = (db, tabId) => {
     setSelectedDatabase(db);
-    localStorage.setItem("rebase_sql_selected_db", db);
+    localStorage.setItem(`rebase_sql_selected_db_${projectPrefix}`, db);
     setTabs((prev_2) => prev_2.map((t_6) => t_6.id === activeTabId ? {
       ...t_6,
       database: db
@@ -1011,7 +1017,7 @@ const SQLEditor = () => {
   };
   const handleRoleChange = (role, tabId_0) => {
     setSelectedRole(role);
-    localStorage.setItem("rebase_sql_selected_role", role);
+    localStorage.setItem(`rebase_sql_selected_role_${projectPrefix}`, role);
     setTabs((prev_3) => prev_3.map((t_7) => t_7.id === activeTabId ? {
       ...t_7,
       role
@@ -1023,7 +1029,7 @@ const SQLEditor = () => {
     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_4) => prev_4.map((t_9) => t_9.id === newTabId ? {
@@ -1033,7 +1039,7 @@ const SQLEditor = () => {
       }
       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_5) => prev_5.map((t_10) => t_10.id === newTabId ? {
           ...t_10,
@@ -1041,7 +1047,7 @@ const SQLEditor = () => {
         } : t_10));
       }
     }
-  }, [tabs, selectedDatabase, selectedRole]);
+  }, [tabs, selectedDatabase, selectedRole, projectPrefix]);
   const fetchSchema = useCallback(async () => {
     if (!databaseAdmin?.executeSql) {
       setSchemaError(t("studio_sql_sql_not_supported"));
@@ -1241,7 +1247,8 @@ const SQLEditor = () => {
     }
   }, [editingCell, schemas, activeTab.lastExecutedSql, activeTab.results, databaseAdmin, updateActiveTab, snackbarController, selectedDatabase, selectedRole]);
   const [columnWidths, setColumnWidths] = useState(() => {
-    const saved_0 = localStorage.getItem("rebase_sql_column_widths");
+    const projectPrefixSync_3 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+    const saved_0 = localStorage.getItem(`rebase_sql_column_widths_${projectPrefixSync_3}`);
     return saved_0 ? JSON.parse(saved_0) : {};
   });
   const [snippets, setSnippets] = useState([]);
@@ -1249,11 +1256,19 @@ const SQLEditor = () => {
   const [isSaveDialogOpen, setIsSaveDialogOpen] = useState(false);
   const [newSnippetName, setNewSnippetName] = useState("");
   useEffect(() => {
-    const savedSnippets = localStorage.getItem("rebase_sql_snippets");
-    if (savedSnippets) setSnippets(JSON.parse(savedSnippets));
-    const savedHistory = localStorage.getItem("rebase_sql_history");
-    if (savedHistory) setHistory(JSON.parse(savedHistory));
-  }, []);
+    const savedSnippets = localStorage.getItem(`rebase_sql_snippets_${projectPrefix}`);
+    if (savedSnippets) {
+      setSnippets(JSON.parse(savedSnippets));
+    } else {
+      setSnippets([]);
+    }
+    const savedHistory = localStorage.getItem(`rebase_sql_history_${projectPrefix}`);
+    if (savedHistory) {
+      setHistory(JSON.parse(savedHistory));
+    } else {
+      setHistory([]);
+    }
+  }, [projectPrefix]);
   useEffect(() => {
     const sanitizedTabs = tabs.map((t_12) => ({
       id: t_12.id,
@@ -1262,18 +1277,18 @@ const SQLEditor = () => {
       database: t_12.database,
       role: t_12.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) => {
     setSnippets(newSnippets);
-    localStorage.setItem("rebase_sql_snippets", JSON.stringify(newSnippets));
+    localStorage.setItem(`rebase_sql_snippets_${projectPrefix}`, JSON.stringify(newSnippets));
   };
   const saveHistory = (newHistory) => {
     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) => {
     saveSnippets(snippets.filter((s) => s.id !== id));
@@ -1328,10 +1343,10 @@ const SQLEditor = () => {
           [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 = () => {
     const formatted = activeTab.sql.replace(/\s+/g, " ").replace(/\s?,\s?/g, ", ").replace(/\s?=\s?/g, " = ").trim();
     setSql(formatted);
@@ -1410,7 +1425,7 @@ const SQLEditor = () => {
         loading: false
       });
     }
-  }, [activeTab.sql, autoLimit, databaseAdmin, history, updateActiveTab]);
+  }, [activeTab.sql, autoLimit, databaseAdmin, history, updateActiveTab, selectedDatabase, selectedRole]);
   const handleRun = useCallback(async (selectedText) => {
     const sqlTarget = selectedText || activeTab.sql;
     if (!sqlTarget.trim()) return;
@@ -1656,7 +1671,8 @@ const SQLEditor = () => {
   };
   const [sidebarSize, setSidebarSize] = useState(() => {
     try {
-      const saved_1 = localStorage.getItem("rebase_sql_editor_sidebar_size");
+      const projectPrefixSync_4 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+      const saved_1 = localStorage.getItem(`rebase_sql_editor_sidebar_size_${projectPrefixSync_4}`);
       return saved_1 !== null ? parseFloat(saved_1) : 20;
     } catch (e_9) {
       return 20;
@@ -1664,7 +1680,8 @@ const SQLEditor = () => {
   });
   const [editorHeight, setEditorHeight] = useState(() => {
     try {
-      const saved_2 = localStorage.getItem("rebase_sql_editor_height");
+      const projectPrefixSync_5 = client?.baseUrl ? client.baseUrl.replace(/^https?:\/\//, "").replace(/[^a-zA-Z0-9]/g, "_") : "default";
+      const saved_2 = localStorage.getItem(`rebase_sql_editor_height_${projectPrefixSync_5}`);
       return saved_2 !== null ? parseFloat(saved_2) : 50;
     } catch (e_10) {
       return 50;
@@ -1672,16 +1689,16 @@ const SQLEditor = () => {
   });
   useEffect(() => {
     try {
-      localStorage.setItem("rebase_sql_editor_sidebar_size", sidebarSize.toString());
+      localStorage.setItem(`rebase_sql_editor_sidebar_size_${projectPrefix}`, sidebarSize.toString());
     } catch (e_11) {
     }
-  }, [sidebarSize]);
+  }, [sidebarSize, projectPrefix]);
   useEffect(() => {
     try {
-      localStorage.setItem("rebase_sql_editor_height", editorHeight.toString());
+      localStorage.setItem(`rebase_sql_editor_height_${projectPrefix}`, editorHeight.toString());
     } catch (e_12) {
     }
-  }, [editorHeight]);
+  }, [editorHeight, projectPrefix]);
   const activeSnippet = snippets.find((s_0) => s_0.sql === activeTab.sql);
   const isFavorite = activeSnippet?.isFavorite || false;
   return /* @__PURE__ */ jsxs("div", { className: "flex h-full w-full bg-white dark:bg-surface-950 overflow-hidden text-text-primary dark:text-text-primary-dark", children: [
@@ -1724,7 +1741,7 @@ const SQLEditor = () => {
           /* @__PURE__ */ jsx("div", { className: "h-4 w-px bg-surface-200 dark:bg-surface-950 mx-1" }),
           /* @__PURE__ */ jsx(Menu, { trigger: /* @__PURE__ */ jsxs(Button, { size: "small", variant: "outlined", className: "text-text-secondary dark:text-text-secondary-dark font-medium mr-2", children: [
             /* @__PURE__ */ jsx(DatabaseIcon, { size: iconSize.small, className: "mr-1.5 text-text-disabled dark:text-text-disabled-dark" }),
-            /* @__PURE__ */ jsx("span", { className: "max-w-[80px] truncate", children: isLoadingConfig ? "..." : selectedDatabase || t("studio_sql_select_db") })
+            /* @__PURE__ */ jsx("span", { className: "max-w-[160px] truncate", children: isLoadingConfig ? "..." : `${selectedDatabase || t("studio_sql_select_db")}${selectedRole ? ` (${selectedRole})` : ""}` })
           ] }), children: /* @__PURE__ */ jsxs("div", { className: "max-h-64 overflow-y-auto", children: [
             /* @__PURE__ */ jsx("div", { className: "px-3 py-1.5 border-b border-surface-200 dark:border-surface-950 mb-1", children: /* @__PURE__ */ jsx(Typography, { variant: "caption", className: "font-bold uppercase tracking-wider text-[9px] text-text-disabled dark:text-text-disabled-dark", children: t("studio_sql_database") }) }),
             isLoadingConfig ? /* @__PURE__ */ jsx("div", { className: "flex items-center justify-center p-4", children: /* @__PURE__ */ jsx(CircularProgress, { size: "small" }) }) : connectionConfigError ? /* @__PURE__ */ jsx("div", { className: "px-3 py-2 text-xs text-red-500 dark:text-red-400 max-w-[200px] break-words", children: connectionConfigError }) : /* @__PURE__ */ jsxs(Fragment, { children: [
@@ -1777,4 +1794,4 @@ function _temp(e) {
 export {
   SQLEditor
 };
-//# sourceMappingURL=SQLEditor-Cr9Kg_Qg.js.map
+//# sourceMappingURL=SQLEditor-BELYJQRP.js.map