@quillsql/admin 1.6.0 → 1.6.2

package/dist/esm/forms/client_onboard/CreateVirtualTables.js

@@ -0,0 +1,388 @@
+import { createElement as _createElement } from "react";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useEffect, useRef, useState } from 'react';
+import { ButtonPrimitive, ModalPrimitive, SecondaryButtonPrimitive, TextInputPrimitive, } from '../../primitives';
+import { createVirtualTable, getQueryFromAiWithConnection, getVirtualTableData, } from '../../api/ConnectionClient';
+import VirtualTableTile from '../../components/VirtualTableTile';
+import SqlTextEditor from '../../components/SqlTextEditor';
+import { SchemaListComponent, Table } from '@quillsql/react';
+import { LoadingSpinner, QuillToolTip } from '../../components/UiComponents';
+import { getDuplicateColumns } from '../../public_components/VirtualTableManager';
+import { processColumnReference } from '../../utils/columnProcessing';
+import { useAdmin } from '../../AdminProvider';
+import { SINGLE_TENANT } from '../../utils/constants';
+import { Highlight, themes } from 'prism-react-renderer';
+export default function CreateVirtualTables({ containerStyle, header1Style, header2Style, environmentDetails, databaseSchema, completeOnboarding, client, state, tenantInit, isLoading, sqlTables, setSqlTables, }) {
+    const [queryInfo, setQueryInfo] = useState({
+        name: '',
+        query: '',
+    });
+    const [aiPrompt, setAIPrompt] = useState('');
+    const [errorInfo, setErrorInfo] = useState({ status: false, msg: '' });
+    const [sqlQueryError, setSqlQueryInfo] = useState({ status: false, msg: '' });
+    const [tableData, setTableData] = useState(undefined);
+    const [virtualTableModalOpen, setVirtualTableModalOpen] = useState(false);
+    const [openModalButtonLoading, setOpenModalButtonLoading] = useState(false);
+    const [askAIButtonLoading, setAskAIButtonLoading] = useState(false);
+    const [runQueryButtonLoading, setRunQueryButtonLoading] = useState(false);
+    const [failingTenant, setFailingTenant] = useState(undefined);
+    const [tableSearchQuery, setTableSearchQuery] = useState('');
+    const [displayedTableData, setDisplayedTableData] = useState(databaseSchema);
+    const middleColumn = useRef(null);
+    const { getToken, quillFetchWithToken } = useAdmin();
+    useEffect(() => {
+        const scrollToBottom = () => {
+            // Scroll to the bottom of the container (or page)
+            document
+                .getElementById('create-virtual-table')
+                ?.scrollIntoView({ behavior: 'smooth' });
+        };
+        // Ensuring that the scroll happens after layout updates
+        const timeoutId = setTimeout(scrollToBottom, 0);
+        return () => clearTimeout(timeoutId); // Cleanup the timeout if the component unmounts
+    }, []);
+    useEffect(() => {
+        if (!tableData || !middleColumn.current) {
+            return;
+        }
+        middleColumn.current.scrollTop = middleColumn.current.scrollHeight;
+    }, [tableData]);
+    const addVirtualTable = async (override) => {
+        const tenantField = tenantInit.ownerTenant?.tenantField ??
+            client.allTenantTypes?.[0]?.tenantField;
+        if (!tenantField) {
+            throw new Error('Your tenant is not set up correctly, please redo step 2');
+        }
+        const results = await createVirtualTable(queryInfo, client, tableData, sqlTables?.map((e) => {
+            return e.name;
+        }) ?? [], client.clientId, tenantField === SINGLE_TENANT || override ? [] : [tenantField], state.queryEndpoint, state.queryHeaders, getToken);
+        if (results.failingTenant) {
+            setFailingTenant(results.failingTenant);
+            return;
+        }
+        if (results.success && results.tables) {
+            setSqlQueryInfo({ msg: '', status: false });
+            setSqlTables(results.tables);
+            setQueryInfo({ name: '', query: '' });
+        }
+        else {
+            setSqlQueryInfo({ msg: results.message, status: true });
+        }
+        setTableData(undefined);
+    };
+    const handleClickSchemaItem = async (event) => {
+        const name = event.target.textContent;
+        if (!name) {
+            return;
+        }
+        let databaseType = 'postgresql';
+        if (environmentDetails && environmentDetails.type) {
+            databaseType = environmentDetails.type;
+        }
+        const newQuery = `SELECT * FROM ${processColumnReference(name, databaseType)};`;
+        if (!queryInfo.query || queryInfo.query.length === 0) {
+            setQueryInfo({ name: queryInfo.name, query: newQuery });
+        }
+        await navigator.clipboard.writeText(newQuery);
+    };
+    return (_jsxs("div", { style: {
+            ...containerStyle,
+            display: 'flex',
+            flexDirection: 'column',
+            minWidth: 900,
+            maxWidth: '95vw',
+            width: '100%',
+        }, id: "create-virtual-table", children: [_jsxs("div", { style: {
+                    padding: 20,
+                    paddingBottom: 12,
+                    height: 92,
+                    maxHeight: 92,
+                    minHeight: 92,
+                }, children: [_jsx("h1", { style: header1Style, children: "Create Virtual Tables" }), _jsx("h2", { style: { ...header2Style }, children: "Create a cleaned schema that makes it easier to write queries." })] }), _jsxs("div", { style: {
+                    display: 'flex',
+                    flexDirection: 'row',
+                    borderTop: 'solid 1px #e7e7e7',
+                    height: 'calc(100vh - 150px)',
+                }, children: [_jsxs("div", { style: {
+                            display: 'flex',
+                            flexDirection: 'column',
+                            height: '100%',
+                            paddingBottom: 10,
+                        }, children: [_jsx("div", { style: {
+                                    paddingLeft: '20px',
+                                    paddingRight: '30px',
+                                    height: 70,
+                                    display: 'flex',
+                                    alignItems: 'center',
+                                    paddingTop: 6,
+                                }, children: _jsx(TextInputPrimitive, { placeholder: "Filter tables...", 
+                                    // @ts-ignore
+                                    onChange: (e) => {
+                                        setTableSearchQuery(e.target.value);
+                                        setDisplayedTableData(databaseSchema.filter((table) => {
+                                            return (table.tableName
+                                                ?.toLowerCase()
+                                                ?.includes(e.target.value.toLowerCase().trim()) ||
+                                                table.columns.some((col) => col.field
+                                                    ?.toLowerCase()
+                                                    ?.includes(e.target.value.toLowerCase().trim())));
+                                        }));
+                                    }, value: tableSearchQuery, width: "200px" }) }), _jsx(SchemaListComponent, { schema: displayedTableData, theme: state.theme, width: '250px', loading: false, LoadingComponent: LoadingSpinner, onClick: handleClickSchemaItem, style: { overflowY: 'auto' } })] }), _jsxs("div", { id: "form-editor-table", ref: middleColumn, style: {
+                            display: 'flex',
+                            flexDirection: 'column',
+                            width: 'calc(100% - 500px)',
+                            minWidth: 300,
+                            // height: 'calc(100vh - 302px)',
+                            overflowY: 'scroll',
+                        }, children: [_jsxs("div", { style: {
+                                    display: 'flex',
+                                    flexDirection: 'row',
+                                    alignItems: 'center',
+                                    justifyContent: 'space-between',
+                                    gap: 12,
+                                    paddingLeft: 16,
+                                    paddingRight: 16,
+                                    height: 70,
+                                }, children: [_jsx(TextInputPrimitive, { onChange: (e) => {
+                                            setAIPrompt(e.target.value);
+                                        }, placeholder: 'Describe a query...', value: aiPrompt }), _jsx(QuillToolTip, { text: aiPrompt.trim().length > 500
+                                            ? 'Prompt must be less than 500 characters'
+                                            : '', enabled: aiPrompt.trim().length > 500, textStyle: {
+                                            maxWidth: '90px',
+                                            whiteSpace: 'normal',
+                                        }, displayBelow: true, children: _jsx(ButtonPrimitive, { label: 'Ask AI', isLoading: askAIButtonLoading, onClick: async () => {
+                                                setAskAIButtonLoading(true);
+                                                const results = await getQueryFromAiWithConnection(state.queryEndpoint, aiPrompt, client.clientId, getToken, client.schemaNames);
+                                                setQueryInfo({ name: queryInfo.name, query: results });
+                                                setAskAIButtonLoading(false);
+                                            }, style: { width: '90px' }, disabled: aiPrompt.trim().length > 500 }) })] }), _jsx("div", { "data-testid": "quill-sql-text-editor", children: _jsx(SqlTextEditor, { value: queryInfo.query, setValue: (e) => {
+                                        setQueryInfo({ name: queryInfo.name, query: e });
+                                    }, setEditorMounted: () => { }, containerStyle: {
+                                        paddingLeft: 16,
+                                        paddingRight: 16,
+                                        marginTop: 2,
+                                    }, schema: displayedTableData, databaseType: environmentDetails?.type, resizable: false }) }), _jsxs("div", { style: {
+                                    display: 'flex',
+                                    flexDirection: 'row',
+                                    paddingTop: 12,
+                                    paddingLeft: 16,
+                                    paddingRight: 16,
+                                    height: 70,
+                                }, children: [_jsx(ButtonPrimitive, { label: 'Run query', isLoading: runQueryButtonLoading, onClick: async () => {
+                                            setRunQueryButtonLoading(true);
+                                            const getSqlResults = await getVirtualTableData(client.clientId, queryInfo.query, environmentDetails.type, state.queryEndpoint, getToken);
+                                            if (!getSqlResults.success) {
+                                                setErrorInfo({ status: true, msg: getSqlResults.error });
+                                            }
+                                            else {
+                                                setErrorInfo({ status: false, msg: '' });
+                                                setTableData(getSqlResults);
+                                            }
+                                            setRunQueryButtonLoading(false);
+                                        } }), errorInfo.status ? (_jsx("h2", { style: {
+                                            ...header2Style,
+                                            color: 'red',
+                                            paddingLeft: '12px',
+                                            height: 40,
+                                            alignItems: 'center',
+                                            display: 'flex',
+                                            justifyContent: 'center',
+                                        }, children: errorInfo.msg })) : (_jsx("div", {}))] }), !errorInfo.status && tableData && (_jsxs("div", { style: {
+                                    paddingLeft: 16,
+                                    paddingRight: 16,
+                                    height: 'calc(100vh - 539px)',
+                                }, children: [_jsx(Table, { rows: tableData.rows, columns: tableData.fields, containerStyle: {
+                                            width: '100%',
+                                            height: 270,
+                                        }, hideCSVDownloadButton: true, isLoading: runQueryButtonLoading }), _jsxs("div", { style: {
+                                            width: '100%',
+                                            display: 'flex',
+                                            justifyContent: 'space-between',
+                                            alignItems: 'center',
+                                            height: 70,
+                                        }, children: [sqlQueryError.status ? (_jsx("h2", { style: {
+                                                    ...header2Style,
+                                                    color: 'red',
+                                                    paddingTop: '12px',
+                                                    // textTransform: 'capitalize',
+                                                }, children: sqlQueryError.msg })) : (_jsx("div", {})), _jsx("div", { "data-testid": "add-virtual-table-button", children: _jsx(ButtonPrimitive, { label: "Add Virtual Table", isLoading: openModalButtonLoading, disabled: !!failingTenant, onClick: async () => {
+                                                        setOpenModalButtonLoading(true);
+                                                        const duplicateColumns2 = getDuplicateColumns(tableData?.fields);
+                                                        if (duplicateColumns2.length > 0) {
+                                                            alert(`Ambiguous column names found. Either use aliases or remove the column to make every column name unique: \n\n${duplicateColumns2.join('\n')}`);
+                                                            return;
+                                                        }
+                                                        const clientId = client?.publicKey || client?.clientId;
+                                                        if (!clientId) {
+                                                            // Shouldn't have reached this point without a public key
+                                                            // If we were to call astify without a public key, the database type would be unknown
+                                                            return;
+                                                        }
+                                                        const localURL = client.queryEndpoint ?? state.queryEndpoint;
+                                                        const { data: queryAstRaw } = await quillFetchWithToken({
+                                                            client: {
+                                                                queryEndpoint: localURL,
+                                                                clientId,
+                                                                withCredentials: !!state.withCredentials,
+                                                                queryHeaders: state.queryHeaders,
+                                                            },
+                                                            task: 'astify',
+                                                            metadata: {
+                                                                query: queryInfo.query,
+                                                                clientId: client?.publicKey || client?.clientId,
+                                                                useNewNodeSql: true,
+                                                            },
+                                                            adminMode: true,
+                                                        });
+                                                        if (queryAstRaw.success !== false) {
+                                                            const queryAst = queryAstRaw.ast[0]
+                                                                ? queryAstRaw.ast[0]
+                                                                : queryAstRaw.ast;
+                                                            const queryName = queryAst.from
+                                                                .map((ast) => ast.table.toLowerCase())
+                                                                .join('_');
+                                                            setQueryInfo({
+                                                                name: queryName,
+                                                                query: queryInfo.query,
+                                                            });
+                                                        }
+                                                        setOpenModalButtonLoading(false);
+                                                        setVirtualTableModalOpen(true);
+                                                    } }) })] })] }))] }), _jsxs("div", { style: {
+                            width: '400px',
+                            // background: '#fafafa',
+                            borderLeft: 'solid 1px #e7e7e7',
+                            // height: '100%',
+                        }, children: [_jsx("h1", { style: {
+                                    ...header1Style,
+                                    color: '#212121',
+                                    paddingLeft: 16,
+                                    height: 70,
+                                    background: 'white',
+                                    borderBottom: 'solid 1px #e7e7e7',
+                                    display: 'flex',
+                                    flexDirection: 'column',
+                                    justifyContent: 'center',
+                                    alignItems: 'flex-start',
+                                    marginBottom: 0,
+                                }, children: "Virtual Tables" }), sqlTables?.length > 0 ? (_jsxs("div", { style: {
+                                    display: 'flex',
+                                    flexDirection: 'column',
+                                    gap: '10px',
+                                    overflow: 'scroll',
+                                    height: 'calc(100vh - 302px)',
+                                    paddingTop: '12px',
+                                    paddingLeft: '20px',
+                                    paddingRight: '20px',
+                                }, children: [sqlTables?.map((e) => {
+                                        return (_jsx("div", { children: _jsx(VirtualTableTile, { name: e.name, query: e.viewQuery, theme: state.theme }) }, e.name));
+                                    }), _jsx("div", { style: { height: 20 } })] })) : (_jsx("div", { style: { paddingLeft: 12, paddingRight: 12 }, children: _jsx("div", { style: {
+                                        display: 'flex',
+                                        justifyContent: 'center',
+                                        alignItems: 'center',
+                                        border: 'dashed 1px #C5C5C5',
+                                        marginTop: '12px',
+                                        borderRadius: 6,
+                                        height: 200,
+                                    }, children: _jsxs("h2", { style: {
+                                            ...header2Style,
+                                            display: 'flex',
+                                            justifyContent: 'center',
+                                            alignItems: 'center',
+                                            color: '#565656',
+                                            textAlign: 'center',
+                                        }, children: ['No virtual tables created yet.', _jsx("br", {}), 'Create one with the SQL editor.'] }) }) })), (sqlTables?.length ?? 0) > 0 ? (_jsx("div", { style: {
+                                    display: 'flex',
+                                    justifyContent: 'flex-end',
+                                    flexDirection: 'row',
+                                    alignItems: 'center',
+                                    height: 70,
+                                    paddingRight: 12,
+                                    background: 'white',
+                                    borderTop: 'solid 1px #e7e7e7',
+                                }, children: _jsx(ButtonPrimitive, { label: "Looks good", onClick: () => {
+                                        completeOnboarding();
+                                    }, style: {
+                                    // marginBottom: -8,
+                                    }, isLoading: isLoading }) })) : null] })] }), _jsx(TenantFieldModal, { isOpen: !!failingTenant, setClosed: () => setFailingTenant(undefined), tenant: failingTenant, onOverride: () => addVirtualTable(true) }), _jsx(VirtualTableModal, { isOpen: virtualTableModalOpen, setIsOpen: setVirtualTableModalOpen, queryInfo: queryInfo, setQueryInfo: setQueryInfo, addVirtualTable: () => addVirtualTable(false) })] }));
+}
+export function VirtualTableModal({ isOpen, setIsOpen, queryInfo, setQueryInfo, addVirtualTable, }) {
+    return (_jsx(ModalPrimitive, { isOpen: isOpen, onClose: () => {
+            setIsOpen(false);
+        }, children: _jsxs("div", { style: { display: 'flex', flexDirection: 'column', width: '100%' }, children: [_jsx("h1", { style: {
+                        fontSize: 24,
+                        fontWeight: 600,
+                        color: '#212121',
+                        paddingBottom: 20,
+                    }, children: "Add Virtual Table" }), _jsx("h3", { style: {
+                        fontWeight: '600',
+                        fontSize: '14px',
+                        marginBottom: '6px',
+                        color: '#212121',
+                    }, children: "Name" }), _jsx(TextInputPrimitive, { ref: (e) => {
+                        e?.focus();
+                    }, value: queryInfo.name, onChange: (e) => {
+                        setQueryInfo({ name: e.target.value, query: queryInfo.query });
+                    }, placeholder: "Name..." }), _jsx("h3", { style: {
+                        fontWeight: '600',
+                        fontSize: '14px',
+                        marginBottom: '6px',
+                        marginTop: '12px',
+                        color: '#212121',
+                    }, children: "Query" }), _jsx(Highlight, { theme: themes.nightOwlLight, code: queryInfo.query || '', language: "sql", children: ({ className, tokens, style, getLineProps, getTokenProps }) => (_jsx("pre", { className: className, style: {
+                            ...style,
+                            textAlign: 'left',
+                            padding: '1em',
+                            overflow: 'auto',
+                            height: 132,
+                            maxWidth: 460,
+                            borderRadius: 8,
+                            position: 'relative',
+                        }, children: tokens.map((line, i) => (_createElement("div", { ...getLineProps({ line }), key: i, style: {
+                                wordWrap: 'normal',
+                                whiteSpace: 'normal',
+                            } }, line.map((token, key) => (_jsx("span", { ...getTokenProps({ token }) }, key)))))) })) }), _jsxs("div", { style: {
+                        display: 'flex',
+                        flexDirection: 'row',
+                        gap: '12px',
+                        justifyContent: 'flex-end',
+                        marginTop: '12px',
+                    }, children: [_jsx(SecondaryButtonPrimitive, { label: "Cancel", onClick: () => setIsOpen(false) }), _jsx(ButtonPrimitive, { label: "Confirm", variant: "primary", onClick: () => {
+                                if (!/^[a-z0-9_]+$/.test(queryInfo.name)) {
+                                    alert('Names can only include lowercase letters, numbers, and underscores');
+                                    return;
+                                }
+                                addVirtualTable(false);
+                                setIsOpen(false);
+                            } })] })] }) }));
+}
+export function TenantFieldModal({ isOpen, setClosed, tenant, onOverride, }) {
+    return (_jsx(ModalPrimitive, { isOpen: isOpen, onClose: () => setClosed(), children: _jsxs("div", { style: { display: 'flex', flexDirection: 'column', width: '500px' }, children: [_jsx("h1", { style: {
+                        fontSize: 24,
+                        fontWeight: 600,
+                        color: '#212121',
+                        paddingBottom: 14,
+                    }, children: "Foreign Key Missing" }), _jsxs("div", { style: { marginBottom: '12px' }, children: [_jsx("span", { style: {
+                                color: '#212121',
+                                fontWeight: 600,
+                                backgroundColor: '#f7f7f7',
+                                padding: '2px 4px',
+                                borderRadius: 4,
+                                border: '1px solid #e0e0e0',
+                            }, children: tenant?.name }), "'s foreign key", ' ', _jsx("span", { style: {
+                                color: 'red',
+                                fontWeight: 600,
+                                backgroundColor: '#f7f7f7',
+                                padding: '2px 4px',
+                                borderRadius: 4,
+                                border: '1px solid #e0e0e0',
+                            }, children: tenant?.tenantField.replaceAll('"', '') }), ' ', "is not present in the current query to allow for filtering.", onOverride ? (_jsxs("span", { children: [' ', "Only override if this data is meant to be accessible by", ' ', _jsx("strong", { children: "every tenant" }), " (this is rare)"] })) : (_jsxs("span", { children: [' ', "If this tenant is not meant to filter this data, remove it from the owners dropdown"] }))] }), _jsx("br", {}), _jsxs("div", { style: {
+                        display: 'flex',
+                        flexDirection: 'row',
+                        gap: '12px',
+                        justifyContent: 'flex-end',
+                    }, children: [onOverride && (_jsx(SecondaryButtonPrimitive, { label: "Override", onClick: () => {
+                                onOverride();
+                                setClosed();
+                            } })), _jsx(ButtonPrimitive, { label: "Got It", onClick: () => setClosed() })] })] }) }));
+}

package/dist/esm/forms/client_onboard/__tests__/ConnectDatabase.test.d.ts

@@ -0,0 +1,2 @@
+import '@testing-library/jest-dom';
+//# sourceMappingURL=ConnectDatabase.test.d.ts.map

package/dist/esm/forms/client_onboard/__tests__/ConnectDatabase.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConnectDatabase.test.d.ts","sourceRoot":"","sources":["../../../../../src/forms/client_onboard/__tests__/ConnectDatabase.test.tsx"],"names":[],"mappings":"AAEA,OAAO,2BAA2B,CAAC"}