@sqlrooms/sql-editor 0.26.1-rc.1 → 0.26.1-rc.11

package/dist/SqlEditor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SqlEditor.d.ts","sourceRoot":"","sources":["../src/SqlEditor.tsx"],"names":[],"mappings":"AAQA,OAAO,KAA8B,MAAM,OAAO,CAAC;AAGnD,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAEL,wBAAwB,EACzB,MAAM,kCAAkC,CAAC;AAE1C,MAAM,MAAM,cAAc,GAAG;IAC3B;;0EAEsE;IACtE,MAAM,CAAC,EAAE,wBAAwB,CAAC,QAAQ,CAAC,CAAC;IAC5C,kDAAkD;IAClD,MAAM,EAAE,OAAO,CAAC;IAChB,uEAAuE;IACvE,kBAAkB,CAAC,EAAE,KAAK,CAAC,SAAS,CAAC;IACrC;;;OAGG;IACH,gBAAgB,CAAC,EAAE,IAAI,CACrB,KAAK,CAAC,cAAc,CAAC,OAAO,gBAAgB,CAAC,EAC7C,YAAY,GAAG,kBAAkB,CAClC,CAAC;IACF,0DAA0D;IAC1D,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB,CAAC;AAEF,QAAA,MAAM,SAAS,4CAwFb,CAAC;AAEH,eAAe,SAAS,CAAC"}
+{"version":3,"file":"SqlEditor.d.ts","sourceRoot":"","sources":["../src/SqlEditor.tsx"],"names":[],"mappings":"AAOA,OAAO,KAA8B,MAAM,OAAO,CAAC;AAGnD,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAEL,wBAAwB,EACzB,MAAM,kCAAkC,CAAC;AAG1C,MAAM,MAAM,cAAc,GAAG;IAC3B;;0EAEsE;IACtE,MAAM,CAAC,EAAE,wBAAwB,CAAC,QAAQ,CAAC,CAAC;IAC5C,kDAAkD;IAClD,MAAM,EAAE,OAAO,CAAC;IAChB,uEAAuE;IACvE,kBAAkB,CAAC,EAAE,KAAK,CAAC,SAAS,CAAC;IACrC;;;OAGG;IACH,gBAAgB,CAAC,EAAE,IAAI,CACrB,KAAK,CAAC,cAAc,CAAC,OAAO,gBAAgB,CAAC,EAC7C,YAAY,GAAG,kBAAkB,CAClC,CAAC;IACF,0DAA0D;IAC1D,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB,CAAC;AAEF,QAAA,MAAM,SAAS,4CA8Eb,CAAC;AAGH,eAAe,SAAS,CAAC"}

package/dist/SqlEditor.js

@@ -1,5 +1,4 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import { useBaseRoomShellStore } from '@sqlrooms/room-shell';
 import { Button, ResizableHandle, ResizablePanel, ResizablePanelGroup, } from '@sqlrooms/ui';
 import { PlusIcon } from 'lucide-react';
 import React, { useCallback, useState } from 'react';
@@ -11,15 +10,7 @@ import { TableStructurePanel, } from './components/TableStructurePanel';
 import { useStoreWithSqlEditor } from './SqlEditorSlice';
 const SqlEditor = React.memo((props) => {
     const { schema = '*', documentationPanel, queryResultProps } = props;
-    // Store access
-    const addOrUpdateSqlQueryDataSource = useBaseRoomShellStore((state) => state.room.addOrUpdateSqlQueryDataSource);
-    const lastQueryStatement = useStoreWithSqlEditor((s) => s.sqlEditor.queryResult?.status === 'success' &&
-        s.sqlEditor.queryResult?.type === 'select'
-        ? s.sqlEditor.queryResult.lastQueryStatement
-        : '');
-    // const currentQuery = useStoreWithSqlEditor((s) =>
-    //   s.sqlEditor.getCurrentQuery(),
-    // );
+    const lastQuery = useStoreWithSqlEditor(({ sqlEditor: { queryResult: qr } }) => qr?.status === 'success' && qr?.type === 'select' ? qr.query : '');
     // UI state
     const [showDocs, setShowDocs] = useState(false);
     const [createTableModalOpen, setCreateTableModalOpen] = useState(false);
@@ -30,7 +21,8 @@ const SqlEditor = React.memo((props) => {
     const handleCreateTable = useCallback(() => {
         setCreateTableModalOpen(true);
     }, []);
-    return (_jsxs("div", { className: "relative flex h-full w-full flex-col overflow-hidden", children: [_jsx(SqlEditorHeader, { title: "SQL Editor", showDocs: showDocs, documentationPanel: documentationPanel, onToggleDocs: handleToggleDocs }), _jsx("div", { className: "bg-muted h-full flex-grow", children: _jsxs(ResizablePanelGroup, { direction: "horizontal", className: "h-full", children: [_jsx(ResizablePanel, { defaultSize: showDocs ? 70 : 100, children: _jsxs(ResizablePanelGroup, { direction: "vertical", className: "h-full", children: [_jsx(ResizablePanel, { defaultSize: 50, className: "flex flex-row", children: _jsxs(ResizablePanelGroup, { direction: "horizontal", children: [_jsx(ResizablePanel, { defaultSize: 20, children: _jsx(TableStructurePanel, { schema: schema }) }), _jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 80, children: _jsx(QueryEditorPanel, {}) })] }) }), _jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 50, children: _jsx(QueryResultPanel, { onRowClick: queryResultProps?.onRowClick, onRowDoubleClick: queryResultProps?.onRowDoubleClick, renderActions: () => (_jsx("div", { className: "flex gap-2", children: _jsxs(Button, { size: "xs", onClick: handleCreateTable, children: [_jsx(PlusIcon, { className: "h-4 w-4" }), "New table"] }) })) }) })] }) }), showDocs && (_jsxs(_Fragment, { children: [_jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 30, children: documentationPanel })] }))] }) }), _jsx(CreateTableModal, { query: lastQueryStatement, isOpen: createTableModalOpen, onClose: () => setCreateTableModalOpen(false), onAddOrUpdateSqlQuery: addOrUpdateSqlQueryDataSource })] }));
+    return (_jsxs("div", { className: "relative flex h-full w-full flex-col overflow-hidden", children: [_jsx(SqlEditorHeader, { title: "SQL Editor", showDocs: showDocs, documentationPanel: documentationPanel, onToggleDocs: handleToggleDocs }), _jsx("div", { className: "bg-muted h-full flex-grow", children: _jsxs(ResizablePanelGroup, { direction: "horizontal", className: "h-full", children: [_jsx(ResizablePanel, { defaultSize: showDocs ? 70 : 100, children: _jsxs(ResizablePanelGroup, { direction: "vertical", className: "h-full", children: [_jsx(ResizablePanel, { defaultSize: 50, className: "flex flex-row", children: _jsxs(ResizablePanelGroup, { direction: "horizontal", children: [_jsx(ResizablePanel, { defaultSize: 20, children: _jsx(TableStructurePanel, { schema: schema }) }), _jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 80, children: _jsx(QueryEditorPanel, {}) })] }) }), _jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 50, children: _jsx(QueryResultPanel, { onRowClick: queryResultProps?.onRowClick, onRowDoubleClick: queryResultProps?.onRowDoubleClick, renderActions: () => (_jsx("div", { className: "flex gap-2", children: _jsxs(Button, { size: "xs", onClick: handleCreateTable, children: [_jsx(PlusIcon, { className: "h-4 w-4" }), "New table"] }) })) }) })] }) }), showDocs && (_jsxs(_Fragment, { children: [_jsx(ResizableHandle, { withHandle: true }), _jsx(ResizablePanel, { defaultSize: 30, children: documentationPanel })] }))] }) }), _jsx(CreateTableModal, { query: lastQuery, isOpen: createTableModalOpen, onClose: () => setCreateTableModalOpen(false), allowMultipleStatements: true, showSchemaSelection: true })] }));
 });
+SqlEditor.displayName = 'SqlEditor';
 export default SqlEditor;
 //# sourceMappingURL=SqlEditor.js.map

package/dist/SqlEditor.js.map

@@ -1 +1 @@
-{"version":3,"file":"SqlEditor.js","sourceRoot":"","sources":["../src/SqlEditor.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,qBAAqB,EAAC,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EACL,MAAM,EACN,eAAe,EACf,cAAc,EACd,mBAAmB,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAC,QAAQ,EAAC,MAAM,cAAc,CAAC;AACtC,OAAO,KAAK,EAAE,EAAC,WAAW,EAAE,QAAQ,EAAC,MAAM,OAAO,CAAC;AACnD,OAAO,gBAAgB,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EACL,mBAAmB,GAEpB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAC,qBAAqB,EAAC,MAAM,kBAAkB,CAAC;AAsBvD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAiB,CAAC,KAAK,EAAE,EAAE;IACrD,MAAM,EAAC,MAAM,GAAG,GAAG,EAAE,kBAAkB,EAAE,gBAAgB,EAAC,GAAG,KAAK,CAAC;IAEnE,eAAe;IACf,MAAM,6BAA6B,GAAG,qBAAqB,CACzD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,6BAA6B,CACpD,CAAC;IACF,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,CAAC,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,KAAK,SAAS;QAC7C,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,KAAK,QAAQ;QACxC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,kBAAkB;QAC5C,CAAC,CAAC,EAAE,CACP,CAAC;IACF,oDAAoD;IACpD,mCAAmC;IACnC,KAAK;IAEL,WAAW;IACX,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChD,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExE,WAAW;IACX,MAAM,gBAAgB,GAAG,WAAW,CAAC,CAAC,IAAa,EAAE,EAAE;QACrD,WAAW,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;QACzC,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,eAAK,SAAS,EAAC,sDAAsD,aACnE,KAAC,eAAe,IACd,KAAK,EAAC,YAAY,EAClB,QAAQ,EAAE,QAAQ,EAClB,kBAAkB,EAAE,kBAAkB,EACtC,YAAY,EAAE,gBAAgB,GAC9B,EACF,cAAK,SAAS,EAAC,2BAA2B,YACxC,MAAC,mBAAmB,IAAC,SAAS,EAAC,YAAY,EAAC,SAAS,EAAC,QAAQ,aAC5D,KAAC,cAAc,IAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,YAC9C,MAAC,mBAAmB,IAAC,SAAS,EAAC,UAAU,EAAC,SAAS,EAAC,QAAQ,aAC1D,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,EAAE,SAAS,EAAC,eAAe,YACxD,MAAC,mBAAmB,IAAC,SAAS,EAAC,YAAY,aACzC,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,mBAAmB,IAAC,MAAM,EAAE,MAAM,GAAI,GACxB,EACjB,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,gBAAgB,KAAG,GACL,IACG,GACP,EACjB,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,gBAAgB,IACf,UAAU,EAAE,gBAAgB,EAAE,UAAU,EACxC,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB,EACpD,aAAa,EAAE,GAAG,EAAE,CAAC,CACnB,cAAK,SAAS,EAAC,YAAY,YACzB,MAAC,MAAM,IAAC,IAAI,EAAC,IAAI,EAAC,OAAO,EAAE,iBAAiB,aAC1C,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,iBAEzB,GACL,CACP,GACD,GACa,IACG,GACP,EAChB,QAAQ,IAAI,CACX,8BACE,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC5B,kBAAkB,GACJ,IAChB,CACJ,IACmB,GAClB,EACN,KAAC,gBAAgB,IACf,KAAK,EAAE,kBAAkB,EACzB,MAAM,EAAE,oBAAoB,EAC5B,OAAO,EAAE,GAAG,EAAE,CAAC,uBAAuB,CAAC,KAAK,CAAC,EAC7C,qBAAqB,EAAE,6BAA6B,GACpD,IACE,CACP,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,eAAe,SAAS,CAAC","sourcesContent":["import {useBaseRoomShellStore} from '@sqlrooms/room-shell';\nimport {\n  Button,\n  ResizableHandle,\n  ResizablePanel,\n  ResizablePanelGroup,\n} from '@sqlrooms/ui';\nimport {PlusIcon} from 'lucide-react';\nimport React, {useCallback, useState} from 'react';\nimport CreateTableModal from './components/CreateTableModal';\nimport {QueryEditorPanel} from './components/QueryEditorPanel';\nimport {QueryResultPanel} from './components/QueryResultPanel';\nimport {SqlEditorHeader} from './components/SqlEditorHeader';\nimport {\n  TableStructurePanel,\n  TableStructurePanelProps,\n} from './components/TableStructurePanel';\nimport {useStoreWithSqlEditor} from './SqlEditorSlice';\nexport type SqlEditorProps = {\n  /** The database schema to use. Defaults to '*'.\n   * If '*' is provided, all tables will be shown.\n   * If a function is provided, it will be used to filter the tables. */\n  schema?: TableStructurePanelProps['schema'];\n  /** Whether the SQL editor is currently visible */\n  isOpen: boolean;\n  /** Optional component to render SQL documentation in the side panel */\n  documentationPanel?: React.ReactNode;\n  /**\n   * Props forwarded to `QueryResultPanel` to configure result behavior.\n   * This provides a single entry point for table interactions.\n   */\n  queryResultProps?: Pick<\n    React.ComponentProps<typeof QueryResultPanel>,\n    'onRowClick' | 'onRowDoubleClick'\n  >;\n  /** Callback fired when the SQL editor should be closed */\n  onClose: () => void;\n};\n\nconst SqlEditor = React.memo<SqlEditorProps>((props) => {\n  const {schema = '*', documentationPanel, queryResultProps} = props;\n\n  // Store access\n  const addOrUpdateSqlQueryDataSource = useBaseRoomShellStore(\n    (state) => state.room.addOrUpdateSqlQueryDataSource,\n  );\n  const lastQueryStatement = useStoreWithSqlEditor((s) =>\n    s.sqlEditor.queryResult?.status === 'success' &&\n    s.sqlEditor.queryResult?.type === 'select'\n      ? s.sqlEditor.queryResult.lastQueryStatement\n      : '',\n  );\n  // const currentQuery = useStoreWithSqlEditor((s) =>\n  //   s.sqlEditor.getCurrentQuery(),\n  // );\n\n  // UI state\n  const [showDocs, setShowDocs] = useState(false);\n  const [createTableModalOpen, setCreateTableModalOpen] = useState(false);\n\n  // Handlers\n  const handleToggleDocs = useCallback((show: boolean) => {\n    setShowDocs(show);\n  }, []);\n\n  const handleCreateTable = useCallback(() => {\n    setCreateTableModalOpen(true);\n  }, []);\n\n  return (\n    <div className=\"relative flex h-full w-full flex-col overflow-hidden\">\n      <SqlEditorHeader\n        title=\"SQL Editor\"\n        showDocs={showDocs}\n        documentationPanel={documentationPanel}\n        onToggleDocs={handleToggleDocs}\n      />\n      <div className=\"bg-muted h-full flex-grow\">\n        <ResizablePanelGroup direction=\"horizontal\" className=\"h-full\">\n          <ResizablePanel defaultSize={showDocs ? 70 : 100}>\n            <ResizablePanelGroup direction=\"vertical\" className=\"h-full\">\n              <ResizablePanel defaultSize={50} className=\"flex flex-row\">\n                <ResizablePanelGroup direction=\"horizontal\">\n                  <ResizablePanel defaultSize={20}>\n                    <TableStructurePanel schema={schema} />\n                  </ResizablePanel>\n                  <ResizableHandle withHandle />\n                  <ResizablePanel defaultSize={80}>\n                    <QueryEditorPanel />\n                  </ResizablePanel>\n                </ResizablePanelGroup>\n              </ResizablePanel>\n              <ResizableHandle withHandle />\n              <ResizablePanel defaultSize={50}>\n                <QueryResultPanel\n                  onRowClick={queryResultProps?.onRowClick}\n                  onRowDoubleClick={queryResultProps?.onRowDoubleClick}\n                  renderActions={() => (\n                    <div className=\"flex gap-2\">\n                      <Button size=\"xs\" onClick={handleCreateTable}>\n                        <PlusIcon className=\"h-4 w-4\" />\n                        New table\n                      </Button>\n                    </div>\n                  )}\n                />\n              </ResizablePanel>\n            </ResizablePanelGroup>\n          </ResizablePanel>\n          {showDocs && (\n            <>\n              <ResizableHandle withHandle />\n              <ResizablePanel defaultSize={30}>\n                {documentationPanel}\n              </ResizablePanel>\n            </>\n          )}\n        </ResizablePanelGroup>\n      </div>\n      <CreateTableModal\n        query={lastQueryStatement}\n        isOpen={createTableModalOpen}\n        onClose={() => setCreateTableModalOpen(false)}\n        onAddOrUpdateSqlQuery={addOrUpdateSqlQueryDataSource}\n      />\n    </div>\n  );\n});\n\nexport default SqlEditor;\n"]}
+{"version":3,"file":"SqlEditor.js","sourceRoot":"","sources":["../src/SqlEditor.tsx"],"names":[],"mappings":";AAAA,OAAO,EACL,MAAM,EACN,eAAe,EACf,cAAc,EACd,mBAAmB,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAC,QAAQ,EAAC,MAAM,cAAc,CAAC;AACtC,OAAO,KAAK,EAAE,EAAC,WAAW,EAAE,QAAQ,EAAC,MAAM,OAAO,CAAC;AACnD,OAAO,gBAAgB,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAC,gBAAgB,EAAC,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EACL,mBAAmB,GAEpB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAC,qBAAqB,EAAC,MAAM,kBAAkB,CAAC;AAuBvD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAiB,CAAC,KAAK,EAAE,EAAE;IACrD,MAAM,EAAC,MAAM,GAAG,GAAG,EAAE,kBAAkB,EAAE,gBAAgB,EAAC,GAAG,KAAK,CAAC;IAEnE,MAAM,SAAS,GAAG,qBAAqB,CAAC,CAAC,EAAC,SAAS,EAAE,EAAC,WAAW,EAAE,EAAE,EAAC,EAAC,EAAE,EAAE,CACzE,EAAE,EAAE,MAAM,KAAK,SAAS,IAAI,EAAE,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAClE,CAAC;IACF,WAAW;IACX,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChD,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExE,WAAW;IACX,MAAM,gBAAgB,GAAG,WAAW,CAAC,CAAC,IAAa,EAAE,EAAE;QACrD,WAAW,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;QACzC,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,eAAK,SAAS,EAAC,sDAAsD,aACnE,KAAC,eAAe,IACd,KAAK,EAAC,YAAY,EAClB,QAAQ,EAAE,QAAQ,EAClB,kBAAkB,EAAE,kBAAkB,EACtC,YAAY,EAAE,gBAAgB,GAC9B,EACF,cAAK,SAAS,EAAC,2BAA2B,YACxC,MAAC,mBAAmB,IAAC,SAAS,EAAC,YAAY,EAAC,SAAS,EAAC,QAAQ,aAC5D,KAAC,cAAc,IAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,YAC9C,MAAC,mBAAmB,IAAC,SAAS,EAAC,UAAU,EAAC,SAAS,EAAC,QAAQ,aAC1D,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,EAAE,SAAS,EAAC,eAAe,YACxD,MAAC,mBAAmB,IAAC,SAAS,EAAC,YAAY,aACzC,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,mBAAmB,IAAC,MAAM,EAAE,MAAM,GAAI,GACxB,EACjB,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,gBAAgB,KAAG,GACL,IACG,GACP,EACjB,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC7B,KAAC,gBAAgB,IACf,UAAU,EAAE,gBAAgB,EAAE,UAAU,EACxC,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB,EACpD,aAAa,EAAE,GAAG,EAAE,CAAC,CACnB,cAAK,SAAS,EAAC,YAAY,YACzB,MAAC,MAAM,IAAC,IAAI,EAAC,IAAI,EAAC,OAAO,EAAE,iBAAiB,aAC1C,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,iBAEzB,GACL,CACP,GACD,GACa,IACG,GACP,EAChB,QAAQ,IAAI,CACX,8BACE,KAAC,eAAe,IAAC,UAAU,SAAG,EAC9B,KAAC,cAAc,IAAC,WAAW,EAAE,EAAE,YAC5B,kBAAkB,GACJ,IAChB,CACJ,IACmB,GAClB,EACN,KAAC,gBAAgB,IACf,KAAK,EAAE,SAAS,EAChB,MAAM,EAAE,oBAAoB,EAC5B,OAAO,EAAE,GAAG,EAAE,CAAC,uBAAuB,CAAC,KAAK,CAAC,EAC7C,uBAAuB,EAAE,IAAI,EAC7B,mBAAmB,EAAE,IAAI,GACzB,IACE,CACP,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,SAAS,CAAC,WAAW,GAAG,WAAW,CAAC;AAEpC,eAAe,SAAS,CAAC","sourcesContent":["import {\n  Button,\n  ResizableHandle,\n  ResizablePanel,\n  ResizablePanelGroup,\n} from '@sqlrooms/ui';\nimport {PlusIcon} from 'lucide-react';\nimport React, {useCallback, useState} from 'react';\nimport CreateTableModal from './components/CreateTableModal';\nimport {QueryEditorPanel} from './components/QueryEditorPanel';\nimport {QueryResultPanel} from './components/QueryResultPanel';\nimport {SqlEditorHeader} from './components/SqlEditorHeader';\nimport {\n  TableStructurePanel,\n  TableStructurePanelProps,\n} from './components/TableStructurePanel';\nimport {useStoreWithSqlEditor} from './SqlEditorSlice';\n\nexport type SqlEditorProps = {\n  /** The database schema to use. Defaults to '*'.\n   * If '*' is provided, all tables will be shown.\n   * If a function is provided, it will be used to filter the tables. */\n  schema?: TableStructurePanelProps['schema'];\n  /** Whether the SQL editor is currently visible */\n  isOpen: boolean;\n  /** Optional component to render SQL documentation in the side panel */\n  documentationPanel?: React.ReactNode;\n  /**\n   * Props forwarded to `QueryResultPanel` to configure result behavior.\n   * This provides a single entry point for table interactions.\n   */\n  queryResultProps?: Pick<\n    React.ComponentProps<typeof QueryResultPanel>,\n    'onRowClick' | 'onRowDoubleClick'\n  >;\n  /** Callback fired when the SQL editor should be closed */\n  onClose: () => void;\n};\n\nconst SqlEditor = React.memo<SqlEditorProps>((props) => {\n  const {schema = '*', documentationPanel, queryResultProps} = props;\n\n  const lastQuery = useStoreWithSqlEditor(({sqlEditor: {queryResult: qr}}) =>\n    qr?.status === 'success' && qr?.type === 'select' ? qr.query : '',\n  );\n  // UI state\n  const [showDocs, setShowDocs] = useState(false);\n  const [createTableModalOpen, setCreateTableModalOpen] = useState(false);\n\n  // Handlers\n  const handleToggleDocs = useCallback((show: boolean) => {\n    setShowDocs(show);\n  }, []);\n\n  const handleCreateTable = useCallback(() => {\n    setCreateTableModalOpen(true);\n  }, []);\n\n  return (\n    <div className=\"relative flex h-full w-full flex-col overflow-hidden\">\n      <SqlEditorHeader\n        title=\"SQL Editor\"\n        showDocs={showDocs}\n        documentationPanel={documentationPanel}\n        onToggleDocs={handleToggleDocs}\n      />\n      <div className=\"bg-muted h-full flex-grow\">\n        <ResizablePanelGroup direction=\"horizontal\" className=\"h-full\">\n          <ResizablePanel defaultSize={showDocs ? 70 : 100}>\n            <ResizablePanelGroup direction=\"vertical\" className=\"h-full\">\n              <ResizablePanel defaultSize={50} className=\"flex flex-row\">\n                <ResizablePanelGroup direction=\"horizontal\">\n                  <ResizablePanel defaultSize={20}>\n                    <TableStructurePanel schema={schema} />\n                  </ResizablePanel>\n                  <ResizableHandle withHandle />\n                  <ResizablePanel defaultSize={80}>\n                    <QueryEditorPanel />\n                  </ResizablePanel>\n                </ResizablePanelGroup>\n              </ResizablePanel>\n              <ResizableHandle withHandle />\n              <ResizablePanel defaultSize={50}>\n                <QueryResultPanel\n                  onRowClick={queryResultProps?.onRowClick}\n                  onRowDoubleClick={queryResultProps?.onRowDoubleClick}\n                  renderActions={() => (\n                    <div className=\"flex gap-2\">\n                      <Button size=\"xs\" onClick={handleCreateTable}>\n                        <PlusIcon className=\"h-4 w-4\" />\n                        New table\n                      </Button>\n                    </div>\n                  )}\n                />\n              </ResizablePanel>\n            </ResizablePanelGroup>\n          </ResizablePanel>\n          {showDocs && (\n            <>\n              <ResizableHandle withHandle />\n              <ResizablePanel defaultSize={30}>\n                {documentationPanel}\n              </ResizablePanel>\n            </>\n          )}\n        </ResizablePanelGroup>\n      </div>\n      <CreateTableModal\n        query={lastQuery}\n        isOpen={createTableModalOpen}\n        onClose={() => setCreateTableModalOpen(false)}\n        allowMultipleStatements={true}\n        showSchemaSelection={true}\n      />\n    </div>\n  );\n});\nSqlEditor.displayName = 'SqlEditor';\n\nexport default SqlEditor;\n"]}

package/dist/SqlEditorSlice.d.ts

@@ -14,10 +14,12 @@ export type QueryResult = {
     status: 'success';
     type: 'pragma' | 'explain' | 'select';
     result: arrow.Table | undefined;
+    query: string;
     lastQueryStatement: string;
 } | {
     status: 'success';
     type: 'exec';
+    query: string;
     lastQueryStatement: string;
 };
 export declare function isQueryWithResult(queryResult: QueryResult | undefined): queryResult is QueryResult & {
@@ -85,9 +87,14 @@ export type SqlEditorSliceState = {
         closeQueryTab(queryId: string): void;
         /**
          * Open a closed tab id.
-         * @param queryId - The ID of the query to remove.
+         * @param queryId - The ID of the query to open.
          */
         openQueryTab(queryId: string): void;
+        /**
+         * Set the list of open tab IDs. Used for reordering or opening tabs.
+         * @param tabIds - The new list of open tab IDs.
+         */
+        setOpenTabs(tabIds: string[]): void;
         /**
          * Update the SQL text for a query.
          * @param queryId - The ID of the query to update.

package/dist/SqlEditorSlice.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SqlEditorSlice.d.ts","sourceRoot":"","sources":["../src/SqlEditorSlice.tsx"],"names":[],"mappings":"AAOA,OAAO,EAGL,mBAAmB,EACnB,YAAY,EAEb,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,oBAAoB,EACrB,MAAM,6BAA6B,CAAC;AAErC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAKtC,MAAM,MAAM,WAAW,GACnB;IAAC,MAAM,EAAE,SAAS,CAAC;IAAC,cAAc,CAAC,EAAE,OAAO,CAAC;IAAC,UAAU,EAAE,eAAe,CAAA;CAAC,GAC1E;IAAC,MAAM,EAAE,SAAS,CAAA;CAAC,GACnB;IAAC,MAAM,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAC,GAChC;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;IACtC,MAAM,EAAE,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IAChC,kBAAkB,EAAE,MAAM,CAAC;CAC5B,GACD;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,kBAAkB,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEN,wBAAgB,iBAAiB,CAC/B,WAAW,EAAE,WAAW,GAAG,SAAS,GACnC,WAAW,IAAI,WAAW,GAAG;IAC9B,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;CACvC,CAOA;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,SAAS,EAAE;QACT,MAAM,EAAE,oBAAoB,CAAC;QAE7B,WAAW,CAAC,EAAE,WAAW,CAAC;QAC1B,mBAAmB;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,6FAA6F;QAC7F,eAAe,EAAE,OAAO,CAAC;QACzB,kBAAkB;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;QAErB,gBAAgB,EAAE,MAAM,CAAC;QACzB,4CAA4C;QAC5C,uBAAuB,EAAE,MAAM,EAAE,CAAC;QAElC;;WAEG;QACH,SAAS,CAAC,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAAC;QAE9C;;WAEG;QACH,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAE/C;;WAEG;QACH,uBAAuB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;QAEzC;;WAEG;QACH,iBAAiB,IAAI,IAAI,CAAC;QAE1B;;;WAGG;QACH,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAElE;;;WAGG;QACH,cAAc,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG;YACrC,EAAE,EAAE,MAAM,CAAC;YACX,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,CAAC;QAEF;;;WAGG;QACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEtC;;;;WAIG;QACH,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEvD;;;WAGG;QACH,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACrC;;;WAGG;QACH,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEpC;;;;WAIG;QACH,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QAE1D;;;WAGG;QACH,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAE1C;;WAEG;QACH,eAAe,IAAI,MAAM,CAAC;QAE1B,kBAAkB;QAClB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAAC;QAE7C,iBAAiB,IAAI,IAAI,CAAC;QAE1B,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;KAC1C,CAAC;CACH,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,EACnC,MAAuC,EACvC,gBAAsB,EACtB,uBAA0C,GAC3C,GAAE;IACD,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uBAAuB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,GAAG,YAAY,CAAC,mBAAmB,CAAC,CAyVzC;AAED,KAAK,sBAAsB,GAAG,mBAAmB,GAAG,mBAAmB,CAAC;AAExE,wBAAgB,qBAAqB,CAAC,CAAC,EACrC,QAAQ,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,CAAC,GAC7C,CAAC,CAIH"}
+{"version":3,"file":"SqlEditorSlice.d.ts","sourceRoot":"","sources":["../src/SqlEditorSlice.tsx"],"names":[],"mappings":"AAQA,OAAO,EAGL,mBAAmB,EACnB,YAAY,EAEb,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,oBAAoB,EACrB,MAAM,6BAA6B,CAAC;AAErC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAKtC,MAAM,MAAM,WAAW,GACnB;IAAC,MAAM,EAAE,SAAS,CAAC;IAAC,cAAc,CAAC,EAAE,OAAO,CAAC;IAAC,UAAU,EAAE,eAAe,CAAA;CAAC,GAC1E;IAAC,MAAM,EAAE,SAAS,CAAA;CAAC,GACnB;IAAC,MAAM,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAC,GAChC;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;IACtC,MAAM,EAAE,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,kBAAkB,EAAE,MAAM,CAAC;CAC5B,GACD;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,kBAAkB,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEN,wBAAgB,iBAAiB,CAC/B,WAAW,EAAE,WAAW,GAAG,SAAS,GACnC,WAAW,IAAI,WAAW,GAAG;IAC9B,MAAM,EAAE,SAAS,CAAC;IAClB,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;CACvC,CAOA;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,SAAS,EAAE;QACT,MAAM,EAAE,oBAAoB,CAAC;QAE7B,WAAW,CAAC,EAAE,WAAW,CAAC;QAC1B,mBAAmB;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,6FAA6F;QAC7F,eAAe,EAAE,OAAO,CAAC;QACzB,kBAAkB;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;QAErB,gBAAgB,EAAE,MAAM,CAAC;QACzB,4CAA4C;QAC5C,uBAAuB,EAAE,MAAM,EAAE,CAAC;QAElC;;WAEG;QACH,SAAS,CAAC,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAAC;QAE9C;;WAEG;QACH,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAE/C;;WAEG;QACH,uBAAuB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;QAEzC;;WAEG;QACH,iBAAiB,IAAI,IAAI,CAAC;QAE1B;;;WAGG;QACH,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAElE;;;WAGG;QACH,cAAc,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG;YACrC,EAAE,EAAE,MAAM,CAAC;YACX,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,CAAC;QAEF;;;WAGG;QACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEtC;;;;WAIG;QACH,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEvD;;;WAGG;QACH,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAErC;;;WAGG;QACH,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAEpC;;;WAGG;QACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QAEpC;;;;WAIG;QACH,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QAE1D;;;WAGG;QACH,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QAE1C;;WAEG;QACH,eAAe,IAAI,MAAM,CAAC;QAE1B,kBAAkB;QAClB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAAC;QAE7C,iBAAiB,IAAI,IAAI,CAAC;QAE1B,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;KAC1C,CAAC;CACH,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,EACnC,MAAuC,EACvC,gBAAsB,EACtB,uBAA0C,GAC3C,GAAE;IACD,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uBAAuB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,GAAG,YAAY,CAAC,mBAAmB,CAAC,CAqWzC;AAED,KAAK,sBAAsB,GAAG,mBAAmB,GAAG,mBAAmB,CAAC;AAExE,wBAAgB,qBAAqB,CAAC,CAAC,EACrC,QAAQ,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,CAAC,GAC7C,CAAC,CAIH"}

package/dist/SqlEditorSlice.js

@@ -1,5 +1,5 @@
 import { createId } from '@paralleldrive/cuid2';
-import { getSqlErrorWithPointer, makeLimitQuery, splitSqlStatements, } from '@sqlrooms/duckdb';
+import { getSqlErrorWithPointer, joinStatements, makeLimitQuery, separateLastStatement, } from '@sqlrooms/duckdb';
 import { createSlice, useBaseRoomShellStore, } from '@sqlrooms/room-shell';
 import { createDefaultSqlEditorConfig, } from '@sqlrooms/sql-editor-config';
 import { generateUniqueName } from '@sqlrooms/utils';
@@ -43,6 +43,7 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                     };
                     set((state) => produce(state, (draft) => {
                         draft.sqlEditor.config.queries.push(newQuery);
+                        draft.sqlEditor.config.openTabs.push(newQuery.id);
                         draft.sqlEditor.config.selectedQueryId = newQuery.id;
                     }));
                     return newQuery;
@@ -50,25 +51,38 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                 deleteQueryTab: (queryId) => {
                     const sqlEditorConfig = get().sqlEditor.config;
                     const queries = sqlEditorConfig.queries;
+                    const openTabs = sqlEditorConfig.openTabs;
                     if (queries.length <= 1) {
                         // Don't delete the last query
                         return;
                     }
-                    const index = queries.findIndex((q) => q.id === queryId);
-                    if (index === -1)
-                        return;
-                    const isSelected = sqlEditorConfig.selectedQueryId === queryId;
+                    const wasSelected = sqlEditorConfig.selectedQueryId === queryId;
+                    const deletingOpenIndex = openTabs.indexOf(queryId);
                     const filteredQueries = queries.filter((q) => q.id !== queryId);
                     set((state) => produce(state, (draft) => {
                         draft.sqlEditor.config.queries = filteredQueries;
-                        // If we're deleting the selected tab, select the previous one or the first one
-                        if (isSelected && filteredQueries.length > 0) {
-                            const newSelectedIndex = Math.max(0, index - 1);
-                            // Safely access the ID with fallback to the first query if needed
-                            const newSelectedId = filteredQueries[newSelectedIndex]?.id ??
-                                filteredQueries[0]?.id;
-                            if (newSelectedId) {
-                                draft.sqlEditor.config.selectedQueryId = newSelectedId;
+                        draft.sqlEditor.config.openTabs = openTabs.filter((id) => id !== queryId);
+                        // If we deleted the selected query, select another one
+                        if (wasSelected) {
+                            const newOpenTabs = draft.sqlEditor.config.openTabs;
+                            const remainingQueries = draft.sqlEditor.config.queries;
+                            if (newOpenTabs.length > 0) {
+                                // Select from remaining open tabs
+                                const newIndex = deletingOpenIndex === 0
+                                    ? 0
+                                    : Math.min(deletingOpenIndex - 1, newOpenTabs.length - 1);
+                                const newSelectedId = newOpenTabs[newIndex];
+                                if (newSelectedId) {
+                                    draft.sqlEditor.config.selectedQueryId = newSelectedId;
+                                }
+                            }
+                            else if (remainingQueries.length > 0) {
+                                // No open tabs left, open a closed query
+                                const queryToOpen = remainingQueries[0];
+                                if (queryToOpen) {
+                                    draft.sqlEditor.config.openTabs.push(queryToOpen.id);
+                                    draft.sqlEditor.config.selectedQueryId = queryToOpen.id;
+                                }
                             }
                         }
                     }));
@@ -83,22 +97,23 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                 },
                 closeQueryTab: (queryId) => {
                     set((state) => produce(state, (draft) => {
-                        draft.sqlEditor.config.closedTabIds.push(queryId);
-                        const openedTabs = draft.sqlEditor.config.queries.filter((q) => !draft.sqlEditor.config.closedTabIds.includes(q.id));
-                        if (draft.sqlEditor.config.selectedQueryId === queryId &&
-                            openedTabs.length > 0 &&
-                            openedTabs[0]) {
-                            draft.sqlEditor.config.selectedQueryId = openedTabs[0].id;
-                        }
+                        draft.sqlEditor.config.openTabs =
+                            draft.sqlEditor.config.openTabs.filter((id) => id !== queryId);
                     }));
                 },
                 openQueryTab: (queryId) => {
                     set((state) => produce(state, (draft) => {
-                        draft.sqlEditor.config.closedTabIds =
-                            draft.sqlEditor.config.closedTabIds?.filter((id) => id !== queryId);
+                        if (!draft.sqlEditor.config.openTabs.includes(queryId)) {
+                            draft.sqlEditor.config.openTabs.push(queryId);
+                        }
                         draft.sqlEditor.config.selectedQueryId = queryId;
                     }));
                 },
+                setOpenTabs: (tabIds) => {
+                    set((state) => produce(state, (draft) => {
+                        draft.sqlEditor.config.openTabs = tabIds;
+                    }));
+                },
                 updateQueryText: (queryId, queryText) => {
                     set((state) => produce(state, (draft) => {
                         const query = draft.sqlEditor.config.queries.find((q) => q.id === queryId);
@@ -174,12 +189,8 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                     try {
                         const connector = await get().db.getConnector();
                         const signal = queryController.signal;
-                        const statements = splitSqlStatements(query);
-                        const allButLastStatements = statements.slice(0, -1);
-                        const lastQueryStatement = statements[statements.length - 1];
-                        if (!statements?.length) {
-                            throw new Error('Empty query');
-                        }
+                        const { precedingStatements, lastStatement: lastQueryStatement } = separateLastStatement(query);
+                        const hasMultipleStatements = precedingStatements.length > 0;
                         if (signal.aborted) {
                             throw new Error('Query aborted');
                         }
@@ -190,17 +201,16 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                         const isValidSelectQuery = !parsedLastStatement.error;
                         if (isValidSelectQuery) {
                             // Add limit to the last statement
-                            const queryWithLimit = [
-                                ...allButLastStatements,
-                                makeLimitQuery(lastQueryStatement, {
-                                    sanitize: false, // should already be sanitized
-                                    limit: get().sqlEditor.queryResultLimit,
-                                }),
-                            ].join(';\n');
+                            const limitedLastStatement = makeLimitQuery(lastQueryStatement, {
+                                sanitize: false, // should already be sanitized
+                                limit: get().sqlEditor.queryResultLimit,
+                            });
+                            const queryWithLimit = joinStatements(precedingStatements, limitedLastStatement);
                             const result = await connector.query(queryWithLimit, { signal });
                             queryResult = {
                                 status: 'success',
                                 type: 'select',
+                                query,
                                 lastQueryStatement,
                                 result,
                             };
@@ -220,6 +230,7 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                                 queryResult = {
                                     status: 'success',
                                     type: 'explain',
+                                    query,
                                     lastQueryStatement,
                                     result,
                                 };
@@ -228,6 +239,7 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                                 queryResult = {
                                     status: 'success',
                                     type: 'pragma',
+                                    query,
                                     lastQueryStatement,
                                     result,
                                 };
@@ -236,6 +248,7 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                                 queryResult = {
                                     status: 'success',
                                     type: 'exec',
+                                    query,
                                     lastQueryStatement,
                                 };
                             }
@@ -245,7 +258,7 @@ export function createSqlEditorSlice({ config = createDefaultSqlEditorConfig(),
                         }
                         // Refresh table schemas if there are multiple statements or if the
                         // last statement is not a select query
-                        if (statements.length > 1 || !isValidSelectQuery) {
+                        if (hasMultipleStatements || !isValidSelectQuery) {
                             get().db.refreshTableSchemas();
                         }
                         if (signal.aborted) {

package/dist/SqlEditorSlice.js.map

@@ -1 +1 @@
-{"version":3,"file":"SqlEditorSlice.js","sourceRoot":"","sources":["../src/SqlEditorSlice.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAEL,sBAAsB,EACtB,cAAc,EACd,kBAAkB,GACnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,WAAW,EAGX,qBAAqB,GACtB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,4BAA4B,GAE7B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAC,kBAAkB,EAAC,MAAM,iBAAiB,CAAC;AAEnD,OAAO,EAAC,SAAS,EAAC,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAC,MAAM,EAAC,MAAM,YAAY,CAAC;AAClC,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAkB9B,MAAM,UAAU,iBAAiB,CAC/B,WAAoC;IAKpC,OAAO,CACL,WAAW,EAAE,MAAM,KAAK,SAAS;QACjC,CAAC,WAAW,CAAC,IAAI,KAAK,QAAQ;YAC5B,WAAW,CAAC,IAAI,KAAK,SAAS;YAC9B,WAAW,CAAC,IAAI,KAAK,QAAQ,CAAC,CACjC,CAAC;AACJ,CAAC;AAyGD,MAAM,UAAU,oBAAoB,CAAC,EACnC,MAAM,GAAG,4BAA4B,EAAE,EACvC,gBAAgB,GAAG,GAAG,EACtB,uBAAuB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,MAKxC,EAAE;IACJ,OAAO,WAAW,CAGhB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACb,OAAO;YACL,SAAS,EAAE;gBACT,MAAM;gBACN,2BAA2B;gBAC3B,eAAe,EAAE,KAAK;gBACtB,gBAAgB;gBAChB,uBAAuB;gBAEvB,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE;oBACpB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC;oBAClC,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,kBAAkB,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE;oBACxC,IAAI,CAAC,OAAO;wBAAE,OAAO;oBACrB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE;wBACpD,IAAI,EAAE,0BAA0B;qBACjC,CAAC,CAAC;oBACH,MAAM,CAAC,IAAI,EAAE,QAAQ,IAAI,UAAU,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;gBACvE,CAAC;gBAED,cAAc,EAAE,CAAC,YAAY,GAAG,EAAE,EAAE,EAAE;oBACpC,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,QAAQ,GAAG;wBACf,EAAE,EAAE,QAAQ,EAAE;wBACd,IAAI,EAAE,kBAAkB,CACtB,UAAU,EACV,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAC3C;wBACD,KAAK,EAAE,YAAY;qBACpB,CAAC;oBAEF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAC9C,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,QAAQ,CAAC,EAAE,CAAC;oBACvD,CAAC,CAAC,CACH,CAAC;oBAEF,OAAO,QAAQ,CAAC;gBAClB,CAAC;gBAED,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC1B,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC;oBAExC,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;wBACxB,8BAA8B;wBAC9B,OAAO;oBACT,CAAC;oBAED,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;oBACzD,IAAI,KAAK,KAAK,CAAC,CAAC;wBAAE,OAAO;oBAEzB,MAAM,UAAU,GAAG,eAAe,CAAC,eAAe,KAAK,OAAO,CAAC;oBAC/D,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;oBAEhE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,GAAG,eAAe,CAAC;wBAEjD,+EAA+E;wBAC/E,IAAI,UAAU,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC7C,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;4BAChD,kEAAkE;4BAClE,MAAM,aAAa,GACjB,eAAe,CAAC,gBAAgB,CAAC,EAAE,EAAE;gCACrC,eAAe,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;4BACzB,IAAI,aAAa,EAAE,CAAC;gCAClB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,aAAa,CAAC;4BACzD,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,cAAc,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;oBACnC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CACxB,CAAC;wBACF,IAAI,KAAK,EAAE,CAAC;4BACV,KAAK,CAAC,IAAI,GAAG,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC;wBACrC,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,aAAa,EAAE,CAAC,OAAO,EAAE,EAAE;oBACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;wBAClD,MAAM,UAAU,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CACtD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAC3D,CAAC;wBAEF,IACE,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,KAAK,OAAO;4BAClD,UAAU,CAAC,MAAM,GAAG,CAAC;4BACrB,UAAU,CAAC,CAAC,CAAC,EACb,CAAC;4BACD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC5D,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,YAAY,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY;4BACjC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CACzC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,OAAO,CACvB,CAAC;wBACJ,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,OAAO,CAAC;oBACnD,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,eAAe,EAAE,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE;oBACtC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CACxB,CAAC;wBACF,IAAI,KAAK,EAAE,CAAC;4BACV,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC1B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,kBAAkB,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC9B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,OAAO,CAAC;oBACnD,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,eAAe,EAAE,GAAG,EAAE;oBACpB,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,UAAU,GAAG,eAAe,CAAC,eAAe,CAAC;oBACnD,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAC3B,CAAC;oBACF,OAAO,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC;gBAC5B,CAAC;gBAED,kBAAkB;gBAClB,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE;oBACrB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,aAAa,GAAG,KAAK,CAAC;oBACxC,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,iBAAiB,EAAE,GAAG,EAAE;oBACtB,sDAAsD;oBACtD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACd,GAAG,KAAK;wBACR,SAAS,EAAE;4BACT,GAAG,KAAK,CAAC,SAAS;4BAClB,YAAY,EAAE,IAAI;4BAClB,UAAU,EAAE,SAAS;yBACtB;qBACF,CAAC,CAAC,CAAC;gBACN,CAAC;gBAED,uBAAuB,EAAE,KAAK,IAAmB,EAAE,CACjD,GAAG,EAAE,CAAC,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;gBAErE,iBAAiB,EAAE,GAAG,EAAE;oBACtB,MAAM,aAAa,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC;oBAClD,IAAI,aAAa,EAAE,MAAM,KAAK,SAAS,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;wBACpE,aAAa,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnC,CAAC;oBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,IAAI,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;4BACtD,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,GAAG,IAAI,CAAC;wBACpD,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;oBAC7B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,gBAAgB,GAAG,KAAK,CAAC;oBAC3C,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,gBAAgB,EAAE,KAAK,EAAE,KAAK,EAAiB,EAAE;oBAC/C,IAAI,GAAG,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;wBACtD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;oBAC3C,CAAC;oBACD,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;wBAClB,OAAO;oBACT,CAAC;oBAED,mDAAmD;oBACnD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;oBAE9C,+CAA+C;oBAC/C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,aAAa,GAAG,SAAS,CAAC;wBAC1C,KAAK,CAAC,SAAS,CAAC,WAAW,GAAG;4BAC5B,MAAM,EAAE,SAAS;4BACjB,cAAc,EAAE,KAAK;4BACrB,UAAU,EAAE,eAAe;yBAC5B,CAAC;oBACJ,CAAC,CAAC,CACH,CAAC;oBAEF,IAAI,WAAwB,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;wBAChD,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC;wBAEtC,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;wBAC7C,MAAM,oBAAoB,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;wBACrD,MAAM,kBAAkB,GAAG,UAAU,CACnC,UAAU,CAAC,MAAM,GAAG,CAAC,CACZ,CAAC;wBAEZ,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;4BACxB,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;wBACjC,CAAC;wBAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBAED,MAAM,mBAAmB,GACvB,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;wBAErD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBAED,MAAM,kBAAkB,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC;wBAEtD,IAAI,kBAAkB,EAAE,CAAC;4BACvB,kCAAkC;4BAClC,MAAM,cAAc,GAAG;gCACrB,GAAG,oBAAoB;gCACvB,cAAc,CAAC,kBAAkB,EAAE;oCACjC,QAAQ,EAAE,KAAK,EAAE,8BAA8B;oCAC/C,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,gBAAgB;iCACxC,CAAC;6BACH,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACd,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,cAAc,EAAE,EAAC,MAAM,EAAC,CAAC,CAAC;4BAC/D,WAAW,GAAG;gCACZ,MAAM,EAAE,SAAS;gCACjB,IAAI,EAAE,QAAQ;gCACd,kBAAkB;gCAClB,MAAM;6BACP,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,kCAAkC;4BAClC,IACE,mBAAmB,CAAC,KAAK;gCACzB,mBAAmB,CAAC,UAAU,KAAK,iBAAiB,EACpD,CAAC;gCACD,MAAM,CACJ,GAAG,mBAAmB,CAAC,UAAU,IAAI,mBAAmB,CAAC,aAAa,KAAK,mBAAmB,CAAC,aAAa,EAAE;oCAC9G,KAAK,sBAAsB,CAAC,kBAAkB,EAAE,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,CAClG,CAAC;4BACJ,CAAC;4BAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,EAAC,MAAM,EAAC,CAAC,CAAC;4BACtD,wDAAwD;4BACxD,8CAA8C;4BAC9C,sDAAsD;4BACtD,IAAI,aAAa,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;gCAC3C,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,SAAS;oCACf,kBAAkB;oCAClB,MAAM;iCACP,CAAC;4BACJ,CAAC;iCAAM,IAAI,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;gCACjD,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,QAAQ;oCACd,kBAAkB;oCAClB,MAAM;iCACP,CAAC;4BACJ,CAAC;iCAAM,CAAC;gCACN,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,MAAM;oCACZ,kBAAkB;iCACnB,CAAC;4BACJ,CAAC;wBACH,CAAC;wBACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBACD,mEAAmE;wBACnE,uCAAuC;wBACvC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;4BACjD,GAAG,EAAE,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC;wBACjC,CAAC;wBACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;oBACH,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;wBACjB,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;wBAChE,IACE,YAAY,KAAK,eAAe;4BAChC,eAAe,CAAC,MAAM,CAAC,OAAO,EAC9B,CAAC;4BACD,WAAW,GAAG,EAAC,MAAM,EAAE,SAAS,EAAC,CAAC;wBACpC,CAAC;6BAAM,CAAC;4BACN,WAAW,GAAG;gCACZ,MAAM,EAAE,OAAO;gCACf,KAAK,EAAE,YAAY;6BACpB,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACd,GAAG,KAAK;wBACR,SAAS,EAAE,EAAC,GAAG,KAAK,CAAC,SAAS,EAAE,WAAW,EAAC;qBAC7C,CAAC,CAAC,CAAC;gBACN,CAAC;aACF;SAC4B,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC;AAID,MAAM,UAAU,qBAAqB,CACnC,QAA8C;IAE9C,OAAO,qBAAqB,CAAyB,CAAC,KAAK,EAAE,EAAE,CAC7D,QAAQ,CAAC,KAA0C,CAAC,CACrD,CAAC;AACJ,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {\n  DuckDbSliceState,\n  getSqlErrorWithPointer,\n  makeLimitQuery,\n  splitSqlStatements,\n} from '@sqlrooms/duckdb';\nimport {\n  BaseRoomStoreState,\n  createSlice,\n  RoomShellSliceState,\n  StateCreator,\n  useBaseRoomShellStore,\n} from '@sqlrooms/room-shell';\nimport {\n  createDefaultSqlEditorConfig,\n  SqlEditorSliceConfig,\n} from '@sqlrooms/sql-editor-config';\nimport {generateUniqueName} from '@sqlrooms/utils';\nimport * as arrow from 'apache-arrow';\nimport {csvFormat} from 'd3-dsv';\nimport {saveAs} from 'file-saver';\nimport {produce} from 'immer';\n\nexport type QueryResult =\n  | {status: 'loading'; isBeingAborted?: boolean; controller: AbortController}\n  | {status: 'aborted'}\n  | {status: 'error'; error: string}\n  | {\n      status: 'success';\n      type: 'pragma' | 'explain' | 'select';\n      result: arrow.Table | undefined;\n      lastQueryStatement: string;\n    }\n  | {\n      status: 'success';\n      type: 'exec';\n      lastQueryStatement: string;\n    };\n\nexport function isQueryWithResult(\n  queryResult: QueryResult | undefined,\n): queryResult is QueryResult & {\n  status: 'success';\n  type: 'pragma' | 'explain' | 'select';\n} {\n  return (\n    queryResult?.status === 'success' &&\n    (queryResult.type === 'pragma' ||\n      queryResult.type === 'explain' ||\n      queryResult.type === 'select')\n  );\n}\n\nexport type SqlEditorSliceState = {\n  sqlEditor: {\n    config: SqlEditorSliceConfig;\n    // Runtime state\n    queryResult?: QueryResult;\n    /** @deprecated  */\n    selectedTable?: string;\n    /** @deprecated Use `useStoreWithSqlEditor((s) => s.db.isRefreshingTableSchemas)` instead. */\n    isTablesLoading: boolean;\n    /** @deprecated */\n    tablesError?: string;\n\n    queryResultLimit: number;\n    /** Options for the result limit dropdown */\n    queryResultLimitOptions: number[];\n\n    /**\n     * Set the config for the sql editor slice.\n     */\n    setConfig(config: SqlEditorSliceConfig): void;\n\n    /**\n     * Run the currently selected query.\n     */\n    parseAndRunQuery(query: string): Promise<void>;\n\n    /**\n     * Run the currently selected query.\n     */\n    parseAndRunCurrentQuery(): Promise<void>;\n\n    /**\n     * Abort the currently running query.\n     */\n    abortCurrentQuery(): void;\n\n    /**\n     * Export query results to CSV.\n     * @deprecated Use `useExportToCsv` from `@sqlrooms/duckdb` instead.\n     */\n    exportResultsToCsv(results: arrow.Table, filename?: string): void;\n\n    /**\n     * Create a new query tab.\n     * @param initialQuery - Optional initial query text.\n     */\n    createQueryTab(initialQuery?: string): {\n      id: string;\n      name: string;\n      query: string;\n    };\n\n    /**\n     * Delete a query tab.\n     * @param queryId - The ID of the query to delete.\n     */\n    deleteQueryTab(queryId: string): void;\n\n    /**\n     * Rename a query tab.\n     * @param queryId - The ID of the query to rename.\n     * @param newName - The new name for the query.\n     */\n    renameQueryTab(queryId: string, newName: string): void;\n\n    /**\n     * Close a query tab.\n     * @param queryId - The ID of the query to close.\n     */\n    closeQueryTab(queryId: string): void;\n    /**\n     * Open a closed tab id.\n     * @param queryId - The ID of the query to remove.\n     */\n    openQueryTab(queryId: string): void;\n\n    /**\n     * Update the SQL text for a query.\n     * @param queryId - The ID of the query to update.\n     * @param queryText - The new SQL text.\n     */\n    updateQueryText(queryId: string, queryText: string): void;\n\n    /**\n     * Set the selected query tab.\n     * @param queryId - The ID of the query to select.\n     */\n    setSelectedQueryId(queryId: string): void;\n\n    /**\n     * Get the currently selected query's SQL text.\n     */\n    getCurrentQuery(): string;\n\n    /** @deprecated */\n    selectTable(table: string | undefined): void;\n\n    clearQueryResults(): void;\n\n    setQueryResultLimit(limit: number): void;\n  };\n};\n\nexport function createSqlEditorSlice({\n  config = createDefaultSqlEditorConfig(),\n  queryResultLimit = 100,\n  queryResultLimitOptions = [100, 500, 1000],\n}: {\n  config?: SqlEditorSliceConfig;\n  queryResultLimit?: number;\n  queryResultLimitOptions?: number[];\n} = {}): StateCreator<SqlEditorSliceState> {\n  return createSlice<\n    SqlEditorSliceState,\n    BaseRoomStoreState & DuckDbSliceState & SqlEditorSliceState\n  >((set, get) => {\n    return {\n      sqlEditor: {\n        config,\n        // Initialize runtime state\n        isTablesLoading: false,\n        queryResultLimit,\n        queryResultLimitOptions,\n\n        setConfig: (config) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config = config;\n            }),\n          );\n        },\n\n        exportResultsToCsv: (results, filename) => {\n          if (!results) return;\n          const blob = new Blob([csvFormat(results.toArray())], {\n            type: 'text/plain;charset=utf-8',\n          });\n          saveAs(blob, filename || `export-${createId().substring(0, 5)}.csv`);\n        },\n\n        createQueryTab: (initialQuery = '') => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const newQuery = {\n            id: createId(),\n            name: generateUniqueName(\n              'Untitled',\n              sqlEditorConfig.queries.map((q) => q.name),\n            ),\n            query: initialQuery,\n          };\n\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.queries.push(newQuery);\n              draft.sqlEditor.config.selectedQueryId = newQuery.id;\n            }),\n          );\n\n          return newQuery;\n        },\n\n        deleteQueryTab: (queryId) => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const queries = sqlEditorConfig.queries;\n\n          if (queries.length <= 1) {\n            // Don't delete the last query\n            return;\n          }\n\n          const index = queries.findIndex((q) => q.id === queryId);\n          if (index === -1) return;\n\n          const isSelected = sqlEditorConfig.selectedQueryId === queryId;\n          const filteredQueries = queries.filter((q) => q.id !== queryId);\n\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.queries = filteredQueries;\n\n              // If we're deleting the selected tab, select the previous one or the first one\n              if (isSelected && filteredQueries.length > 0) {\n                const newSelectedIndex = Math.max(0, index - 1);\n                // Safely access the ID with fallback to the first query if needed\n                const newSelectedId =\n                  filteredQueries[newSelectedIndex]?.id ??\n                  filteredQueries[0]?.id;\n                if (newSelectedId) {\n                  draft.sqlEditor.config.selectedQueryId = newSelectedId;\n                }\n              }\n            }),\n          );\n        },\n\n        renameQueryTab: (queryId, newName) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const query = draft.sqlEditor.config.queries.find(\n                (q) => q.id === queryId,\n              );\n              if (query) {\n                query.name = newName || query.name;\n              }\n            }),\n          );\n        },\n\n        closeQueryTab: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.closedTabIds.push(queryId);\n              const openedTabs = draft.sqlEditor.config.queries.filter(\n                (q) => !draft.sqlEditor.config.closedTabIds.includes(q.id),\n              );\n\n              if (\n                draft.sqlEditor.config.selectedQueryId === queryId &&\n                openedTabs.length > 0 &&\n                openedTabs[0]\n              ) {\n                draft.sqlEditor.config.selectedQueryId = openedTabs[0].id;\n              }\n            }),\n          );\n        },\n\n        openQueryTab: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.closedTabIds =\n                draft.sqlEditor.config.closedTabIds?.filter(\n                  (id) => id !== queryId,\n                );\n              draft.sqlEditor.config.selectedQueryId = queryId;\n            }),\n          );\n        },\n\n        updateQueryText: (queryId, queryText) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const query = draft.sqlEditor.config.queries.find(\n                (q) => q.id === queryId,\n              );\n              if (query) {\n                query.query = queryText;\n              }\n            }),\n          );\n        },\n\n        setSelectedQueryId: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.selectedQueryId = queryId;\n            }),\n          );\n        },\n\n        getCurrentQuery: () => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const selectedId = sqlEditorConfig.selectedQueryId;\n          const query = sqlEditorConfig.queries.find(\n            (q) => q.id === selectedId,\n          );\n          return query?.query || '';\n        },\n\n        /** @deprecated */\n        selectTable: (table) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.selectedTable = table;\n            }),\n          );\n        },\n\n        clearQueryResults: () => {\n          // Update state without using Immer for the Table type\n          set((state) => ({\n            ...state,\n            sqlEditor: {\n              ...state.sqlEditor,\n              queryResults: null,\n              queryError: undefined,\n            },\n          }));\n        },\n\n        parseAndRunCurrentQuery: async (): Promise<void> =>\n          get().sqlEditor.parseAndRunQuery(get().sqlEditor.getCurrentQuery()),\n\n        abortCurrentQuery: () => {\n          const currentResult = get().sqlEditor.queryResult;\n          if (currentResult?.status === 'loading' && currentResult.controller) {\n            currentResult.controller.abort();\n          }\n\n          set((state) =>\n            produce(state, (draft) => {\n              if (draft.sqlEditor.queryResult?.status === 'loading') {\n                draft.sqlEditor.queryResult.isBeingAborted = true;\n              }\n            }),\n          );\n        },\n\n        setQueryResultLimit: (limit) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.queryResultLimit = limit;\n            }),\n          );\n        },\n\n        parseAndRunQuery: async (query): Promise<void> => {\n          if (get().sqlEditor.queryResult?.status === 'loading') {\n            throw new Error('Query already running');\n          }\n          if (!query.trim()) {\n            return;\n          }\n\n          // Create abort controller for this query execution\n          const queryController = new AbortController();\n\n          // First update loading state and clear results\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.selectedTable = undefined;\n              draft.sqlEditor.queryResult = {\n                status: 'loading',\n                isBeingAborted: false,\n                controller: queryController,\n              };\n            }),\n          );\n\n          let queryResult: QueryResult;\n          try {\n            const connector = await get().db.getConnector();\n            const signal = queryController.signal;\n\n            const statements = splitSqlStatements(query);\n            const allButLastStatements = statements.slice(0, -1);\n            const lastQueryStatement = statements[\n              statements.length - 1\n            ] as string;\n\n            if (!statements?.length) {\n              throw new Error('Empty query');\n            }\n\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n\n            const parsedLastStatement =\n              await get().db.sqlSelectToJson(lastQueryStatement);\n\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n\n            const isValidSelectQuery = !parsedLastStatement.error;\n\n            if (isValidSelectQuery) {\n              // Add limit to the last statement\n              const queryWithLimit = [\n                ...allButLastStatements,\n                makeLimitQuery(lastQueryStatement, {\n                  sanitize: false, // should already be sanitized\n                  limit: get().sqlEditor.queryResultLimit,\n                }),\n              ].join(';\\n');\n              const result = await connector.query(queryWithLimit, {signal});\n              queryResult = {\n                status: 'success',\n                type: 'select',\n                lastQueryStatement,\n                result,\n              };\n            } else {\n              // Run the complete query as it is\n              if (\n                parsedLastStatement.error &&\n                parsedLastStatement.error_type !== 'not implemented'\n              ) {\n                throw (\n                  `${parsedLastStatement.error_type} ${parsedLastStatement.error_subtype}: ${parsedLastStatement.error_message}` +\n                  `\\n${getSqlErrorWithPointer(lastQueryStatement, Number(parsedLastStatement.position)).formatted}`\n                );\n              }\n\n              const result = await connector.query(query, {signal});\n              // EXPLAIN and PRAGMA are not detected as select queries\n              // and we cannot wrap them in a SELECT * FROM,\n              // but we can still execute them and return the result\n              if (/^(EXPLAIN)/i.test(lastQueryStatement)) {\n                queryResult = {\n                  status: 'success',\n                  type: 'explain',\n                  lastQueryStatement,\n                  result,\n                };\n              } else if (/^(PRAGMA)/i.test(lastQueryStatement)) {\n                queryResult = {\n                  status: 'success',\n                  type: 'pragma',\n                  lastQueryStatement,\n                  result,\n                };\n              } else {\n                queryResult = {\n                  status: 'success',\n                  type: 'exec',\n                  lastQueryStatement,\n                };\n              }\n            }\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n            // Refresh table schemas if there are multiple statements or if the\n            // last statement is not a select query\n            if (statements.length > 1 || !isValidSelectQuery) {\n              get().db.refreshTableSchemas();\n            }\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n          } catch (e) {\n            console.error(e);\n            const errorMessage = e instanceof Error ? e.message : String(e);\n            if (\n              errorMessage === 'Query aborted' ||\n              queryController.signal.aborted\n            ) {\n              queryResult = {status: 'aborted'};\n            } else {\n              queryResult = {\n                status: 'error',\n                error: errorMessage,\n              };\n            }\n          }\n\n          set((state) => ({\n            ...state,\n            sqlEditor: {...state.sqlEditor, queryResult},\n          }));\n        },\n      },\n    } satisfies SqlEditorSliceState;\n  });\n}\n\ntype RoomStateWithSqlEditor = RoomShellSliceState & SqlEditorSliceState;\n\nexport function useStoreWithSqlEditor<T>(\n  selector: (state: RoomStateWithSqlEditor) => T,\n): T {\n  return useBaseRoomShellStore<RoomShellSliceState, T>((state) =>\n    selector(state as unknown as RoomStateWithSqlEditor),\n  );\n}\n"]}
+{"version":3,"file":"SqlEditorSlice.js","sourceRoot":"","sources":["../src/SqlEditorSlice.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAEL,sBAAsB,EACtB,cAAc,EACd,cAAc,EACd,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,WAAW,EAGX,qBAAqB,GACtB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,4BAA4B,GAE7B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAC,kBAAkB,EAAC,MAAM,iBAAiB,CAAC;AAEnD,OAAO,EAAC,SAAS,EAAC,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAC,MAAM,EAAC,MAAM,YAAY,CAAC;AAClC,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAoB9B,MAAM,UAAU,iBAAiB,CAC/B,WAAoC;IAKpC,OAAO,CACL,WAAW,EAAE,MAAM,KAAK,SAAS;QACjC,CAAC,WAAW,CAAC,IAAI,KAAK,QAAQ;YAC5B,WAAW,CAAC,IAAI,KAAK,SAAS;YAC9B,WAAW,CAAC,IAAI,KAAK,QAAQ,CAAC,CACjC,CAAC;AACJ,CAAC;AAgHD,MAAM,UAAU,oBAAoB,CAAC,EACnC,MAAM,GAAG,4BAA4B,EAAE,EACvC,gBAAgB,GAAG,GAAG,EACtB,uBAAuB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,MAKxC,EAAE;IACJ,OAAO,WAAW,CAGhB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACb,OAAO;YACL,SAAS,EAAE;gBACT,MAAM;gBACN,2BAA2B;gBAC3B,eAAe,EAAE,KAAK;gBACtB,gBAAgB;gBAChB,uBAAuB;gBAEvB,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE;oBACpB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC;oBAClC,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,kBAAkB,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE;oBACxC,IAAI,CAAC,OAAO;wBAAE,OAAO;oBACrB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE;wBACpD,IAAI,EAAE,0BAA0B;qBACjC,CAAC,CAAC;oBACH,MAAM,CAAC,IAAI,EAAE,QAAQ,IAAI,UAAU,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;gBACvE,CAAC;gBAED,cAAc,EAAE,CAAC,YAAY,GAAG,EAAE,EAAE,EAAE;oBACpC,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,QAAQ,GAAG;wBACf,EAAE,EAAE,QAAQ,EAAE;wBACd,IAAI,EAAE,kBAAkB,CACtB,UAAU,EACV,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAC3C;wBACD,KAAK,EAAE,YAAY;qBACpB,CAAC;oBAEF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAC9C,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;wBAClD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,QAAQ,CAAC,EAAE,CAAC;oBACvD,CAAC,CAAC,CACH,CAAC;oBAEF,OAAO,QAAQ,CAAC;gBAClB,CAAC;gBAED,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC1B,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC;oBACxC,MAAM,QAAQ,GAAG,eAAe,CAAC,QAAQ,CAAC;oBAE1C,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;wBACxB,8BAA8B;wBAC9B,OAAO;oBACT,CAAC;oBAED,MAAM,WAAW,GAAG,eAAe,CAAC,eAAe,KAAK,OAAO,CAAC;oBAChE,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBACpD,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;oBAEhE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,GAAG,eAAe,CAAC;wBACjD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAC/C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,OAAO,CACvB,CAAC;wBAEF,uDAAuD;wBACvD,IAAI,WAAW,EAAE,CAAC;4BAChB,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC;4BACpD,MAAM,gBAAgB,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC;4BAExD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCAC3B,kCAAkC;gCAClC,MAAM,QAAQ,GACZ,iBAAiB,KAAK,CAAC;oCACrB,CAAC,CAAC,CAAC;oCACH,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,EAAE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gCAC9D,MAAM,aAAa,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;gCAC5C,IAAI,aAAa,EAAE,CAAC;oCAClB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,aAAa,CAAC;gCACzD,CAAC;4BACH,CAAC;iCAAM,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCACvC,yCAAyC;gCACzC,MAAM,WAAW,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;gCACxC,IAAI,WAAW,EAAE,CAAC;oCAChB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;oCACrD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,WAAW,CAAC,EAAE,CAAC;gCAC1D,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,cAAc,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;oBACnC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CACxB,CAAC;wBACF,IAAI,KAAK,EAAE,CAAC;4BACV,KAAK,CAAC,IAAI,GAAG,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC;wBACrC,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,aAAa,EAAE,CAAC,OAAO,EAAE,EAAE;oBACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ;4BAC7B,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;oBACnE,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,YAAY,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;4BACvD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;wBAChD,CAAC;wBACD,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,OAAO,CAAC;oBACnD,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,WAAW,EAAE,CAAC,MAAM,EAAE,EAAE;oBACtB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC;oBAC3C,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,eAAe,EAAE,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE;oBACtC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CACxB,CAAC;wBACF,IAAI,KAAK,EAAE,CAAC;4BACV,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC1B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,kBAAkB,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC9B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,GAAG,OAAO,CAAC;oBACnD,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,eAAe,EAAE,GAAG,EAAE;oBACpB,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/C,MAAM,UAAU,GAAG,eAAe,CAAC,eAAe,CAAC;oBACnD,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAC3B,CAAC;oBACF,OAAO,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC;gBAC5B,CAAC;gBAED,kBAAkB;gBAClB,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE;oBACrB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,aAAa,GAAG,KAAK,CAAC;oBACxC,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,iBAAiB,EAAE,GAAG,EAAE;oBACtB,sDAAsD;oBACtD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACd,GAAG,KAAK;wBACR,SAAS,EAAE;4BACT,GAAG,KAAK,CAAC,SAAS;4BAClB,YAAY,EAAE,IAAI;4BAClB,UAAU,EAAE,SAAS;yBACtB;qBACF,CAAC,CAAC,CAAC;gBACN,CAAC;gBAED,uBAAuB,EAAE,KAAK,IAAmB,EAAE,CACjD,GAAG,EAAE,CAAC,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;gBAErE,iBAAiB,EAAE,GAAG,EAAE;oBACtB,MAAM,aAAa,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC;oBAClD,IAAI,aAAa,EAAE,MAAM,KAAK,SAAS,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;wBACpE,aAAa,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnC,CAAC;oBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,IAAI,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;4BACtD,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,GAAG,IAAI,CAAC;wBACpD,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;oBAC7B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,gBAAgB,GAAG,KAAK,CAAC;oBAC3C,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAED,gBAAgB,EAAE,KAAK,EAAE,KAAK,EAAiB,EAAE;oBAC/C,IAAI,GAAG,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;wBACtD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;oBAC3C,CAAC;oBACD,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;wBAClB,OAAO;oBACT,CAAC;oBAED,mDAAmD;oBACnD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;oBAE9C,+CAA+C;oBAC/C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,SAAS,CAAC,aAAa,GAAG,SAAS,CAAC;wBAC1C,KAAK,CAAC,SAAS,CAAC,WAAW,GAAG;4BAC5B,MAAM,EAAE,SAAS;4BACjB,cAAc,EAAE,KAAK;4BACrB,UAAU,EAAE,eAAe;yBAC5B,CAAC;oBACJ,CAAC,CAAC,CACH,CAAC;oBAEF,IAAI,WAAwB,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;wBAChD,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC;wBAEtC,MAAM,EAAC,mBAAmB,EAAE,aAAa,EAAE,kBAAkB,EAAC,GAC5D,qBAAqB,CAAC,KAAK,CAAC,CAAC;wBAC/B,MAAM,qBAAqB,GAAG,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAAC;wBAE7D,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBAED,MAAM,mBAAmB,GACvB,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;wBAErD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBAED,MAAM,kBAAkB,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC;wBAEtD,IAAI,kBAAkB,EAAE,CAAC;4BACvB,kCAAkC;4BAClC,MAAM,oBAAoB,GAAG,cAAc,CAAC,kBAAkB,EAAE;gCAC9D,QAAQ,EAAE,KAAK,EAAE,8BAA8B;gCAC/C,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,gBAAgB;6BACxC,CAAC,CAAC;4BACH,MAAM,cAAc,GAAG,cAAc,CACnC,mBAAmB,EACnB,oBAAoB,CACrB,CAAC;4BACF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,cAAc,EAAE,EAAC,MAAM,EAAC,CAAC,CAAC;4BAC/D,WAAW,GAAG;gCACZ,MAAM,EAAE,SAAS;gCACjB,IAAI,EAAE,QAAQ;gCACd,KAAK;gCACL,kBAAkB;gCAClB,MAAM;6BACP,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,kCAAkC;4BAClC,IACE,mBAAmB,CAAC,KAAK;gCACzB,mBAAmB,CAAC,UAAU,KAAK,iBAAiB,EACpD,CAAC;gCACD,MAAM,CACJ,GAAG,mBAAmB,CAAC,UAAU,IAAI,mBAAmB,CAAC,aAAa,KAAK,mBAAmB,CAAC,aAAa,EAAE;oCAC9G,KAAK,sBAAsB,CAAC,kBAAkB,EAAE,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,CAClG,CAAC;4BACJ,CAAC;4BAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,EAAC,MAAM,EAAC,CAAC,CAAC;4BACtD,wDAAwD;4BACxD,8CAA8C;4BAC9C,sDAAsD;4BACtD,IAAI,aAAa,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;gCAC3C,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,SAAS;oCACf,KAAK;oCACL,kBAAkB;oCAClB,MAAM;iCACP,CAAC;4BACJ,CAAC;iCAAM,IAAI,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;gCACjD,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,QAAQ;oCACd,KAAK;oCACL,kBAAkB;oCAClB,MAAM;iCACP,CAAC;4BACJ,CAAC;iCAAM,CAAC;gCACN,WAAW,GAAG;oCACZ,MAAM,EAAE,SAAS;oCACjB,IAAI,EAAE,MAAM;oCACZ,KAAK;oCACL,kBAAkB;iCACnB,CAAC;4BACJ,CAAC;wBACH,CAAC;wBACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;wBACD,mEAAmE;wBACnE,uCAAuC;wBACvC,IAAI,qBAAqB,IAAI,CAAC,kBAAkB,EAAE,CAAC;4BACjD,GAAG,EAAE,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC;wBACjC,CAAC;wBACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;wBACnC,CAAC;oBACH,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;wBACjB,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;wBAChE,IACE,YAAY,KAAK,eAAe;4BAChC,eAAe,CAAC,MAAM,CAAC,OAAO,EAC9B,CAAC;4BACD,WAAW,GAAG,EAAC,MAAM,EAAE,SAAS,EAAC,CAAC;wBACpC,CAAC;6BAAM,CAAC;4BACN,WAAW,GAAG;gCACZ,MAAM,EAAE,OAAO;gCACf,KAAK,EAAE,YAAY;6BACpB,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBACd,GAAG,KAAK;wBACR,SAAS,EAAE,EAAC,GAAG,KAAK,CAAC,SAAS,EAAE,WAAW,EAAC;qBAC7C,CAAC,CAAC,CAAC;gBACN,CAAC;aACF;SAC4B,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC;AAID,MAAM,UAAU,qBAAqB,CACnC,QAA8C;IAE9C,OAAO,qBAAqB,CAAyB,CAAC,KAAK,EAAE,EAAE,CAC7D,QAAQ,CAAC,KAA0C,CAAC,CACrD,CAAC;AACJ,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {\n  DuckDbSliceState,\n  getSqlErrorWithPointer,\n  joinStatements,\n  makeLimitQuery,\n  separateLastStatement,\n} from '@sqlrooms/duckdb';\nimport {\n  BaseRoomStoreState,\n  createSlice,\n  RoomShellSliceState,\n  StateCreator,\n  useBaseRoomShellStore,\n} from '@sqlrooms/room-shell';\nimport {\n  createDefaultSqlEditorConfig,\n  SqlEditorSliceConfig,\n} from '@sqlrooms/sql-editor-config';\nimport {generateUniqueName} from '@sqlrooms/utils';\nimport * as arrow from 'apache-arrow';\nimport {csvFormat} from 'd3-dsv';\nimport {saveAs} from 'file-saver';\nimport {produce} from 'immer';\n\nexport type QueryResult =\n  | {status: 'loading'; isBeingAborted?: boolean; controller: AbortController}\n  | {status: 'aborted'}\n  | {status: 'error'; error: string}\n  | {\n      status: 'success';\n      type: 'pragma' | 'explain' | 'select';\n      result: arrow.Table | undefined;\n      query: string;\n      lastQueryStatement: string;\n    }\n  | {\n      status: 'success';\n      type: 'exec';\n      query: string;\n      lastQueryStatement: string;\n    };\n\nexport function isQueryWithResult(\n  queryResult: QueryResult | undefined,\n): queryResult is QueryResult & {\n  status: 'success';\n  type: 'pragma' | 'explain' | 'select';\n} {\n  return (\n    queryResult?.status === 'success' &&\n    (queryResult.type === 'pragma' ||\n      queryResult.type === 'explain' ||\n      queryResult.type === 'select')\n  );\n}\n\nexport type SqlEditorSliceState = {\n  sqlEditor: {\n    config: SqlEditorSliceConfig;\n    // Runtime state\n    queryResult?: QueryResult;\n    /** @deprecated  */\n    selectedTable?: string;\n    /** @deprecated Use `useStoreWithSqlEditor((s) => s.db.isRefreshingTableSchemas)` instead. */\n    isTablesLoading: boolean;\n    /** @deprecated */\n    tablesError?: string;\n\n    queryResultLimit: number;\n    /** Options for the result limit dropdown */\n    queryResultLimitOptions: number[];\n\n    /**\n     * Set the config for the sql editor slice.\n     */\n    setConfig(config: SqlEditorSliceConfig): void;\n\n    /**\n     * Run the currently selected query.\n     */\n    parseAndRunQuery(query: string): Promise<void>;\n\n    /**\n     * Run the currently selected query.\n     */\n    parseAndRunCurrentQuery(): Promise<void>;\n\n    /**\n     * Abort the currently running query.\n     */\n    abortCurrentQuery(): void;\n\n    /**\n     * Export query results to CSV.\n     * @deprecated Use `useExportToCsv` from `@sqlrooms/duckdb` instead.\n     */\n    exportResultsToCsv(results: arrow.Table, filename?: string): void;\n\n    /**\n     * Create a new query tab.\n     * @param initialQuery - Optional initial query text.\n     */\n    createQueryTab(initialQuery?: string): {\n      id: string;\n      name: string;\n      query: string;\n    };\n\n    /**\n     * Delete a query tab.\n     * @param queryId - The ID of the query to delete.\n     */\n    deleteQueryTab(queryId: string): void;\n\n    /**\n     * Rename a query tab.\n     * @param queryId - The ID of the query to rename.\n     * @param newName - The new name for the query.\n     */\n    renameQueryTab(queryId: string, newName: string): void;\n\n    /**\n     * Close a query tab.\n     * @param queryId - The ID of the query to close.\n     */\n    closeQueryTab(queryId: string): void;\n\n    /**\n     * Open a closed tab id.\n     * @param queryId - The ID of the query to open.\n     */\n    openQueryTab(queryId: string): void;\n\n    /**\n     * Set the list of open tab IDs. Used for reordering or opening tabs.\n     * @param tabIds - The new list of open tab IDs.\n     */\n    setOpenTabs(tabIds: string[]): void;\n\n    /**\n     * Update the SQL text for a query.\n     * @param queryId - The ID of the query to update.\n     * @param queryText - The new SQL text.\n     */\n    updateQueryText(queryId: string, queryText: string): void;\n\n    /**\n     * Set the selected query tab.\n     * @param queryId - The ID of the query to select.\n     */\n    setSelectedQueryId(queryId: string): void;\n\n    /**\n     * Get the currently selected query's SQL text.\n     */\n    getCurrentQuery(): string;\n\n    /** @deprecated */\n    selectTable(table: string | undefined): void;\n\n    clearQueryResults(): void;\n\n    setQueryResultLimit(limit: number): void;\n  };\n};\n\nexport function createSqlEditorSlice({\n  config = createDefaultSqlEditorConfig(),\n  queryResultLimit = 100,\n  queryResultLimitOptions = [100, 500, 1000],\n}: {\n  config?: SqlEditorSliceConfig;\n  queryResultLimit?: number;\n  queryResultLimitOptions?: number[];\n} = {}): StateCreator<SqlEditorSliceState> {\n  return createSlice<\n    SqlEditorSliceState,\n    BaseRoomStoreState & DuckDbSliceState & SqlEditorSliceState\n  >((set, get) => {\n    return {\n      sqlEditor: {\n        config,\n        // Initialize runtime state\n        isTablesLoading: false,\n        queryResultLimit,\n        queryResultLimitOptions,\n\n        setConfig: (config) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config = config;\n            }),\n          );\n        },\n\n        exportResultsToCsv: (results, filename) => {\n          if (!results) return;\n          const blob = new Blob([csvFormat(results.toArray())], {\n            type: 'text/plain;charset=utf-8',\n          });\n          saveAs(blob, filename || `export-${createId().substring(0, 5)}.csv`);\n        },\n\n        createQueryTab: (initialQuery = '') => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const newQuery = {\n            id: createId(),\n            name: generateUniqueName(\n              'Untitled',\n              sqlEditorConfig.queries.map((q) => q.name),\n            ),\n            query: initialQuery,\n          };\n\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.queries.push(newQuery);\n              draft.sqlEditor.config.openTabs.push(newQuery.id);\n              draft.sqlEditor.config.selectedQueryId = newQuery.id;\n            }),\n          );\n\n          return newQuery;\n        },\n\n        deleteQueryTab: (queryId) => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const queries = sqlEditorConfig.queries;\n          const openTabs = sqlEditorConfig.openTabs;\n\n          if (queries.length <= 1) {\n            // Don't delete the last query\n            return;\n          }\n\n          const wasSelected = sqlEditorConfig.selectedQueryId === queryId;\n          const deletingOpenIndex = openTabs.indexOf(queryId);\n          const filteredQueries = queries.filter((q) => q.id !== queryId);\n\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.queries = filteredQueries;\n              draft.sqlEditor.config.openTabs = openTabs.filter(\n                (id) => id !== queryId,\n              );\n\n              // If we deleted the selected query, select another one\n              if (wasSelected) {\n                const newOpenTabs = draft.sqlEditor.config.openTabs;\n                const remainingQueries = draft.sqlEditor.config.queries;\n\n                if (newOpenTabs.length > 0) {\n                  // Select from remaining open tabs\n                  const newIndex =\n                    deletingOpenIndex === 0\n                      ? 0\n                      : Math.min(deletingOpenIndex - 1, newOpenTabs.length - 1);\n                  const newSelectedId = newOpenTabs[newIndex];\n                  if (newSelectedId) {\n                    draft.sqlEditor.config.selectedQueryId = newSelectedId;\n                  }\n                } else if (remainingQueries.length > 0) {\n                  // No open tabs left, open a closed query\n                  const queryToOpen = remainingQueries[0];\n                  if (queryToOpen) {\n                    draft.sqlEditor.config.openTabs.push(queryToOpen.id);\n                    draft.sqlEditor.config.selectedQueryId = queryToOpen.id;\n                  }\n                }\n              }\n            }),\n          );\n        },\n\n        renameQueryTab: (queryId, newName) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const query = draft.sqlEditor.config.queries.find(\n                (q) => q.id === queryId,\n              );\n              if (query) {\n                query.name = newName || query.name;\n              }\n            }),\n          );\n        },\n\n        closeQueryTab: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.openTabs =\n                draft.sqlEditor.config.openTabs.filter((id) => id !== queryId);\n            }),\n          );\n        },\n\n        openQueryTab: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              if (!draft.sqlEditor.config.openTabs.includes(queryId)) {\n                draft.sqlEditor.config.openTabs.push(queryId);\n              }\n              draft.sqlEditor.config.selectedQueryId = queryId;\n            }),\n          );\n        },\n\n        setOpenTabs: (tabIds) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.openTabs = tabIds;\n            }),\n          );\n        },\n\n        updateQueryText: (queryId, queryText) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const query = draft.sqlEditor.config.queries.find(\n                (q) => q.id === queryId,\n              );\n              if (query) {\n                query.query = queryText;\n              }\n            }),\n          );\n        },\n\n        setSelectedQueryId: (queryId) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.config.selectedQueryId = queryId;\n            }),\n          );\n        },\n\n        getCurrentQuery: () => {\n          const sqlEditorConfig = get().sqlEditor.config;\n          const selectedId = sqlEditorConfig.selectedQueryId;\n          const query = sqlEditorConfig.queries.find(\n            (q) => q.id === selectedId,\n          );\n          return query?.query || '';\n        },\n\n        /** @deprecated */\n        selectTable: (table) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.selectedTable = table;\n            }),\n          );\n        },\n\n        clearQueryResults: () => {\n          // Update state without using Immer for the Table type\n          set((state) => ({\n            ...state,\n            sqlEditor: {\n              ...state.sqlEditor,\n              queryResults: null,\n              queryError: undefined,\n            },\n          }));\n        },\n\n        parseAndRunCurrentQuery: async (): Promise<void> =>\n          get().sqlEditor.parseAndRunQuery(get().sqlEditor.getCurrentQuery()),\n\n        abortCurrentQuery: () => {\n          const currentResult = get().sqlEditor.queryResult;\n          if (currentResult?.status === 'loading' && currentResult.controller) {\n            currentResult.controller.abort();\n          }\n\n          set((state) =>\n            produce(state, (draft) => {\n              if (draft.sqlEditor.queryResult?.status === 'loading') {\n                draft.sqlEditor.queryResult.isBeingAborted = true;\n              }\n            }),\n          );\n        },\n\n        setQueryResultLimit: (limit) => {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.queryResultLimit = limit;\n            }),\n          );\n        },\n\n        parseAndRunQuery: async (query): Promise<void> => {\n          if (get().sqlEditor.queryResult?.status === 'loading') {\n            throw new Error('Query already running');\n          }\n          if (!query.trim()) {\n            return;\n          }\n\n          // Create abort controller for this query execution\n          const queryController = new AbortController();\n\n          // First update loading state and clear results\n          set((state) =>\n            produce(state, (draft) => {\n              draft.sqlEditor.selectedTable = undefined;\n              draft.sqlEditor.queryResult = {\n                status: 'loading',\n                isBeingAborted: false,\n                controller: queryController,\n              };\n            }),\n          );\n\n          let queryResult: QueryResult;\n          try {\n            const connector = await get().db.getConnector();\n            const signal = queryController.signal;\n\n            const {precedingStatements, lastStatement: lastQueryStatement} =\n              separateLastStatement(query);\n            const hasMultipleStatements = precedingStatements.length > 0;\n\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n\n            const parsedLastStatement =\n              await get().db.sqlSelectToJson(lastQueryStatement);\n\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n\n            const isValidSelectQuery = !parsedLastStatement.error;\n\n            if (isValidSelectQuery) {\n              // Add limit to the last statement\n              const limitedLastStatement = makeLimitQuery(lastQueryStatement, {\n                sanitize: false, // should already be sanitized\n                limit: get().sqlEditor.queryResultLimit,\n              });\n              const queryWithLimit = joinStatements(\n                precedingStatements,\n                limitedLastStatement,\n              );\n              const result = await connector.query(queryWithLimit, {signal});\n              queryResult = {\n                status: 'success',\n                type: 'select',\n                query,\n                lastQueryStatement,\n                result,\n              };\n            } else {\n              // Run the complete query as it is\n              if (\n                parsedLastStatement.error &&\n                parsedLastStatement.error_type !== 'not implemented'\n              ) {\n                throw (\n                  `${parsedLastStatement.error_type} ${parsedLastStatement.error_subtype}: ${parsedLastStatement.error_message}` +\n                  `\\n${getSqlErrorWithPointer(lastQueryStatement, Number(parsedLastStatement.position)).formatted}`\n                );\n              }\n\n              const result = await connector.query(query, {signal});\n              // EXPLAIN and PRAGMA are not detected as select queries\n              // and we cannot wrap them in a SELECT * FROM,\n              // but we can still execute them and return the result\n              if (/^(EXPLAIN)/i.test(lastQueryStatement)) {\n                queryResult = {\n                  status: 'success',\n                  type: 'explain',\n                  query,\n                  lastQueryStatement,\n                  result,\n                };\n              } else if (/^(PRAGMA)/i.test(lastQueryStatement)) {\n                queryResult = {\n                  status: 'success',\n                  type: 'pragma',\n                  query,\n                  lastQueryStatement,\n                  result,\n                };\n              } else {\n                queryResult = {\n                  status: 'success',\n                  type: 'exec',\n                  query,\n                  lastQueryStatement,\n                };\n              }\n            }\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n            // Refresh table schemas if there are multiple statements or if the\n            // last statement is not a select query\n            if (hasMultipleStatements || !isValidSelectQuery) {\n              get().db.refreshTableSchemas();\n            }\n            if (signal.aborted) {\n              throw new Error('Query aborted');\n            }\n          } catch (e) {\n            console.error(e);\n            const errorMessage = e instanceof Error ? e.message : String(e);\n            if (\n              errorMessage === 'Query aborted' ||\n              queryController.signal.aborted\n            ) {\n              queryResult = {status: 'aborted'};\n            } else {\n              queryResult = {\n                status: 'error',\n                error: errorMessage,\n              };\n            }\n          }\n\n          set((state) => ({\n            ...state,\n            sqlEditor: {...state.sqlEditor, queryResult},\n          }));\n        },\n      },\n    } satisfies SqlEditorSliceState;\n  });\n}\n\ntype RoomStateWithSqlEditor = RoomShellSliceState & SqlEditorSliceState;\n\nexport function useStoreWithSqlEditor<T>(\n  selector: (state: RoomStateWithSqlEditor) => T,\n): T {\n  return useBaseRoomShellStore<RoomShellSliceState, T>((state) =>\n    selector(state as unknown as RoomStateWithSqlEditor),\n  );\n}\n"]}

package/dist/SqlMonacoEditor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SqlMonacoEditor.d.ts","sourceRoot":"","sources":["../src/SqlMonacoEditor.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAgD,MAAM,OAAO,CAAC;AAErE,OAAO,KAAK,EAAC,iBAAiB,EAAC,MAAM,yBAAyB,CAAC;AAQ/D,OAAO,KAAK,EAAC,SAAS,EAAE,eAAe,EAAC,MAAM,kBAAkB,CAAC;AAGjE,MAAM,WAAW,oBACf,SAAQ,IAAI,CAAC,iBAAiB,EAAE,UAAU,CAAC;IAC3C,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,CAAC,EAAE,SAAS,EAAE,CAAC;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,MAAM;QACvB,YAAY,EAAE,SAAS,EAAE,CAAC;KAC3B,CAAC;CACH;AAQD;;;GAGG;AACH,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAyQ1D,CAAC"}
+{"version":3,"file":"SqlMonacoEditor.d.ts","sourceRoot":"","sources":["../src/SqlMonacoEditor.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAgD,MAAM,OAAO,CAAC;AAErE,OAAO,KAAK,EAAC,iBAAiB,EAAC,MAAM,yBAAyB,CAAC;AAQ/D,OAAO,KAAK,EAAC,SAAS,EAAE,eAAe,EAAC,MAAM,kBAAkB,CAAC;AAGjE,MAAM,WAAW,oBAAqB,SAAQ,IAAI,CAChD,iBAAiB,EACjB,UAAU,CACX;IACC,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,CAAC,EAAE,SAAS,EAAE,CAAC;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,MAAM;QACvB,YAAY,EAAE,SAAS,EAAE,CAAC;KAC3B,CAAC;CACH;AAQD;;;GAGG;AACH,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAgR1D,CAAC"}

package/dist/SqlMonacoEditor.js

@@ -18,6 +18,12 @@ export const SqlMonacoEditor = ({ connector, customKeywords = [], customFunction
     const editorRef = useRef(null);
     const monacoRef = useRef(null);
     const disposableRef = useRef(null);
+    // Store getLatestSchemas in a ref to avoid re-registering completion provider
+    // when the callback reference changes (e.g., non-memoized inline functions)
+    const getLatestSchemasRef = useRef(getLatestSchemas);
+    useEffect(() => {
+        getLatestSchemasRef.current = getLatestSchemas;
+    }, [getLatestSchemas]);
     // Function to register the completion provider
     const registerCompletionProvider = useCallback(() => {
         if (!editorRef.current || !monacoRef.current)
@@ -34,8 +40,8 @@ export const SqlMonacoEditor = ({ connector, customKeywords = [], customFunction
                 try {
                     // Get the latest schemas if the callback is provided
                     let currentSchemas = tableSchemas;
-                    if (getLatestSchemas) {
-                        const latest = getLatestSchemas();
+                    if (getLatestSchemasRef.current) {
+                        const latest = getLatestSchemasRef.current();
                         currentSchemas = latest.tableSchemas;
                     }
                     const suggestions = [];
@@ -176,7 +182,7 @@ export const SqlMonacoEditor = ({ connector, customKeywords = [], customFunction
         });
         // Store the disposable to clean up later
         disposableRef.current = disposable;
-    }, [customKeywords, customFunctions, tableSchemas, getLatestSchemas]);
+    }, [connector, customKeywords, customFunctions, tableSchemas]);
     // Re-register completion provider when tableSchemas change
     useEffect(() => {
         if (editorRef.current && monacoRef.current) {

package/dist/SqlMonacoEditor.js.map

@@ -1 +1 @@
-{"version":3,"file":"SqlMonacoEditor.js","sourceRoot":"","sources":["../src/SqlMonacoEditor.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAC,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAC,MAAM,OAAO,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,yBAAyB,CAAC;AAIrD,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,0BAA0B,GAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAC,EAAE,EAAC,MAAM,cAAc,CAAC;AAChC,OAAO,EAAC,sBAAsB,EAAC,MAAM,iCAAiC,CAAC;AAyBvE,MAAM,cAAc,GAAiC;IACnD,aAAa,EAAE,IAAI;IACnB,YAAY,EAAE,IAAI;IAClB,QAAQ,EAAE,IAAI;CACf,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,eAAe,GAAmC,CAAC,EAC9D,SAAS,EACT,cAAc,GAAG,EAAE,EACnB,eAAe,GAAG,EAAE,EACpB,YAAY,GAAG,EAAE,EACjB,gBAAgB,EAChB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,EACb,EAAE,EAAE;IACH,wCAAwC;IACxC,MAAM,SAAS,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IACpC,MAAM,SAAS,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IACpC,MAAM,aAAa,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IAExC,+CAA+C;IAC/C,MAAM,0BAA0B,GAAG,WAAW,CAAC,GAAG,EAAE;QAClD,IAAI,CAAC,SAAS,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO;YAAE,OAAO;QAErD,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;QAEjC,yCAAyC;QACzC,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;YAC1B,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAClC,CAAC;QAED,mCAAmC;QACnC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC,8BAA8B,CAAC,KAAK,EAAE;YACxE,iBAAiB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;YAC5C,sBAAsB,EAAE,KAAK,EAAE,KAAU,EAAE,QAAa,EAAE,EAAE;gBAC1D,IAAI,CAAC;oBACH,qDAAqD;oBACrD,IAAI,cAAc,GAAG,YAAY,CAAC;oBAElC,IAAI,gBAAgB,EAAE,CAAC;wBACrB,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;wBAClC,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC;oBACvC,CAAC;oBAED,MAAM,WAAW,GAAsC,EAAE,CAAC;oBAC1D,MAAM,IAAI,GAAG,KAAK,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBAClD,MAAM,KAAK,GAAG;wBACZ,eAAe,EAAE,QAAQ,CAAC,UAAU;wBACpC,aAAa,EAAE,QAAQ,CAAC,UAAU;wBAClC,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;qBAC1B,CAAC;oBAEF,sDAAsD;oBACtD,MAAM,WAAW,GAAG,KAAK,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAC9D,MAAM,gBAAgB,GAAG,WAAW;yBACjC,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;yBACjC,IAAI,EAAE;yBACN,WAAW,EAAE,CAAC;oBAEjB,uFAAuF;oBACvF,MAAM,cAAc,GAAG,wCAAwC,CAAC,IAAI,CAClE,gBAAgB,CACjB,CAAC;oBAEF,gFAAgF;oBAChF,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAE/D,kDAAkD;oBAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,cAAc,CAAC,CAAC;oBACzD,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;oBAE5D,yDAAyD;oBACzD,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;4BAC3B,WAAW,CAAC,IAAI,CAAC;gCACf,KAAK,EAAE,OAAO;gCACd,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO;gCACjD,UAAU,EAAE,OAAO;gCACnB,KAAK,EAAE,KAAK;gCACZ,MAAM,EAAE,SAAS;gCACjB,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,OAAO,EAAE,kCAAkC;6BAC7F,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;oBACL,CAAC;oBAED,0DAA0D;oBAC1D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;4BACzB,WAAW,CAAC,IAAI,CAAC;gCACf,KAAK,EAAE,IAAI;gCACX,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ;gCAClD,UAAU,EAAE,IAAI;gCAChB,KAAK,EAAE,KAAK;gCACZ,MAAM,EAAE,UAAU;gCAClB,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,EAAE,kCAAkC;6BACvF,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;wBACH,IAAI,SAAS,EAAE,CAAC;4BACd,MAAM,mBAAmB,GAAG,MAAM,sBAAsB,CACtD,SAAS,EACT,IAAI,CAAC,IAAI,CACV,CAAC;4BACF,KAAK,MAAM,EAAC,IAAI,EAAE,aAAa,EAAC,IAAI,mBAAmB,EAAE,CAAC;gCACxD,WAAW,CAAC,IAAI,CAAC;oCACf,KAAK,EAAE,IAAI;oCACX,UAAU,EAAE,IAAI;oCAChB,aAAa,EAAE;wCACb,KAAK,EAAE,aAAa;wCACpB,SAAS,EAAE,IAAI;wCACf,WAAW,EAAE,IAAI;qCAClB;oCACD,KAAK,EAAE,KAAK;oCACZ,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ;oCAClD,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,EAAE,kCAAkC;iCACvF,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,gDAAgD;oBAChD,cAAc,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;wBAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;wBAElC,uBAAuB;wBACvB,WAAW,CAAC,IAAI,CAAC;4BACf,KAAK,EAAE,SAAS;4BAChB,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;4BAC/C,UAAU,EAAE,SAAS;4BACrB,KAAK,EAAE,KAAK;4BACZ,MAAM,EAAE,OAAO;4BACf,aAAa,EAAE;gCACb,KAAK,EAAE,UAAU,SAAS,EAAE;gCAC5B,SAAS,EAAE,IAAI;6BAChB;4BACD,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,EAAE,mCAAmC;yBAClG,CAAC,CAAC;wBAEH,+DAA+D;wBAC/D,IAAI,gBAAgB,GAAG,EAAE,CAAC;wBAC1B,IAAI,eAAe,EAAE,CAAC;4BACpB,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;4BACtD,IAAI,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gCACtB,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;4BAC9B,CAAC;wBACH,CAAC;wBAED,sEAAsE;wBACtE,IAAI,CAAC,eAAe,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;4BACvD,yBAAyB;4BACzB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gCAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;gCAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;gCAE/B,WAAW,CAAC,IAAI,CAAC;oCACf,KAAK,EAAE,UAAU;oCACjB,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;oCAC/C,UAAU,EAAE,UAAU;oCACtB,KAAK,EAAE,KAAK;oCACZ,MAAM,EAAE,WAAW,UAAU,GAAG;oCAChC,aAAa,EAAE;wCACb,KAAK,EAAE,qBAAqB,SAAS,EAAE;wCACvC,SAAS,EAAE,IAAI;qCAChB;oCACD,QAAQ,EACN,eAAe,IAAI,gBAAgB,KAAK,SAAS;wCAC/C,CAAC,CAAC,GAAG,GAAG,UAAU;wCAClB,CAAC,CAAC,GAAG,GAAG,UAAU;iCACvB,CAAC,CAAC;gCAEH,+DAA+D;gCAC/D,IAAI,CAAC,eAAe,EAAE,CAAC;oCACrB,WAAW,CAAC,IAAI,CAAC;wCACf,KAAK,EAAE,GAAG,SAAS,IAAI,UAAU,EAAE;wCACnC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;wCAC/C,UAAU,EAAE,GAAG,SAAS,IAAI,UAAU,EAAE;wCACxC,KAAK,EAAE,KAAK;wCACZ,MAAM,EAAE,WAAW,UAAU,GAAG;wCAChC,aAAa,EAAE;4CACb,KAAK,EAAE,qBAAqB,SAAS,EAAE;4CACvC,SAAS,EAAE,IAAI;yCAChB;wCACD,QAAQ,EAAE,GAAG,GAAG,SAAS,GAAG,UAAU;qCACvC,CAAC,CAAC;gCACL,CAAC;4BACH,CAAC,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,OAAO;wBACL,WAAW;qBACZ,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;oBAC1D,OAAO,EAAC,WAAW,EAAE,EAAE,EAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,yCAAyC;QACzC,aAAa,CAAC,OAAO,GAAG,UAAU,CAAC;IACrC,CAAC,EAAE,CAAC,cAAc,EAAE,eAAe,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAEtE,2DAA2D;IAC3D,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YAC3C,0BAA0B,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,YAAY,EAAE,0BAA0B,CAAC,CAAC,CAAC;IAE/C,4DAA4D;IAC5D,MAAM,oBAAoB,GAAG,WAAW,CACtC,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;QACjB,mBAAmB;QACnB,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;QAC3B,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;QAE3B,kDAAkD;QAClD,IACE,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,EACvE,CAAC;YACD,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAC,EAAE,EAAE,KAAK,EAAC,CAAC,CAAC;QACzC,CAAC;QAED,kDAAkD;QAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,cAAc,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;QAE5D,iCAAiC;QACjC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,KAAK,EAAE;YAC/C,GAAG,0BAA0B;YAC7B,QAAQ;YACR,gBAAgB,EAAE,SAAS;SACrB,CAAC,CAAC,CAAC,mDAAmD;QAE9D,mCAAmC;QACnC,0BAA0B,EAAE,CAAC;QAE7B,mDAAmD;QACnD,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE;YACvB,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC1B,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,EACD,CAAC,cAAc,EAAE,eAAe,EAAE,OAAO,EAAE,0BAA0B,CAAC,CACvE,CAAC;IAEF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAiC,EAAE,CAAC,CAAC;QACnC,GAAG,cAAc;QACjB,GAAG,OAAO;KACX,CAAC,EACF,CAAC,OAAO,CAAC,CACV,CAAC;IACF,OAAO,CACL,KAAC,YAAY,IACX,QAAQ,EAAC,KAAK,EACd,OAAO,EAAE,oBAAoB,EAC7B,SAAS,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC,EAClC,OAAO,EAAE,eAAe,KACpB,SAAS,GACb,CACH,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import React, {useCallback, useEffect, useMemo, useRef} from 'react';\nimport {MonacoEditor} from '@sqlrooms/monaco-editor';\nimport type {MonacoEditorProps} from '@sqlrooms/monaco-editor';\nimport type {OnMount} from '@monaco-editor/react';\nimport type * as Monaco from 'monaco-editor';\nimport {\n  DUCKDB_KEYWORDS,\n  DUCKDB_FUNCTIONS,\n  SQL_LANGUAGE_CONFIGURATION,\n} from './constants/duckdb-dialect';\nimport type {DataTable, DuckDbConnector} from '@sqlrooms/duckdb';\nimport {cn} from '@sqlrooms/ui';\nimport {getFunctionSuggestions} from './constants/functionSuggestions';\nexport interface SqlMonacoEditorProps\n  extends Omit<MonacoEditorProps, 'language'> {\n  connector?: DuckDbConnector;\n  /**\n   * Custom SQL keywords to add to the completion provider\n   */\n  customKeywords?: string[];\n  /**\n   * Custom SQL functions to add to the completion provider\n   */\n  customFunctions?: string[];\n  /**\n   * Table schemas for autocompletion\n   */\n  tableSchemas?: DataTable[];\n  /**\n   * Callback to get the latest table schemas\n   * This is called from within provideCompletionItems to ensure we have the latest data\n   */\n  getLatestSchemas?: () => {\n    tableSchemas: DataTable[];\n  };\n}\n\nconst EDITOR_OPTIONS: MonacoEditorProps['options'] = {\n  formatOnPaste: true,\n  formatOnType: true,\n  wordWrap: 'on',\n};\n\n/**\n * A Monaco editor for editing SQL with DuckDB syntax highlighting and autocompletion\n * This is an internal component used by SqlEditor\n */\nexport const SqlMonacoEditor: React.FC<SqlMonacoEditorProps> = ({\n  connector,\n  customKeywords = [],\n  customFunctions = [],\n  tableSchemas = [],\n  getLatestSchemas,\n  onMount,\n  className,\n  options,\n  ...restProps\n}) => {\n  // Store references to editor and monaco\n  const editorRef = useRef<any>(null);\n  const monacoRef = useRef<any>(null);\n  const disposableRef = useRef<any>(null);\n\n  // Function to register the completion provider\n  const registerCompletionProvider = useCallback(() => {\n    if (!editorRef.current || !monacoRef.current) return;\n\n    const monaco = monacoRef.current;\n\n    // Dispose previous provider if it exists\n    if (disposableRef.current) {\n      disposableRef.current.dispose();\n    }\n\n    // Register SQL completion provider\n    const disposable = monaco.languages.registerCompletionItemProvider('sql', {\n      triggerCharacters: [' ', '.', ',', '(', '='],\n      provideCompletionItems: async (model: any, position: any) => {\n        try {\n          // Get the latest schemas if the callback is provided\n          let currentSchemas = tableSchemas;\n\n          if (getLatestSchemas) {\n            const latest = getLatestSchemas();\n            currentSchemas = latest.tableSchemas;\n          }\n\n          const suggestions: Monaco.languages.CompletionItem[] = [];\n          const word = model.getWordUntilPosition(position);\n          const range = {\n            startLineNumber: position.lineNumber,\n            endLineNumber: position.lineNumber,\n            startColumn: word.startColumn,\n            endColumn: word.endColumn,\n          };\n\n          // Get the text before the cursor to determine context\n          const lineContent = model.getLineContent(position.lineNumber);\n          const textBeforeCursor = lineContent\n            .substring(0, position.column - 1)\n            .trim()\n            .toLowerCase();\n\n          // Check if we're after a FROM, JOIN, or similar clause to prioritize table suggestions\n          const isTableContext = /\\b(from|join|into|update|table)\\s+\\w*$/.test(\n            textBeforeCursor,\n          );\n\n          // Check if we're after a table name and period to prioritize column suggestions\n          const isColumnContext = /\\b(\\w+)\\.\\w*$/.test(textBeforeCursor);\n\n          // Combine keywords and functions with custom ones\n          const keywords = [...DUCKDB_KEYWORDS, ...customKeywords];\n          const functions = [...DUCKDB_FUNCTIONS, ...customFunctions];\n\n          // Add keyword suggestions (if not in a specific context)\n          if (!isColumnContext) {\n            keywords.forEach((keyword) => {\n              suggestions.push({\n                label: keyword,\n                kind: monaco.languages.CompletionItemKind.Keyword,\n                insertText: keyword,\n                range: range,\n                detail: 'Keyword',\n                sortText: isTableContext ? 'z' + keyword : 'a' + keyword, // Lower priority in table context\n              });\n            });\n          }\n\n          // Add function suggestions (if not in a specific context)\n          if (!isColumnContext) {\n            functions.forEach((func) => {\n              suggestions.push({\n                label: func,\n                kind: monaco.languages.CompletionItemKind.Function,\n                insertText: func,\n                range: range,\n                detail: 'Function',\n                sortText: isTableContext ? 'z' + func : 'b' + func, // Lower priority in table context\n              });\n            });\n            if (connector) {\n              const functionSuggestions = await getFunctionSuggestions(\n                connector,\n                word.word,\n              );\n              for (const {name, documentation} of functionSuggestions) {\n                suggestions.push({\n                  label: name,\n                  insertText: name,\n                  documentation: {\n                    value: documentation,\n                    isTrusted: true,\n                    supportHtml: true,\n                  },\n                  range: range,\n                  kind: monaco.languages.CompletionItemKind.Function,\n                  sortText: isTableContext ? 'z' + name : 'b' + name, // Lower priority in table context\n                });\n              }\n            }\n          }\n\n          // Add table and column suggestions from schemas\n          currentSchemas.forEach((table) => {\n            const tableName = table.tableName;\n\n            // Add table suggestion\n            suggestions.push({\n              label: tableName,\n              kind: monaco.languages.CompletionItemKind.Class,\n              insertText: tableName,\n              range: range,\n              detail: 'Table',\n              documentation: {\n                value: `Table: ${tableName}`,\n                isTrusted: true,\n              },\n              sortText: isTableContext ? 'a' + tableName : 'c' + tableName, // Higher priority in table context\n            });\n\n            // Extract table name from context if we're in a column context\n            let contextTableName = '';\n            if (isColumnContext) {\n              const match = textBeforeCursor.match(/\\b(\\w+)\\.\\w*$/);\n              if (match && match[1]) {\n                contextTableName = match[1];\n              }\n            }\n\n            // Only add columns for the current table if we're in a column context\n            if (!isColumnContext || contextTableName === tableName) {\n              // Add column suggestions\n              table.columns.forEach((column) => {\n                const columnName = column.name;\n                const columnType = column.type;\n\n                suggestions.push({\n                  label: columnName,\n                  kind: monaco.languages.CompletionItemKind.Field,\n                  insertText: columnName,\n                  range: range,\n                  detail: `Column (${columnType})`,\n                  documentation: {\n                    value: `Column from table ${tableName}`,\n                    isTrusted: true,\n                  },\n                  sortText:\n                    isColumnContext && contextTableName === tableName\n                      ? 'a' + columnName\n                      : 'd' + columnName,\n                });\n\n                // Only add table.column suggestions if not in a column context\n                if (!isColumnContext) {\n                  suggestions.push({\n                    label: `${tableName}.${columnName}`,\n                    kind: monaco.languages.CompletionItemKind.Field,\n                    insertText: `${tableName}.${columnName}`,\n                    range: range,\n                    detail: `Column (${columnType})`,\n                    documentation: {\n                      value: `Column from table ${tableName}`,\n                      isTrusted: true,\n                    },\n                    sortText: 'e' + tableName + columnName,\n                  });\n                }\n              });\n            }\n          });\n\n          return {\n            suggestions,\n          };\n        } catch (error) {\n          console.error('Error in SQL completion provider:', error);\n          return {suggestions: []};\n        }\n      },\n    });\n\n    // Store the disposable to clean up later\n    disposableRef.current = disposable;\n  }, [customKeywords, customFunctions, tableSchemas, getLatestSchemas]);\n\n  // Re-register completion provider when tableSchemas change\n  useEffect(() => {\n    if (editorRef.current && monacoRef.current) {\n      registerCompletionProvider();\n    }\n  }, [tableSchemas, registerCompletionProvider]);\n\n  // Handle editor mounting to configure SQL language features\n  const handleEditorDidMount = useCallback<OnMount>(\n    (editor, monaco) => {\n      // Store references\n      editorRef.current = editor;\n      monacoRef.current = monaco;\n\n      // Register SQL language if not already registered\n      if (\n        !monaco.languages.getLanguages().some((lang: any) => lang.id === 'sql')\n      ) {\n        monaco.languages.register({id: 'sql'});\n      }\n\n      // Combine keywords and functions with custom ones\n      const keywords = [...DUCKDB_KEYWORDS, ...customKeywords];\n      const functions = [...DUCKDB_FUNCTIONS, ...customFunctions];\n\n      // Set the language configuration\n      monaco.languages.setMonarchTokensProvider('sql', {\n        ...SQL_LANGUAGE_CONFIGURATION,\n        keywords,\n        builtinFunctions: functions,\n      } as any); // Using 'as any' to bypass the type checking issue\n\n      // Register the completion provider\n      registerCompletionProvider();\n\n      // Store the disposable to clean up later if needed\n      editor.onDidDispose(() => {\n        if (disposableRef.current) {\n          disposableRef.current.dispose();\n        }\n      });\n\n      // Call the original onMount if provided\n      if (onMount) {\n        onMount(editor, monaco);\n      }\n    },\n    [customKeywords, customFunctions, onMount, registerCompletionProvider],\n  );\n\n  const combinedOptions = useMemo(\n    (): MonacoEditorProps['options'] => ({\n      ...EDITOR_OPTIONS,\n      ...options,\n    }),\n    [options],\n  );\n  return (\n    <MonacoEditor\n      language=\"sql\"\n      onMount={handleEditorDidMount}\n      className={cn('h-full', className)}\n      options={combinedOptions}\n      {...restProps}\n    />\n  );\n};\n"]}
+{"version":3,"file":"SqlMonacoEditor.js","sourceRoot":"","sources":["../src/SqlMonacoEditor.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAC,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAC,MAAM,OAAO,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,yBAAyB,CAAC;AAIrD,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,0BAA0B,GAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAC,EAAE,EAAC,MAAM,cAAc,CAAC;AAChC,OAAO,EAAC,sBAAsB,EAAC,MAAM,iCAAiC,CAAC;AA2BvE,MAAM,cAAc,GAAiC;IACnD,aAAa,EAAE,IAAI;IACnB,YAAY,EAAE,IAAI;IAClB,QAAQ,EAAE,IAAI;CACf,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,eAAe,GAAmC,CAAC,EAC9D,SAAS,EACT,cAAc,GAAG,EAAE,EACnB,eAAe,GAAG,EAAE,EACpB,YAAY,GAAG,EAAE,EACjB,gBAAgB,EAChB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,EACb,EAAE,EAAE;IACH,wCAAwC;IACxC,MAAM,SAAS,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IACpC,MAAM,SAAS,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IACpC,MAAM,aAAa,GAAG,MAAM,CAAM,IAAI,CAAC,CAAC;IAExC,8EAA8E;IAC9E,4EAA4E;IAC5E,MAAM,mBAAmB,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;IACrD,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,GAAG,gBAAgB,CAAC;IACjD,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAEvB,+CAA+C;IAC/C,MAAM,0BAA0B,GAAG,WAAW,CAAC,GAAG,EAAE;QAClD,IAAI,CAAC,SAAS,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO;YAAE,OAAO;QAErD,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;QAEjC,yCAAyC;QACzC,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;YAC1B,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAClC,CAAC;QAED,mCAAmC;QACnC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC,8BAA8B,CAAC,KAAK,EAAE;YACxE,iBAAiB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;YAC5C,sBAAsB,EAAE,KAAK,EAAE,KAAU,EAAE,QAAa,EAAE,EAAE;gBAC1D,IAAI,CAAC;oBACH,qDAAqD;oBACrD,IAAI,cAAc,GAAG,YAAY,CAAC;oBAElC,IAAI,mBAAmB,CAAC,OAAO,EAAE,CAAC;wBAChC,MAAM,MAAM,GAAG,mBAAmB,CAAC,OAAO,EAAE,CAAC;wBAC7C,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC;oBACvC,CAAC;oBAED,MAAM,WAAW,GAAsC,EAAE,CAAC;oBAC1D,MAAM,IAAI,GAAG,KAAK,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBAClD,MAAM,KAAK,GAAG;wBACZ,eAAe,EAAE,QAAQ,CAAC,UAAU;wBACpC,aAAa,EAAE,QAAQ,CAAC,UAAU;wBAClC,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;qBAC1B,CAAC;oBAEF,sDAAsD;oBACtD,MAAM,WAAW,GAAG,KAAK,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAC9D,MAAM,gBAAgB,GAAG,WAAW;yBACjC,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;yBACjC,IAAI,EAAE;yBACN,WAAW,EAAE,CAAC;oBAEjB,uFAAuF;oBACvF,MAAM,cAAc,GAAG,wCAAwC,CAAC,IAAI,CAClE,gBAAgB,CACjB,CAAC;oBAEF,gFAAgF;oBAChF,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAE/D,kDAAkD;oBAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,cAAc,CAAC,CAAC;oBACzD,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;oBAE5D,yDAAyD;oBACzD,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;4BAC3B,WAAW,CAAC,IAAI,CAAC;gCACf,KAAK,EAAE,OAAO;gCACd,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO;gCACjD,UAAU,EAAE,OAAO;gCACnB,KAAK,EAAE,KAAK;gCACZ,MAAM,EAAE,SAAS;gCACjB,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,OAAO,EAAE,kCAAkC;6BAC7F,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;oBACL,CAAC;oBAED,0DAA0D;oBAC1D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;4BACzB,WAAW,CAAC,IAAI,CAAC;gCACf,KAAK,EAAE,IAAI;gCACX,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ;gCAClD,UAAU,EAAE,IAAI;gCAChB,KAAK,EAAE,KAAK;gCACZ,MAAM,EAAE,UAAU;gCAClB,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,EAAE,kCAAkC;6BACvF,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;wBACH,IAAI,SAAS,EAAE,CAAC;4BACd,MAAM,mBAAmB,GAAG,MAAM,sBAAsB,CACtD,SAAS,EACT,IAAI,CAAC,IAAI,CACV,CAAC;4BACF,KAAK,MAAM,EAAC,IAAI,EAAE,aAAa,EAAC,IAAI,mBAAmB,EAAE,CAAC;gCACxD,WAAW,CAAC,IAAI,CAAC;oCACf,KAAK,EAAE,IAAI;oCACX,UAAU,EAAE,IAAI;oCAChB,aAAa,EAAE;wCACb,KAAK,EAAE,aAAa;wCACpB,SAAS,EAAE,IAAI;wCACf,WAAW,EAAE,IAAI;qCAClB;oCACD,KAAK,EAAE,KAAK;oCACZ,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ;oCAClD,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,EAAE,kCAAkC;iCACvF,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,gDAAgD;oBAChD,cAAc,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;wBAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;wBAElC,uBAAuB;wBACvB,WAAW,CAAC,IAAI,CAAC;4BACf,KAAK,EAAE,SAAS;4BAChB,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;4BAC/C,UAAU,EAAE,SAAS;4BACrB,KAAK,EAAE,KAAK;4BACZ,MAAM,EAAE,OAAO;4BACf,aAAa,EAAE;gCACb,KAAK,EAAE,UAAU,SAAS,EAAE;gCAC5B,SAAS,EAAE,IAAI;6BAChB;4BACD,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,EAAE,mCAAmC;yBAClG,CAAC,CAAC;wBAEH,+DAA+D;wBAC/D,IAAI,gBAAgB,GAAG,EAAE,CAAC;wBAC1B,IAAI,eAAe,EAAE,CAAC;4BACpB,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;4BACtD,IAAI,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gCACtB,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;4BAC9B,CAAC;wBACH,CAAC;wBAED,sEAAsE;wBACtE,IAAI,CAAC,eAAe,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;4BACvD,yBAAyB;4BACzB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gCAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;gCAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;gCAE/B,WAAW,CAAC,IAAI,CAAC;oCACf,KAAK,EAAE,UAAU;oCACjB,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;oCAC/C,UAAU,EAAE,UAAU;oCACtB,KAAK,EAAE,KAAK;oCACZ,MAAM,EAAE,WAAW,UAAU,GAAG;oCAChC,aAAa,EAAE;wCACb,KAAK,EAAE,qBAAqB,SAAS,EAAE;wCACvC,SAAS,EAAE,IAAI;qCAChB;oCACD,QAAQ,EACN,eAAe,IAAI,gBAAgB,KAAK,SAAS;wCAC/C,CAAC,CAAC,GAAG,GAAG,UAAU;wCAClB,CAAC,CAAC,GAAG,GAAG,UAAU;iCACvB,CAAC,CAAC;gCAEH,+DAA+D;gCAC/D,IAAI,CAAC,eAAe,EAAE,CAAC;oCACrB,WAAW,CAAC,IAAI,CAAC;wCACf,KAAK,EAAE,GAAG,SAAS,IAAI,UAAU,EAAE;wCACnC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,KAAK;wCAC/C,UAAU,EAAE,GAAG,SAAS,IAAI,UAAU,EAAE;wCACxC,KAAK,EAAE,KAAK;wCACZ,MAAM,EAAE,WAAW,UAAU,GAAG;wCAChC,aAAa,EAAE;4CACb,KAAK,EAAE,qBAAqB,SAAS,EAAE;4CACvC,SAAS,EAAE,IAAI;yCAChB;wCACD,QAAQ,EAAE,GAAG,GAAG,SAAS,GAAG,UAAU;qCACvC,CAAC,CAAC;gCACL,CAAC;4BACH,CAAC,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,OAAO;wBACL,WAAW;qBACZ,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;oBAC1D,OAAO,EAAC,WAAW,EAAE,EAAE,EAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,yCAAyC;QACzC,aAAa,CAAC,OAAO,GAAG,UAAU,CAAC;IACrC,CAAC,EAAE,CAAC,SAAS,EAAE,cAAc,EAAE,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;IAE/D,2DAA2D;IAC3D,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YAC3C,0BAA0B,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,YAAY,EAAE,0BAA0B,CAAC,CAAC,CAAC;IAE/C,4DAA4D;IAC5D,MAAM,oBAAoB,GAAG,WAAW,CACtC,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;QACjB,mBAAmB;QACnB,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;QAC3B,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;QAE3B,kDAAkD;QAClD,IACE,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,EACvE,CAAC;YACD,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAC,EAAE,EAAE,KAAK,EAAC,CAAC,CAAC;QACzC,CAAC;QAED,kDAAkD;QAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,cAAc,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;QAE5D,iCAAiC;QACjC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,KAAK,EAAE;YAC/C,GAAG,0BAA0B;YAC7B,QAAQ;YACR,gBAAgB,EAAE,SAAS;SACrB,CAAC,CAAC,CAAC,mDAAmD;QAE9D,mCAAmC;QACnC,0BAA0B,EAAE,CAAC;QAE7B,mDAAmD;QACnD,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE;YACvB,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC1B,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,EACD,CAAC,cAAc,EAAE,eAAe,EAAE,OAAO,EAAE,0BAA0B,CAAC,CACvE,CAAC;IAEF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAiC,EAAE,CAAC,CAAC;QACnC,GAAG,cAAc;QACjB,GAAG,OAAO;KACX,CAAC,EACF,CAAC,OAAO,CAAC,CACV,CAAC;IACF,OAAO,CACL,KAAC,YAAY,IACX,QAAQ,EAAC,KAAK,EACd,OAAO,EAAE,oBAAoB,EAC7B,SAAS,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC,EAClC,OAAO,EAAE,eAAe,KACpB,SAAS,GACb,CACH,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import React, {useCallback, useEffect, useMemo, useRef} from 'react';\nimport {MonacoEditor} from '@sqlrooms/monaco-editor';\nimport type {MonacoEditorProps} from '@sqlrooms/monaco-editor';\nimport type {OnMount} from '@monaco-editor/react';\nimport type * as Monaco from 'monaco-editor';\nimport {\n  DUCKDB_KEYWORDS,\n  DUCKDB_FUNCTIONS,\n  SQL_LANGUAGE_CONFIGURATION,\n} from './constants/duckdb-dialect';\nimport type {DataTable, DuckDbConnector} from '@sqlrooms/duckdb';\nimport {cn} from '@sqlrooms/ui';\nimport {getFunctionSuggestions} from './constants/functionSuggestions';\nexport interface SqlMonacoEditorProps extends Omit<\n  MonacoEditorProps,\n  'language'\n> {\n  connector?: DuckDbConnector;\n  /**\n   * Custom SQL keywords to add to the completion provider\n   */\n  customKeywords?: string[];\n  /**\n   * Custom SQL functions to add to the completion provider\n   */\n  customFunctions?: string[];\n  /**\n   * Table schemas for autocompletion\n   */\n  tableSchemas?: DataTable[];\n  /**\n   * Callback to get the latest table schemas\n   * This is called from within provideCompletionItems to ensure we have the latest data\n   */\n  getLatestSchemas?: () => {\n    tableSchemas: DataTable[];\n  };\n}\n\nconst EDITOR_OPTIONS: MonacoEditorProps['options'] = {\n  formatOnPaste: true,\n  formatOnType: true,\n  wordWrap: 'on',\n};\n\n/**\n * A Monaco editor for editing SQL with DuckDB syntax highlighting and autocompletion\n * This is an internal component used by SqlEditor\n */\nexport const SqlMonacoEditor: React.FC<SqlMonacoEditorProps> = ({\n  connector,\n  customKeywords = [],\n  customFunctions = [],\n  tableSchemas = [],\n  getLatestSchemas,\n  onMount,\n  className,\n  options,\n  ...restProps\n}) => {\n  // Store references to editor and monaco\n  const editorRef = useRef<any>(null);\n  const monacoRef = useRef<any>(null);\n  const disposableRef = useRef<any>(null);\n\n  // Store getLatestSchemas in a ref to avoid re-registering completion provider\n  // when the callback reference changes (e.g., non-memoized inline functions)\n  const getLatestSchemasRef = useRef(getLatestSchemas);\n  useEffect(() => {\n    getLatestSchemasRef.current = getLatestSchemas;\n  }, [getLatestSchemas]);\n\n  // Function to register the completion provider\n  const registerCompletionProvider = useCallback(() => {\n    if (!editorRef.current || !monacoRef.current) return;\n\n    const monaco = monacoRef.current;\n\n    // Dispose previous provider if it exists\n    if (disposableRef.current) {\n      disposableRef.current.dispose();\n    }\n\n    // Register SQL completion provider\n    const disposable = monaco.languages.registerCompletionItemProvider('sql', {\n      triggerCharacters: [' ', '.', ',', '(', '='],\n      provideCompletionItems: async (model: any, position: any) => {\n        try {\n          // Get the latest schemas if the callback is provided\n          let currentSchemas = tableSchemas;\n\n          if (getLatestSchemasRef.current) {\n            const latest = getLatestSchemasRef.current();\n            currentSchemas = latest.tableSchemas;\n          }\n\n          const suggestions: Monaco.languages.CompletionItem[] = [];\n          const word = model.getWordUntilPosition(position);\n          const range = {\n            startLineNumber: position.lineNumber,\n            endLineNumber: position.lineNumber,\n            startColumn: word.startColumn,\n            endColumn: word.endColumn,\n          };\n\n          // Get the text before the cursor to determine context\n          const lineContent = model.getLineContent(position.lineNumber);\n          const textBeforeCursor = lineContent\n            .substring(0, position.column - 1)\n            .trim()\n            .toLowerCase();\n\n          // Check if we're after a FROM, JOIN, or similar clause to prioritize table suggestions\n          const isTableContext = /\\b(from|join|into|update|table)\\s+\\w*$/.test(\n            textBeforeCursor,\n          );\n\n          // Check if we're after a table name and period to prioritize column suggestions\n          const isColumnContext = /\\b(\\w+)\\.\\w*$/.test(textBeforeCursor);\n\n          // Combine keywords and functions with custom ones\n          const keywords = [...DUCKDB_KEYWORDS, ...customKeywords];\n          const functions = [...DUCKDB_FUNCTIONS, ...customFunctions];\n\n          // Add keyword suggestions (if not in a specific context)\n          if (!isColumnContext) {\n            keywords.forEach((keyword) => {\n              suggestions.push({\n                label: keyword,\n                kind: monaco.languages.CompletionItemKind.Keyword,\n                insertText: keyword,\n                range: range,\n                detail: 'Keyword',\n                sortText: isTableContext ? 'z' + keyword : 'a' + keyword, // Lower priority in table context\n              });\n            });\n          }\n\n          // Add function suggestions (if not in a specific context)\n          if (!isColumnContext) {\n            functions.forEach((func) => {\n              suggestions.push({\n                label: func,\n                kind: monaco.languages.CompletionItemKind.Function,\n                insertText: func,\n                range: range,\n                detail: 'Function',\n                sortText: isTableContext ? 'z' + func : 'b' + func, // Lower priority in table context\n              });\n            });\n            if (connector) {\n              const functionSuggestions = await getFunctionSuggestions(\n                connector,\n                word.word,\n              );\n              for (const {name, documentation} of functionSuggestions) {\n                suggestions.push({\n                  label: name,\n                  insertText: name,\n                  documentation: {\n                    value: documentation,\n                    isTrusted: true,\n                    supportHtml: true,\n                  },\n                  range: range,\n                  kind: monaco.languages.CompletionItemKind.Function,\n                  sortText: isTableContext ? 'z' + name : 'b' + name, // Lower priority in table context\n                });\n              }\n            }\n          }\n\n          // Add table and column suggestions from schemas\n          currentSchemas.forEach((table) => {\n            const tableName = table.tableName;\n\n            // Add table suggestion\n            suggestions.push({\n              label: tableName,\n              kind: monaco.languages.CompletionItemKind.Class,\n              insertText: tableName,\n              range: range,\n              detail: 'Table',\n              documentation: {\n                value: `Table: ${tableName}`,\n                isTrusted: true,\n              },\n              sortText: isTableContext ? 'a' + tableName : 'c' + tableName, // Higher priority in table context\n            });\n\n            // Extract table name from context if we're in a column context\n            let contextTableName = '';\n            if (isColumnContext) {\n              const match = textBeforeCursor.match(/\\b(\\w+)\\.\\w*$/);\n              if (match && match[1]) {\n                contextTableName = match[1];\n              }\n            }\n\n            // Only add columns for the current table if we're in a column context\n            if (!isColumnContext || contextTableName === tableName) {\n              // Add column suggestions\n              table.columns.forEach((column) => {\n                const columnName = column.name;\n                const columnType = column.type;\n\n                suggestions.push({\n                  label: columnName,\n                  kind: monaco.languages.CompletionItemKind.Field,\n                  insertText: columnName,\n                  range: range,\n                  detail: `Column (${columnType})`,\n                  documentation: {\n                    value: `Column from table ${tableName}`,\n                    isTrusted: true,\n                  },\n                  sortText:\n                    isColumnContext && contextTableName === tableName\n                      ? 'a' + columnName\n                      : 'd' + columnName,\n                });\n\n                // Only add table.column suggestions if not in a column context\n                if (!isColumnContext) {\n                  suggestions.push({\n                    label: `${tableName}.${columnName}`,\n                    kind: monaco.languages.CompletionItemKind.Field,\n                    insertText: `${tableName}.${columnName}`,\n                    range: range,\n                    detail: `Column (${columnType})`,\n                    documentation: {\n                      value: `Column from table ${tableName}`,\n                      isTrusted: true,\n                    },\n                    sortText: 'e' + tableName + columnName,\n                  });\n                }\n              });\n            }\n          });\n\n          return {\n            suggestions,\n          };\n        } catch (error) {\n          console.error('Error in SQL completion provider:', error);\n          return {suggestions: []};\n        }\n      },\n    });\n\n    // Store the disposable to clean up later\n    disposableRef.current = disposable;\n  }, [connector, customKeywords, customFunctions, tableSchemas]);\n\n  // Re-register completion provider when tableSchemas change\n  useEffect(() => {\n    if (editorRef.current && monacoRef.current) {\n      registerCompletionProvider();\n    }\n  }, [tableSchemas, registerCompletionProvider]);\n\n  // Handle editor mounting to configure SQL language features\n  const handleEditorDidMount = useCallback<OnMount>(\n    (editor, monaco) => {\n      // Store references\n      editorRef.current = editor;\n      monacoRef.current = monaco;\n\n      // Register SQL language if not already registered\n      if (\n        !monaco.languages.getLanguages().some((lang: any) => lang.id === 'sql')\n      ) {\n        monaco.languages.register({id: 'sql'});\n      }\n\n      // Combine keywords and functions with custom ones\n      const keywords = [...DUCKDB_KEYWORDS, ...customKeywords];\n      const functions = [...DUCKDB_FUNCTIONS, ...customFunctions];\n\n      // Set the language configuration\n      monaco.languages.setMonarchTokensProvider('sql', {\n        ...SQL_LANGUAGE_CONFIGURATION,\n        keywords,\n        builtinFunctions: functions,\n      } as any); // Using 'as any' to bypass the type checking issue\n\n      // Register the completion provider\n      registerCompletionProvider();\n\n      // Store the disposable to clean up later if needed\n      editor.onDidDispose(() => {\n        if (disposableRef.current) {\n          disposableRef.current.dispose();\n        }\n      });\n\n      // Call the original onMount if provided\n      if (onMount) {\n        onMount(editor, monaco);\n      }\n    },\n    [customKeywords, customFunctions, onMount, registerCompletionProvider],\n  );\n\n  const combinedOptions = useMemo(\n    (): MonacoEditorProps['options'] => ({\n      ...EDITOR_OPTIONS,\n      ...options,\n    }),\n    [options],\n  );\n  return (\n    <MonacoEditor\n      language=\"sql\"\n      onMount={handleEditorDidMount}\n      className={cn('h-full', className)}\n      options={combinedOptions}\n      {...restProps}\n    />\n  );\n};\n"]}