fix drill members inheritance (cases of non-owned members)

Author: KSDaemon

packages/cubejs-schema-compiler/src/compiler/CubeSymbols.ts

@@ -185,6 +185,15 @@ type MemberSets = {
   allMembers: Set<string>;
 };
 
+type ViewResolvedMember = {
+  member: string;
+  name: string;
+};
+
+type ViewExcludedMember = {
+  member: string;
+};
+
 const FunctionRegex = /function\s+\w+\(([A-Za-z0-9_,]*)|\(([\s\S]*?)\)\s*=>|\(?(\w+)\)?\s*=>/;
 export const CONTEXT_SYMBOLS = {
   SECURITY_CONTEXT: 'securityContext',
@@ -569,8 +578,10 @@ export class CubeSymbols implements TranspilerSymbolResolver {
     // because drillMembers processing for views in generateIncludeMembers() relies on this
     const types = ['hierarchies', 'dimensions', 'measures', 'segments'];
 
+    const viewAllMembers: ViewResolvedMember[] = [];
+
     for (const type of types) {
-      let cubeIncludes: any[] = [];
+      let cubeIncludes: ViewResolvedMember[] = [];
 
       // If the hierarchy is included all members from it should be included as well
       // Extend `includes` with members from hierarchies that should be auto-included
@@ -596,6 +607,7 @@ export class CubeSymbols implements TranspilerSymbolResolver {
       }) : includedCubes;
 
       cubeIncludes = this.membersFromCubes(cube, cubes, type, errorReporter, splitViews, memberSets) || [];
+      viewAllMembers.push(...cubeIncludes);
 
       if (type === 'hierarchies') {
         for (const member of cubeIncludes) {
@@ -613,7 +625,7 @@ export class CubeSymbols implements TranspilerSymbolResolver {
         }
       }
 
-      const includeMembers = this.generateIncludeMembers(cubeIncludes, type, cube);
+      const includeMembers = this.generateIncludeMembers(cubeIncludes, type, cube, viewAllMembers);
       this.applyIncludeMembers(includeMembers, cube, type, errorReporter);
 
       const existing = cube.includedMembers ?? [];
@@ -665,8 +677,8 @@ export class CubeSymbols implements TranspilerSymbolResolver {
     errorReporter: ErrorReporter,
     splitViews: SplitViews,
     memberSets: MemberSets
-  ) {
-    const result: any[] = [];
+  ): ViewResolvedMember[] {
+    const result: ViewResolvedMember[] = [];
     const seen = new Set<string>();
 
     for (const cubeInclude of cubes) {
@@ -764,7 +776,8 @@ export class CubeSymbols implements TranspilerSymbolResolver {
           splitViewDef = splitViews[viewName];
         }
 
-        const includeMembers = this.generateIncludeMembers(finalIncludes, type, splitViewDef);
+        const viewAllMembers: ViewResolvedMember[] = [];
+        const includeMembers = this.generateIncludeMembers(finalIncludes, type, splitViewDef, viewAllMembers);
         this.applyIncludeMembers(includeMembers, splitViewDef, type, errorReporter);
       } else {
         for (const member of finalIncludes) {
@@ -780,7 +793,7 @@ export class CubeSymbols implements TranspilerSymbolResolver {
     return result;
   }
 
-  protected diffByMember(includes: any[], excludes: any[]) {
+  protected diffByMember(includes: ViewResolvedMember[], excludes: ViewExcludedMember[]) {
     const excludesMap = new Map();
 
     for (const exclude of excludes) {
@@ -794,15 +807,7 @@ export class CubeSymbols implements TranspilerSymbolResolver {
     return this.symbols[cubeName]?.cubeObj()?.[type]?.[memberName];
   }
 
-  protected generateIncludeMembers(members: any[], type: string, targetCube: CubeDefinitionExtended) {
-    const availableDimMembers = new Set<string>();
-
-    if (type === 'measures') {
-      Object.keys(targetCube.dimensions || {}).forEach(dimName => {
-        availableDimMembers.add(`${targetCube.name}.${dimName}`);
-      });
-    }
-
+  protected generateIncludeMembers(members: any[], type: string, targetCube: CubeDefinitionExtended, viewAllMembers: ViewResolvedMember[]) {
     return members.map(memberRef => {
       const path = memberRef.member.split('.');
       const resolvedMember = this.getResolvedMember(type, path[path.length - 2], path[path.length - 1]);
@@ -824,15 +829,16 @@ export class CubeSymbols implements TranspilerSymbolResolver {
 
         const drillMembersArray = (Array.isArray(evaluatedDrillMembers)
           ? evaluatedDrillMembers
-          : [evaluatedDrillMembers]).map(member => {
-          const memberParts = member.split('.');
-          if (memberParts[0] === sourceCubeName) {
-            return `${targetCube.name}.${memberParts[1]}`;
+          : [evaluatedDrillMembers]);
+
+        const filteredDrillMembers = drillMembersArray.flatMap(member => {
+          const found = viewAllMembers.find(v => v.member.endsWith(member));
+          if (!found) {
+            return [];
           }
-          return member; // Keep as-is if not from source cube
-        });
 
-        const filteredDrillMembers = drillMembersArray.filter(member => availableDimMembers.has(member));
+          return [`${targetCube.name}.${found.name}`];
+        });
 
         processedDrillMembers = () => filteredDrillMembers;
       }