Added API for track columns and pass default columns

src/QueryBuilder/QueryBuilder/QueryBuilderModal/QueryBuilderModal.js

@@ -37,6 +37,8 @@ export const QueryBuilderModal = ({
   onQueryRunFail,
   onQueryExecutionSuccess,
   onQueryExecutionFail,
+  onSetDefaultVisibleColumns,
+  recordColumns,
   getParamsSource,
   recordsLimit,
   additionalControls,
@@ -212,6 +214,8 @@ export const QueryBuilderModal = ({
             testQuery={testQuery}
             isTestQueryLoading={isTestQueryLoading}
             fqlQuery={fqlQuery}
+            recordColumns={recordColumns}
+            onSetDefaultVisibleColumns={onSetDefaultVisibleColumns}
             testQueryDataSource={testQueryDataSource}
             entityTypeDataSource={entityTypeDataSource}
             queryDetailsDataSource={queryDetailsDataSource}

src/QueryBuilder/QueryBuilder/TestQuery/TestQuery.js

@@ -20,6 +20,8 @@ export const TestQuery = ({
   onQueryExecutionSuccess,
   onQueryExecutionFail,
   onQueryRetrieved,
+  recordColumns = [],
+  onSetDefaultVisibleColumns,
   fqlQuery,
   entityTypeId,
   isPreviewLoading,
@@ -31,7 +33,7 @@ export const TestQuery = ({
   const queryClient = useQueryClient();
 
   const [columns, setColumns] = useState([]);
-  const [visibleColumns, setVisibleColumns] = useState([]);
+  const [visibleColumns, setVisibleColumns] = useState(recordColumns);
   const [includeContent, setIncludeContent] = useState(true);
   const [recordsLimitExceeded, setRecordsLimitExceeded] = useState(false);
 
@@ -106,11 +108,15 @@ export const TestQuery = ({
     setIsPreviewLoading(false);
   };
 
-  const handleColumnChange = ({ values }) => setVisibleColumns(values);
+  const handleColumnChange = ({ values }) => {
+    onSetDefaultVisibleColumns(values)
+    return setVisibleColumns(values)
+  };
 
   const handleDefaultVisibleColumnsChange = (values) => {
     if (!visibleColumns.length) {
       setVisibleColumns(values);
+      onSetDefaultVisibleColumns(values)
     }
   };
 

src/QueryBuilder/propTypes.js

@@ -7,8 +7,10 @@ export const queryBuilderModalPropTypes = {
   testQueryDataSource: PropTypes.func,
   cancelQueryDataSource: PropTypes.func,
   queryDetailsDataSource: PropTypes.func,
+  recordColumns: PropTypes.arrayOf(PropTypes.string),
   initialValues: PropTypes.object,
   saveBtnLabel: PropTypes.oneOfType([PropTypes.element, PropTypes.string]),
+  onSetDefaultVisibleColumns: PropTypes.func,
   entityTypeDataSource: PropTypes.func,
   getParamsSource: PropTypes.func,
   onQueryRunSuccess: PropTypes.func,

src/index.js

@@ -1 +1,2 @@
 export { QueryBuilderPlugin as default } from './QueryBuilder';
+export { QueryBuilderPlugin } from './QueryBuilder';