@quillsql/admin 1.3.7 → 1.3.9

package/dist/cjs/AdminProvider.js

@@ -1,4 +1,7 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.useAdmin = exports.AdminProvider = void 0;
 const jsx_runtime_1 = require("react/jsx-runtime");
@@ -6,6 +9,7 @@ const react_1 = require("react");
 const Admin_1 = require("./Admin");
 const react_2 = require("@quillsql/react");
 const constants_1 = require("./utils/constants");
+const CreateEnvironment_1 = __importDefault(require("./public_components/CreateEnvironment"));
 const AdminContext = (0, react_1.createContext)(undefined);
 function reducer(state, action) {
     switch (action.type) {
@@ -25,8 +29,6 @@ function reducer(state, action) {
             return { ...state, clients: [...state.clients, action.payload] };
         case 'SET_ENVIRONMENT':
             return { ...state, environment: action.payload };
-        case 'SET_ORGANIZATION':
-            return { ...state, organization: action.payload };
         case 'SET_ORGANIZATIONS':
             return { ...state, organizations: action.payload };
         case 'SET_ORGANIZATION_ID':
@@ -59,8 +61,8 @@ function reducer(state, action) {
             return { ...state, navigateToDashboardManager: action.payload };
         case 'SET_NAVIGATE_TO_DASHBOARD_BUILDER':
             return { ...state, navigateToDashboardBuilder: action.payload };
-        case 'SET_ORGANIZATION_ID_SET':
-            return { ...state, organizationIdSet: action.payload };
+        case 'SET_DATABASE_TYPE_MISMATCH':
+            return { ...state, databaseTypeMismatch: action.payload };
         default:
             return state;
     }
@@ -86,13 +88,10 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
                 return {};
             }
             return {
-                activeComponent: savedState.activeComponent || 'Dashboards',
                 selectedDashboard: savedState.selectedDashboard || null,
                 organizationId: savedState.organizationId || null,
-                organizationIdSet: savedState.organizationIdSet || false,
-                organizations: savedState.organizations || [],
-                organization: savedState.organization || null,
                 client: savedState.client || null,
+                publicKey: savedState.publicKey || null,
             };
         }
         catch (err) {
@@ -125,21 +124,17 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
         organizationIdSet: false,
         dateFilter: {},
         domainName: domainName || '',
+        databaseTypeMismatch: { show: false, backendDatabaseType: '' },
         ...loadState(),
     };
     const [state, dispatch] = (0, react_1.useReducer)(reducer, initialState);
     (0, react_1.useEffect)(() => {
         try {
             const stateToSave = {
-                activeComponent: state.activeComponent,
                 selectedDashboard: state.selectedDashboard,
                 organizationId: state.organizationId,
-                organizationIdSet: state.organizationIdSet,
-                organizations: state.organizations,
-                organization: state.organization,
                 publicKey: state.publicKey,
                 client: state.client,
-                domainName: state.domainName,
             };
             const serializedState = JSON.stringify(stateToSave);
             sessionStorage.setItem('adminState', serializedState);
@@ -148,15 +143,10 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
             console.warn('Error writing state to session storage');
         }
     }, [
-        state.activeComponent,
         state.selectedDashboard,
         state.organizationId,
         state.client,
-        state.organizationIdSet,
-        state.organizations,
-        state.organization,
         state.publicKey,
-        state.domainName,
     ]);
     (0, react_1.useEffect)(() => {
         async function fetchClient(publicKey) {
@@ -164,14 +154,26 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
                 return;
             }
             try {
-                const response = await fetch(`${constants_1.QUILL_SERVER}/client/${publicKey}/`, {
-                    method: 'GET',
-                    headers: { Authorization: `Bearer ` },
+                const response = await fetch(`${queryEndpoint}`, {
+                    method: 'POST',
+                    headers: {
+                        ...queryHeaders,
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({
+                        metadata: {
+                            clientId: publicKey,
+                            publicKey,
+                            task: 'client',
+                        },
+                    }),
                 });
                 const data = await response.json();
-                if (!state.client) {
-                    dispatch({ type: 'SET_CLIENT', payload: data.client });
-                }
+                dispatch({
+                    type: 'SET_CLIENT',
+                    payload: initialState.client || data.data.client,
+                });
+                dispatch({ type: 'SET_CLIENTS', payload: data.data.clients });
             }
             catch (error) {
                 console.error('ERROR:', error);
@@ -185,16 +187,27 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
                 return;
             }
             try {
-                const response = await fetch(`${constants_1.QUILL_SERVER}/schema2/${clientId}/`, {
-                    method: 'GET',
-                    headers: { Authorization: `Bearer ` },
+                const response = await fetch(`${queryEndpoint}`, {
+                    method: 'POST',
+                    headers: {
+                        ...queryHeaders,
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({
+                        metadata: {
+                            clientId: publicKey,
+                            publicKey,
+                            task: 'schema',
+                            databaseType: state.client.databaseType,
+                        },
+                    }),
                 });
                 if (!response.ok) {
                     throw new Error('Failed to fetch tables data');
                 }
                 const data = await response.json();
-                dispatch({ type: 'SET_TABLES', payload: data.tables });
-                dispatch({ type: 'SET_SCHEMA', payload: data.tables });
+                dispatch({ type: 'SET_TABLES', payload: data.data.tables });
+                dispatch({ type: 'SET_SCHEMA', payload: data.data.tables });
             }
             catch (error) {
                 console.error('ERROR:', error);
@@ -204,31 +217,6 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
             fetchSchemaAndTables(state.client._id);
         }
     }, [state?.client, publicKey]);
-    (0, react_1.useEffect)(() => {
-        if (publicKey === '' || publicKey === undefined || publicKey === null) {
-            throw new Error('publicKey is required');
-        }
-        const getClients = async () => {
-            try {
-                const url = `${constants_1.QUILL_SERVER}/clients/${publicKey}/`;
-                const response = await fetch(url, {
-                    method: 'GET',
-                    headers: {
-                        Authorization: `Bearer `,
-                    },
-                });
-                if (!response.ok) {
-                    throw new Error(`HTTP error! status: ${response.status}`);
-                }
-                const data = await response.json();
-                dispatch({ type: 'SET_CLIENTS', payload: data });
-            }
-            catch (error) {
-                console.error('Error fetching clients:', error);
-            }
-        };
-        getClients();
-    }, [publicKey]);
     (0, react_1.useEffect)(() => {
         let isSubscribed = true;
         async function getCustomers() {
@@ -241,6 +229,7 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
             };
             try {
                 let response;
+                dispatch({ type: 'SET_LOADING', payload: true });
                 if (state.queryEndpoint) {
                     response = await fetch(state.queryEndpoint, {
                         method: 'POST',
@@ -249,7 +238,11 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
                             'Content-Type': 'application/json',
                         },
                         body: JSON.stringify({
-                            metadata: { clientId: state.client._id, task: 'orgs' },
+                            metadata: {
+                                clientId: state.client._id,
+                                task: 'orgs',
+                                databaseType: state.client.databaseType,
+                            },
                         }),
                         credentials: state.withCredentials ? 'include' : 'omit', // If withCredentials is true, set to 'include'. Otherwise, set to 'omit'.
                     });
@@ -262,38 +255,45 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
                 }
                 const data = await response.json();
                 if (data.status === 'error') {
+                    dispatch({ type: 'SET_LOADING', payload: false });
                     throw new Error(data.error);
                 }
+                if (data.queries.dbMismatched) {
+                    dispatch({
+                        type: 'SET_DATABASE_TYPE_MISMATCH',
+                        payload: {
+                            show: true,
+                            backendDatabaseType: data.queries.backendDatbaseType, // TYPO NEED TO FIX IN NODE and Python SDK
+                        },
+                    });
+                    dispatch({ type: 'SET_LOADING', payload: false });
+                    return;
+                }
                 if (state.queryEndpoint) {
                     dispatch({
                         type: 'SET_ORGANIZATIONS',
-                        payload: [...data.data.rows, { id: '', name: 'All Organizations' }],
+                        payload: [...data.data.rows],
                     });
                 }
                 else {
                     dispatch({
                         type: 'SET_ORGANIZATIONS',
-                        payload: [...data.orgs, { id: '', name: 'All Organizations' }],
+                        payload: [...data.orgs],
                     });
                 }
-                if (!state.organizationIdSet) {
-                    if (state.environment === 'STAGING') {
-                        dispatch({
-                            type: 'SET_ORGANIZATION_ID',
-                            payload: state.client.stagingAdminCustomerId,
-                        });
-                    }
-                    else {
-                        dispatch({
-                            type: 'SET_ORGANIZATION_ID',
-                            payload: state.client.adminCustomerId,
-                        });
-                    }
-                    dispatch({ type: 'SET_ORGANIZATION_ID_SET', payload: true });
-                }
+                dispatch({
+                    type: 'SET_DATABASE_TYPE_MISMATCH',
+                    payload: { show: false, backendDatabaseType: '' },
+                });
+                dispatch({
+                    type: 'SET_ORGANIZATION_ID',
+                    payload: state.organizationId || state.client.adminCustomerId,
+                });
+                dispatch({ type: 'SET_LOADING', payload: false });
             }
             catch (error) {
                 console.error('Error fetching data:', error);
+                dispatch({ type: 'SET_LOADING', payload: false });
             }
         }
         if (isSubscribed && state.client) {
@@ -303,17 +303,6 @@ function AdminProvider({ children, publicKey, queryEndpoint, queryHeaders, withC
             isSubscribed = false;
         };
     }, [state.environment, state.client, state.organizationId, publicKey]);
-    (0, react_1.useEffect)(() => {
-        if (state.client &&
-            state.clients &&
-            state.organizations &&
-            state.organizationIdSet) {
-            dispatch({ type: 'SET_LOADING', payload: false });
-        }
-        else {
-            dispatch({ type: 'SET_LOADING', payload: true });
-        }
-    }, [state.client, state.clients, state.organizations, state.organizationId]);
-    return ((0, jsx_runtime_1.jsx)(AdminContext.Provider, { value: { state, dispatch }, children: !state.loading && ((0, jsx_runtime_1.jsx)(react_2.QuillProvider, { organizationId: state.organizationId, queryEndpoint: state.queryEndpoint ?? '', withCredentials: state.withCredentials, queryHeaders: state.queryHeaders ?? {}, publicKey: state.client._id, environment: state.environment, theme: state.theme, children: children })) }));
+    return ((0, jsx_runtime_1.jsxs)(AdminContext.Provider, { value: { state, dispatch }, children: [!state.client && !state.loading && (0, jsx_runtime_1.jsx)(CreateEnvironment_1.default, {}), state.client && ((0, jsx_runtime_1.jsx)(react_2.QuillProvider, { organizationId: state.organizationId, queryEndpoint: state.queryEndpoint ?? '', withCredentials: state.withCredentials, queryHeaders: state.queryHeaders ?? {}, publicKey: state.client._id, environment: state.environment, theme: state.theme, children: children }))] }));
 }
 exports.AdminProvider = AdminProvider;

package/dist/cjs/api/ConnectionClient.d.ts

@@ -1,24 +1,26 @@
 import { ConnectionError, ConnectionType } from '../utils/databases';
 import { SchemaDetails } from '../utils/schema';
 import { Column, ColumnsByTable } from '../utils/table';
-export declare function testConnectionString(connectionString: string): Promise<any>;
-export declare function getSchemaTableInfo(connectionString: string, schema: string): Promise<SchemaDetails | ConnectionError>;
-export declare function getTablesBySchema(connectionString: string, schema: string): Promise<string[] | undefined>;
-export declare function getTableInfo(connectionString: string, schema: string, table: string): Promise<SchemaDetails>;
-export declare function getOrgInfo(connectionString: string, schema: string, table: string, orgIdColumn: string, orgColumn: string): Promise<{
+export declare function testConnectionString(connectionString: string, databaseType: string, clientId?: string): Promise<any>;
+export declare function getSchemaTableInfo(clientId: string, schema: string, databaseType: string): Promise<SchemaDetails | ConnectionError>;
+export declare function getTablesBySchema(clientId: string, schema: string): Promise<string[] | undefined>;
+export declare function getTableInfo(clientId: string, schema: string, table: string): Promise<SchemaDetails>;
+export declare function getOrgInfo(clientId: string, schema: string, table: string, orgIdColumn: string, orgColumn: string): Promise<{
     name: string;
     id: any;
 }[]>;
-export declare function getFKInfo(connectionString: string, schema: string, table: string, primaryKey: string): Promise<string[]>;
-export declare function getTableSchema(connectionString: string, tables: string[]): Promise<ColumnsByTable[]>;
+export declare function getFKInfo(clientId: string, schema: string, table: string, primaryKey: string): Promise<string[]>;
+export declare function getTableSchema(clientId: string, tables: string[], schemaName: string): Promise<ColumnsByTable[]>;
+export declare function getTableColumnsBySchema(clientId: string, schemaNames: string[], databaseType: string, queryEndpoint?: string): Promise<ColumnsByTable[]>;
 export declare function getSQLViews(tableSchemas: any, customerNameField: string): Promise<any>;
-export declare function getSqlViewData(connectionString: string, query: string): Promise<any>;
+export declare function getSqlViewData(clientId: string, query: string, databaseType: string, queryEndpoint?: string): Promise<any>;
 export declare function getQueryFromAI(aiPrompt: string, columnsByTable: ColumnsByTable[]): Promise<string>;
+export declare function getQueryFromAiWithConnection(aiPrompt: string, clientId: string, schemaNames?: string[]): Promise<string>;
 export declare function createSQLView(connectionType: ConnectionType, allSchemaInfo: SchemaDetails, columnsByTable: ColumnsByTable[], viewInfo: {
     name: string;
     query: string;
 }, client: any, domainName: string | undefined, tableData: {
     rows: object[];
-    columns: Column[];
-} | undefined, currentTableNames: string[], override?: boolean): Promise<any>;
+    fields: Column[];
+} | undefined, currentTableNames: string[], override: boolean | undefined, clientId: string): Promise<any>;
 //# sourceMappingURL=ConnectionClient.d.ts.map

package/dist/cjs/api/ConnectionClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ConnectionClient.d.ts","sourceRoot":"","sources":["../../../src/api/ConnectionClient.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EACf,cAAc,EAEf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAExD,wBAAsB,oBAAoB,CAAC,gBAAgB,EAAE,MAAM,gBAclE;AAED,wBAAsB,kBAAkB,CACtC,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,aAAa,GAAG,eAAe,CAAC,CAe1C;AACD,wBAAsB,iBAAiB,CACrC,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC,CAe/B;AAED,wBAAsB,YAAY,CAChC,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,aAAa,CAAC,CAgBxB;AAED,wBAAsB,UAAU,CAC9B,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,GAAG,CAAA;CAAE,EAAE,CAAC,CAkBtC;AAED,wBAAsB,SAAS,CAC7B,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,MAAM,EAAE,CAAC,CA6BnB;AAED,wBAAsB,cAAc,CAClC,gBAAgB,EAAE,MAAM,EACxB,MAAM,EAAE,MAAM,EAAE,GACf,OAAO,CAAC,cAAc,EAAE,CAAC,CAe3B;AAED,wBAAsB,WAAW,CAC/B,YAAY,EAAE,GAAG,EACjB,iBAAiB,EAAE,MAAM,GACxB,OAAO,CAAC,GAAG,CAAC,CAqBd;AAED,wBAAsB,cAAc,CAClC,gBAAgB,EAAE,MAAM,EACxB,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,GAAG,CAAC,CAed;AAED,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,cAAc,EAAE,GAC/B,OAAO,CAAC,MAAM,CAAC,CAejB;AAED,wBAAsB,aAAa,CACjC,cAAc,EAAE,cAAc,EAC9B,aAAa,EAAE,aAAa,EAC5B,cAAc,EAAE,cAAc,EAAE,EAChC,QAAQ,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,EACzC,MAAM,EAAE,GAAG,EACX,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,SAAS,EACL;IACE,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,EAAE,MAAM,EAAE,CAAC;CACnB,GACD,SAAS,EACb,iBAAiB,EAAE,MAAM,EAAE,EAC3B,QAAQ,UAAQ,GACf,OAAO,CAAC,GAAG,CAAC,CAuEd"}
+{"version":3,"file":"ConnectionClient.d.ts","sourceRoot":"","sources":["../../../src/api/ConnectionClient.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EACf,cAAc,EAEf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAExD,wBAAsB,oBAAoB,CACxC,gBAAgB,EAAE,MAAM,EACxB,YAAY,EAAE,MAAM,EACpB,QAAQ,CAAC,EAAE,MAAM,gBAiBlB;AAED,wBAAsB,kBAAkB,CACtC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,aAAa,GAAG,eAAe,CAAC,CAgB1C;AACD,wBAAsB,iBAAiB,CACrC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC,CAe/B;AAED,wBAAsB,YAAY,CAChC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,aAAa,CAAC,CAgBxB;AAED,wBAAsB,UAAU,CAC9B,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,GAAG,CAAA;CAAE,EAAE,CAAC,CAkBtC;AAED,wBAAsB,SAAS,CAC7B,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,MAAM,EAAE,CAAC,CA6BnB;AAED,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EAAE,EAChB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,cAAc,EAAE,CAAC,CAgB3B;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,MAAM,EAAE,EACrB,YAAY,EAAE,MAAM,EACpB,aAAa,CAAC,EAAE,MAAM,GACrB,OAAO,CAAC,cAAc,EAAE,CAAC,CA4B3B;AAED,wBAAsB,WAAW,CAC/B,YAAY,EAAE,GAAG,EACjB,iBAAiB,EAAE,MAAM,GACxB,OAAO,CAAC,GAAG,CAAC,CAqBd;AAED,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,YAAY,EAAE,MAAM,EACpB,aAAa,CAAC,EAAE,MAAM,GACrB,OAAO,CAAC,GAAG,CAAC,CAmCd;AAED,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,cAAc,EAAE,GAC/B,OAAO,CAAC,MAAM,CAAC,CAejB;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,MAAM,EAAE,GACrB,OAAO,CAAC,MAAM,CAAC,CAgBjB;AAED,wBAAsB,aAAa,CACjC,cAAc,EAAE,cAAc,EAC9B,aAAa,EAAE,aAAa,EAC5B,cAAc,EAAE,cAAc,EAAE,EAChC,QAAQ,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,EACzC,MAAM,EAAE,GAAG,EACX,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,SAAS,EACL;IACE,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,GACD,SAAS,EACb,iBAAiB,EAAE,MAAM,EAAE,EAC3B,QAAQ,qBAAQ,EAChB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,GAAG,CAAC,CA2Ed"}

package/dist/cjs/api/ConnectionClient.js

@@ -1,9 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createSQLView = exports.getQueryFromAI = exports.getSqlViewData = exports.getSQLViews = exports.getTableSchema = exports.getFKInfo = exports.getOrgInfo = exports.getTableInfo = exports.getTablesBySchema = exports.getSchemaTableInfo = exports.testConnectionString = void 0;
+exports.createSQLView = exports.getQueryFromAiWithConnection = exports.getQueryFromAI = exports.getSqlViewData = exports.getSQLViews = exports.getTableColumnsBySchema = exports.getTableSchema = exports.getFKInfo = exports.getOrgInfo = exports.getTableInfo = exports.getTablesBySchema = exports.getSchemaTableInfo = exports.testConnectionString = void 0;
 const constants_1 = require("../utils/constants");
-const databases_1 = require("../utils/databases");
-async function testConnectionString(connectionString) {
+async function testConnectionString(connectionString, databaseType, clientId) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/test-connection`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -12,14 +11,16 @@ async function testConnectionString(connectionString) {
         },
         body: JSON.stringify({
             connectionString,
+            clientId,
+            databaseType,
             task: 'test-connection',
         }),
     });
     const results = await response.json();
-    return results.success;
+    return results;
 }
 exports.testConnectionString = testConnectionString;
-async function getSchemaTableInfo(connectionString, schema) {
+async function getSchemaTableInfo(clientId, schema, databaseType) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/schema-info`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -27,8 +28,9 @@ async function getSchemaTableInfo(connectionString, schema) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             schema,
+            databaseType,
             task: 'schema-info',
         }),
     });
@@ -36,7 +38,7 @@ async function getSchemaTableInfo(connectionString, schema) {
     return results;
 }
 exports.getSchemaTableInfo = getSchemaTableInfo;
-async function getTablesBySchema(connectionString, schema) {
+async function getTablesBySchema(clientId, schema) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/tables-by-schema`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -44,8 +46,8 @@ async function getTablesBySchema(connectionString, schema) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
-            schema,
+            clientId,
+            schemaNames: [schema],
             task: 'tables-by-schema',
         }),
     });
@@ -53,7 +55,7 @@ async function getTablesBySchema(connectionString, schema) {
     return results.tables;
 }
 exports.getTablesBySchema = getTablesBySchema;
-async function getTableInfo(connectionString, schema, table) {
+async function getTableInfo(clientId, schema, table) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/table-info`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -61,7 +63,7 @@ async function getTableInfo(connectionString, schema, table) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             schema,
             table,
             task: 'table-info',
@@ -71,7 +73,7 @@ async function getTableInfo(connectionString, schema, table) {
     return results;
 }
 exports.getTableInfo = getTableInfo;
-async function getOrgInfo(connectionString, schema, table, orgIdColumn, orgColumn) {
+async function getOrgInfo(clientId, schema, table, orgIdColumn, orgColumn) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/org-info`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -79,7 +81,7 @@ async function getOrgInfo(connectionString, schema, table, orgIdColumn, orgColum
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             schema,
             table,
             orgColumn,
@@ -91,7 +93,7 @@ async function getOrgInfo(connectionString, schema, table, orgIdColumn, orgColum
     return results.orgStrings;
 }
 exports.getOrgInfo = getOrgInfo;
-async function getFKInfo(connectionString, schema, table, primaryKey) {
+async function getFKInfo(clientId, schema, table, primaryKey) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/fk-info`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -99,7 +101,7 @@ async function getFKInfo(connectionString, schema, table, primaryKey) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             schema,
             table,
             primaryKey,
@@ -119,7 +121,7 @@ async function getFKInfo(connectionString, schema, table, primaryKey) {
     return results.foreignKeysString;
 }
 exports.getFKInfo = getFKInfo;
-async function getTableSchema(connectionString, tables) {
+async function getTableSchema(clientId, tables, schemaName) {
     const localURL = `${constants_1.QUILL_SERVER}/sdk/table-schemas`;
     const response = await fetch(localURL, {
         method: 'POST',
@@ -127,8 +129,9 @@ async function getTableSchema(connectionString, tables) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             tables,
+            schemaName,
             task: 'table-schemas',
         }),
     });
@@ -136,6 +139,36 @@ async function getTableSchema(connectionString, tables) {
     return results;
 }
 exports.getTableSchema = getTableSchema;
+async function getTableColumnsBySchema(clientId, schemaNames, databaseType, queryEndpoint) {
+    const localURL = queryEndpoint ?? `${constants_1.QUILL_SERVER}/table-info-by-schema`;
+    const response = await fetch(localURL, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+            clientId,
+            schemaNames,
+            task: 'table-schemas',
+            metadata: {
+                task: 'table-schemas',
+                preQueries: ['temp'],
+                clientId,
+                runQueryConfig: {
+                    overridePost: true,
+                    convertDatatypes: true,
+                    getTables: true,
+                    schema: schemaNames[0],
+                    schemaNames: schemaNames,
+                },
+                databaseType: databaseType,
+            },
+        }),
+    });
+    const results = await response.json();
+    return queryEndpoint ? results.data.queryResults : results;
+}
+exports.getTableColumnsBySchema = getTableColumnsBySchema;
 async function getSQLViews(tableSchemas, customerNameField) {
     const localURL = `${constants_1.QUILL_SERVER}/sqlviewai`;
     const response = await fetch(localURL, {
@@ -159,21 +192,40 @@ async function getSQLViews(tableSchemas, customerNameField) {
     return results.notSureWhatThisIs.sqlQueries;
 }
 exports.getSQLViews = getSQLViews;
-async function getSqlViewData(connectionString, query) {
-    const localURL = `${constants_1.QUILL_SERVER}/sdk/query-view`;
-    const response = await fetch(localURL, {
+async function getSqlViewData(clientId, query, databaseType, queryEndpoint) {
+    const url = queryEndpoint ?? `${constants_1.QUILL_SERVER}/sdk/query-view`;
+    const response = await fetch(url, {
         method: 'POST',
         headers: {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            connectionString,
+            clientId,
             query,
+            metadata: {
+                task: 'query-view',
+                preQueries: [query],
+                clientId,
+                runQueryConfig: {
+                    overridePost: true,
+                    convertDatatypes: true,
+                    limitThousand: true,
+                },
+                databaseType: databaseType,
+            },
             task: 'query-view',
         }),
     });
     const results = await response.json();
-    return results;
+    if (results.status === 'error') {
+        return { ...results, success: false };
+    }
+    return queryEndpoint
+        ? {
+            ...results.data.queryResults[0],
+            success: results.status === 'success' ? true : false,
+        }
+        : results;
 }
 exports.getSqlViewData = getSqlViewData;
 async function getQueryFromAI(aiPrompt, columnsByTable) {
@@ -193,7 +245,25 @@ async function getQueryFromAI(aiPrompt, columnsByTable) {
     return results.message;
 }
 exports.getQueryFromAI = getQueryFromAI;
-async function createSQLView(connectionType, allSchemaInfo, columnsByTable, viewInfo, client, domainName, tableData, currentTableNames, override = false) {
+async function getQueryFromAiWithConnection(aiPrompt, clientId, schemaNames) {
+    const localURL = `${constants_1.QUILL_SERVER}/ai-from-client-schema`;
+    const response = await fetch(localURL, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+            prompt: aiPrompt,
+            clientId,
+            schemaNames,
+            task: 'sql-generator',
+        }),
+    });
+    const results = await response.json();
+    return results.message;
+}
+exports.getQueryFromAiWithConnection = getQueryFromAiWithConnection;
+async function createSQLView(connectionType, allSchemaInfo, columnsByTable, viewInfo, client, domainName, tableData, currentTableNames, override = false, clientId) {
     let queryName = viewInfo.name;
     let validQuery = false;
     let localURL = `${constants_1.QUILL_SERVER}/astify`;
@@ -204,8 +274,9 @@ async function createSQLView(connectionType, allSchemaInfo, columnsByTable, view
         },
         body: JSON.stringify({ query: viewInfo.query }),
     });
-    tableData?.columns.forEach((column) => {
-        if (column.name === allSchemaInfo.selectedFields.foreignKey) {
+    tableData?.fields.forEach((column) => {
+        if (column.name ===
+            allSchemaInfo.selectedFields.foreignKey.replaceAll('"', '')) {
             validQuery = true;
         }
     });
@@ -223,27 +294,28 @@ async function createSQLView(connectionType, allSchemaInfo, columnsByTable, view
     if (currentTableNames.includes(queryName)) {
         return { success: false, message: 'Query name already exists' };
     }
-    const columns = columnsByTable.find((table) => table.tableName === allSchemaInfo.selectedFields.table).columns;
+    const columns = columnsByTable.find((table) => table.tableName === `${allSchemaInfo.selectedFields.schema}.${allSchemaInfo.selectedFields.table}`).columns;
     const customerFieldType = columns.find((column) => column.columnName === allSchemaInfo.selectedFields.organizationIdField).fieldType;
     const fuzzyType = customerFieldType === 'string' || customerFieldType === 'uuid'
         ? 'text'
         : 'number';
     localURL = `${constants_1.QUILL_SERVER}/sdk/create-environment`;
     const payload = {
-        databaseConnectionString: (0, databases_1.formConnectionString)(connectionType),
-        databaseType: connectionType.type,
         customerFieldName: allSchemaInfo.selectedFields.foreignKey,
-        customerTableName: allSchemaInfo.selectedFields.table,
+        customerTableName: `${allSchemaInfo.selectedFields.schema}.${allSchemaInfo.selectedFields.table}`,
         customerTableFieldName: allSchemaInfo.selectedFields.organizationIdField,
         adminCustomerId: allSchemaInfo.selectedFields.defaultOrg?.id,
         customerTableTitleFieldName: allSchemaInfo.selectedFields.customerNameField,
-        columns: tableData?.columns,
+        columns: tableData?.fields,
         task: 'create-environment',
         viewQuery: viewInfo.query,
         customerFieldType: fuzzyType,
         name: queryName,
         client,
         domainName,
+        schemaName: allSchemaInfo.selectedFields.schema,
+        schemaNames: [allSchemaInfo.selectedFields.schema],
+        clientId,
     };
     const response = await fetch(localURL, {
         method: 'POST',

package/dist/cjs/assets/ArrowDownHeadIcon.d.ts

@@ -0,0 +1,5 @@
+declare const ArrowDownHeadIcon: ({ ...props }: {
+    [x: string]: any;
+}) => import("react/jsx-runtime").JSX.Element;
+export default ArrowDownHeadIcon;
+//# sourceMappingURL=ArrowDownHeadIcon.d.ts.map

package/dist/cjs/assets/ArrowDownHeadIcon.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ArrowDownHeadIcon.d.ts","sourceRoot":"","sources":["../../../src/assets/ArrowDownHeadIcon.tsx"],"names":[],"mappings":"AAGA,QAAA,MAAM,iBAAiB;;6CAKtB,CAAC;AAEF,eAAe,iBAAiB,CAAC"}

package/dist/cjs/assets/ArrowDownHeadIcon.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jsx_runtime_1 = require("react/jsx-runtime");
+const ArrowDownHeadIcon = ({ ...props }) => ((0, jsx_runtime_1.jsx)("svg", { ...props, xmlns: "http://www.w3.org/2000/svg", viewBox: "0 0 24 24", width: "24", height: "24", children: (0, jsx_runtime_1.jsx)("path", { fill: "currentColor", d: "M12 13.172l4.95-4.95 1.414 1.414L12 16 5.636 9.636 7.05 8.222z" }) }));
+exports.default = ArrowDownHeadIcon;

package/dist/cjs/assets/XIcon.d.ts

@@ -0,0 +1,5 @@
+declare const XIcon: ({ ...props }: {
+    [x: string]: any;
+}) => import("react/jsx-runtime").JSX.Element;
+export default XIcon;
+//# sourceMappingURL=XIcon.d.ts.map

package/dist/cjs/assets/XIcon.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"XIcon.d.ts","sourceRoot":"","sources":["../../../src/assets/XIcon.tsx"],"names":[],"mappings":"AAGA,QAAA,MAAM,KAAK;;6CAcV,CAAC;AACF,eAAe,KAAK,CAAC"}

package/dist/cjs/assets/XIcon.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jsx_runtime_1 = require("react/jsx-runtime");
+const XIcon = ({ ...props }) => ((0, jsx_runtime_1.jsx)("svg", { ...props, xmlns: "http://www.w3.org/2000/svg", viewBox: "0 0 24 24", height: "20", width: "20", children: (0, jsx_runtime_1.jsx)("path", { fillRule: "evenodd", d: "M5.47 5.47a.75.75 0 011.06 0L12 10.94l5.47-5.47a.75.75 0 111.06 1.06L13.06 12l5.47 5.47a.75.75 0 11-1.06 1.06L12 13.06l-5.47 5.47a.75.75 0 01-1.06-1.06L10.94 12 5.47 6.53a.75.75 0 010-1.06z", clipRule: "evenodd" }) }));
+exports.default = XIcon;