@quillsql/react 2.13.26 → 2.13.28

package/dist/cjs/ReportBuilder.js

@@ -74,12 +74,13 @@ const constants_2 = require("./utils/constants");
  * ### Report Builder API
  * @see https://docs.quillsql.com/components/report-builder
  */
-function ReportBuilder({ initialTableName = '', onSubmitEditReport = () => void null, onSubmitCreateReport = () => void null, onDiscardChanges = undefined, onSaveChanges = undefined, onCloseChartBuilder = undefined, destinationDashboard = undefined, organizationName = '', ButtonComponent = UiComponents_1.MemoizedButton, SecondaryButtonComponent = UiComponents_1.MemoizedSecondaryButton, DeleteButtonComponent = UiComponents_1.MemoizedDeleteButton, ModalComponent = UiComponents_1.MemoizedModal, TextInputComponent = UiComponents_2.QuillTextInput, SelectComponent = QuillSelect_1.QuillSelectComponent, MultiSelectComponent = QuillMultiSelectWithCombo_1.QuillMultiSelectComponentWithCombo, TableComponent = UiComponents_1.QuillTableReportBuilderComponent, PopoverComponent = UiComponents_1.MemoizedPopover, TabsComponent = UiComponents_1.QuillTabs, 
+function ReportBuilder({ initialTableName = '', onSubmitEditReport = () => void null, onSubmitCreateReport = () => void null, onDiscardChanges = undefined, onSaveChanges = undefined, onCloseChartBuilder = undefined, destinationDashboard = undefined, chartBuilderTitle = undefined, organizationName = '', ButtonComponent = UiComponents_1.MemoizedButton, SecondaryButtonComponent = UiComponents_1.MemoizedSecondaryButton, DeleteButtonComponent = UiComponents_1.MemoizedDeleteButton, ModalComponent = UiComponents_1.MemoizedModal, TextInputComponent = UiComponents_2.QuillTextInput, SelectComponent = QuillSelect_1.QuillSelectComponent, MultiSelectComponent = QuillMultiSelectWithCombo_1.QuillMultiSelectComponentWithCombo, TableComponent = UiComponents_1.QuillTableReportBuilderComponent, PopoverComponent = UiComponents_1.MemoizedPopover, TabsComponent = UiComponents_1.QuillTabs, 
 // CheckboxComponent = MemoizedCheckbox,
-SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer, SelectColumnComponent = ui_1.QuillSelectColumn, DraggableColumnComponent = ui_1.QuillDraggableColumn, SidebarHeadingComponent = ui_1.QuillSidebarHeading, FilterPopoverComponent = ui_1.QuillFilterPopover, SortPopoverComponent = ui_1.QuillSortPopover, LimitPopoverComponent = ui_1.QuillLimitPopover, CardComponent = QuillCard_1.QuillCard, LabelComponent = UiComponents_1.MemoizedLabel, HeaderComponent = UiComponents_1.MemoizedHeader, SubHeaderComponent = UiComponents_1.MemoizedSubHeader, TextComponent = UiComponents_1.MemoizedText, ErrorMessageComponent = UiComponents_1.QuillErrorMessageComponent, ChartBuilderInputRowContainer = UiComponents_1.QuillChartBuilderInputRowContainer, ChartBuilderInputColumnContainer = UiComponents_1.QuillChartBuilderInputColumnContainer, PivotRowContainer = UiComponents_1.QuillPivotRowContainer, PivotColumnContainer = UiComponents_1.QuillPivotColumnContainer, LoadingComponent = UiComponents_1.QuillLoadingComponent, ColumnSearchEmptyState = UiComponents_1.QuillColumnSearchEmptyState, ChartBuilderFormContainer = UiComponents_1.QuillChartBuilderFormContainer, ChartBuilderModalComponent = UiComponents_1.MemoizedModal, isAdminEnabled = false, isAIEnabled = true, containerStyle, className, pivotRecommendationsEnabled = true, reportId, hideCopySQL = true, isChartBuilderHorizontalView = true, onClickChartElement, }) {
+SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer, SelectColumnComponent = ui_1.QuillSelectColumn, DraggableColumnComponent = ui_1.QuillDraggableColumn, SidebarHeadingComponent = ui_1.QuillSidebarHeading, FilterPopoverComponent = ui_1.QuillFilterPopover, SortPopoverComponent = ui_1.QuillSortPopover, LimitPopoverComponent = ui_1.QuillLimitPopover, CardComponent = QuillCard_1.QuillCard, LabelComponent = UiComponents_1.MemoizedLabel, HeaderComponent = UiComponents_1.MemoizedHeader, SubHeaderComponent = UiComponents_1.MemoizedSubHeader, TextComponent = UiComponents_1.MemoizedText, ErrorMessageComponent = UiComponents_1.QuillErrorMessageComponent, ChartBuilderInputRowContainer = UiComponents_1.QuillChartBuilderInputRowContainer, ChartBuilderInputColumnContainer = UiComponents_1.QuillChartBuilderInputColumnContainer, PivotRowContainer = UiComponents_1.QuillPivotRowContainer, PivotColumnContainer = UiComponents_1.QuillPivotColumnContainer, LoadingComponent = UiComponents_1.QuillLoadingComponent, ColumnSearchEmptyState = UiComponents_1.QuillColumnSearchEmptyState, ChartBuilderFormContainer = UiComponents_1.QuillChartBuilderFormContainer, ChartBuilderModalComponent = UiComponents_1.MemoizedModal, isAdminEnabled = false, isAIEnabled = true, containerStyle, className, pivotRecommendationsEnabled = true, reportId, hideCopySQL = true, isChartBuilderHorizontalView = true, onClickChartElement, flagsList, }) {
     const [dashboard, dashboardDispatch] = (0, react_1.useContext)(Context_1.DashboardContext);
     const [schemaData] = (0, react_1.useContext)(Context_1.SchemaDataContext);
     const { dashboardFilters } = (0, react_1.useContext)(Context_1.DashboardFiltersContext);
+    const { tenants } = (0, react_1.useContext)(Context_1.TenantContext);
     const specificDashboardFilters = (0, react_1.useMemo)(() => {
         if (!reportId)
             return [];
@@ -88,6 +89,63 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             return [];
         return Object.values(dashboardFilters[dashboardName] ?? {}).map((f) => f.filter);
     }, [dashboardFilters, reportId, dashboard]);
+    const [client] = (0, react_1.useContext)(Context_1.ClientContext);
+    // const { dashboardConfig } = useContext(DashboardConfigContext);
+    // const flagsToAdd: { [tenantField: string]: string[] } = useMemo(() => {
+    //   if (client && asSingleDatabaseTenant(client))
+    //     return { 'QUILL_SINGLE_TENANT': flags ?? {} } as { [tenantField: string]: string[] };
+    //   const tenantKeys =
+    //     dashboardConfig[
+    //       dashboard[reportId ?? '']?.dashboardName ?? destinationDashboard ?? ''
+    //     ]?.config.tenantKeys ?? [];
+    //   if (!tenantKeys.length) return {};
+    //   if (
+    //     tenants &&
+    //     (typeof tenants[0] === 'string' || typeof tenants[0] === 'number')
+    //   ) {
+    //     const dashboardOwner = tenantKeys[0]!;
+    //     const tenantIds = (client?.allTenantTypes as QuillTenant[])?.find(
+    //       (tt) => dashboardOwner === tt.tenantField,
+    //     )?.tenantIds;
+    //     return {
+    //       [dashboardOwner]:
+    //         tenantIds
+    //           ?.filter(
+    //             (tt) =>
+    //               (tenants as (string | number)[]).includes(tt.id) && tt.flag,
+    //           )
+    //           ?.map((tt) => tt.flag!) ?? [],
+    //     };
+    //   }
+    //   const tenantKeysSet = new Set(tenantKeys);
+    //   return (
+    //     (
+    //       tenants as
+    //         | { tenantField: string; tenantIds: (string | number)[] }[]
+    //         | undefined
+    //     )
+    //       ?.filter((t) => tenantKeysSet.has(t.tenantField))
+    //       .reduce(
+    //         (acc, t) => {
+    //           const tenantIds = new Set(t.tenantIds);
+    //           acc[t.tenantField] =
+    //             (client?.allTenantTypes as QuillTenant[])
+    //               ?.find((tt) => t.tenantField === tt.tenantField)
+    //               ?.tenantIds?.filter((tt) => tenantIds.has(tt.id) && tt.flag)
+    //               ?.map((tt) => tt.flag!) ?? [];
+    //           return acc;
+    //         },
+    //         {} as { [tenantField: string]: string[] },
+    //       ) ?? {}
+    //   );
+    // }, [
+    //   tenants,
+    //   flags,
+    //   dashboardConfig,
+    //   destinationDashboard,
+    //   dashboard[reportId ?? '']?.dashboardName,
+    //   client?.allTenantTypes,
+    // ]);
     const [reportInfo, setReportInfo] = (0, react_1.useState)(null);
     const [aiPrompt, setAiPrompt] = (0, react_1.useState)('');
     const [errorMessage, setErrorMessage] = (0, react_1.useState)('');
@@ -136,12 +194,10 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
     /* eslint-disable-next-line */
     const [pivotError, setPivotError] = (0, react_1.useState)(undefined);
     const [dateRanges, setDateRanges] = (0, react_1.useState)(null);
-    const [client] = (0, react_1.useContext)(Context_1.ClientContext);
-    const [currentOrganizationId, setCurrentOrganizationId] = (0, react_1.useState)(client.organizationId);
     const [fieldValuesMap, setFieldValuesMap] = (0, react_1.useState)({}); // Mapping of unique values per field, used in string filter 'in' and 'not in'
     const [globalUniqueValues, setGlobalUniqueValues] = (0, react_1.useState)({});
     const [globalUniqueValuesIsLoading, setGlobalUniqueValuesIsLoading] = (0, react_1.useState)(false);
-    const { filterTree, filterStack } = (0, useAstToFilterTree_1.default)(formData, client); // Stores the state of filters
+    const { filterTree, filterStack } = (0, useAstToFilterTree_1.default)(formData, client, columns); // Stores the state of filters
     const [removingFilter, setRemovingFilter] = (0, react_1.useState)(false);
     const [pivotRecommendationsEnabledState, setPivotRecommendationsEnabledState,] = (0, react_1.useState)(pivotRecommendationsEnabled);
     const REPORT_BUILDER_PAGINATION = {
@@ -150,6 +206,15 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         rowsPerRequest: 1000,
     };
     const [unresolvedReportMessage, setUnresolvedReportMessage] = (0, react_1.useState)('');
+    const dashboardName = (0, react_1.useMemo)(() => {
+        if (destinationDashboard) {
+            return destinationDashboard;
+        }
+        if (!reportId) {
+            return undefined;
+        }
+        return dashboard[reportId]?.dashboardName;
+    }, [reportId, dashboard, destinationDashboard]);
     const [filtersEnabled, setFiltersEnabled] = (0, react_1.useState)(!!reportId);
     (0, react_1.useEffect)(() => {
         if (!client) {
@@ -197,9 +262,9 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             window.removeEventListener('resize', handleResize);
         };
     }, []);
-    const updatePivot = async (changeField, fieldKey) => {
+    const updatePivot = async (changeField, fieldKey, prevValue) => {
         const newPivot = (0, util_1.deepCopy)(pivot);
-        setTableLoading(true);
+        setPivotError(undefined);
         // @ts-ignore
         newPivot[fieldKey] = changeField;
         if (fieldKey === 'rowField') {
@@ -217,12 +282,20 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                 newPivot.sort = undefined;
             }
         }
+        const { valid, reason } = (0, pivotProcessing_1.isValidPivot)(newPivot);
+        if (!valid) {
+            setPivotError(reason);
+            return;
+        }
+        setTableLoading(true);
         let dateBucket = undefined;
         const tempDateRange = dateRanges && dateRanges[newPivot.rowField || ''];
         if (tempDateRange) {
             dateBucket = (0, dates_1.getDateBucketFromRange)(tempDateRange.dateRange);
         }
         let distinctValuesForQuery = {};
+        const prevPivot = pivot;
+        const prevPivotData = pivotData;
         setPivot(newPivot);
         setPivotHint('');
         if (newPivot.columnField) {
@@ -232,7 +305,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                     label: newPivot.columnField,
                     format: 'string',
                 },
-            ], activeQuery, [], client, schemaData.customFields ?? []);
+            ], activeQuery, [], client, tenants, schemaData.customFields ?? [], undefined, dashboardName);
         }
         try {
             const pivotedData = await (0, PivotModal_1.generatePivotTable)({
@@ -241,12 +314,33 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                 report: tempReport,
                 client,
                 uniqueValues: distinctValuesForQuery,
+                dashboardName,
+                tenants,
             });
             setPivotData(pivotedData || []);
             const formattedRows = formatRows(pivotedData.rows, columns, true, newPivot.aggregationType);
             setFormattedRows(formattedRows);
         }
         catch (e) {
+            switch (fieldKey) {
+                case 'rowField':
+                    setPivotRowField(prevValue);
+                    break;
+                case 'columnField':
+                    setPivotColumnField(prevValue);
+                    break;
+                case 'valueField':
+                    setPivotValueField(prevValue);
+                    break;
+                case 'valueField2':
+                    setPivotValueField2(prevValue);
+                    break;
+                case 'aggregationType':
+                    setPivotAggregation(prevValue);
+                    break;
+            }
+            setPivot(prevPivot);
+            setPivotData(prevPivotData);
             if (e instanceof Error)
                 setPivotError(e.message);
         }
@@ -284,9 +378,12 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             setDataDisplayed(false);
             setRows([]);
             setColumns([]);
+            setSelectedOrderedColumns([]);
             setErrorMessage('');
+            setPivotError(undefined);
             setFormattedRows([]);
-            // setUniqueValues({});
+            setCurrentTable('');
+            setUniqueValues({});
             setPivot(null);
             setPivotHint('');
             setPivotData(null);
@@ -295,7 +392,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
     };
     const fetchGlobalUniqueValues = async (columns, currentTable, client) => {
         setGlobalUniqueValuesIsLoading(true);
-        const global = await (0, tableProcessing_1.getUniqueStringValues)(columns, currentTable, client, schemaData.customFields, undefined, true);
+        const global = await (0, tableProcessing_1.getUniqueStringValues)(columns, currentTable, client, tenants, schemaData.customFields, undefined, true, undefined, dashboardName);
         setGlobalUniqueValues((0, filterProcessing_1.uniqueValuesToStringMap)(global));
         setGlobalUniqueValuesIsLoading(false);
     };
@@ -418,7 +515,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         const dateColumnNames = dateColumns.map((column) => {
             return column.field;
         });
-        const dateRanges = await (0, tableProcessing_1.getQueryDateRangeByColumns)(dateColumns, `Select ${dateColumnNames.join(', ')} from ${tableName}`, client, schemaData.customFields ?? []);
+        const dateRanges = await (0, tableProcessing_1.getQueryDateRangeByColumns)(dateColumns, `Select ${dateColumnNames.join(', ')} from ${tableName}`, client, tenants, schemaData.customFields ?? [], dashboardName);
         return dateRanges;
     };
     // It's just like getColumnsInPivot but we expand the columnField
@@ -448,7 +545,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         const tableInfo = tables.find((tableInfo) => tableInfo.name === initialTableName);
         if (tableInfo) {
             setUniqueValuesIsLoading(true);
-            const newUniqueValues = await (0, tableProcessing_1.getUniqueStringValues)(tableInfo.columns, initialTableName, client, schemaData.customFields, uniqueValues, true);
+            const newUniqueValues = await (0, tableProcessing_1.getUniqueStringValues)(tableInfo.columns, initialTableName, client, tenants, schemaData.customFields, uniqueValues, true, undefined, dashboardName);
             if ((0, crypto_1.hashCode)(uniqueValues) !== (0, crypto_1.hashCode)(newUniqueValues)) {
                 setUniqueValues(newUniqueValues);
                 updateFieldValuesMap(newUniqueValues, initialTableName);
@@ -611,16 +708,6 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
     const [numberOfRows, setNumberOfRows] = (0, react_1.useState)(0);
     const [rowCountIsLoading, setRowCountIsLoading] = (0, react_1.useState)(false);
     const [tableLoading, setTableLoading] = (0, react_1.useState)(false);
-    // const initialData = useMemo(() => ({
-    //   rows,
-    //   columns,
-    //   itemQuery: tempReport.itemQuery ?? [],
-    //   rowCount: numberOfRows ?? 0,
-    //   dashboardName: destinationDashboard ?? reportInfo?.dashboardName,
-    //   query: activeQuery,
-    //   pivot: pivot,
-    //   pivotData: pivotData,
-    // }), [rows, columns, tempReport, activeQuery, pivot, pivotData, numberOfRows, destinationDashboard, reportInfo]);
     const onPageChange = (page, initiator = 'ReportBuilder') => {
         const pagination = initiator === 'ReportBuilder'
             ? REPORT_BUILDER_PAGINATION
@@ -636,36 +723,30 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             setPreviousPage(page);
         }
     };
-    const onSortChange = (sort, persistSort = false) => {
-        // if (shouldSortInMemory(REPORT_BUILDER_PAGINATION, numberOfRows, !!pivot)) {
-        //   return;
-        // }
-        const updatedProcessing = { page: REPORT_BUILDER_PAGINATION, sort };
-        if (persistSort) {
-            const newAst = { ...baseAst };
-            if (!newAst.orderby) {
-                newAst.orderby = [];
-            }
-            const existingSortIndex = newAst.orderby.findIndex((item) => (0, astFilterProcessing_1.getFieldFromExpression)(item.expr) === sort.field);
-            if (existingSortIndex !== -1) {
-                newAst.orderby[existingSortIndex] = {
-                    expr: { type: 'column_ref', column: sort.field },
-                    type: sort.direction.toUpperCase(),
-                };
-            }
-            else {
-                newAst.orderby.push({
-                    expr: { type: 'column_ref', column: sort.field },
-                    type: sort.direction.toUpperCase(),
-                });
-            }
-            setBaseAst((0, util_1.deepCopy)(newAst));
-            fetchReportFromASTHelper(newAst, newAst.where);
+    const onSortChange = (sort) => {
+        if ((0, paginationProcessing_1.shouldSortInMemory)(REPORT_BUILDER_PAGINATION, numberOfRows) ||
+            (pivot && pivot.rowField !== sort.field)) {
+            return;
+        }
+        const newAst = { ...baseAst };
+        if (!newAst.orderby) {
+            newAst.orderby = [];
+        }
+        const existingSortIndex = newAst.orderby.findIndex((item) => (0, astFilterProcessing_1.getFieldFromExpression)(item.expr) === sort.field);
+        if (existingSortIndex !== -1) {
+            newAst.orderby[existingSortIndex] = {
+                expr: { type: 'column_ref', column: sort.field },
+                type: sort.direction.toUpperCase(),
+            };
         }
         else {
-            handleRunQuery(updatedProcessing, true);
+            newAst.orderby.push({
+                expr: { type: 'column_ref', column: sort.field },
+                type: sort.direction.toUpperCase(),
+            });
         }
-        setCurrentProcessing(updatedProcessing);
+        setBaseAst((0, util_1.deepCopy)(newAst));
+        fetchReportFromASTHelper(newAst, newAst.where);
         setPreviousPage(0);
     };
     const fetchRowCount = async (processing, includeFilters) => {
@@ -673,9 +754,9 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             return;
         }
         setRowCountIsLoading(true);
-        const tableInfo = await (0, tableProcessing_1.fetchResultsByQuery)(activeQuery, client, processing, schemaData.customFields, includeFilters ? specificDashboardFilters : undefined, includeFilters
+        const tableInfo = await (0, tableProcessing_1.fetchResultsByQuery)(activeQuery, client, tenants, processing, schemaData.customFields, includeFilters ? specificDashboardFilters : undefined, includeFilters
             ? (tempReport.dateField ?? reportInfo?.dateField)
-            : undefined, false, true);
+            : undefined, false, true, undefined, dashboardName);
         if (tableInfo.rowCount) {
             setNumberOfRows(tableInfo.rowCount);
             // @ts-ignore
@@ -688,7 +769,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
     };
     const fetchRowCountFromAST = async (ast, where) => {
         setRowCountIsLoading(true);
-        const tableData = await (0, tableProcessing_1.fetchTableByAST)({ ...ast, where }, client, { page: REPORT_BUILDER_PAGINATION }, undefined, false, true);
+        const tableData = await (0, tableProcessing_1.fetchTableByAST)({ ...ast, where }, client, tenants, { page: REPORT_BUILDER_PAGINATION }, undefined, false, true, dashboardName);
         if (tableData.rowCount) {
             setNumberOfRows(tableData.rowCount);
             // @ts-ignore
@@ -703,7 +784,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         try {
             setErrorMessage('');
             setTableLoading(true);
-            const tableInfo = await (0, tableProcessing_1.fetchResultsByQuery)(activeQuery, client, processing, schemaData.customFields, undefined, undefined, true);
+            const tableInfo = await (0, tableProcessing_1.fetchResultsByQuery)(activeQuery, client, tenants, processing, schemaData.customFields, undefined, undefined, true, undefined, undefined, dashboardName);
             if (tableInfo.error) {
                 throw new Error(tableInfo.error);
             }
@@ -711,7 +792,6 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                 throw new Error('No data found');
             }
             fetchRowCount(processing, includeFilters);
-            setCurrentProcessing(processing);
             let tempRows = [...rows, ...tableInfo.rows];
             if (resetRows) {
                 tempRows = tableInfo.rows;
@@ -770,7 +850,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         let reportBuilderInfo = undefined;
         try {
             setLoading(true);
-            reportBuilderInfo = await (0, report_1.fetchReportBuilderDataFromAST)(baseAst, curFormData, curSchema ?? schemaData.schema, client, curPivot ?? pivot, formData, currentTable, { uniqueStrings: uniqueValues, dateRanges: dateRanges ?? {} }, previousReport ?? reportInfo ?? undefined, schemaData.customFields, true, true, { page: REPORT_BUILDER_PAGINATION });
+            reportBuilderInfo = await (0, report_1.fetchReportBuilderDataFromAST)(baseAst, curFormData, curSchema ?? schemaData.schema, client, tenants, curPivot ?? pivot, formData, currentTable, { uniqueStrings: uniqueValues, dateRanges: dateRanges ?? {} }, previousReport ?? reportInfo ?? undefined, schemaData.customFields, true, true, { page: REPORT_BUILDER_PAGINATION }, dashboardName);
             if (reportBuilderInfo.error) {
                 throw new Error(reportBuilderInfo.error);
             }
@@ -786,6 +866,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             return { error: true, message: 'Failed to fetch', rows: [] };
         }
         if (!reportBuilderInfo) {
+            setLoading(false);
             setErrorMessage('Failed to fetch');
             return;
         }
@@ -797,7 +878,6 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         });
         const prevFormData = formData;
         const prevTable = currentTable;
-        const prevOrganization = currentOrganizationId;
         setRows(reportBuilderInfo.rows);
         setPreviousPage(0);
         if (!(client.databaseType.toLowerCase() === 'bigquery') ||
@@ -805,6 +885,9 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             setColumns(reportBuilderInfo.columns);
             setSelectedColumns(reportBuilderInfoColumns);
         }
+        if (prevTable !== reportBuilderInfo.table) {
+            setSelectedOrderedColumns([]); // reset selected ordered columns
+        }
         setNumberOfRows(reportBuilderInfo.rowCount);
         setPivot(reportBuilderInfo.pivot);
         if (!keepPivotHint) {
@@ -815,7 +898,6 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         setFormattedRows(reportBuilderInfo.formattedRows);
         setDataDisplayed(true);
         setCurrentTable(reportBuilderInfo.table);
-        setCurrentOrganizationId(client.organizationId);
         setFormData(curFormData);
         if (reportBuilderInfo.pivot) {
             setPivotRowField(reportBuilderInfo.pivot.rowField);
@@ -851,8 +933,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         }
         // fetch unique values after everything else since it is the most expensive
         if (prevTable !== reportBuilderInfo.table ||
-            !(0, fast_deep_equal_1.default)(prevFormData, curFormData) ||
-            prevOrganization !== client.organizationId) {
+            !(0, fast_deep_equal_1.default)(prevFormData, curFormData)) {
             try {
                 setUniqueValuesIsLoading(true);
                 if (reportBuilderInfo.pivot) {
@@ -861,7 +942,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                 if (!tableInfo) {
                     throw new Error('Table info not found');
                 }
-                const uniqueStrings = await (0, tableProcessing_1.getUniqueStringValues)(tableInfo.columns, reportBuilderInfo.table, client, schemaData.customFields, undefined, true, query);
+                const uniqueStrings = await (0, tableProcessing_1.getUniqueStringValues)(tableInfo.columns, reportBuilderInfo.table, client, tenants, schemaData.customFields, undefined, true, query, dashboardName);
                 const newUnique = updateUniqueValue(uniqueStrings, reportBuilderInfo.table);
                 let pivotChanged = false;
                 let newPivot;
@@ -915,7 +996,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             setLoading(true);
             setAskAILoading(true);
             setErrorMessage('');
-            astInfo = await (0, astProcessing_1.fetchAndProcessASTFromPrompt)(prompt, schemaData.schema, client, pivot, activeQuery, currentTable);
+            astInfo = await (0, astProcessing_1.fetchAndProcessASTFromPrompt)(prompt, schemaData.schema, client, pivot, activeQuery, currentTable, dashboardName);
             if (astInfo.error) {
                 throw new Error(astInfo.error);
             }
@@ -935,8 +1016,13 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
             astInfo.pivot = null;
             astInfo.ast.groupby = null;
         }
+        const _tables = (0, ast_1.getTableNames)(astInfo.ast);
+        const _table = _tables[0] ? _tables[0] : '';
+        const _columns = schemaData.schema.find((table) => {
+            return table.name === _table;
+        })?.columns;
         // parse the whereAst first
-        const filterTree = (0, astFilterProcessing_1.astToFilterTree)(astInfo.whereAST, client.databaseType?.toLowerCase());
+        const filterTree = (0, astFilterProcessing_1.astToFilterTree)(astInfo.whereAST, client.databaseType?.toLowerCase(), _columns);
         const cleanAst = filterTree
             ? (0, astFilterProcessing_1.filterTreeToAst)(filterTree, client.databaseType?.toLowerCase())
             : null;
@@ -978,7 +1064,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
         return { pivot: null, hint: '' };
     };
     const isValidPivotForReport = (pivot, uniqueValuesForPivot, reportTable, reportColumns) => {
-        if (!(0, pivotProcessing_1.isValidPivot)(pivot)) {
+        if (!(0, pivotProcessing_1.isValidPivot)(pivot).valid) {
             return false;
         }
         const uniqueValuesToCheck = uniqueValuesForPivot ?? uniqueValues;
@@ -1161,7 +1247,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                             flexDirection: 'column',
                                             gap: 8,
                                             marginBottom: 12,
-                                        }, children: (0, jsx_runtime_1.jsx)(FilterStack_1.default, { client: client, filterStack: filterStack, filterTree: filterTree, defaultAST: constants_1.defaultAST, baseAst: baseAst, setBaseAst: setBaseAst, setFormData: setFormData, fetchSqlQuery: fetchSqlQuery, schemaData: schemaData, defaultColumn: constants_1.defaultColumn, defaultTable: constants_1.defaultTable, currentTable: currentTable, initialTableName: initialTableName, globalUniqueValues: globalUniqueValues, globalUniqueValuesIsLoading: globalUniqueValuesIsLoading, columns: columns, removingFilter: removingFilter, setRemovingFilter: setRemovingFilter, TabsComponent: TabsComponent, FilterPopoverComponent: FilterPopoverComponent, FilterModal: FilterModal_1.default, ButtonComponent: ButtonComponent, SecondaryButtonComponent: SecondaryButtonComponent, SelectComponent: SelectComponent, TextInputComponent: TextInputComponent, MultiSelectComponent: MultiSelectComponent, actionsEnabled: !tableLoading && !loading }) })), (0, jsx_runtime_1.jsxs)("div", { style: {
+                                        }, children: (0, jsx_runtime_1.jsx)(FilterStack_1.default, { client: client, filterStack: filterStack, filterTree: filterTree, defaultAST: constants_1.defaultAST, baseAst: baseAst, setBaseAst: setBaseAst, setFormData: setFormData, fetchSqlQuery: fetchSqlQuery, schemaData: schemaData, defaultColumn: constants_1.defaultColumn, defaultTable: constants_1.defaultTable, currentTable: currentTable, initialTableName: initialTableName, globalUniqueValues: globalUniqueValues, globalUniqueValuesIsLoading: globalUniqueValuesIsLoading, columns: columns, removingFilter: removingFilter, setRemovingFilter: setRemovingFilter, TabsComponent: TabsComponent, FilterPopoverComponent: FilterPopoverComponent, FilterModal: FilterModal_1.default, ButtonComponent: ButtonComponent, SecondaryButtonComponent: SecondaryButtonComponent, SelectComponent: SelectComponent, TextInputComponent: TextInputComponent, MultiSelectComponent: MultiSelectComponent, actionsEnabled: !tableLoading && !loading, dashboardName: dashboardName }) })), (0, jsx_runtime_1.jsxs)("div", { style: {
                                             display: 'flex',
                                             flexDirection: 'column',
                                             alignItems: 'flex-start',
@@ -1234,6 +1320,8 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                                         report: tempReport,
                                                         client,
                                                         uniqueValues,
+                                                        dashboardName,
+                                                        tenants,
                                                     });
                                                 }
                                                 setPivotData(pivotTable || []);
@@ -1249,23 +1337,29 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                             finally {
                                                 setTableLoading(false);
                                             }
-                                        }, selectPivotOnEdit: true, showTrigger: !pivot, theme: theme, LabelComponent: LabelComponent, HeaderComponent: HeaderComponent, dateRange: undefined, pivotCountRequest: 4, query: activeQuery, initialUniqueValues: uniqueValues[currentTable], uniqueValuesIsLoading: uniqueValuesIsLoading, disabled: !baseAst || !dataDisplayed || tableLoading || loading, pivotRecommendationsEnabled: pivotRecommendationsEnabledState, report: tempReport }), pivot && ((0, jsx_runtime_1.jsx)(PivotForm_1.default, { columns: columns, uniqueValues: uniqueValues[currentTable], uniqueValuesIsLoading: uniqueValuesIsLoading, setPivotRowField: (value) => {
+                                        }, selectPivotOnEdit: true, showTrigger: !pivot, theme: theme, LabelComponent: LabelComponent, HeaderComponent: HeaderComponent, dateRange: undefined, pivotCountRequest: 4, query: activeQuery, initialUniqueValues: uniqueValues[currentTable], uniqueValuesIsLoading: uniqueValuesIsLoading, disabled: !baseAst || !dataDisplayed || tableLoading || loading, pivotRecommendationsEnabled: pivotRecommendationsEnabledState, report: tempReport, dashboardName: dashboardName }), pivot && ((0, jsx_runtime_1.jsx)(PivotForm_1.default, { columns: columns, uniqueValues: uniqueValues[currentTable], uniqueValuesIsLoading: uniqueValuesIsLoading, setPivotRowField: (value) => {
+                                            const prev = pivotRowField;
                                             setPivotRowField(value);
-                                            updatePivot(value, 'rowField');
+                                            updatePivot(value, 'rowField', prev);
                                         }, setPivotColumnField: (value) => {
+                                            const prev = pivotColumnField;
                                             setPivotColumnField(value);
-                                            updatePivot(value, 'columnField');
+                                            updatePivot(value, 'columnField', prev);
                                         }, setPivotValueField: (value) => {
+                                            const prev = pivotValueField;
                                             setPivotValueField(value);
-                                            updatePivot(value, 'valueField');
+                                            updatePivot(value, 'valueField', prev);
                                         }, setPivotValueField2: (value) => {
+                                            const prev = pivotValueField2;
                                             setPivotValueField2(value);
-                                            updatePivot(value, 'valueField2');
+                                            updatePivot(value, 'valueField2', prev);
                                         }, setPivotAggregation: (value) => {
+                                            const prev = pivotAggregation;
                                             setPivotAggregation(value);
-                                            updatePivot(value, 'aggregationType');
+                                            updatePivot(value, 'aggregationType', prev);
                                         }, onDelete: () => {
                                             setPivot(null);
+                                            setPivotError(undefined);
                                             setPivotHint('');
                                             setPivotData([]);
                                             const formattedRows = formatRows(rows, columns, false);
@@ -1309,6 +1403,8 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                                         report: tempReport,
                                                         client,
                                                         uniqueValues: uniqueValues[currentTable],
+                                                        dashboardName,
+                                                        tenants,
                                                     });
                                                     setPivotData(pivotedData || []);
                                                     const formattedRows = formatRows(pivotedData.rows, columns, true, pivot.aggregationType, dateBucket);
@@ -1356,6 +1452,8 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                                         report: tempReport,
                                                         client,
                                                         uniqueValues: uniqueValues[currentTable],
+                                                        dashboardName,
+                                                        tenants,
                                                     });
                                                     setPivotData(pivotedData || []);
                                                     const formattedRows = formatRows(pivotedData.rows, columns, true, pivot.aggregationType, dateBucket);
@@ -1468,6 +1566,8 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                                                 report: tempReport,
                                                                 client,
                                                                 uniqueValues: uniqueValues[currentTable],
+                                                                dashboardName,
+                                                                tenants,
                                                             });
                                                             setErrorMessage('');
                                                             setPivotData(pivotedData || []);
@@ -1581,7 +1681,7 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                             field: c,
                                         };
                                     }), onPageChange: onPageChange, onSortChange: (sort) => {
-                                    onSortChange(sort, true);
+                                    onSortChange(sort);
                                 }, containerStyle: {
                                     maxHeight: Math.max(window.innerHeight - 290, 75 + Math.min(Math.max(10, rows.length), 20) * 37),
                                 } })), (0, jsx_runtime_1.jsxs)("div", { style: {
@@ -1590,14 +1690,15 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                     gap: '12px',
                                     width: '100%',
                                     marginTop: 'auto',
-                                }, children: [errorMessage ? ((0, jsx_runtime_1.jsxs)("div", { style: {
+                                    alignItems: 'center',
+                                }, children: [errorMessage || pivotError ? ((0, jsx_runtime_1.jsxs)("div", { style: {
                                             display: 'flex',
                                             flexDirection: 'row',
                                             overflow: 'hidden',
                                             width: '100%',
                                             gap: 12,
                                             alignItems: 'center',
-                                        }, children: [(0, jsx_runtime_1.jsx)(ErrorMessageComponent, { errorMessage: errorMessage }), (0, jsx_runtime_1.jsx)(SecondaryButtonComponent, { onClick: () => {
+                                        }, children: [(0, jsx_runtime_1.jsx)(ErrorMessageComponent, { errorMessage: errorMessage || pivotError || '' }), (0, jsx_runtime_1.jsx)(SecondaryButtonComponent, { onClick: () => {
                                                     fetchAstFromPromptHelper();
                                                 }, label: 'Retry' }), (0, jsx_runtime_1.jsx)(SecondaryButtonComponent, { onClick: clearAllState, label: 'Reset' })] })) : ((0, jsx_runtime_1.jsx)("div", { style: { width: '100%' } })), baseAst && dataDisplayed && ((0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [onDiscardChanges && ((0, jsx_runtime_1.jsx)(SecondaryButtonComponent, { onClick: onDiscardChanges, label: "Discard changes" })), !hideCopySQL && ((0, jsx_runtime_1.jsx)(SecondaryButtonComponent, { label: isCopying ? 'Copied' : 'Copy SQL', onClick: () => copySQLToClipboard() })), (0, jsx_runtime_1.jsx)(ButtonComponent, { onClick: async () => {
                                                     onSaveChanges && onSaveChanges();
@@ -1615,27 +1716,44 @@ SidebarComponent = ui_1.QuillSidebar, ContainerComponent = ui_1.CustomContainer,
                                                                     yAxisFields: reportInfo?.pivot && !pivot
                                                                         ? []
                                                                         : reportInfo?.yAxisFields,
-                                                                    columns: reportInfo?.columns.filter((col) => {
-                                                                        return columns.find((c) => {
-                                                                            return col.field === c.field;
-                                                                        });
-                                                                    }),
+                                                                    columns: selectedOrderedColumns.length > 0
+                                                                        ? selectedOrderedColumns
+                                                                            .map((columnName) => {
+                                                                            return columns.find((col) => col.field === columnName);
+                                                                        })
+                                                                            .filter((col) => col !== undefined)
+                                                                            .map((c) => reportInfo?.columns.find((col) => col.field === c.field))
+                                                                            .filter((col) => col !== undefined)
+                                                                        : reportInfo?.columns.filter((col) => {
+                                                                            return columns.find((c) => {
+                                                                                return col.field === c.field;
+                                                                            });
+                                                                        }),
                                                                     queryString: activeQuery,
                                                                     rows: rows,
                                                                     pivotRows: pivotData?.rows,
                                                                     pivotColumns: pivotData?.columns,
+                                                                    flags: reportInfo?.flags ??
+                                                                        tempReport?.flags
+                                                                    // flagsToAdd,
                                                                 }
                                                                 : {
                                                                     ...tempReport,
                                                                     id: Report_1.TEMP_REPORT_ID,
+                                                                    flags: tempReport?.flags,
                                                                 }),
                                                         },
                                                     });
                                                     setIsChartBuilderOpen(true);
                                                 }, disabled: !!errorMessage ||
+                                                    !!pivotError ||
                                                     tableLoading ||
                                                     loading ||
-                                                    unresolvedReportMessage, label: reportId ? 'Save changes' : 'Add to dashboard', tooltipText: unresolvedReportMessage })] }))] })] }), (0, jsx_runtime_1.jsx)("style", { children: `body{margin:0;}` })] })), (!isChartBuilderHorizontalView || isChartBuilderOpen) && ((0, jsx_runtime_1.jsx)(ChartBuilder_1.ChartBuilderWithModal, { reportId: reportId, isAdmin: isAdminEnabled, title: reportId ? 'Save changes' : 'Add to dashboard', isHorizontalView: true, isOpen: isChartBuilderOpen, setIsOpen: setIsChartBuilderOpen, onAddToDashboardComplete: reportId ? onSubmitEditReport : onSubmitCreateReport, organizationName: organizationName, initialUniqueValues: uniqueValues[currentTable], initialUniqueValuesIsLoading: uniqueValuesIsLoading, pivotRecommendationsEnabled: pivotRecommendationsEnabledState, SelectComponent: SelectComponent, TextInputComponent: TextInputComponent, ButtonComponent: ButtonComponent, SecondaryButtonComponent: SecondaryButtonComponent, HeaderComponent: HeaderComponent, SubHeaderComponent: SubHeaderComponent, LabelComponent: LabelComponent, TextComponent: TextComponent, CardComponent: CardComponent, ModalComponent: ChartBuilderModalComponent, PopoverComponent: PopoverComponent, TableComponent: TableComponent, DeleteButtonComponent: DeleteButtonComponent, LoadingComponent: LoadingComponent, ChartBuilderInputRowContainer: ChartBuilderInputRowContainer, ChartBuilderInputColumnContainer: ChartBuilderInputColumnContainer, FormContainer: ChartBuilderFormContainer, hideDateRangeFilter: true, buttonLabel: !!reportId ? 'Save changes' : 'Add to dashboard', onClickChartElement: onClickChartElement, filtersEnabled: filtersEnabled, onFiltersEnabledChanged: (enabled) => {
+                                                    unresolvedReportMessage, label: reportId ? 'Save changes' : 'Add to dashboard', tooltipText: unresolvedReportMessage })] }))] })] }), (0, jsx_runtime_1.jsx)("style", { children: `body{margin:0;}` })] })), (!isChartBuilderHorizontalView || isChartBuilderOpen) && ((0, jsx_runtime_1.jsx)(ChartBuilder_1.ChartBuilderWithModal, { reportId: reportId, isAdmin: isAdminEnabled, title: chartBuilderTitle
+                    ? chartBuilderTitle
+                    : reportId
+                        ? 'Save changes'
+                        : 'Add to dashboard', isHorizontalView: true, isOpen: isChartBuilderOpen, setIsOpen: setIsChartBuilderOpen, onAddToDashboardComplete: reportId ? onSubmitEditReport : onSubmitCreateReport, organizationName: organizationName, initialUniqueValues: uniqueValues[currentTable], initialUniqueValuesIsLoading: uniqueValuesIsLoading, pivotRecommendationsEnabled: pivotRecommendationsEnabledState, SelectComponent: SelectComponent, TextInputComponent: TextInputComponent, ButtonComponent: ButtonComponent, SecondaryButtonComponent: SecondaryButtonComponent, HeaderComponent: HeaderComponent, SubHeaderComponent: SubHeaderComponent, LabelComponent: LabelComponent, TextComponent: TextComponent, CardComponent: CardComponent, ModalComponent: ChartBuilderModalComponent, PopoverComponent: PopoverComponent, TableComponent: TableComponent, DeleteButtonComponent: DeleteButtonComponent, LoadingComponent: LoadingComponent, ChartBuilderInputRowContainer: ChartBuilderInputRowContainer, ChartBuilderInputColumnContainer: ChartBuilderInputColumnContainer, FormContainer: ChartBuilderFormContainer, hideDateRangeFilter: true, buttonLabel: !!reportId ? 'Save changes' : 'Add to dashboard', onClickChartElement: onClickChartElement, filtersEnabled: filtersEnabled, onFiltersEnabledChanged: (enabled) => {
                     setFiltersEnabled(enabled);
-                }, isEditingMode: true }))] }));
+                }, isEditingMode: true, flagsList: flagsList }))] }));
 }

package/dist/cjs/SQLEditor.d.ts

@@ -249,6 +249,10 @@ export interface SQLEditorProps {
      * Styles the top-level container of the SQLEditor.
      */
     containerStyle?: React.CSSProperties;
+    /**
+     * A list of flags that can be applied to the resulting report (for single tenant per database setups)
+     */
+    flagsList?: string[];
 }
 /**
  * ### Quill SQLEditor
@@ -275,7 +279,7 @@ export interface SQLEditorProps {
  * ### SQLEditor API
  * @see https://docs.quillsql.com/components/sql-editor
  */
-export default function SQLEditor({ ButtonComponent, SecondaryButtonComponent, DeleteButtonComponent, TextInputComponent, SelectComponent, TableComponent, isNewQueryEnabled, LoadingComponent, ModalComponent, PopoverComponent, CardComponent, LabelComponent, HeaderComponent, SubHeaderComponent, TextComponent, ErrorMessageComponent, ChartBuilderInputRowContainer, ChartBuilderInputColumnContainer, PivotRowContainer, PivotColumnContainer, ChartBuilderFormContainer, defaultQuery, destinationDashboard, onChangeQuery, onChangeData, onChangeColumns, onChangeFields, onDiscardChanges, onSaveChanges, onCloseChartBuilder, isChartBuilderEnabled, isAdminEnabled, chartBuilderTitle, runQueryOnMount, onAddToDashboardComplete, addToDashboardButtonLabel, report, organizationName, isChartBuilderHorizontalView, containerStyle, className, onClickChartElement, }: SQLEditorProps): import("react/jsx-runtime").JSX.Element;
+export default function SQLEditor({ ButtonComponent, SecondaryButtonComponent, DeleteButtonComponent, TextInputComponent, SelectComponent, TableComponent, isNewQueryEnabled, LoadingComponent, ModalComponent, PopoverComponent, CardComponent, LabelComponent, HeaderComponent, SubHeaderComponent, TextComponent, ErrorMessageComponent, ChartBuilderInputRowContainer, ChartBuilderInputColumnContainer, PivotRowContainer, PivotColumnContainer, ChartBuilderFormContainer, defaultQuery, destinationDashboard, onChangeQuery, onChangeData, onChangeColumns, onChangeFields, onDiscardChanges, onSaveChanges, onCloseChartBuilder, isChartBuilderEnabled, isAdminEnabled, chartBuilderTitle, runQueryOnMount, onAddToDashboardComplete, addToDashboardButtonLabel, report, organizationName, isChartBuilderHorizontalView, containerStyle, className, onClickChartElement, flagsList, }: SQLEditorProps): import("react/jsx-runtime").JSX.Element;
 export declare const SchemaListComponent: ({ schema, theme, loading, LoadingComponent, width, onClick, style, }: {
     schema: any;
     theme: any;