buehler#20

Author: alex-shul

src/tsStructureParser.ts

@@ -102,7 +102,7 @@ export function parseStruct(content: string, modules: {[path: string]: Module},
             let isExport = false;
             let params: {name: string, type: string, mandatory: boolean}[] = [];
             if (name) {
-                const modifierContainer = isArrow
+                let modifierContainer = isArrow
                     ? (functionDeclaration.parent as ts.VariableDeclaration).initializer
                     : functionDeclaration;
                 if (modifierContainer && modifierContainer.modifiers) {
@@ -116,6 +116,20 @@ export function parseStruct(content: string, modules: {[path: string]: Module},
                     });
                 }
 
+                if (isArrow && !isExport) {
+                  do {
+                    modifierContainer = modifierContainer.parent as ts.Expression;
+                  } while (modifierContainer && modifierContainer.kind !== ts.SyntaxKind.VariableStatement);
+
+                  if (modifierContainer && modifierContainer.modifiers) {
+                    modifierContainer.modifiers.forEach(modi => {
+                        if (modi.kind === ts.SyntaxKind.ExportKeyword) {
+                            isExport = true;
+                        }
+                    });
+                  }
+                }
+
                 functionDeclaration.parameters.forEach(param => {
                     params.push({
                         name: param.name.getText(),

test/src/model/hero/testFunction.json

@@ -45,7 +45,7 @@
   },
   {
     "isArrow": true,
-    "isExport": false,
+    "isExport": true,
     "isAsync": false,
     "name": "arrowFunctionLikeVariable",
     "params": []
@@ -59,7 +59,7 @@
   },
   {
     "isArrow": true,
-    "isExport": false,
+    "isExport": true,
     "isAsync": true,
     "name": "arrowAsyncFunctionLikeVariable",
     "params":  [ { "name": "params1", "type": "string", "mandatory": true } ]
@@ -74,7 +74,7 @@
   {
      "isArrow": true,
      "isAsync": true,
-      "isExport": false,
+      "isExport": true,
       "name": "analogIdsTransformer",
       "params": [
         {