@quillsql/admin 1.7.0 → 1.7.2

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

@@ -1,426 +0,0 @@
-import { createElement as _createElement } from "react";
-import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } 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';
-import { useLongLoading } from '../../hooks/useLongLoading';
-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);
-    useLongLoading(openModalButtonLoading, {
-        origin: 'CreateVirtualTables',
-        loadDescription: 'Opening virtual table modal',
-    });
-    useLongLoading(askAIButtonLoading, {
-        origin: 'CreateVirtualTables',
-        loadDescription: 'Asking AI',
-    });
-    useLongLoading(runQueryButtonLoading, {
-        origin: 'CreateVirtualTables',
-        loadDescription: 'Running query',
-    });
-    const [failingTenant, setFailingTenant] = useState(undefined);
-    const [failingTenantErrorType, setFailingTenantErrorType] = useState(undefined);
-    const [tableSearchQuery, setTableSearchQuery] = useState('');
-    const [displayedTableData, setDisplayedTableData] = useState(databaseSchema);
-    const middleColumn = useRef(null);
-    const { getToken, quillFetchWithToken, eventTracking } = 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);
-            setFailingTenantErrorType(results.failingTenantErrorType);
-            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.isSelfHosted, databaseSchema, state.queryEndpoint, aiPrompt, client.clientId, getToken, client.schemaNames, state.queryHeaders);
-                                                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: databaseSchema, 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, eventTracking, client.queryHeaders);
-                                            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')}`);
-                                                            setOpenModalButtonLoading(false);
-                                                            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), errorType: failingTenantErrorType }), _jsx(VirtualTableModal, { isOpen: virtualTableModalOpen, setIsOpen: setVirtualTableModalOpen, queryInfo: queryInfo, setQueryInfo: setQueryInfo, addVirtualTable: () => addVirtualTable(false) })] }));
-}
-export function VirtualTableModal({ isOpen, setIsOpen, queryInfo, setQueryInfo, addVirtualTable, }) {
-    const { state: { theme }, } = useAdmin();
-    return (_jsx(ModalPrimitive, { isOpen: isOpen, onClose: () => {
-            setIsOpen(false);
-        }, children: _jsxs("div", { style: { display: 'flex', flexDirection: 'column', width: '100%' }, children: [_jsx("h1", { style: {
-                        fontSize: 24,
-                        fontFamily: theme.fontFamily,
-                        fontWeight: 600,
-                        color: '#212121',
-                        paddingBottom: 20,
-                    }, children: "Add Virtual Table" }), _jsx("h3", { style: {
-                        fontWeight: '600',
-                        fontFamily: theme.fontFamily,
-                        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',
-                        fontFamily: theme.fontFamily,
-                        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, errorType = 'missing', }) {
-    const { state: { theme }, } = useAdmin();
-    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,
-                        fontFamily: theme.fontFamily,
-                    }, children: errorType === 'missing'
-                        ? 'Foreign Key Missing'
-                        : 'Foreign Key Type Mismatch' }), _jsx("div", { style: { marginBottom: '12px', fontFamily: theme.fontFamily }, children: errorType === 'missing' ? (_jsxs(_Fragment, { 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"] }))] })) : (_jsxs(_Fragment, { children: ["The foreign key", ' ', _jsx("span", { style: {
-                                    color: 'red',
-                                    fontWeight: 600,
-                                    backgroundColor: '#f7f7f7',
-                                    padding: '2px 4px',
-                                    borderRadius: 4,
-                                    border: '1px solid #e0e0e0',
-                                }, children: tenant?.tenantField.replaceAll('"', '') }), ' ', "in your query has a type that does not match the tenant ID type for", ' ', _jsx("span", { style: {
-                                    color: '#212121',
-                                    fontWeight: 600,
-                                    backgroundColor: '#f7f7f7',
-                                    padding: '2px 4px',
-                                    borderRadius: 4,
-                                    border: '1px solid #e0e0e0',
-                                }, children: tenant?.name }), ". Please modify your query to ensure the foreign key type matches the tenant ID type."] })) }), _jsx("br", {}), _jsxs("div", { style: {
-                        display: 'flex',
-                        flexDirection: 'row',
-                        gap: '12px',
-                        justifyContent: 'flex-end',
-                    }, children: [onOverride && errorType === 'missing' && (_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

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

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

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