move ContextEvaluator to ts

Author: KSDaemon

packages/cubejs-schema-compiler/src/compiler/ContextEvaluator.js

@@ -0,0 +1,47 @@
+import type { CubeEvaluator } from './CubeEvaluator';
+import type { ErrorReporter } from './ErrorReporter';
+import { CompilerInterface } from './PrepareCompiler';
+
+export interface ContextInput {
+  name: string;
+  contextMembers: any;
+}
+
+export interface CompiledContext {
+  name: string;
+  contextMembers: any;
+}
+
+export class ContextEvaluator implements CompilerInterface {
+  private cubeEvaluator: CubeEvaluator;
+
+  private contextDefinitions: Record<string, CompiledContext>;
+
+  public constructor(cubeEvaluator: CubeEvaluator) {
+    this.cubeEvaluator = cubeEvaluator;
+    this.contextDefinitions = {};
+  }
+
+  public compile(contexts: ContextInput[], _errorReporter?: ErrorReporter): void {
+    if (contexts.length === 0) {
+      return;
+    }
+
+    // TODO: handle duplications, context names must be uniq
+    this.contextDefinitions = {};
+    for (const context of contexts) {
+      this.contextDefinitions[context.name] = this.compileContext(context);
+    }
+  }
+
+  private compileContext(context: ContextInput): CompiledContext {
+    return {
+      name: context.name,
+      contextMembers: this.cubeEvaluator.evaluateReferences(null, context.contextMembers)
+    };
+  }
+
+  public get contextList(): string[] {
+    return Object.keys(this.contextDefinitions);
+  }
+}

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

@@ -914,9 +914,7 @@ export class CubeSymbols implements TranspilerSymbolResolver, CompilerInterface
    * refactoring.
    */
   protected evaluateContextFunction(cube: any, contextFn: any, context: any = {}) {
-    const cubeEvaluator = this;
-
-    return cubeEvaluator.resolveSymbolsCall(contextFn, (name: string) => {
+    return this.resolveSymbolsCall(contextFn, (name: string) => {
       const resolvedSymbol = this.resolveSymbol(cube, name);
       if (resolvedSymbol) {
         return resolvedSymbol;
@@ -931,7 +929,7 @@ export class CubeSymbols implements TranspilerSymbolResolver, CompilerInterface
     });
   }
 
-  protected evaluateReferences<T extends ToString | Array<ToString>>(
+  public evaluateReferences<T extends ToString | Array<ToString>>(
     cube: string | null,
     referencesFn: (...args: Array<unknown>) => T,
     options: { collectJoinHints?: boolean, originalSorting?: boolean } = {}