@quillsql/admin 1.6.3 → 1.6.4

package/dist/cjs/forms/client_onboard/ConnectDatabase.js

@@ -17,6 +17,7 @@ const QuillMultiSelectWithCombo_1 = require("../../components/QuillMultiSelectWi
 const CardSection_1 = __importDefault(require("../../components/CardSection"));
 // import InputLabel from '../../components/InputLabel';
 const AdminProvider_1 = require("../../AdminProvider");
+const dataFetcher_1 = require("../../utils/dataFetcher");
 function ConnectDatabase({ containerStyle, header1Style, header2Style, environmentDetails, setEnvironmentDetails, fetchSchemasLoading, onConnectDatabase, }) {
     const { state, getToken } = (0, AdminProvider_1.useAdmin)();
     const [error, setError] = (0, react_1.useState)(undefined);
@@ -126,8 +127,8 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
         }
         try {
             setIsConnecting(true);
-            await (0, ConnectionClient_1.setSchemaNames)(state.queryEndpoint, clientId, selectedSchemas, getToken);
-            const databaseSchema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.queryEndpoint, clientId, getToken, selectedSchemas).then((res) => res.columnsByTable);
+            await (0, ConnectionClient_1.setSchemaNames)(state.queryEndpoint, clientId, selectedSchemas, getToken, state.queryHeaders);
+            const databaseSchema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.isSelfHosted, state.queryEndpoint, clientId, databaseConnectionDetails.type, getToken, selectedSchemas, state.queryHeaders).then((res) => res.columnsByTable);
             setEnvironmentDetails(databaseConnectionDetails, databaseSchema, selectedSchemas, clientId);
         }
         catch (e) {
@@ -139,11 +140,38 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
             }, 1000);
         }
     };
+    const getSchemaNames = async () => {
+        setIsConnecting(true);
+        const tempClientId = clientId ?? state.client?.clientId;
+        const { data, status, queries } = await (0, dataFetcher_1.quillFetch)({
+            client: {
+                queryEndpoint: state.queryEndpoint,
+                queryHeaders: state.queryHeaders,
+                withCredentials: !!state.withCredentials,
+                clientId: tempClientId,
+            },
+            task: 'get-schema-names',
+            metadata: {
+                databaseType: databaseConnectionDetails.type,
+                newClientId: clientId,
+            },
+            getToken: getToken,
+        });
+        if (status !== 'success') {
+            setIsConnecting(false);
+            setError('Error fetching schemas');
+            return;
+        }
+        const schemas = queries?.queryResults?.[0]?.rows?.map((row) => row['schema_name'] || row['SCHEMA_NAME']);
+        setClientId(data?.clientId);
+        setAllSchemas(schemas);
+        setIsConnecting(false);
+    };
     const testConnectionAndFormClient = async () => {
         setError(undefined);
         const connectionString = (0, databases_1.formConnectionString)(databaseConnectionDetails);
         setIsConnecting(true);
-        const result = await (0, ConnectionClient_1.testConnectionString)(state.queryEndpoint, connectionString, databaseConnectionDetails.type, getToken, clientId);
+        const result = await (0, ConnectionClient_1.testConnectionString)(state.queryEndpoint, connectionString, databaseConnectionDetails.type, getToken, clientId, state.queryHeaders);
         if (result.success) {
             try {
                 setConnectionDetails({
@@ -176,50 +204,75 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
                     display: 'flex',
                     flexDirection: 'row',
                     justifyContent: 'space-between',
-                }, children: [(0, jsx_runtime_1.jsx)("h1", { style: header1Style, children: "Connect Database" }), clientId && !!selectedSchemas?.length && ((0, jsx_runtime_1.jsx)("svg", { xmlns: "http://www.w3.org/2000/svg", viewBox: "0 0 20 20", fill: "currentColor", width: "24px", height: "24px", children: (0, jsx_runtime_1.jsx)("path", { fillRule: "evenodd", d: "M10 18a8 8 0 1 0 0-16 8 8 0 0 0 0 16Zm3.857-9.809a.75.75 0 0 0-1.214-.882l-3.483 4.79-1.88-1.88a.75.75 0 1 0-1.06 1.061l2.5 2.5a.75.75 0 0 0 1.137-.089l4-5.5Z", clipRule: "evenodd" }) }))] }), (0, jsx_runtime_1.jsxs)("div", { style: { marginTop: 12 }, children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: 600, color: '#565656' }, children: "Database type" }), (0, jsx_runtime_1.jsx)("div", { style: { display: 'flex', gap: 10, userSelect: 'none' }, children: databases_1.DATABASE_DEFAULT_INFO.map((database) => {
-                            return ((0, jsx_runtime_1.jsx)(DatabaseSelector_1.default, { onDatabaseSelect: onDatabaseSelect, databaseType: databaseConnectionDetails?.type || '', containerStyle: { marginBottom: '12px' }, label: database.label, value: database.value, imageUrl: database.imageUrl }, database.label));
-                        }) })] }), databaseConnectionDetails && ((0, jsx_runtime_1.jsxs)("div", { children: [databaseConnectionDetails.type !== 'BigQuery' ? ((0, jsx_runtime_1.jsxs)("div", { children: [['MySQL', 'PostgreSQL'].includes(databaseConnectionDetails.type) && ((0, jsx_runtime_1.jsxs)("div", { style: { marginTop: 16, marginBottom: 24 }, children: [(0, jsx_runtime_1.jsx)("h2", { style: {
-                                            ...header2Style,
-                                            fontWeight: 600,
-                                            marginBottom: 8,
-                                        }, children: "Connection Method" }), (0, jsx_runtime_1.jsx)(QuillTab_1.default, { width: "486px", modeToggleOptions: credentialTabOptions, toggleMode: credentialTab, setToggleMode: setCredentialTab })] })), credentialTab === 'Database Credentials' ? ((0, jsx_runtime_1.jsx)("div", { id: "database-details-form", children: (0, jsx_runtime_1.jsx)("div", { style: {
-                                        display: 'grid',
-                                        gridTemplateColumns: 'repeat(3, 1fr)',
-                                        gap: '20px',
-                                        justifyContent: 'space-between',
-                                        marginTop: '16px',
-                                    }, children: 
-                                    // get a list of types in the databaseConnectionDetails object and create a form for each
-                                    Object.entries(databaseConnectionDetails).map(([key, value]) => {
-                                        if (key !== 'connectionString' && key !== 'type')
-                                            return ((0, jsx_runtime_1.jsxs)("div", { "data-testid": `quill-input-${key}`, children: [(0, jsx_runtime_1.jsx)("h2", { style: {
-                                                            ...header2Style,
-                                                            fontWeight: '600',
-                                                        }, children: key
-                                                            .split(/(?=[A-Z])/)
-                                                            .map((word) => word.charAt(0).toUpperCase() +
-                                                            word.slice(1))
-                                                            .join(' ') }), (0, jsx_runtime_1.jsx)(primitives_1.TextInputPrimitive, { onChange: (e) => {
-                                                            setConnectionDetails({
-                                                                ...databaseConnectionDetails,
-                                                                [key]: e.target.value,
-                                                                connectionString: '',
-                                                            });
-                                                        }, value: value, disabled: key === 'type', password: key === 'password' ? true : false })] }, key));
-                                    }) }) })) : ((0, jsx_runtime_1.jsxs)("div", { "data-testid": "quill-input-connectionString", children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: 600 }, children: "Connection String" }), (0, jsx_runtime_1.jsx)(primitives_1.TextInputPrimitive, { onChange: (e) => {
-                                            updateConnectionString(e.target.value);
-                                        }, value: databaseConnectionDetails.connectionString })] }))] })) : ((0, jsx_runtime_1.jsx)("div", { style: { minWidth: 790 }, children: (0, jsx_runtime_1.jsx)("div", { id: "database-details-form-bq", style: {
-                                display: 'grid',
-                                gridTemplateColumns: 'repeat(1, 1fr)',
-                                gap: '20px',
-                                justifyContent: 'space-between',
-                                marginTop: '16px',
-                            }, children: (0, jsx_runtime_1.jsxs)("div", { children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: '600' }, children: "Service Account Json" }), (0, jsx_runtime_1.jsxs)("div", { children: [(0, jsx_runtime_1.jsx)("input", { title: "Upload JSON", type: "file", accept: ".json", onChange: handleFileChange, style: { display: 'none' }, id: "fileInput", "data-testid": "quill-input-fileInput", ref: myInputRef }), (0, jsx_runtime_1.jsxs)("div", { style: {
-                                                    display: 'flex',
-                                                    flexDirection: 'row',
-                                                    gap: 12,
-                                                    alignItems: 'center',
-                                                }, children: [(0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { onClick: handleButtonClick, label: 'Upload JSON' }), (0, jsx_runtime_1.jsx)("span", { "data-testid": "quill-file-label", style: { ...header2Style, fontWeight: '600' }, children: fileLabel })] })] })] }) }) })), (0, jsx_runtime_1.jsxs)("div", { style: {
+                }, children: [(0, jsx_runtime_1.jsx)("h1", { style: header1Style, children: "Connect Database" }), clientId && !!selectedSchemas?.length && ((0, jsx_runtime_1.jsx)("svg", { xmlns: "http://www.w3.org/2000/svg", viewBox: "0 0 20 20", fill: "currentColor", width: "24px", height: "24px", children: (0, jsx_runtime_1.jsx)("path", { fillRule: "evenodd", d: "M10 18a8 8 0 1 0 0-16 8 8 0 0 0 0 16Zm3.857-9.809a.75.75 0 0 0-1.214-.882l-3.483 4.79-1.88-1.88a.75.75 0 1 0-1.06 1.061l2.5 2.5a.75.75 0 0 0 1.137-.089l4-5.5Z", clipRule: "evenodd" }) }))] }), !state.isSelfHosted && ((0, jsx_runtime_1.jsxs)("div", { style: { marginTop: 12 }, children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: 600 }, children: "Database type" }), (0, jsx_runtime_1.jsx)("div", { style: { display: 'flex', gap: 10, userSelect: 'none' }, children: databases_1.DATABASE_DEFAULT_INFO.map((database) => {
+                            return ((0, jsx_runtime_1.jsx)(DatabaseSelector_1.default, { onDatabaseSelect: onDatabaseSelect, databaseType: databaseConnectionDetails?.type || '', containerStyle: {
+                                    marginBottom: '12px',
+                                    ...(state.isSelfHosted && {
+                                        cursor: 'not-allowed',
+                                        pointerEvents: 'none',
+                                    }),
+                                    ...(state.isSelfHosted &&
+                                        database.value.toLowerCase() !==
+                                            environmentDetails.connection.type.toLowerCase() && {
+                                        backgroundColor: '#f5f5f5',
+                                    }),
+                                }, label: database.label, value: database.value, imageUrl: database.imageUrl, theme: state.theme }, database.label));
+                        }) })] })), databaseConnectionDetails && ((0, jsx_runtime_1.jsxs)("div", { children: [!state.isSelfHosted && ((0, jsx_runtime_1.jsx)("div", { children: databaseConnectionDetails.type !== 'BigQuery' ? ((0, jsx_runtime_1.jsxs)("div", { children: [['MySQL', 'PostgreSQL'].includes(databaseConnectionDetails.type) && ((0, jsx_runtime_1.jsxs)("div", { style: { marginTop: 16, marginBottom: 24 }, children: [(0, jsx_runtime_1.jsx)("h2", { style: {
+                                                ...header2Style,
+                                                fontWeight: 600,
+                                                marginBottom: 8,
+                                            }, children: "Connection Method" }), (0, jsx_runtime_1.jsx)(QuillTab_1.default, { width: "486px", modeToggleOptions: credentialTabOptions, toggleMode: credentialTab, setToggleMode: setCredentialTab })] })), credentialTab === 'Database Credentials' ? ((0, jsx_runtime_1.jsx)("div", { id: "database-details-form", children: (0, jsx_runtime_1.jsx)("div", { style: {
+                                            display: 'grid',
+                                            gridTemplateColumns: 'repeat(3, 1fr)',
+                                            gap: '20px',
+                                            justifyContent: 'space-between',
+                                            marginTop: '16px',
+                                        }, children: 
+                                        // get a list of types in the databaseConnectionDetails object and create a form for each
+                                        Object.entries(databaseConnectionDetails).map(([key, value]) => {
+                                            if (key !== 'connectionString' && key !== 'type')
+                                                return ((0, jsx_runtime_1.jsxs)("div", { "data-testid": `quill-input-${key}`, children: [(0, jsx_runtime_1.jsx)("h2", { style: {
+                                                                ...header2Style,
+                                                                fontWeight: '600',
+                                                            }, children: key
+                                                                .split(/(?=[A-Z])/)
+                                                                .map((word) => word.charAt(0).toUpperCase() +
+                                                                word.slice(1))
+                                                                .join(' ') }), (0, jsx_runtime_1.jsx)(primitives_1.TextInputPrimitive, { onChange: (e) => {
+                                                                setConnectionDetails({
+                                                                    ...databaseConnectionDetails,
+                                                                    [key]: e.target.value,
+                                                                    connectionString: '',
+                                                                });
+                                                            }, value: value, disabled: key === 'type', password: key === 'password' ? true : false })] }, key));
+                                        }) }) })) : ((0, jsx_runtime_1.jsxs)("div", { "data-testid": "quill-input-connectionString", children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: 600 }, children: "Connection String" }), (0, jsx_runtime_1.jsx)(primitives_1.TextInputPrimitive, { onChange: (e) => {
+                                                updateConnectionString(e.target.value);
+                                            }, value: databaseConnectionDetails.connectionString })] }))] })) : ((0, jsx_runtime_1.jsx)("div", { style: { minWidth: 790 }, children: (0, jsx_runtime_1.jsx)("div", { id: "database-details-form-bq", style: {
+                                    display: 'grid',
+                                    gridTemplateColumns: 'repeat(1, 1fr)',
+                                    gap: '20px',
+                                    justifyContent: 'space-between',
+                                    marginTop: '16px',
+                                }, children: (0, jsx_runtime_1.jsxs)("div", { children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: '600' }, children: "Service Account Json" }), (0, jsx_runtime_1.jsxs)("div", { children: [(0, jsx_runtime_1.jsx)("input", { title: "Upload JSON", type: "file", accept: ".json", onChange: handleFileChange, style: { display: 'none' }, id: "fileInput", "data-testid": "quill-input-fileInput", ref: myInputRef }), (0, jsx_runtime_1.jsxs)("div", { style: {
+                                                        display: 'flex',
+                                                        flexDirection: 'row',
+                                                        gap: 12,
+                                                        alignItems: 'center',
+                                                    }, children: [(0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { onClick: handleButtonClick, label: 'Upload JSON' }), (0, jsx_runtime_1.jsx)("span", { "data-testid": "quill-file-label", style: { ...header2Style, fontWeight: '600' }, children: fileLabel })] })] })] }) }) })) })), state.isSelfHosted ? ((0, jsx_runtime_1.jsxs)("div", { style: {
+                            borderRadius: 6,
+                            padding: 12,
+                            background: '#f5f5f5',
+                            marginBottom: 18,
+                            marginTop: 18,
+                            fontSize: 14,
+                        }, children: [(0, jsx_runtime_1.jsx)("h2", { style: { ...header2Style, fontWeight: 600 }, children: "Self-hosted Database Connection" }), (0, jsx_runtime_1.jsx)("div", { style: {
+                                    display: 'flex',
+                                    flexDirection: 'row',
+                                    alignItems: 'center',
+                                    gap: 4,
+                                    color: state.theme?.secondaryTextColor,
+                                    fontFamily: state.theme?.fontFamily,
+                                }, children: `Using the database credentials passed into the Quill Server SDK at this endpoint: ${state.queryEndpoint}.` })] })) : ((0, jsx_runtime_1.jsxs)("div", { style: {
                             borderRadius: 6,
                             padding: 12,
                             background: '#f5f5f5',
@@ -231,6 +284,8 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
                                     flexDirection: 'row',
                                     alignItems: 'center',
                                     gap: 4,
+                                    color: state.theme?.secondaryTextColor,
+                                    fontFamily: state.theme?.fontFamily,
                                 }, children: ["Ensure Quill's IP is allowed to connect to your database", ' ', (0, jsx_runtime_1.jsx)(ExternalLink_1.default, { href: databaseConnectionDetails.type.toLowerCase() === 'bigquery'
                                             ? 'https://cloud.google.com/access-context-manager/docs/overview#ip-address'
                                             : databaseConnectionDetails.type.toLowerCase() === 'mysql'
@@ -246,18 +301,26 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
                                                             ? 'https://clickhouse.com/docs/cloud/security/setting-ip-filters/'
                                                             : 'https://www.postgresql.org/docs/current/auth-pg-hba-conf.html' })] }), (0, jsx_runtime_1.jsx)(ClipboardButton_1.default, { text: "34.133.137.225", containerStyle: {
                                     marginTop: 12,
-                                } })] }), (0, jsx_runtime_1.jsx)("div", { style: {
+                                } })] })), (0, jsx_runtime_1.jsx)("div", { style: {
                             width: '100%',
                             display: 'flex',
                             justifyContent: 'space-between',
                             alignItems: 'center',
                             marginTop: '24px',
-                        }, "data-testid": "quill-connect-button", children: (0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { label: 'Connect', onClick: testConnectionAndFormClient, disabled: isButtonDisabled, isLoading: isConnecting }) })] })), (0, jsx_runtime_1.jsx)(primitives_1.ModalPrimitive, { isOpen: !!error, onClose: () => setError(undefined), style: {
+                        }, "data-testid": "quill-connect-button", children: (0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { label: 'Connect', onClick: () => {
+                                if (state.isSelfHosted) {
+                                    getSchemaNames();
+                                }
+                                else {
+                                    testConnectionAndFormClient();
+                                }
+                            }, disabled: isButtonDisabled && !state.isSelfHosted, isLoading: isConnecting }) })] })), (0, jsx_runtime_1.jsx)(primitives_1.ModalPrimitive, { isOpen: !!error, onClose: () => setError(undefined), style: {
                     width: 420,
                 }, children: (0, jsx_runtime_1.jsxs)("div", { style: {
                         display: 'flex',
                         flexDirection: 'column',
                         gap: 20,
+                        fontFamily: state.theme?.fontFamily,
                         width: '100%',
                     }, children: [(0, jsx_runtime_1.jsx)(CardSection_1.default, { children: "Failed to Connect" }), (0, jsx_runtime_1.jsxs)("div", { style: { display: 'flex', flexDirection: 'column', gap: 5 }, children: [(0, jsx_runtime_1.jsx)("div", { style: {
                                         color: 'red',
@@ -286,6 +349,7 @@ function ConnectDatabase({ containerStyle, header1Style, header2Style, environme
                         flexDirection: 'column',
                         gap: 20,
                         width: '100%',
+                        fontFamily: state.theme?.fontFamily,
                     }, children: [(0, jsx_runtime_1.jsx)(CardSection_1.default, { children: "Connection Successful" }), (0, jsx_runtime_1.jsx)("h2", { style: {
                                 fontSize: 15,
                                 fontWeight: 500,

package/dist/cjs/forms/client_onboard/ConnectSchema.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ConnectSchema.d.ts","sourceRoot":"","sources":["../../../../src/forms/client_onboard/ConnectSchema.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAgC,MAAM,OAAO,CAAC;AAOpE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,KAAK,EAAY,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAMhD,MAAM,CAAC,OAAO,UAAU,aAAa,CAAC,EACpC,cAAc,EACd,YAAY,EAEZ,cAAc,EACd,UAAU,EACV,UAAU,EACV,gBAAgB,EAChB,QAAQ,EACR,eAAe,EACf,KAAK,EACL,KAAK,EACL,uBAAuB,GACxB,EAAE;IACD,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,cAAc,EAAE,cAAc,EAAE,CAAC;IACjC,UAAU,EAAE,cAAc,CAAC;IAC3B,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAChC,gBAAgB,EAAE,CAAC,UAAU,EAAE,UAAU,KAAK,IAAI,CAAC;IACnD,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,uBAAuB,EAAE,MAAM,IAAI,CAAC;CACrC,2CA0SA"}
+{"version":3,"file":"ConnectSchema.d.ts","sourceRoot":"","sources":["../../../../src/forms/client_onboard/ConnectSchema.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAgC,MAAM,OAAO,CAAC;AAOpE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,KAAK,EAAY,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAMhD,MAAM,CAAC,OAAO,UAAU,aAAa,CAAC,EACpC,cAAc,EACd,YAAY,EAEZ,cAAc,EACd,UAAU,EACV,UAAU,EACV,gBAAgB,EAChB,QAAQ,EACR,eAAe,EACf,KAAK,EACL,KAAK,EACL,uBAAuB,GACxB,EAAE;IACD,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,cAAc,EAAE,cAAc,EAAE,CAAC;IACjC,UAAU,EAAE,cAAc,CAAC;IAC3B,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAChC,gBAAgB,EAAE,CAAC,UAAU,EAAE,UAAU,KAAK,IAAI,CAAC;IACnD,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,uBAAuB,EAAE,MAAM,IAAI,CAAC;CACrC,2CA4SA"}

package/dist/cjs/forms/client_onboard/ConnectSchema.js

@@ -129,7 +129,8 @@ databaseSchema, connection, tenantInit, onTenantCreation, clientId, selectedSche
                         }, children: [(0, jsx_runtime_1.jsx)("h2", { style: {
                                     fontSize: 15,
                                     fontWeight: 500,
-                                    color: '#6e6e6e',
+                                    color: state.theme?.secondaryTextColor,
+                                    fontFamily: state.theme?.fontFamily,
                                     paddingBottom: 20,
                                 }, children: `How do you represent your customers in your database?` }), (0, jsx_runtime_1.jsx)(EditTenant_1.EditTenant, { tenant: tenantToEdit, setTenant: (tenant) => setTenantToEdit(tenant), setCreateNewTenant: () => { }, onSubmitEditTenant: onSubmitEditTenant, onValidateEditTenant: onValidateEditTenant, SecondaryButtonComponent: primitives_1.SecondaryButtonPrimitive, TextInputComponent: primitives_1.TextInputPrimitive, ButtonComponent: UiComponents_1.MemoizedButton, isSubmitTenantLoading: isSubmitTenantLoading, client: {
                                     id: clientId,
@@ -144,6 +145,7 @@ databaseSchema, connection, tenantInit, onTenantCreation, clientId, selectedSche
                         display: 'flex',
                         flexDirection: 'column',
                         gap: 20,
+                        fontFamily: state.theme?.fontFamily,
                         width: '100%',
                     }, children: [(0, jsx_runtime_1.jsx)(CardSection_1.default, { children: noTables ? 'Schema Issue' : 'Tenant Issue' }), (0, jsx_runtime_1.jsxs)("div", { style: { display: 'flex', flexDirection: 'column', gap: 5 }, children: [(0, jsx_runtime_1.jsx)("div", { style: {
                                         color: 'red',

package/dist/cjs/forms/client_onboard/CreateVirtualTables.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CreateVirtualTables.d.ts","sourceRoot":"","sources":["../../../../src/forms/client_onboard/CreateVirtualTables.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAA+B,MAAM,OAAO,CAAC;AACnE,OAAO,EAAU,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAY3D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAOvD,OAAO,EAAE,KAAK,EAAY,MAAM,qBAAqB,CAAC;AAGtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,MAAM,CAAC,OAAO,UAAU,mBAAmB,CAAC,EAC1C,cAAc,EACd,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACd,kBAAkB,EAClB,MAAM,EACN,KAAK,EACL,UAAU,EACV,SAAS,EACT,SAAS,EACT,YAAY,GACb,EAAE;IACD,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,cAAc,EAAE,cAAc,EAAE,CAAC;IACjC,kBAAkB,CAAC,EAAE,cAAc,CAAC;IACpC,kBAAkB,EAAE,MAAM,IAAI,CAAC;IAC/B,MAAM,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrC,KAAK,EAAE,KAAK,CAAC;IACb,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAChC,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IACjD,YAAY,EAAE,CAAC,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,KAAK,IAAI,CAAC;CAClE,2CAoiBA;AAED,wBAAgB,iBAAiB,CAAC,EAChC,MAAM,EACN,SAAS,EACT,SAAS,EACT,YAAY,EACZ,eAAe,GAChB,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IAChC,SAAS,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3C,YAAY,EAAE,CAAC,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;IAC3D,eAAe,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD,2CAqHA;AAED,wBAAgB,gBAAgB,CAAC,EAC/B,MAAM,EACN,SAAS,EACT,MAAM,EACN,UAAU,GACX,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;CACzB,2CA8EA"}
+{"version":3,"file":"CreateVirtualTables.d.ts","sourceRoot":"","sources":["../../../../src/forms/client_onboard/CreateVirtualTables.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAA+B,MAAM,OAAO,CAAC;AACnE,OAAO,EAAU,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAY3D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAOvD,OAAO,EAAE,KAAK,EAAY,MAAM,qBAAqB,CAAC;AAGtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,MAAM,CAAC,OAAO,UAAU,mBAAmB,CAAC,EAC1C,cAAc,EACd,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACd,kBAAkB,EAClB,MAAM,EACN,KAAK,EACL,UAAU,EACV,SAAS,EACT,SAAS,EACT,YAAY,GACb,EAAE;IACD,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,cAAc,EAAE,cAAc,EAAE,CAAC;IACjC,kBAAkB,CAAC,EAAE,cAAc,CAAC;IACpC,kBAAkB,EAAE,MAAM,IAAI,CAAC;IAC/B,MAAM,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrC,KAAK,EAAE,KAAK,CAAC;IACb,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAChC,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IACjD,YAAY,EAAE,CAAC,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,KAAK,IAAI,CAAC;CAClE,2CAwiBA;AAED,wBAAgB,iBAAiB,CAAC,EAChC,MAAM,EACN,SAAS,EACT,SAAS,EACT,YAAY,EACZ,eAAe,GAChB,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IAChC,SAAS,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3C,YAAY,EAAE,CAAC,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;IAC3D,eAAe,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD,2CA2HA;AAED,wBAAgB,gBAAgB,CAAC,EAC/B,MAAM,EACN,SAAS,EACT,MAAM,EACN,UAAU,GACX,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;CACzB,2CA8EA"}

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

@@ -160,7 +160,7 @@ function CreateVirtualTables({ containerStyle, header1Style, header2Style, envir
                                             whiteSpace: 'normal',
                                         }, displayBelow: true, children: (0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { label: 'Ask AI', isLoading: askAIButtonLoading, onClick: async () => {
                                                 setAskAIButtonLoading(true);
-                                                const results = await (0, ConnectionClient_1.getQueryFromAiWithConnection)(state.queryEndpoint, aiPrompt, client.clientId, getToken, client.schemaNames);
+                                                const results = await (0, ConnectionClient_1.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 }) })] }), (0, jsx_runtime_1.jsx)("div", { "data-testid": "quill-sql-text-editor", children: (0, jsx_runtime_1.jsx)(SqlTextEditor_1.default, { value: queryInfo.query, setValue: (e) => {
@@ -178,7 +178,7 @@ function CreateVirtualTables({ containerStyle, header1Style, header2Style, envir
                                     height: 70,
                                 }, children: [(0, jsx_runtime_1.jsx)(primitives_1.ButtonPrimitive, { label: 'Run query', isLoading: runQueryButtonLoading, onClick: async () => {
                                             setRunQueryButtonLoading(true);
-                                            const getSqlResults = await (0, ConnectionClient_1.getVirtualTableData)(client.clientId, queryInfo.query, environmentDetails.type, state.queryEndpoint, getToken);
+                                            const getSqlResults = await (0, ConnectionClient_1.getVirtualTableData)(client.clientId, queryInfo.query, environmentDetails.type, state.queryEndpoint, getToken, client.queryHeaders);
                                             if (!getSqlResults.success) {
                                                 setErrorInfo({ status: true, msg: getSqlResults.error });
                                             }
@@ -315,15 +315,18 @@ function CreateVirtualTables({ containerStyle, header1Style, header2Style, envir
                                     }, isLoading: isLoading }) })) : null] })] }), (0, jsx_runtime_1.jsx)(TenantFieldModal, { isOpen: !!failingTenant, setClosed: () => setFailingTenant(undefined), tenant: failingTenant, onOverride: () => addVirtualTable(true) }), (0, jsx_runtime_1.jsx)(VirtualTableModal, { isOpen: virtualTableModalOpen, setIsOpen: setVirtualTableModalOpen, queryInfo: queryInfo, setQueryInfo: setQueryInfo, addVirtualTable: () => addVirtualTable(false) })] }));
 }
 function VirtualTableModal({ isOpen, setIsOpen, queryInfo, setQueryInfo, addVirtualTable, }) {
+    const { state: { theme }, } = (0, AdminProvider_1.useAdmin)();
     return ((0, jsx_runtime_1.jsx)(primitives_1.ModalPrimitive, { isOpen: isOpen, onClose: () => {
             setIsOpen(false);
         }, children: (0, jsx_runtime_1.jsxs)("div", { style: { display: 'flex', flexDirection: 'column', width: '100%' }, children: [(0, jsx_runtime_1.jsx)("h1", { style: {
                         fontSize: 24,
+                        fontFamily: theme.fontFamily,
                         fontWeight: 600,
                         color: '#212121',
                         paddingBottom: 20,
                     }, children: "Add Virtual Table" }), (0, jsx_runtime_1.jsx)("h3", { style: {
                         fontWeight: '600',
+                        fontFamily: theme.fontFamily,
                         fontSize: '14px',
                         marginBottom: '6px',
                         color: '#212121',
@@ -333,6 +336,7 @@ function VirtualTableModal({ isOpen, setIsOpen, queryInfo, setQueryInfo, addVirt
                         setQueryInfo({ name: e.target.value, query: queryInfo.query });
                     }, placeholder: "Name..." }), (0, jsx_runtime_1.jsx)("h3", { style: {
                         fontWeight: '600',
+                        fontFamily: theme.fontFamily,
                         fontSize: '14px',
                         marginBottom: '6px',
                         marginTop: '12px',

package/dist/cjs/forms/client_onboard/__tests__/ConnectSchema.test.js

@@ -69,6 +69,7 @@ describe('ConnectSchema', () => {
     const mockGetToken = jest.fn();
     const mockQuillFetchWithToken = jest.fn();
     const mockState = {
+        isSelfHosted: false,
         queryEndpoint: 'http://test-endpoint',
         loading: true,
         publicKey: 'test-key',

package/dist/cjs/forms/client_onboard/__tests__/CreateVirtualTables.test.js

@@ -52,6 +52,7 @@ describe('CreateVirtualTables', () => {
     const mockGetToken = jest.fn();
     const mockQuillFetchWithToken = jest.fn();
     const mockState = {
+        isSelfHosted: false,
         queryEndpoint: 'http://test-endpoint',
         loading: true,
         publicKey: 'test-key',

package/dist/cjs/forms/virtual_tables/CreateEditVirtualTable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CreateEditVirtualTable.d.ts","sourceRoot":"","sources":["../../../../src/forms/virtual_tables/CreateEditVirtualTable.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EAGrB,MAAM,iBAAiB,CAAC;AAQzB,OAAO,EAOL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAU,cAAc,EAAE,KAAK,IAAI,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAO/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAIlD,UAAU,2BAA2B;IACnC,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC;IACpC,mBAAmB,CAAC,EAAE,SAAS,CAAC;IAChC,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC/D,wBAAwB,CAAC,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IACxE,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,uBAAuB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC5D,gBAAgB,CAAC,EAAE,MAAM,GAAG,CAAC,OAAO,CAAC;IACrC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAClC,mBAAmB,EAAE,CACnB,OAAO,EAAE,KAAK,GAAG,MAAM,EACvB,KAAK,EAAE,MAAM,EACb,iBAAiB,EAAE,MAAM,EAAE,EAC3B,IAAI,CAAC,EAAE,MAAM,EACb,EAAE,CAAC,EAAE,MAAM,EACX,OAAO,CAAC,EAAE,GAAG,EAAE,EACf,QAAQ,CAAC,EAAE,OAAO,KACf,IAAI,CAAC;IACV,YAAY,EAAE,cAAc,EAAE,CAAC;IAC/B,eAAe,EAAE,OAAO,CAAC;IACzB,MAAM,EAAE,mBAAmB,CAAC;CAC7B;AAED,MAAM,CAAC,OAAO,UAAU,sBAAsB,CAAC,EAC7C,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EAAE,qBAAqB,EAC5C,mBAAmB,EAAE,mBAAmB,EACxC,gBAA2C,EAC3C,eAAgC,EAChC,wBAAkD,EAClD,SAA6B,EAC7B,YAAY,EACZ,eAAe,EACf,MAAM,GACP,EAAE,2BAA2B,GAAG,GAAG,CAAC,OAAO,CAqgB3C"}
+{"version":3,"file":"CreateEditVirtualTable.d.ts","sourceRoot":"","sources":["../../../../src/forms/virtual_tables/CreateEditVirtualTable.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EAGrB,MAAM,iBAAiB,CAAC;AAQzB,OAAO,EAOL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAU,cAAc,EAAE,KAAK,IAAI,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAO/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAIlD,UAAU,2BAA2B;IACnC,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC;IACpC,mBAAmB,CAAC,EAAE,SAAS,CAAC;IAChC,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC/D,wBAAwB,CAAC,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IACxE,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,uBAAuB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC5D,gBAAgB,CAAC,EAAE,MAAM,GAAG,CAAC,OAAO,CAAC;IACrC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAClC,mBAAmB,EAAE,CACnB,OAAO,EAAE,KAAK,GAAG,MAAM,EACvB,KAAK,EAAE,MAAM,EACb,iBAAiB,EAAE,MAAM,EAAE,EAC3B,IAAI,CAAC,EAAE,MAAM,EACb,EAAE,CAAC,EAAE,MAAM,EACX,OAAO,CAAC,EAAE,GAAG,EAAE,EACf,QAAQ,CAAC,EAAE,OAAO,KACf,IAAI,CAAC;IACV,YAAY,EAAE,cAAc,EAAE,CAAC;IAC/B,eAAe,EAAE,OAAO,CAAC;IACzB,MAAM,EAAE,mBAAmB,CAAC;CAC7B;AAED,MAAM,CAAC,OAAO,UAAU,sBAAsB,CAAC,EAC7C,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EAAE,qBAAqB,EAC5C,mBAAmB,EAAE,mBAAmB,EACxC,gBAA2C,EAC3C,eAAgC,EAChC,wBAAkD,EAClD,SAA6B,EAC7B,YAAY,EACZ,eAAe,EACf,MAAM,GACP,EAAE,2BAA2B,GAAG,GAAG,CAAC,OAAO,CA+gB3C"}

package/dist/cjs/forms/virtual_tables/CreateEditVirtualTable.js

@@ -38,6 +38,7 @@ function CreateEditVirtualTable({ containerStyle, initialVirtualTable, closeEdit
     const [tableData, setTableData] = (0, react_2.useState)(undefined);
     const [runningQuery, setRunningQuery] = (0, react_2.useState)(false);
     const { state, getToken, quillFetchWithToken } = (0, AdminProvider_1.useAdmin)();
+    const containerRef = (0, react_2.useRef)(null);
     const addEditProcessVirtualTable = async (override = false) => {
         const failingTenant = virtualTableOwners?.find((field) => tableData &&
             !tableData.fields.find((col) => col.name === field?.replaceAll('"', '')));
@@ -66,7 +67,7 @@ function CreateEditVirtualTable({ containerStyle, initialVirtualTable, closeEdit
         setErrorInfo({ show: false, message: '' });
         setVirtualTableAddable(false);
         setAskAIButtonLoading(true);
-        const results = await (0, ConnectionClient_1.getQueryFromAiWithConnection)(state.queryEndpoint, aiPrompt, state.client._id, getToken, state.client.schemaNames);
+        const results = await (0, ConnectionClient_1.getQueryFromAiWithConnection)(state.isSelfHosted, allTableData, state.queryEndpoint, aiPrompt, state.client._id, getToken, state.client.schemaNames, state.queryHeaders);
         setEditVirtualTableQuery(results);
         setAskAIButtonLoading(false);
     };
@@ -111,7 +112,7 @@ function CreateEditVirtualTable({ containerStyle, initialVirtualTable, closeEdit
             return;
         }
         setIsLoading(true);
-        const resp = await (0, ConnectionClient_1.getVirtualTableData)(state.client._id, query, state.client.databaseType, state.queryEndpoint, getToken);
+        const resp = await (0, ConnectionClient_1.getVirtualTableData)(state.client._id, query, state.client.databaseType, state.queryEndpoint, getToken, state.queryHeaders);
         if (!resp.success) {
             setErrorInfo({ show: true, message: resp.error });
             setTableData(undefined);
@@ -175,13 +176,16 @@ function CreateEditVirtualTable({ containerStyle, initialVirtualTable, closeEdit
         setIsLoading(false);
         setRunningQuery(false);
     };
-    return ((0, jsx_runtime_1.jsx)("div", { style: {
-            ...containerStyle,
+    return ((0, jsx_runtime_1.jsx)("div", { ref: containerRef, style: {
             gap: 6,
-            height: 'calc(100vh - 130px)',
+            height: containerRef.current
+                ? `calc(100vh - ${containerRef.current.getBoundingClientRect().top ?? 0}px - 80px)`
+                : '100%',
+            // overflow: 'scroll',
+            ...containerStyle,
         }, children: (0, jsx_runtime_1.jsxs)("div", { style: { display: 'flex', height: '100%', flexDirection: 'column' }, children: [(0, jsx_runtime_1.jsx)(DynamicBanner_1.default, { header: initialVirtualTable
                         ? `Edit "${initialVirtualTable.name}" Virtual Table`
-                        : 'Create Virtual Table', onExit: initialVirtualTable ? undefined : closeEditVirtualTable }), (0, jsx_runtime_1.jsxs)("div", { style: {
+                        : 'Create Virtual Table', onExit: initialVirtualTable ? undefined : closeEditVirtualTable, theme: state.theme }), (0, jsx_runtime_1.jsxs)("div", { style: {
                         width: '100%',
                         display: 'flex',
                         height: '100%',

package/dist/cjs/hooks/useDatabaseSchema.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useDatabaseSchema.d.ts","sourceRoot":"","sources":["../../../src/hooks/useDatabaseSchema.tsx"],"names":[],"mappings":"AAIA,eAAO,MAAM,iBAAiB;;;CAuD7B,CAAC"}
+{"version":3,"file":"useDatabaseSchema.d.ts","sourceRoot":"","sources":["../../../src/hooks/useDatabaseSchema.tsx"],"names":[],"mappings":"AAIA,eAAO,MAAM,iBAAiB;;;CAgE7B,CAAC"}

package/dist/cjs/hooks/useDatabaseSchema.js

@@ -7,12 +7,17 @@ const ConnectionClient_1 = require("../api/ConnectionClient");
 const useDatabaseSchema = () => {
     const { state, dispatch, getToken } = (0, AdminProvider_1.useAdmin)();
     const [isLoading, setIsLoading] = (0, react_1.useState)(state.databaseSchema?.id !== state.client.id);
+    const idRef = (0, react_1.useRef)(null);
     (0, react_1.useEffect)(() => {
         const fetchSchema = async () => {
             if ((state.databaseSchema?.schema?.length ||
                 state.databaseSchema?.loading) &&
                 state.databaseSchema?.id === state.client.id)
                 return;
+            if (idRef.current === state.client.id) {
+                return;
+            }
+            idRef.current = state.client.id;
             try {
                 dispatch({
                     type: 'SET_DATABASE_SCHEMA',
@@ -23,7 +28,7 @@ const useDatabaseSchema = () => {
                     },
                 });
                 setIsLoading(true);
-                const schema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.queryEndpoint, state.client.id, getToken);
+                const schema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.isSelfHosted, state.queryEndpoint, state.client.id, state.client.databaseType, getToken, state.client.schemaNames ?? [], state.queryHeaders);
                 dispatch({
                     type: 'SET_DATABASE_SCHEMA',
                     payload: {

package/dist/cjs/modals/EditEnvironmentModal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"EditEnvironmentModal.d.ts","sourceRoot":"","sources":["../../../src/modals/EditEnvironmentModal.tsx"],"names":[],"mappings":"AAAA,OAAc,EACZ,yBAAyB,EACzB,aAAa,EAGd,MAAM,OAAO,CAAC;AAGf,OAAO,EAAE,uBAAuB,EAAE,MAAM,kCAAkC,CAAC;AAQ3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAsB1E,wBAAgB,oBAAoB,CAAC,EACnC,MAAM,EACN,SAAS,EACT,MAAM,EACN,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,eAAe,EACf,wBAAwB,GACzB,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACrC,MAAM,EAAE,MAAM,IAAI,CAAC;IACnB,cAAc,EAAE,GAAG,CAAC;IACpB,kBAAkB,EACd,yBAAyB,CACvB,uBAAuB,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAC1D,GACD,CAAC,CAAC,KAAK,EAAE,uBAAuB,KAAK,GAAG,CAAC,OAAO,CAAC,CAAC;IACtD,eAAe,EAAE,GAAG,CAAC;IACrB,eAAe,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC9D,eAAe,EAAE,GAAG,CAAC;IACrB,wBAAwB,EAAE,GAAG,CAAC;CAC/B,2CAg4BA"}
+{"version":3,"file":"EditEnvironmentModal.d.ts","sourceRoot":"","sources":["../../../src/modals/EditEnvironmentModal.tsx"],"names":[],"mappings":"AAAA,OAAc,EACZ,yBAAyB,EACzB,aAAa,EAGd,MAAM,OAAO,CAAC;AAGf,OAAO,EAAE,uBAAuB,EAAE,MAAM,kCAAkC,CAAC;AAQ3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAsB1E,wBAAgB,oBAAoB,CAAC,EACnC,MAAM,EACN,SAAS,EACT,MAAM,EACN,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,eAAe,EACf,wBAAwB,GACzB,EAAE;IACD,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACrC,MAAM,EAAE,MAAM,IAAI,CAAC;IACnB,cAAc,EAAE,GAAG,CAAC;IACpB,kBAAkB,EACd,yBAAyB,CACvB,uBAAuB,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAC1D,GACD,CAAC,CAAC,KAAK,EAAE,uBAAuB,KAAK,GAAG,CAAC,OAAO,CAAC,CAAC;IACtD,eAAe,EAAE,GAAG,CAAC;IACrB,eAAe,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,GAAG,CAAC,OAAO,CAAC;IAC9D,eAAe,EAAE,GAAG,CAAC;IACrB,wBAAwB,EAAE,GAAG,CAAC;CAC/B,2CAy4BA"}

package/dist/cjs/modals/EditEnvironmentModal.js

@@ -274,7 +274,7 @@ function EditEnvironmentModal({ isOpen, setIsOpen, onSave, ModalComponent, TextI
                     loading: true,
                 },
             });
-            const schema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.queryEndpoint, state.client._id, getToken);
+            const schema = await (0, ConnectionClient_1.getTableColumnsBySchema)(state.isSelfHosted, state.queryEndpoint, state.client._id, state.client.databaseType, getToken, state.client.schemaNames ?? [], state.queryHeaders);
             // If any of the tenants point to a table that no longer exists, throw an error
             const tenantTables = state.client.allTenantTypes?.filter((t) => t.tenantTable !== undefined);
             const tables = schema.columnsByTable.map((t) => t.tableName);
@@ -352,6 +352,10 @@ function EditEnvironmentModal({ isOpen, setIsOpen, onSave, ModalComponent, TextI
         }
     };
     const handleDeleteEnvironment = async () => {
+        if (state.clients.length === 1) {
+            alert('You cannot delete the only environment');
+            return;
+        }
         if (confirm('Are you sure? All dashboards in this environment will also be deleted. This action cannot be undone.')) {
             await quillFetchWithToken({
                 client: {
@@ -371,7 +375,7 @@ function EditEnvironmentModal({ isOpen, setIsOpen, onSave, ModalComponent, TextI
                 payload: state.clients.filter((c) => c._id !== state.client._id),
             });
             setSelectedDashboard(null);
-            setClient(state.clients[0]);
+            setClient(state.clients.filter((c) => c._id !== state.client._id)[0]);
             setIsOpen(false);
             onSave();
         }
@@ -413,6 +417,7 @@ function EditEnvironmentModal({ isOpen, setIsOpen, onSave, ModalComponent, TextI
                                             // padding: '2px 8px',
                                             fontSize: 12,
                                             font: 'monospace',
+                                            fontFamily: state.theme.fontFamily,
                                             fontWeight: 300,
                                             width: 'fit-content',
                                         }, children: "Dashboard Owner" })), (0, jsx_runtime_1.jsx)("div", { style: {

package/dist/cjs/modals/EditFiltersModal.js

@@ -633,7 +633,7 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                                             // ) ?? initialRange,
                                                                             selectablePresetOptions.find((option) => option.value === e) ??
                                                                                 initialRange);
-                                                                        }, options: selectablePresetOptions, theme: Admin_1.defaultTheme, hideEmptyOption: true })] }) }), (0, jsx_runtime_1.jsx)("div", { style: {
+                                                                        }, options: selectablePresetOptions, theme: state.theme, hideEmptyOption: true })] }) }), (0, jsx_runtime_1.jsx)("div", { style: {
                                                                 display: 'flex',
                                                                 flexDirection: 'row',
                                                                 gap: '10px',
@@ -674,12 +674,12 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                     }, children: (0, jsx_runtime_1.jsxs)("div", { style: {
                                                             display: 'flex',
                                                             flexDirection: 'column',
-                                                            fontFamily: Admin_1.defaultTheme.fontFamily,
+                                                            fontFamily: state.theme.fontFamily,
                                                             gap: 8,
                                                         }, children: [(0, jsx_runtime_1.jsx)("h3", { style: {
                                                                     fontSize: 14,
                                                                     fontWeight: 600,
-                                                                    color: Admin_1.defaultTheme.primaryTextColor,
+                                                                    color: state.theme.primaryTextColor,
                                                                 }, children: "Recurring" }), (0, jsx_runtime_1.jsxs)("div", { style: {
                                                                     display: 'flex',
                                                                     flexDirection: 'row',
@@ -763,7 +763,7 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                                                 }, width: "78px", number: true, max: 9999, min: 0, step: 1 })] }))] }), (0, jsx_runtime_1.jsx)("h3", { style: {
                                                                     fontSize: 14,
                                                                     fontWeight: 600,
-                                                                    color: Admin_1.defaultTheme.primaryTextColor,
+                                                                    color: state.theme.primaryTextColor,
                                                                 }, children: "Standard" }), (0, jsx_runtime_1.jsxs)("div", { style: {
                                                                     display: 'flex',
                                                                     flexDirection: 'row',
@@ -873,7 +873,7 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                                                     ...newDateFilter,
                                                                                     presetOptions: updatedOptions,
                                                                                 });
-                                                                            }, theme: Admin_1.defaultTheme, hideEmptyOption: true })] }), (0, jsx_runtime_1.jsxs)("div", { style: {
+                                                                            }, theme: state.theme, hideEmptyOption: true })] }), (0, jsx_runtime_1.jsxs)("div", { style: {
                                                                         display: 'flex',
                                                                         flexDirection: 'column',
                                                                     }, children: [index === 0 && (0, jsx_runtime_1.jsx)(InputLabel_1.default, { children: "Value" }), interval.type === 'week' && ((0, jsx_runtime_1.jsx)("div", { style: { width: 200, visibility: 'hidden' } })), interval.type === 'relative' && ((0, jsx_runtime_1.jsxs)("div", { style: {
@@ -927,7 +927,7 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                                                             ...newDateFilter,
                                                                                             presetOptions: updatedOptions,
                                                                                         });
-                                                                                    }, theme: Admin_1.defaultTheme, hideEmptyOption: true, style: {
+                                                                                    }, theme: state.theme, hideEmptyOption: true, style: {
                                                                                         width: 120,
                                                                                         maxWidth: 120,
                                                                                         minWidth: 120,
@@ -1271,7 +1271,7 @@ function EditFiltersModal({ isOpen, selectedDashboardName, setIsOpen, onSave, Mo
                                                                         alignItems: 'center',
                                                                         width: '100%',
                                                                         gap: 10,
-                                                                    }, children: (0, jsx_runtime_1.jsx)(SegmentedControl_1.SegmentedControl, { theme: Admin_1.defaultTheme, value: filter && filter.stringFilterType
+                                                                    }, children: (0, jsx_runtime_1.jsx)(SegmentedControl_1.SegmentedControl, { theme: state.theme, value: filter && filter.stringFilterType
                                                                             ? filter.stringFilterType
                                                                             : 'default', onChange: (e) => {
                                                                             const updatedFilters = newFilters.map((otherFilter, i) => index === i

package/dist/cjs/primitives/ButtonPrimitive.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ButtonPrimitive.d.ts","sourceRoot":"","sources":["../../../src/primitives/ButtonPrimitive.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAgB,MAAM,UAAU,CAAC;AAGpD,MAAM,MAAM,oBAAoB,GAAG;IACjC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,iBAAiB,EAAE,UAAU,CAAC,KAAK,IAAI,CAAC;IAC1E,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,KAAK,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC;IAC5B,OAAO,CAAC,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;IAC7C,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,IAAI,CAAC,EAAE,QAAQ,GAAG,QAAQ,GAAG,OAAO,CAAC;CACtC,CAAC;AAEF,QAAA,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAC7B,oBAAoB,GAAG;IACrB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAgGF,CAAC;AAEF,eAAe,eAAe,CAAC"}
+{"version":3,"file":"ButtonPrimitive.d.ts","sourceRoot":"","sources":["../../../src/primitives/ButtonPrimitive.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAItC,MAAM,MAAM,oBAAoB,GAAG;IACjC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,iBAAiB,EAAE,UAAU,CAAC,KAAK,IAAI,CAAC;IAC1E,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,KAAK,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC;IAC5B,OAAO,CAAC,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;IAC7C,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,IAAI,CAAC,EAAE,QAAQ,GAAG,QAAQ,GAAG,OAAO,CAAC;CACtC,CAAC;AAEF,QAAA,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAC7B,oBAAoB,GAAG;IACrB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAsGF,CAAC;AAEF,eAAe,eAAe,CAAC"}

package/dist/cjs/primitives/ButtonPrimitive.js

@@ -1,9 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const jsx_runtime_1 = require("react/jsx-runtime");
-const Admin_1 = require("../Admin");
 const UiComponents_1 = require("../components/UiComponents");
-const ButtonPrimitive = ({ onClick, isLoading = false, label, theme = Admin_1.defaultTheme, style, disabled, variant = 'primary', type = 'submit', }) => {
+const AdminProvider_1 = require("../AdminProvider");
+const ButtonPrimitive = ({ onClick, isLoading = false, label, theme, style, disabled, variant = 'primary', type = 'submit', }) => {
+    const { state: { theme: stateTheme }, } = (0, AdminProvider_1.useAdmin)();
+    theme = theme ?? stateTheme;
     if (variant === 'danger') {
         return ((0, jsx_runtime_1.jsx)("button", { type: type, style: {
                 height: 40,
@@ -26,7 +28,6 @@ const ButtonPrimitive = ({ onClick, isLoading = false, label, theme = Admin_1.de
     }
     return ((0, jsx_runtime_1.jsxs)("button", { className: disabled ? 'quill-button-disabled' : 'quill-button', type: type, style: {
             height: 40,
-            color: theme?.backgroundColor ?? 'white',
             display: 'flex',
             borderRadius: 6,
             alignItems: 'center',
@@ -45,13 +46,16 @@ const ButtonPrimitive = ({ onClick, isLoading = false, label, theme = Admin_1.de
             ...style,
         }, onClick: disabled || isLoading ? () => { } : onClick, children: [(0, jsx_runtime_1.jsx)("style", { children: `
       .quill-button {
-        background-color: #212121;
+        background-color: ${theme?.primaryButtonColor ?? '#212121'};
+        color: ${theme?.primaryButtonTextColor ?? 'white'}
       }
       .quill-button-disabled {
-        background-color: rgba(56, 65, 81, 0.9);
+        background-color: ${theme?.primaryButtonColor ?? '#212121'};
+        color: ${theme?.primaryButtonTextColor ?? 'white'}
       }
       .quill-button:hover {
-        background-color: rgba(56, 65, 81, 0.9);
+        background-color: ${theme?.hoverPrimaryButtonColor ?? 'rgba(56, 65, 81, 0.9)'};
+        color: ${theme?.hoverPrimaryButtonTextColor ?? 'white'}
       }` }), isLoading ? ((0, jsx_runtime_1.jsx)("div", { style: { position: 'absolute' }, children: (0, jsx_runtime_1.jsx)(UiComponents_1.LoadingSpinner, {}) })) : null, (0, jsx_runtime_1.jsx)("div", { style: {
                     visibility: isLoading ? 'hidden' : 'visible',
                 }, children: label })] }));

package/dist/cjs/primitives/HeaderPrimitive.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"HeaderPrimitive.d.ts","sourceRoot":"","sources":["../../../src/primitives/HeaderPrimitive.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAgB,MAAM,UAAU,CAAC;AAEpD,KAAK,oBAAoB,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB,CAAC;AAEF,QAAA,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAcnD,CAAC;AAEF,eAAe,eAAe,CAAC"}
+{"version":3,"file":"HeaderPrimitive.d.ts","sourceRoot":"","sources":["../../../src/primitives/HeaderPrimitive.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAGtC,KAAK,oBAAoB,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB,CAAC;AAEF,QAAA,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAiBnD,CAAC;AAEF,eAAe,eAAe,CAAC"}