fix(schema-compiler): ExportNamedDeclaration for variables object (cube-js#9390)

* fix(schema-compiler): ExportNamedDeclaration for variables object

* small fix

* add tests

---------

Co-authored-by: Arkadiusz Świerczek <[email protected]>

Author: 2 people

packages/cubejs-schema-compiler/src/compiler/transpilers/ImportExportTranspiler.ts

@@ -50,10 +50,9 @@ export class ImportExportTranspiler implements TranspilerInterface {
           }
         });
         const addExportCall = t.callExpression(t.identifier('addExport'), [t.objectExpression(<t.ObjectProperty[]>declarations)]);
-        if (path.get('declaration')) {
+        if ('declaration' in path.node && path.node.declaration) {
           path.replaceWithMultiple([
-            // @todo fix without any
-            (<any>path.get('declaration')).node,
+            path.node.declaration,
             t.callExpression(t.identifier('addExport'), [
               t.objectExpression(
                 // @ts-ignore

packages/cubejs-schema-compiler/test/unit/transpilers.test.ts

@@ -1,4 +1,10 @@
+import { parse } from '@babel/parser';
+import babelGenerator from '@babel/generator';
+import babelTraverse from '@babel/traverse';
+
 import { prepareCompiler } from './PrepareCompiler';
+import { ImportExportTranspiler } from '../../src/compiler/transpilers';
+import { ErrorReporter } from '../../src/compiler/ErrorReporter';
 
 describe('Transpilers', () => {
   it('CubeCheckDuplicatePropTranspiler', async () => {
@@ -43,4 +49,36 @@ describe('Transpilers', () => {
 
     await compiler.compile();
   });
+
+  it('ImportExportTranspiler', async () => {
+    const ieTranspiler = new ImportExportTranspiler();
+    const errorsReport = new ErrorReporter();
+    const code = `
+      export const helperFunction = () => 'hello'
+      export { helperFunction as alias }
+      export default helperFunction
+    `;
+    const ast = parse(
+      code,
+      {
+        sourceFilename: 'code.js',
+        sourceType: 'module',
+        plugins: ['objectRestSpread'],
+      },
+    );
+
+    babelTraverse(ast, ieTranspiler.traverseObject(errorsReport));
+    const content = babelGenerator(ast, {}, code).code;
+
+    expect(content).toEqual(`const helperFunction = () => 'hello';
+addExport({
+  helperFunction: helperFunction
+})
+addExport({
+  alias: helperFunction
+});
+setExport(helperFunction);`);
+
+    errorsReport.throwIfAny(); // should not throw
+  });
 });