[minor] enum support

packages/custom-functions-metadata/src/generate.ts

@@ -1,4 +1,4 @@
-import { IFunction, IGenerateResult, IParseTreeResult, parseTree } from "./parseTree";
+import { IFunction, IEnum, IGenerateResult, IParseTreeResult, parseTree } from "./parseTree";
 import { existsSync, readFileSync } from "fs";
 
 /**
@@ -7,51 +7,55 @@ import { existsSync, readFileSync } from "fs";
  * @param outputFileName - Name of the file to create (i.e functions.json)
  */
 export async function generateCustomFunctionsMetadata(
-    input: string | string[],
-    wantConsoleOutput: boolean = false
-  ): Promise<IGenerateResult> {
-    const inputFiles: string[] = Array.isArray(input) ? input : [input];
-    const functions: IFunction[] = [];
-    const generateResults: IGenerateResult = {
-      metadataJson: "",
-      associate: [],
-      errors: [],
-    };
-
-    if (input && inputFiles.length > 0) {
-      inputFiles.forEach((inputFile) => {
-        inputFile = inputFile.trim();
-        if (!inputFile) {
-          // ignore empty strings
-        } else if (!existsSync(inputFile)) {
-          throw new Error(`File not found: ${inputFile}`);
-        } else {
-          const sourceCode = readFileSync(inputFile, "utf-8");
-          const parseTreeResult: IParseTreeResult = parseTree(sourceCode, inputFile);
-          parseTreeResult.extras.forEach((extra) => extra.errors.forEach((err) => generateResults.errors.push(err)));
-
-          if (generateResults.errors.length > 0) {
-            if (wantConsoleOutput) {
-              console.error("Errors in file: " + inputFile);
-              generateResults.errors.forEach((err) => console.error(err));
-            }
-          } else {
-            functions.push(...parseTreeResult.functions);
-            generateResults.associate.push(...parseTreeResult.associate);
+  input: string | string[],
+  wantConsoleOutput: boolean = false
+): Promise<IGenerateResult> {
+  const inputFiles: string[] = Array.isArray(input) ? input : [input];
+  const functions: IFunction[] = [];
+  const enums: IEnum[] = [];
+  const generateResults: IGenerateResult = {
+    metadataJson: "",
+    associate: [],
+    errors: [],
+  };
+
+  if (input && inputFiles.length > 0) {
+    inputFiles.forEach((inputFile) => {
+      inputFile = inputFile.trim();
+      if (!inputFile) {
+        // ignore empty strings
+      } else if (!existsSync(inputFile)) {
+        throw new Error(`File not found: ${inputFile}`);
+      } else {
+        const sourceCode = readFileSync(inputFile, "utf-8");
+        const parseTreeResult: IParseTreeResult = parseTree(sourceCode, inputFile);
+        parseTreeResult.extras.forEach((extra) => extra.errors.forEach((err) => generateResults.errors.push(err)));
+
+        if (generateResults.errors.length > 0) {
+          if (wantConsoleOutput) {
+            console.error("Errors in file: " + inputFile);
+            generateResults.errors.forEach((err) => console.error(err));
           }
+        } else {
+          functions.push(...parseTreeResult.functions);
+          generateResults.associate.push(...parseTreeResult.associate);
+          enums.push(...parseTreeResult.enums);
         }
-      });
-
-      if (functions.length > 0) {
-        const metadata = {
-          allowCustomDataForDataTypeAny: true,
-          functions: functions,
-        }
-        generateResults.metadataJson = JSON.stringify(metadata, null, 4);
       }
+    });
+
+    if (functions.length > 0) {
+      const metadata: { allowCustomDataForDataTypeAny: boolean; functions: IFunction[]; enums?: IEnum[] } = {
+        allowCustomDataForDataTypeAny: true,
+        functions: functions,
+        enums: enums,
+      };
+      if (enums.length == 0) {
+        delete metadata.enums;
+      }
+      generateResults.metadataJson = JSON.stringify(metadata, null, 4);
     }
-
-    return generateResults;
   }
-
-
+
+  return generateResults;
+}