@finos/legend-application-query 5.1.0 → 5.2.3

package/lib/stores/{QueryFunctionsExplorerState.js → explorer/QueryFunctionsExplorerState.js}

@@ -16,22 +16,16 @@
 import { ConcreteFunctionDefinition, Package, Unit, ROOT_PACKAGE_NAME, } from '@finos/legend-graph';
 import { addUniqueEntry, guaranteeNonNullable, isNonNullable, uuid, } from '@finos/legend-shared';
 import { action, makeObservable, observable } from 'mobx';
-export var QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE;
-(function (QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE) {
-    QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE["FUNCTION"] = "FUNCTION";
-    QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE["PACKAGE"] = "PACKAGE";
-})(QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE = QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE || (QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE = {}));
+export const QUERY_BUILDER_FUNCTION_DND_TYPE = 'QUERY_BUILDER_FUNCTION';
 export class QueryBuilderFunctionsExplorerTreeNodeData {
     id;
     label;
-    dndType;
     childrenIds = [];
     isOpen;
     packageableElement;
-    constructor(id, label, dndType, packageableElement) {
+    constructor(id, label, packageableElement) {
         this.id = id;
         this.label = label;
-        this.dndType = dndType;
         this.packageableElement = packageableElement;
     }
 }
@@ -47,9 +41,6 @@ const getValidDisplayablePackageSet = (queryBuilderState, rootPackageName) => {
 };
 export const generateFunctionsExplorerTreeNodeData = (queryBuilderState, element, rootPackageName) => ({
     id: element.path,
-    dndType: element instanceof Package
-        ? QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE.PACKAGE
-        : QUERY_BUILDER_FUNCTIONS_EXPLORER_TREE_DND_TYPE.FUNCTION,
     label: element.name,
     childrenIds: element instanceof Package
         ? element.children

package/lib/stores/explorer/QueryFunctionsExplorerState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryFunctionsExplorerState.js","sourceRoot":"","sources":["../../../src/stores/explorer/QueryFunctionsExplorerState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAEL,0BAA0B,EAC1B,OAAO,EACP,IAAI,EACJ,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,cAAc,EACd,oBAAoB,EACpB,aAAa,EACb,IAAI,GACL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAI1D,MAAM,CAAC,MAAM,+BAA+B,GAAG,wBAAwB,CAAC;AAMxE,MAAM,OAAO,yCAAyC;IACpD,EAAE,CAAS;IACX,KAAK,CAAS;IACd,WAAW,GAAa,EAAE,CAAC;IAC3B,MAAM,CAAuB;IAC7B,kBAAkB,CAAqB;IAEvC,YACE,EAAU,EACV,KAAa,EACb,kBAAsC;QAEtC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IAC/C,CAAC;CACF;AAED,MAAM,6BAA6B,GAAG,CACpC,iBAAoC,EACpC,eAAkC,EACpB,EAAE;IAChB,QAAQ,eAAe,EAAE;QACvB,KAAK,iBAAiB,CAAC,uBAAuB;YAC5C,OAAO,iBAAiB,CAAC,2BAA2B;iBACjD,gCAAgC,CAAC;QACtC;YACE,OAAO,iBAAiB,CAAC,2BAA2B;iBACjD,sBAAsB,CAAC;KAC7B;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qCAAqC,GAAG,CACnD,iBAAoC,EACpC,OAA2B,EAC3B,eAAkC,EACS,EAAE,CAAC,CAAC;IAC/C,EAAE,EAAE,OAAO,CAAC,IAAI;IAChB,KAAK,EAAE,OAAO,CAAC,IAAI;IACnB,WAAW,EACT,OAAO,YAAY,OAAO;QACxB,CAAC,CAAC,OAAO,CAAC,QAAQ;aACb,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC,CAAC;aAC3C,MAAM,CACL,CAAC,KAAK,EAAE,EAAE,CACR,CAAC,KAAK,YAAY,OAAO;YACvB,6BAA6B,CAC3B,iBAAiB,EACjB,eAAe,CAChB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACf,KAAK,YAAY,0BAA0B,CAC9C;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;QAC/B,CAAC,CAAC,EAAE;IACR,kBAAkB,EAAE,OAAO;CAC5B,CAAC,CAAC;AAEH,MAAM,0CAA0C,GAAG,CACjD,iBAAoC,EACpC,IAA+C,EAC/C,IAAyD,EACzD,eAAe,GAAG,iBAAiB,CAAC,IAAI,EAClC,EAAE;IACR,MAAM,0BAA0B,GAAG,6BAA6B,CAC9D,iBAAiB,EACjB,eAAe,CAChB,CAAC;IACF,IAAI,CAAC,WAAW,GAAI,IAAI,CAAC,kBAA8B,CAAC,QAAQ;SAC7D,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC,CAAC;SAC3C,MAAM,CACL,CAAC,KAAK,EAAE,EAAE;IACR,kCAAkC;IAClC,CAAC,KAAK,YAAY,OAAO,IAAI,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnE,KAAK,YAAY,0BAA0B,CAC9C;SACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7B,IAAI,CAAC,kBAA8B,CAAC,QAAQ;SAC1C,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC,CAAC;SAC3C,MAAM,CACL,CAAC,KAAK,EAAE,EAAE;IACR,kCAAkC;IAClC,CAAC,KAAK,YAAY,OAAO,IAAI,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnE,KAAK,YAAY,0BAA0B,CAC9C;SACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACb,qCAAqC,CACnC,iBAAiB,EACjB,KAAK,EACL,eAAe,CAChB,CACF;SACA,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;QACrB,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,WAAW,EAAE;YACf,WAAW,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;YAChD,WAAW,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC;SACrC;aAAM;YACL,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;SACzC;IACH,CAAC,CAAC,CAAC;AACP,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAC1C,IAAa,EACb,iBAAoC,EACpC,eAAe,GAAG,iBAAiB,CAAC,IAAI,EACa,EAAE;IACvD,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,GAAG,EAAqD,CAAC;IAC3E,MAAM,0BAA0B,GAAG,6BAA6B,CAC9D,iBAAiB,EACjB,eAAe,CAChB,CAAC;IACF,QAAQ,eAAe,EAAE;QACvB,KAAK,iBAAiB,CAAC,uBAAuB;YAC5C,IACE,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB,CAAC,SAAS;iBAClE,MAAM,KAAK,CAAC,EACf;gBACA,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;aAC3B;YACD,MAAM;QACR;YACE,IAAI,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE;gBACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;aAC3B;KACJ;IAED,IAAI,CAAC,QAAQ;SACV,KAAK,EAAE;SACP,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC,CAAC;SAC3C,MAAM,CACL,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,YAAY,OAAO,IAAI,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,CACpE;SACA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;SAC5C,IAAI,CACH,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC1E;SACA,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;QACxB,MAAM,iBAAiB,GAAG,qCAAqC,CAC7D,iBAAiB,EACjB,YAAY,EACZ,eAAe,CAChB,CAAC;QACF,cAAc,CAAC,OAAO,EAAE,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAC9C,KAAK,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IACL,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;AAC5B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,oCAAoC,GAAG,CAClD,iBAAoC,EACpC,IAA+C,EAC/C,IAAyD,EACzD,eAAe,GAAG,iBAAiB,CAAC,IAAI,EACK,EAAE;IAC/C,0CAA0C,CACxC,iBAAiB,EACjB,IAAI,EACJ,IAAI,EACJ,eAAe,CAChB,CAAC;IACF,OAAO,IAAI,CAAC,WAAW;SACpB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;SAC/B,MAAM,CAAC,aAAa,CAAC;SACrB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;SAC9C,IAAI,CACH,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACP,CAAC,CAAC,CAAC,kBAAkB,YAAY,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC,kBAAkB,YAAY,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACpD,CAAC;AACN,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,CAAC,OAA2B,EAAa,EAAE;IAC3E,IAAI,OAAO,CAAC,OAAO,EAAE;QACnB,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;KAC7E;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,MAAM,OAAO,0BAA0B;IAC5B,IAAI,GAAG,IAAI,EAAE,CAAC;IACvB,mBAAmB,CAA8B;IACjD,0BAA0B,CAA6B;IAEvD,YACE,mBAAgD,EAChD,0BAAsD;QAEtD,cAAc,CAAC,IAAI,EAAE;YACnB,0BAA0B,EAAE,UAAU;SACvC,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,0BAA0B,GAAG,0BAA0B,CAAC;IAC/D,CAAC;CACF;AAED,MAAM,OAAO,2BAA2B;IACtC,iBAAiB,CAAoB;IACrC,QAAQ,CAAmE;IAC3E,kBAAkB,CAEJ;IACd,sBAAsB,GAAiC,EAAE,CAAC;IAC1D,gCAAgC,GAAiC,EAAE,CAAC;IACpE,sBAAsB,GAAiB,IAAI,GAAG,EAAW,CAAC;IAC1D,gCAAgC,GAAiB,IAAI,GAAG,EAAW,CAAC;IAEpE,YAAY,iBAAoC;QAC9C,cAAc,CAAC,IAAI,EAAE;YACnB,sBAAsB,EAAE,UAAU,CAAC,GAAG;YACtC,gCAAgC,EAAE,UAAU,CAAC,GAAG;YAChD,QAAQ,EAAE,UAAU,CAAC,GAAG;YACxB,kBAAkB,EAAE,UAAU,CAAC,GAAG;YAClC,WAAW,EAAE,MAAM;YACnB,qBAAqB,EAAE,MAAM;YAC7B,WAAW,EAAE,MAAM;YACnB,gBAAgB,EAAE,MAAM;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED,WAAW,CACT,eAAe,GAAG,iBAAiB,CAAC,IAAI;QAExC,QAAQ,eAAe,EAAE;YACvB,KAAK,iBAAiB,CAAC,uBAAuB;gBAC5C,OAAO,IAAI,CAAC,kBAAkB,CAAC;YACjC;gBACE,OAAO,IAAI,CAAC,QAAQ,CAAC;SACxB;IACH,CAAC;IAED,KAAK,CAAC,gCAAgC;QACpC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,YAAY;aACxD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;aACxC,IAAI,EAAE;aACN,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,0CAA0C;QAC9C,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB,CAAC,SAAS;aACvE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;aACxC,IAAI,EAAE;aACN,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,WAAW,CACT,GAAoE;QAEpE,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC;IACtB,CAAC;IAED,qBAAqB,CACnB,GAAoE;QAEpE,IAAI,CAAC,kBAAkB,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,WAAW;QACT,IAAI,IAAI,CAAC,QAAQ,EAAE;YACjB,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;SACtC;QACD,IAAI,IAAI,CAAC,kBAAkB,EAAE;YAC3B,IAAI,CAAC,kBAAkB,GAAG,EAAE,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;SAC1D;IACH,CAAC;IAED,IAAI,mBAAmB;QACrB,OAAO,oBAAoB,CACzB,IAAI,CAAC,QAAQ,EACb,qEAAqE,CACtE,CAAC;IACJ,CAAC;IAED,gBAAgB,GAAG,CACjB,iBAAoC,EACpC,IAA+C,EAC/C,IAAyD,EACzD,eAAe,GAAG,iBAAiB,CAAC,IAAI,EAClC,EAAE;QACR,IAAI,IAAI,CAAC,kBAAkB,YAAY,OAAO,EAAE;YAC9C,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE;gBAC3B,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC3B,0CAA0C,CACxC,iBAAiB,EACjB,IAAI,EACJ,IAAI,EACJ,eAAe,CAChB,CAAC;aACH;YACD,QAAQ,eAAe,EAAE;gBACvB,KAAK,iBAAiB,CAAC,uBAAuB;oBAC5C,IAAI,CAAC,qBAAqB,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;oBACxC,MAAM;gBACR;oBACE,IAAI,CAAC,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;aACjC;SACF;IACH,CAAC,CAAC;IAEF,kBAAkB;QAChB,IAAI,CAAC,gCAAgC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACnD,IAAI,CAAC,WAAW,CACd,4BAA4B,CAC1B,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,EACnD,IAAI,CAAC,iBAAiB,CACvB,CACF,CAAC;YACF,IAAI,CAAC,sBAAsB;gBACzB,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAC7D,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,0BAA0B,CAAC,IAAI,EAAE,CAAC,CAAC,CAC/C,CAAC;QACN,CAAC,CAAC,CAAC;QACH,IACE,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB;aAC7D,eAAe,EAClB;YACA,IAAI,CAAC,0CAA0C,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBAC7D,IAAI,CAAC,qBAAqB,CACxB,4BAA4B,CAC1B,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB;qBAC7D,IAAI,EACP,IAAI,CAAC,iBAAiB,EACtB,iBAAiB,CAAC,uBAAuB,CAC1C,CACF,CAAC;gBACF,IAAI,CAAC,gCAAgC;oBACnC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB,CAAC,SAAS,CAAC,GAAG,CAC5E,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,0BAA0B,CAAC,IAAI,EAAE,CAAC,CAAC,CAC/C,CAAC;YACN,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;CACF"}

package/lib/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import type { QueryBuilderState } from '../QueryBuilderState.js';
+import type { QueryBuilderFetchStructureState } from './QueryBuilderFetchStructureState.js';
+export declare abstract class QueryBuilderFetchStructureImplementationState {
+    queryBuilderState: QueryBuilderState;
+    fetchStructureState: QueryBuilderFetchStructureState;
+    constructor(queryBuilderState: QueryBuilderState, fetchStructureState: QueryBuilderFetchStructureState);
+}
+//# sourceMappingURL=QueryBuilderFetchStructureImplementationState.d.ts.map

package/lib/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderFetchStructureImplementationState.d.ts","sourceRoot":"","sources":["../../../src/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AAE5F,8BAAsB,6CAA6C;IACjE,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,mBAAmB,EAAE,+BAA+B,CAAC;gBAGnD,iBAAiB,EAAE,iBAAiB,EACpC,mBAAmB,EAAE,+BAA+B;CAQvD"}

package/lib/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.js

@@ -0,0 +1,24 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export class QueryBuilderFetchStructureImplementationState {
+    queryBuilderState;
+    fetchStructureState;
+    constructor(queryBuilderState, fetchStructureState) {
+        this.queryBuilderState = queryBuilderState;
+        this.fetchStructureState = fetchStructureState;
+    }
+}
+//# sourceMappingURL=QueryBuilderFetchStructureImplementationState.js.map

package/lib/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderFetchStructureImplementationState.js","sourceRoot":"","sources":["../../../src/stores/fetch-structure/QueryBuilderFetchStructureImplementationState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,MAAM,OAAgB,6CAA6C;IACjE,iBAAiB,CAAoB;IACrC,mBAAmB,CAAkC;IAErD,YACE,iBAAoC,EACpC,mBAAoD;QAEpD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;IACjD,CAAC;CAIF"}

package/lib/stores/fetch-structure/QueryBuilderFetchStructureState.d.ts

@@ -0,0 +1,44 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import type { QueryBuilderState } from '../QueryBuilderState.js';
+import { QueryBuilderGraphFetchTreeState } from './graph-fetch/QueryBuilderGraphFetchTreeState.js';
+import { QueryBuilderProjectionState } from './projection/QueryBuilderProjectionState.js';
+import { type QueryBuilderExplorerTreeNodeData } from '../explorer/QueryBuilderExplorerState.js';
+export declare enum FETCH_STRUCTURE_MODE {
+    PROJECTION = "PROJECTION",
+    GRAPH_FETCH = "GRAPH_FETCH"
+}
+export declare class QueryBuilderFetchStructureState {
+    queryBuilderState: QueryBuilderState;
+    /**
+     * TODO: refactor this so we could reduce this to
+     * `implementationState: QueryBuilderFetchStructureImplementationState`
+     * so we don't have graph fetch and projection at the same time
+     *
+     * TODO?: perhaps it would eventually make sense to default to
+     * graph-fetch since `getAll()` naturally works for graph-fetch case
+     * and graph-fetch allows somewhat an empty tree
+     */
+    fetchStructureMode: FETCH_STRUCTURE_MODE;
+    projectionState: QueryBuilderProjectionState;
+    graphFetchTreeState: QueryBuilderGraphFetchTreeState;
+    constructor(queryBuilderState: QueryBuilderState);
+    setFetchStructureMode(val: FETCH_STRUCTURE_MODE): void;
+    fetchProperty(node: QueryBuilderExplorerTreeNodeData): void;
+    fetchNodeChildrenProperties(node: QueryBuilderExplorerTreeNodeData): void;
+    get validationIssues(): string[] | undefined;
+}
+//# sourceMappingURL=QueryBuilderFetchStructureState.d.ts.map

package/lib/stores/fetch-structure/QueryBuilderFetchStructureState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderFetchStructureState.d.ts","sourceRoot":"","sources":["../../../src/stores/fetch-structure/QueryBuilderFetchStructureState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,+BAA+B,EAAE,MAAM,kDAAkD,CAAC;AACnG,OAAO,EAAE,2BAA2B,EAAE,MAAM,6CAA6C,CAAC;AAC1F,OAAO,EAEL,KAAK,gCAAgC,EAEtC,MAAM,0CAA0C,CAAC;AAMlD,oBAAY,oBAAoB;IAC9B,UAAU,eAAe;IACzB,WAAW,gBAAgB;CAC5B;AAED,qBAAa,+BAA+B;IAC1C,iBAAiB,EAAE,iBAAiB,CAAC;IACrC;;;;;;;;OAQG;IACH,kBAAkB,uBAAmC;IACrD,eAAe,EAAE,2BAA2B,CAAC;IAC7C,mBAAmB,EAAE,+BAA+B,CAAC;gBAEzC,iBAAiB,EAAE,iBAAiB;IAmBhD,qBAAqB,CAAC,GAAG,EAAE,oBAAoB,GAAG,IAAI;IAItD,aAAa,CAAC,IAAI,EAAE,gCAAgC,GAAG,IAAI;IAgC3D,2BAA2B,CAAC,IAAI,EAAE,gCAAgC,GAAG,IAAI;IA0DzE,IAAI,gBAAgB,IAAI,MAAM,EAAE,GAAG,SAAS,CAQ3C;CACF"}

package/lib/stores/fetch-structure/QueryBuilderFetchStructureState.js

@@ -0,0 +1,116 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { action, makeAutoObservable } from 'mobx';
+import { QueryBuilderGraphFetchTreeState } from './graph-fetch/QueryBuilderGraphFetchTreeState.js';
+import { QueryBuilderProjectionState } from './projection/QueryBuilderProjectionState.js';
+import { buildPropertyExpressionFromExplorerTreeNodeData, QueryBuilderExplorerTreePropertyNodeData, } from '../explorer/QueryBuilderExplorerState.js';
+import { Class } from '@finos/legend-graph';
+import { filterByType } from '@finos/legend-shared';
+import { addQueryBuilderPropertyNode } from './graph-fetch/QueryBuilderGraphFetchTreeUtil.js';
+import { QueryBuilderSimpleProjectionColumnState } from './projection/QueryBuilderProjectionColumnState.js';
+export var FETCH_STRUCTURE_MODE;
+(function (FETCH_STRUCTURE_MODE) {
+    FETCH_STRUCTURE_MODE["PROJECTION"] = "PROJECTION";
+    FETCH_STRUCTURE_MODE["GRAPH_FETCH"] = "GRAPH_FETCH";
+})(FETCH_STRUCTURE_MODE = FETCH_STRUCTURE_MODE || (FETCH_STRUCTURE_MODE = {}));
+export class QueryBuilderFetchStructureState {
+    queryBuilderState;
+    /**
+     * TODO: refactor this so we could reduce this to
+     * `implementationState: QueryBuilderFetchStructureImplementationState`
+     * so we don't have graph fetch and projection at the same time
+     *
+     * TODO?: perhaps it would eventually make sense to default to
+     * graph-fetch since `getAll()` naturally works for graph-fetch case
+     * and graph-fetch allows somewhat an empty tree
+     */
+    fetchStructureMode = FETCH_STRUCTURE_MODE.PROJECTION;
+    projectionState;
+    graphFetchTreeState;
+    constructor(queryBuilderState) {
+        makeAutoObservable(this, {
+            queryBuilderState: false,
+            setFetchStructureMode: action,
+        });
+        this.queryBuilderState = queryBuilderState;
+        // TODO: we probably should modularize this a bit better
+        // See https://github.com/finos/legend-studio/issues/731
+        this.projectionState = new QueryBuilderProjectionState(queryBuilderState, this);
+        this.graphFetchTreeState = new QueryBuilderGraphFetchTreeState(queryBuilderState, this);
+    }
+    setFetchStructureMode(val) {
+        this.fetchStructureMode = val;
+    }
+    fetchProperty(node) {
+        if (node instanceof QueryBuilderExplorerTreePropertyNodeData &&
+            !(node.type instanceof Class)) {
+            switch (this.fetchStructureMode) {
+                case FETCH_STRUCTURE_MODE.GRAPH_FETCH: {
+                    this.graphFetchTreeState.addProperty(node);
+                    return;
+                }
+                case FETCH_STRUCTURE_MODE.PROJECTION: {
+                    this.projectionState.addColumn(new QueryBuilderSimpleProjectionColumnState(this.projectionState, buildPropertyExpressionFromExplorerTreeNodeData(this.queryBuilderState.explorerState.nonNullableTreeData, node, this.queryBuilderState.graphManagerState.graph, this.queryBuilderState.explorerState.propertySearchPanelState
+                        .allMappedPropertyNodes), this.queryBuilderState.explorerState.humanizePropertyName));
+                    return;
+                }
+                default:
+                    return;
+            }
+        }
+    }
+    fetchNodeChildrenProperties(node) {
+        if (node.type instanceof Class) {
+            // NOTE: here we require the node to already been expanded so the child nodes are generated
+            // we don't allow adding unopened node. Maybe if it helps, we can show a warning.
+            const nodesToAdd = node.childrenIds
+                .map((childId) => this.queryBuilderState.explorerState.nonNullableTreeData.nodes.get(childId))
+                .filter(filterByType(QueryBuilderExplorerTreePropertyNodeData))
+                .filter((childNode) => !(childNode.type instanceof Class) && childNode.mappingData.mapped);
+            switch (this.fetchStructureMode) {
+                case FETCH_STRUCTURE_MODE.GRAPH_FETCH: {
+                    const graphFetchTreeData = this.graphFetchTreeState.treeData;
+                    if (graphFetchTreeData) {
+                        nodesToAdd.forEach((nodeToAdd) => addQueryBuilderPropertyNode(graphFetchTreeData, this.queryBuilderState.explorerState.nonNullableTreeData, nodeToAdd, this.queryBuilderState));
+                        this.graphFetchTreeState.setGraphFetchTree({
+                            ...graphFetchTreeData,
+                        });
+                    }
+                    return;
+                }
+                case FETCH_STRUCTURE_MODE.PROJECTION: {
+                    nodesToAdd.forEach((nodeToAdd) => {
+                        this.projectionState.addColumn(new QueryBuilderSimpleProjectionColumnState(this.projectionState, buildPropertyExpressionFromExplorerTreeNodeData(this.queryBuilderState.explorerState.nonNullableTreeData, nodeToAdd, this.queryBuilderState.graphManagerState.graph, this.queryBuilderState.explorerState.propertySearchPanelState
+                            .allMappedPropertyNodes), this.queryBuilderState.explorerState.humanizePropertyName));
+                    });
+                    return;
+                }
+                default:
+                    return;
+            }
+        }
+    }
+    get validationIssues() {
+        switch (this.fetchStructureMode) {
+            case FETCH_STRUCTURE_MODE.PROJECTION:
+                return this.projectionState.validationIssues;
+            case FETCH_STRUCTURE_MODE.GRAPH_FETCH:
+            default:
+                return undefined;
+        }
+    }
+}
+//# sourceMappingURL=QueryBuilderFetchStructureState.js.map

package/lib/stores/fetch-structure/QueryBuilderFetchStructureState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderFetchStructureState.js","sourceRoot":"","sources":["../../../src/stores/fetch-structure/QueryBuilderFetchStructureState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,MAAM,CAAC;AAElD,OAAO,EAAE,+BAA+B,EAAE,MAAM,kDAAkD,CAAC;AACnG,OAAO,EAAE,2BAA2B,EAAE,MAAM,6CAA6C,CAAC;AAC1F,OAAO,EACL,+CAA+C,EAE/C,wCAAwC,GACzC,MAAM,0CAA0C,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,2BAA2B,EAAE,MAAM,iDAAiD,CAAC;AAC9F,OAAO,EAAE,uCAAuC,EAAE,MAAM,mDAAmD,CAAC;AAE5G,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,iDAAyB,CAAA;IACzB,mDAA2B,CAAA;AAC7B,CAAC,EAHW,oBAAoB,GAApB,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,+BAA+B;IAC1C,iBAAiB,CAAoB;IACrC;;;;;;;;OAQG;IACH,kBAAkB,GAAG,oBAAoB,CAAC,UAAU,CAAC;IACrD,eAAe,CAA8B;IAC7C,mBAAmB,CAAkC;IAErD,YAAY,iBAAoC;QAC9C,kBAAkB,CAAC,IAAI,EAAE;YACvB,iBAAiB,EAAE,KAAK;YACxB,qBAAqB,EAAE,MAAM;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,wDAAwD;QACxD,wDAAwD;QACxD,IAAI,CAAC,eAAe,GAAG,IAAI,2BAA2B,CACpD,iBAAiB,EACjB,IAAI,CACL,CAAC;QACF,IAAI,CAAC,mBAAmB,GAAG,IAAI,+BAA+B,CAC5D,iBAAiB,EACjB,IAAI,CACL,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,GAAyB;QAC7C,IAAI,CAAC,kBAAkB,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,aAAa,CAAC,IAAsC;QAClD,IACE,IAAI,YAAY,wCAAwC;YACxD,CAAC,CAAC,IAAI,CAAC,IAAI,YAAY,KAAK,CAAC,EAC7B;YACA,QAAQ,IAAI,CAAC,kBAAkB,EAAE;gBAC/B,KAAK,oBAAoB,CAAC,WAAW,CAAC,CAAC;oBACrC,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;oBAC3C,OAAO;iBACR;gBACD,KAAK,oBAAoB,CAAC,UAAU,CAAC,CAAC;oBACpC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,IAAI,uCAAuC,CACzC,IAAI,CAAC,eAAe,EACpB,+CAA+C,CAC7C,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB,EACxD,IAAI,EACJ,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,EAC9C,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,wBAAwB;yBAC1D,sBAAsB,CAC1B,EACD,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,oBAAoB,CAC1D,CACF,CAAC;oBACF,OAAO;iBACR;gBACD;oBACE,OAAO;aACV;SACF;IACH,CAAC;IAED,2BAA2B,CAAC,IAAsC;QAChE,IAAI,IAAI,CAAC,IAAI,YAAY,KAAK,EAAE;YAC9B,2FAA2F;YAC3F,iFAAiF;YACjF,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW;iBAChC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACf,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB,CAAC,KAAK,CAAC,GAAG,CAChE,OAAO,CACR,CACF;iBACA,MAAM,CAAC,YAAY,CAAC,wCAAwC,CAAC,CAAC;iBAC9D,MAAM,CACL,CAAC,SAAS,EAAE,EAAE,CACZ,CAAC,CAAC,SAAS,CAAC,IAAI,YAAY,KAAK,CAAC,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,CACrE,CAAC;YAEJ,QAAQ,IAAI,CAAC,kBAAkB,EAAE;gBAC/B,KAAK,oBAAoB,CAAC,WAAW,CAAC,CAAC;oBACrC,MAAM,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC;oBAC7D,IAAI,kBAAkB,EAAE;wBACtB,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE,CAC/B,2BAA2B,CACzB,kBAAkB,EAClB,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB,EACxD,SAAS,EACT,IAAI,CAAC,iBAAiB,CACvB,CACF,CAAC;wBACF,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,CAAC;4BACzC,GAAG,kBAAkB;yBACtB,CAAC,CAAC;qBACJ;oBACD,OAAO;iBACR;gBACD,KAAK,oBAAoB,CAAC,UAAU,CAAC,CAAC;oBACpC,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;wBAC/B,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,IAAI,uCAAuC,CACzC,IAAI,CAAC,eAAe,EACpB,+CAA+C,CAC7C,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB,EACxD,SAAS,EACT,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,EAC9C,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,wBAAwB;6BAC1D,sBAAsB,CAC1B,EACD,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,oBAAoB,CAC1D,CACF,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,OAAO;iBACR;gBACD;oBACE,OAAO;aACV;SACF;IACH,CAAC;IAED,IAAI,gBAAgB;QAClB,QAAQ,IAAI,CAAC,kBAAkB,EAAE;YAC/B,KAAK,oBAAoB,CAAC,UAAU;gBAClC,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC;YAC/C,KAAK,oBAAoB,CAAC,WAAW,CAAC;YACtC;gBACE,OAAO,SAAS,CAAC;SACpB;IACH,CAAC;CACF"}

package/lib/stores/{QueryBuilderGraphFetchTreeState.d.ts → fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.d.ts}

@@ -13,12 +13,13 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import type { QueryBuilderState } from './QueryBuilderState.js';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
 import { RootGraphFetchTree } from '@finos/legend-graph';
 import { type QueryBuilderGraphFetchTreeData } from './QueryBuilderGraphFetchTreeUtil.js';
-import type { QueryBuilderExplorerTreePropertyNodeData } from './QueryBuilderExplorerState.js';
-export declare class QueryBuilderGraphFetchTreeState {
-    queryBuilderState: QueryBuilderState;
+import type { QueryBuilderExplorerTreePropertyNodeData } from '../../explorer/QueryBuilderExplorerState.js';
+import { QueryBuilderFetchStructureImplementationState } from '../QueryBuilderFetchStructureImplementationState.js';
+import type { QueryBuilderFetchStructureState } from '../QueryBuilderFetchStructureState.js';
+export declare class QueryBuilderGraphFetchTreeState extends QueryBuilderFetchStructureImplementationState {
     treeData?: QueryBuilderGraphFetchTreeData | undefined;
     /**
      * If set to `true` we will use `graphFetchChecked` function instead of `graphFetch`.
@@ -26,7 +27,7 @@ export declare class QueryBuilderGraphFetchTreeState {
      * Hence we default this to `false` for graph fetch to work universally.
      */
     isChecked: boolean;
-    constructor(queryBuilderState: QueryBuilderState);
+    constructor(queryBuilderState: QueryBuilderState, fetchStructureState: QueryBuilderFetchStructureState);
     setGraphFetchTree(val: QueryBuilderGraphFetchTreeData | undefined): void;
     setChecked(val: boolean): void;
     initialize(tree?: RootGraphFetchTree): void;

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchTreeState.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,OAAO,EAEL,kBAAkB,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,8BAA8B,EAGpC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,KAAK,EAAE,wCAAwC,EAAE,MAAM,6CAA6C,CAAC;AAE5G,OAAO,EAAE,6CAA6C,EAAE,MAAM,qDAAqD,CAAC;AACpH,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,uCAAuC,CAAC;AAE7F,qBAAa,+BAAgC,SAAQ,6CAA6C;IAChG,QAAQ,CAAC,EAAE,8BAA8B,GAAG,SAAS,CAAC;IACtD;;;;OAIG;IACH,SAAS,UAAS;gBAGhB,iBAAiB,EAAE,iBAAiB,EACpC,mBAAmB,EAAE,+BAA+B;IAYtD,iBAAiB,CAAC,GAAG,EAAE,8BAA8B,GAAG,SAAS,GAAG,IAAI;IAIxE,UAAU,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAK9B,UAAU,CAAC,IAAI,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAmB3C,WAAW,CAAC,IAAI,EAAE,wCAAwC,GAAG,IAAI;CAgBlE"}

package/lib/stores/{QueryBuilderGraphFetchTreeState.js → fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.js}

@@ -13,11 +13,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { makeAutoObservable, action } from 'mobx';
+import { action, makeObservable, observable } from 'mobx';
 import { PackageableElementExplicitReference, RootGraphFetchTree, } from '@finos/legend-graph';
 import { addQueryBuilderPropertyNode, buildGraphFetchTreeData, } from './QueryBuilderGraphFetchTreeUtil.js';
-export class QueryBuilderGraphFetchTreeState {
-    queryBuilderState;
+import { assertNonNullable } from '@finos/legend-shared';
+import { QueryBuilderFetchStructureImplementationState } from '../QueryBuilderFetchStructureImplementationState.js';
+export class QueryBuilderGraphFetchTreeState extends QueryBuilderFetchStructureImplementationState {
     treeData;
     /**
      * If set to `true` we will use `graphFetchChecked` function instead of `graphFetch`.
@@ -25,13 +26,14 @@ export class QueryBuilderGraphFetchTreeState {
      * Hence we default this to `false` for graph fetch to work universally.
      */
     isChecked = false;
-    constructor(queryBuilderState) {
-        makeAutoObservable(this, {
-            queryBuilderState: false,
+    constructor(queryBuilderState, fetchStructureState) {
+        super(queryBuilderState, fetchStructureState);
+        makeObservable(this, {
+            treeData: observable,
+            isChecked: observable,
             setGraphFetchTree: action,
             setChecked: action,
         });
-        this.queryBuilderState = queryBuilderState;
     }
     setGraphFetchTree(val) {
         this.treeData = val;
@@ -39,27 +41,28 @@ export class QueryBuilderGraphFetchTreeState {
     setChecked(val) {
         this.isChecked = val;
     }
+    // TODO-BEFORE-PR: consider removing this
     initialize(tree) {
-        if (this.queryBuilderState.fetchStructureState.isGraphFetchMode()) {
-            if (tree) {
-                this.setGraphFetchTree(buildGraphFetchTreeData(tree));
+        if (tree) {
+            this.setGraphFetchTree(buildGraphFetchTreeData(tree));
+        }
+        else {
+            const _class = this.queryBuilderState.querySetupState._class;
+            if (_class) {
+                this.setGraphFetchTree(buildGraphFetchTreeData(new RootGraphFetchTree(PackageableElementExplicitReference.create(_class))));
             }
             else {
-                const _class = this.queryBuilderState.querySetupState._class;
-                if (_class) {
-                    this.setGraphFetchTree(buildGraphFetchTreeData(new RootGraphFetchTree(PackageableElementExplicitReference.create(_class))));
-                }
-                else {
-                    this.setGraphFetchTree(undefined);
-                }
+                this.setGraphFetchTree(undefined);
             }
         }
     }
     addProperty(node) {
-        if (this.treeData) {
-            addQueryBuilderPropertyNode(this.treeData, this.queryBuilderState.explorerState.nonNullableTreeData, node, this.queryBuilderState);
-            this.setGraphFetchTree({ ...this.treeData });
+        if (!this.treeData) {
+            this.initialize();
         }
+        assertNonNullable(this.treeData, `Graph-fetch tree has not been properly initialized`);
+        addQueryBuilderPropertyNode(this.treeData, this.queryBuilderState.explorerState.nonNullableTreeData, node, this.queryBuilderState);
+        this.setGraphFetchTree({ ...this.treeData });
     }
 }
 //# sourceMappingURL=QueryBuilderGraphFetchTreeState.js.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchTreeState.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAC1D,OAAO,EACL,mCAAmC,EACnC,kBAAkB,GACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,2BAA2B,EAC3B,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAE7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,6CAA6C,EAAE,MAAM,qDAAqD,CAAC;AAGpH,MAAM,OAAO,+BAAgC,SAAQ,6CAA6C;IAChG,QAAQ,CAA8C;IACtD;;;;OAIG;IACH,SAAS,GAAG,KAAK,CAAC;IAElB,YACE,iBAAoC,EACpC,mBAAoD;QAEpD,KAAK,CAAC,iBAAiB,EAAE,mBAAmB,CAAC,CAAC;QAE9C,cAAc,CAAC,IAAI,EAAE;YACnB,QAAQ,EAAE,UAAU;YACpB,SAAS,EAAE,UAAU;YACrB,iBAAiB,EAAE,MAAM;YACzB,UAAU,EAAE,MAAM;SACnB,CAAC,CAAC;IACL,CAAC;IAED,iBAAiB,CAAC,GAA+C;QAC/D,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC;IACtB,CAAC;IAED,UAAU,CAAC,GAAY;QACrB,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;IACvB,CAAC;IAED,yCAAyC;IACzC,UAAU,CAAC,IAAyB;QAClC,IAAI,IAAI,EAAE;YACR,IAAI,CAAC,iBAAiB,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC;SACvD;aAAM;YACL,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAAC,MAAM,CAAC;YAC7D,IAAI,MAAM,EAAE;gBACV,IAAI,CAAC,iBAAiB,CACpB,uBAAuB,CACrB,IAAI,kBAAkB,CACpB,mCAAmC,CAAC,MAAM,CAAC,MAAM,CAAC,CACnD,CACF,CACF,CAAC;aACH;iBAAM;gBACL,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;aACnC;SACF;IACH,CAAC;IAED,WAAW,CAAC,IAA8C;QACxD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YAClB,IAAI,CAAC,UAAU,EAAE,CAAC;SACnB;QACD,iBAAiB,CACf,IAAI,CAAC,QAAQ,EACb,oDAAoD,CACrD,CAAC;QACF,2BAA2B,CACzB,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB,EACxD,IAAI,EACJ,IAAI,CAAC,iBAAiB,CACvB,CAAC;QACF,IAAI,CAAC,iBAAiB,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC/C,CAAC;CACF"}

package/lib/stores/{QueryBuilderGraphFetchTreeUtil.d.ts → fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.d.ts}

@@ -15,8 +15,8 @@
  */
 import type { TreeNodeData, TreeData } from '@finos/legend-art';
 import { type Type, type RootGraphFetchTree, PropertyGraphFetchTree } from '@finos/legend-graph';
-import { type QueryBuilderExplorerTreeNodeData, QueryBuilderExplorerTreePropertyNodeData } from './QueryBuilderExplorerState.js';
-import type { QueryBuilderState } from './QueryBuilderState.js';
+import { type QueryBuilderExplorerTreeNodeData, QueryBuilderExplorerTreePropertyNodeData } from '../../explorer/QueryBuilderExplorerState.js';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
 export declare class QueryBuilderGraphFetchTreeNodeData implements TreeNodeData {
     isSelected?: boolean | undefined;
     isOpen?: boolean | undefined;

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchTreeUtil.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAGL,KAAK,IAAI,EACT,KAAK,kBAAkB,EAGvB,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,gCAAgC,EACrC,wCAAwC,EAEzC,MAAM,6CAA6C,CAAC;AACrD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAMpE,qBAAa,kCAAmC,YAAW,YAAY;IACrE,UAAU,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjC,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,EAAE,CAAM;IAC3B,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,IAAI,EAAE,sBAAsB,CAAC;gBAG3B,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,GAAG,SAAS,EAC5B,kBAAkB,EAAE,sBAAsB;IAU5C,IAAI,IAAI,IAAI,IAAI,CAEf;CACF;AAED,MAAM,WAAW,8BACf,SAAQ,QAAQ,CAAC,kCAAkC,CAAC;IACpD,IAAI,EAAE,kBAAkB,CAAC;CAC1B;AAsCD;;GAEG;AACH,eAAO,MAAM,uBAAuB,SAC5B,kBAAkB,KACvB,8BASF,CAAC;AAEF,eAAO,MAAM,yBAAyB,aAC1B,8BAA8B,KACvC,OAA8C,CAAC;AA6ClD,eAAO,MAAM,qBAAqB,aACtB,8BAA8B,QAClC,kCAAkC,KACvC,IAGF,CAAC;AAEF,eAAO,MAAM,2BAA2B,aAC5B,8BAA8B,oBACtB,SAAS,gCAAgC,CAAC,QACtD,wCAAwC,qBAC3B,iBAAiB,KACnC,IAgGF,CAAC"}

package/lib/stores/{QueryBuilderGraphFetchTreeUtil.js → fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.js}

@@ -15,7 +15,7 @@
  */
 import { assertType, addUniqueEntry, deleteEntry } from '@finos/legend-shared';
 import { PropertyExplicitReference, Class, PropertyGraphFetchTree, PackageableElementExplicitReference, } from '@finos/legend-graph';
-import { QueryBuilderExplorerTreePropertyNodeData, QueryBuilderExplorerTreeSubTypeNodeData, } from './QueryBuilderExplorerState.js';
+import { QueryBuilderExplorerTreePropertyNodeData, QueryBuilderExplorerTreeSubTypeNodeData, } from '../../explorer/QueryBuilderExplorerState.js';
 import { graphFetchTree_addSubTree, graphFetchTree_removeSubTree, } from '@finos/legend-application';
 export class QueryBuilderGraphFetchTreeNodeData {
     isSelected;

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchTreeUtil.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAE/E,OAAO,EAKL,yBAAyB,EACzB,KAAK,EACL,sBAAsB,EACtB,mCAAmC,GACpC,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,wCAAwC,EACxC,uCAAuC,GACxC,MAAM,6CAA6C,CAAC;AAErD,OAAO,EACL,yBAAyB,EACzB,4BAA4B,GAC7B,MAAM,2BAA2B,CAAC;AAEnC,MAAM,OAAO,kCAAkC;IAC7C,UAAU,CAAuB;IACjC,MAAM,CAAuB;IAC7B,EAAE,CAAS;IACX,KAAK,CAAS;IACd,WAAW,GAAa,EAAE,CAAC;IAC3B,QAAQ,CAAsB;IAC9B,IAAI,CAAyB;IAE7B,YACE,EAAU,EACV,KAAa,EACb,QAA4B,EAC5B,kBAA0C;QAE1C,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;QAC/B,yCAAyC;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;IAC5D,CAAC;CACF;AAOD,MAAM,cAAc,GAAG,CACrB,QAA0B,EAC1B,OAAyB,EACzB,YAAgC,EACxB,EAAE,CACV,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,GACvD,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAC3B,EAAE,CAAC;AAEL,MAAM,sBAAsB,GAAG,CAC7B,IAAoB,EACpB,UAA0D,EAC1D,KAAsD,EAClB,EAAE;IACtC,UAAU,CACR,IAAI,EACJ,sBAAsB,EACtB,0DAA0D,CAC3D,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC;IACpC,MAAM,YAAY,GAAG,UAAU,EAAE,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,kCAAkC,CACjD,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,YAAY,CAAC,EAC/C,QAAQ,CAAC,IAAI,EACb,YAAY,EACZ,IAAI,CACL,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAChC,MAAM,WAAW,GAAG,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACjE,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QACjD,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CACrC,IAAwB,EACQ,EAAE;IAClC,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,GAAG,EAA8C,CAAC;IACpE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAChC,MAAM,WAAW,GAAG,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACtE,cAAc,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QACxC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AAClC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,QAAwC,EAC/B,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC;AAElD,MAAM,gBAAgB,GAAG,CACvB,QAAwC,EACF,EAAE,CACxC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACxC,CAAC,IAAI,EAAE,EAAE;AACP,wBAAwB;AACxB,CAAC,IAAI,CAAC,IAAI,YAAY,sBAAsB;IAC1C,IAAI,CAAC,IAAI,YAAY,KAAK;IAC1B,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,CAAC;IAChC,cAAc;IACd,CAAC,IAAI,CAAC,IAAI,YAAY,sBAAsB;QAC1C,CAAC,CAAC,IAAI,CAAC,IAAI,YAAY,KAAK,CAAC;QAC7B,IAAI,CAAC,QAAQ;QACb,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CACxC,CAAC;AAEJ,MAAM,UAAU,GAAG,CACjB,QAAwC,EACxC,IAAwC,EAClC,EAAE;IACR,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ;QAC9B,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC;QACnC,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,UAAU,EAAE;QACd,WAAW,CAAC,UAAU,CAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7C,4BAA4B,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;KAC1D;SAAM;QACL,WAAW,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,4BAA4B,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;KACxD;IACD,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACjC,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,QAAwC,EAAQ,EAAE;IACvE,IAAI,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC/C,OAAO,aAAa,CAAC,MAAM,EAAE;QAC3B,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAC7B,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;KAC5C;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,QAAwC,EACxC,IAAwC,EAClC,EAAE;IACR,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC3B,aAAa,CAAC,QAAQ,CAAC,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,QAAwC,EACxC,gBAA4D,EAC5D,IAA8C,EAC9C,iBAAoC,EAC9B,EAAE;IACR,qEAAqE;IACrE,qDAAqD;IACrD,MAAM,uBAAuB,GAA6B;QACxD,IAAI,sBAAsB,CACxB,yBAAyB,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAC/C,SAAS,CACV;KACF,CAAC;IACF,IAAI,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvE,OACE,sBAAsB;QACpB,wCAAwC;QAC1C,sBAAsB,YAAY,uCAAuC,EACzE;QACA,IAAI,OAAO,GAAG,SAAS,CAAC;QACxB,IAAI,eAAe,GAAG,KAAK,CAAC;QAC5B,OACE,sBAAsB,YAAY,uCAAuC,EACzE;YACA,IAAI,CAAC,eAAe,EAAE;gBACpB,OAAO,GAAG,mCAAmC,CAAC,MAAM,CAClD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;gBACF,eAAe,GAAG,IAAI,CAAC;aACxB;YACD,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CACjD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;SACH;QACD,IACE,sBAAsB,YAAY,wCAAwC,EAC1E;YACA,MAAM,sBAAsB,GAAG,IAAI,sBAAsB,CACvD,yBAAyB,CAAC,MAAM,CAAC,sBAAsB,CAAC,QAAQ,CAAC,EACjE,OAAO,CACR,CAAC;YACF,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAClC,uBAAuB,CAAC,CAAC,CAA2B,CACrD,CAAC;YACF,uBAAuB,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;YACxD,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CACjD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;SACH;aAAM;YACL,iBAAiB,CAAC,gBAAgB,CAAC,WAAW,CAC5C,mEAAmE,CACpE,CAAC;YACF,OAAO;SACR;KACF;IAED,2EAA2E;IAC3E,uEAAuE;IACvE,0BAA0B;IAC1B,EAAE;IACF,+EAA+E;IAC/E,gDAAgD;IAChD,EAAE;IACF,wEAAwE;IACxE,mFAAmF;IACnF,oEAAoE;IACpE,IAAI,aAAa,GAAuB,SAAS,CAAC;IAClD,IAAI,UAAU,GAAmD,SAAS,CAAC;IAC3E,IAAI,UAA8C,CAAC;IACnD,KAAK,MAAM,sBAAsB,IAAI,uBAAuB,EAAE;QAC5D,aAAa,GAAG,cAAc,CAC5B,sBAAsB,CAAC,QAAQ,CAAC,KAAK,EACrC,sBAAsB,CAAC,OAAO,EAAE,KAAK,EACrC,aAAa,CACd,CAAC;QACF,MAAM,sBAAsB,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjE,IAAI,sBAAsB,EAAE;YAC1B,UAAU,GAAG,sBAAsB,CAAC;SACrC;aAAM;YACL,UAAU,GAAG,sBAAsB,CAAC;YACpC,MAAM;SACP;KACF;IAED,0EAA0E;IAC1E,IAAI,UAAU,EAAE;QACd,MAAM,SAAS,GAAG,sBAAsB,CACtC,UAAU,EACV,UAAU,EACV,QAAQ,CAAC,KAAK,CACf,CAAC;QACF,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC5C,IAAI,UAAU,EAAE;YACd,cAAc,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YACrD,yBAAyB,CAAC,UAAU,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;SAC5D;aAAM;YACL,cAAc,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC/C,yBAAyB,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;SAC1D;KACF;AACH,CAAC,CAAC"}

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { type LambdaFunction } from '@finos/legend-graph';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
+export declare const appendTakeLimit: (queryBuilderState: QueryBuilderState, lambda: LambdaFunction, previewLimit?: number | undefined) => LambdaFunction;
+//# sourceMappingURL=QueryBuilderGraphFetchValueSpecificationBuilder.d.ts.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchValueSpecificationBuilder.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAUL,KAAK,cAAc,EACpB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,eAAO,MAAM,eAAe,sBACP,iBAAiB,UAC5B,cAAc,iBACP,MAAM,GAAG,SAAS,KAChC,cAqDF,CAAC"}

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.js

@@ -0,0 +1,49 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { extractElementNameFromPath, GenericType, GenericTypeExplicitReference, matchFunctionName, PrimitiveInstanceValue, PRIMITIVE_TYPE, SimpleFunctionExpression, TYPICAL_MULTIPLICITY_TYPE, } from '@finos/legend-graph';
+import { guaranteeType } from '@finos/legend-shared';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../QueryBuilder_Const.js';
+export const appendTakeLimit = (queryBuilderState, lambda, previewLimit) => {
+    if (!previewLimit) {
+        return lambda;
+    }
+    const multiplicityOne = queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    if (lambda.expressionSequence.length === 1) {
+        const func = lambda.expressionSequence[0];
+        if (func instanceof SimpleFunctionExpression) {
+            if (matchFunctionName(func.functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.SERIALIZE)) {
+                const limit = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.INTEGER))), multiplicityOne);
+                limit.values = [previewLimit];
+                const takeFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TAKE), multiplicityOne);
+                // NOTE: `take()` does not work on `graphFetch()` or `serialize()` so we will put it
+                // right next to `all()`
+                const serializeFunction = func;
+                const graphFetchFunc = guaranteeType(serializeFunction.parametersValues[0], SimpleFunctionExpression);
+                const getAllFunc = graphFetchFunc
+                    .parametersValues[0];
+                takeFunction.parametersValues[0] = getAllFunc;
+                takeFunction.parametersValues[1] = limit;
+                graphFetchFunc.parametersValues = [
+                    takeFunction,
+                    graphFetchFunc.parametersValues[1],
+                ];
+                return lambda;
+            }
+        }
+    }
+    return lambda;
+};
+//# sourceMappingURL=QueryBuilderGraphFetchValueSpecificationBuilder.js.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchValueSpecificationBuilder.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,0BAA0B,EAC1B,WAAW,EACX,4BAA4B,EAC5B,iBAAiB,EACjB,sBAAsB,EACtB,cAAc,EACd,wBAAwB,EACxB,yBAAyB,GAG1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,iCAAiC,EAAE,MAAM,gCAAgC,CAAC;AAGnF,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,iBAAoC,EACpC,MAAsB,EACtB,YAAiC,EACjB,EAAE;IAClB,IAAI,CAAC,YAAY,EAAE;QACjB,OAAO,MAAM,CAAC;KACf;IACD,MAAM,eAAe,GACnB,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAC9D,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACJ,IAAI,MAAM,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,IAAI,YAAY,wBAAwB,EAAE;YAC5C,IACE,iBAAiB,CACf,IAAI,CAAC,YAAY,EACjB,iCAAiC,CAAC,SAAS,CAC5C,EACD;gBACA,MAAM,KAAK,GAAG,IAAI,sBAAsB,CACtC,4BAA4B,CAAC,MAAM,CACjC,IAAI,WAAW,CACb,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,gBAAgB,CACxD,cAAc,CAAC,OAAO,CACvB,CACF,CACF,EACD,eAAe,CAChB,CAAC;gBACF,KAAK,CAAC,MAAM,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9B,MAAM,YAAY,GAAG,IAAI,wBAAwB,CAC/C,0BAA0B,CAAC,iCAAiC,CAAC,IAAI,CAAC,EAClE,eAAe,CAChB,CAAC;gBAEF,oFAAoF;gBACpF,wBAAwB;gBACxB,MAAM,iBAAiB,GAAG,IAAI,CAAC;gBAC/B,MAAM,cAAc,GAAG,aAAa,CAClC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC,CAAC,EACrC,wBAAwB,CACzB,CAAC;gBACF,MAAM,UAAU,GAAG,cAAc;qBAC9B,gBAAgB,CAAC,CAAC,CAAuB,CAAC;gBAC7C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC;gBAC9C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;gBACzC,cAAc,CAAC,gBAAgB,GAAG;oBAChC,YAAY;oBACZ,cAAc,CAAC,gBAAgB,CAAC,CAAC,CAAuB;iBACzD,CAAC;gBACF,OAAO,MAAM,CAAC;aACf;SACF;KACF;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC"}

package/lib/stores/{QueryBuilderProjectionState.d.ts → fetch-structure/projection/QueryBuilderProjectionColumnState.d.ts}

@@ -14,25 +14,21 @@
  * limitations under the License.
  */
 import { type GeneratorFn } from '@finos/legend-shared';
-import { type QueryBuilderExplorerTreePropertyNodeData } from './QueryBuilderExplorerState.js';
-import { QueryBuilderPropertyExpressionState } from './QueryBuilderPropertyEditorState.js';
-import type { QueryBuilderState } from './QueryBuilderState.js';
-import { type AbstractPropertyExpression, type CompilationError, type Type, RawLambda } from '@finos/legend-graph';
-import { type QueryBuilderAggregateOperator, QueryBuilderAggregationState } from './QueryBuilderAggregationState.js';
+import { type QueryBuilderExplorerTreePropertyNodeData } from '../../explorer/QueryBuilderExplorerState.js';
+import { QueryBuilderPropertyExpressionState } from '../../QueryBuilderPropertyEditorState.js';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
+import { type AbstractPropertyExpression, type Type, RawLambda } from '@finos/legend-graph';
 import { LambdaEditorState } from '@finos/legend-application';
-export declare enum QUERY_BUILDER_PROJECTION_DND_TYPE {
-    PROJECTION_COLUMN = "PROJECTION_COLUMN"
-}
+import type { QueryBuilderProjectionState } from './QueryBuilderProjectionState.js';
+export declare const QUERY_BUILDER_PROJECTION_COLUMN_DND_TYPE = "PROJECTION_COLUMN";
 export interface QueryBuilderProjectionColumnDragSource {
     columnState: QueryBuilderProjectionColumnState;
 }
 export declare abstract class QueryBuilderProjectionColumnState {
     readonly uuid: string;
     projectionState: QueryBuilderProjectionState;
-    isBeingDragged: boolean;
     columnName: string;
     constructor(projectionState: QueryBuilderProjectionState, columnName: string);
-    setIsBeingDragged(val: boolean): void;
     setColumnName(val: string): void;
     abstract getReturnType(): Type | undefined;
 }
@@ -73,34 +69,5 @@ export declare class QueryBuilderDerivationProjectionColumnState extends QueryBu
     fetchDerivationLambdaReturnType(): GeneratorFn<void>;
     getReturnType(): Type | undefined;
 }
-export declare class QueryBuilderProjectionState {
-    queryBuilderState: QueryBuilderState;
-    columns: QueryBuilderProjectionColumnState[];
-    aggregationState: QueryBuilderAggregationState;
-    isConvertDerivationProjectionObjects: boolean;
-    constructor(queryBuilderState: QueryBuilderState, operators: QueryBuilderAggregateOperator[]);
-    convertDerivationProjectionObjects(): GeneratorFn<void>;
-    transformSimpleProjectionToDerivation(simpleProjectionColumnState: QueryBuilderSimpleProjectionColumnState): void;
-    replaceColumn(oldVal: QueryBuilderProjectionColumnState, newVal: QueryBuilderProjectionColumnState): void;
-    removeColumn(val: QueryBuilderProjectionColumnState): void;
-    addColumn(val: QueryBuilderProjectionColumnState, options?: {
-        /**
-         * Often time, we would want to enforce doing a sort when adding new column
-         * to ensure aggregate columns stay at the bottom of the list of projections
-         * But sometimes, we can opt in to use this flag to disable this sorting behavior,
-         * such as when we build/process.
-         */
-        skipSorting?: boolean | undefined;
-    }): void;
-    moveColumn(sourceIndex: number, targetIndex: number): void;
-    addNewBlankDerivation(): void;
-    revealCompilationError(compilationError: CompilationError): boolean;
-    get derivations(): QueryBuilderDerivationProjectionColumnState[];
-    get hasParserError(): boolean;
-    clearCompilationError(): void;
-    previewData(node: QueryBuilderExplorerTreePropertyNodeData): GeneratorFn<void>;
-    isValidProjectionState(): boolean;
-    getValidationErrorMessage(): string | undefined;
-}
 export {};
-//# sourceMappingURL=QueryBuilderProjectionState.d.ts.map
+//# sourceMappingURL=QueryBuilderProjectionColumnState.d.ts.map

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderProjectionColumnState.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EACL,KAAK,WAAW,EAOjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,KAAK,wCAAwC,EAE9C,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAEL,mCAAmC,EACpC,MAAM,0CAA0C,CAAC;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EACL,KAAK,0BAA0B,EAC/B,KAAK,IAAI,EAUT,SAAS,EAEV,MAAM,qBAAqB,CAAC;AAK7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAEpF,eAAO,MAAM,wCAAwC,sBAAsB,CAAC;AAE5E,MAAM,WAAW,sCAAsC;IACrD,WAAW,EAAE,iCAAiC,CAAC;CAChD;AAED,8BAAsB,iCAAiC;IACrD,QAAQ,CAAC,IAAI,SAAU;IACvB,eAAe,EAAE,2BAA2B,CAAC;IAC7C,UAAU,EAAE,MAAM,CAAC;gBAGjB,eAAe,EAAE,2BAA2B,EAC5C,UAAU,EAAE,MAAM;IAapB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAIhC,QAAQ,CAAC,aAAa,IAAI,IAAI,GAAG,SAAS;CAC3C;AAED,qBAAa,uCAAwC,SAAQ,iCAAiC;IAC5F,mBAAmB,EAAE,MAAM,CAAgC;IAC3D,uBAAuB,EAAE,mCAAmC,CAAC;gBAG3D,eAAe,EAAE,2BAA2B,EAC5C,kBAAkB,EAAE,0BAA0B,EAC9C,oBAAoB,EAAE,OAAO;IAqB/B,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAIzC,cAAc,CACZ,IAAI,EAAE,wCAAwC,EAC9C,oBAAoB,EAAE,OAAO,GAC5B,IAAI;IAkBE,aAAa,IAAI,IAAI,GAAG,SAAS;CAI3C;AAED,cAAM,2CAA4C,SAAQ,iBAAiB;IACzE,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,+BAA+B,EAAE,2CAA2C,CAAC;IAC7E;;;OAGG;IACH,kBAAkB,SAAM;gBAGtB,iBAAiB,EAAE,iBAAiB,EACpC,+BAA+B,EAAE,2CAA2C;IAO9E,IAAI,QAAQ,IAAI,MAAM,CAQrB;IAED,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAItC,kCAAkC,IAAI,WAAW,CAAC,IAAI,CAAC;IA2BvD,kCAAkC,CAAC,MAAM,EAAE,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC;CAmCxE;AAED,qBAAa,2CAA4C,SAAQ,iCAAiC;IAChG,2BAA2B,EAAE,2CAA2C,CAAC;IACzE,MAAM,EAAE,SAAS,CAAC;IAClB,UAAU,EAAE,IAAI,GAAG,SAAS,CAAC;gBAEjB,eAAe,EAAE,2BAA2B,EAAE,MAAM,EAAE,SAAS;IAkB3E,SAAS,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI;IAI/B,aAAa,CAAC,GAAG,EAAE,IAAI,GAAG,SAAS,GAAG,IAAI;IAI1C;;;;OAIG;IACF,+BAA+B,IAAI,WAAW,CAAC,IAAI,CAAC;IA4C5C,aAAa,IAAI,IAAI,GAAG,SAAS;CAG3C"}