fix array issue in docgen

Cyace84 · Aug 6, 2023 · dae1a17 · dae1a17
1 parent 87a0d29
commit dae1a17
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/ast-builder/compile-ast.ts b/src/ast-builder/compile-ast.ts
@@ -2,7 +2,7 @@ import fs from "fs";
 import { execSync } from "child_process";
 import { Config } from "../config";
 import { getAstsFromSources, getContractsList } from "./getters";
-import { basename, join, resolve } from "path";
+import { basename, extname, join, resolve } from "path";
 
 const createDirectoryIfNotExists = (dir: string) => {
   if (!fs.existsSync(dir)) {
@@ -51,6 +51,7 @@ export const compileAst = (config: Config) => {
   deleteDirectoryIfExists(ast_cache_path);
   compileExternalAst(config);
   renameAstFiles(astOutputPath);
+  wrapAstInArray(astOutputPath);
 };
 
 /**
@@ -93,3 +94,27 @@ const renameAstFiles = (dir: string) => {
     }
   });
 };
+
+const wrapAstInArray = (dir: string) => {
+  const files = fs.readdirSync(dir);
+
+  files.forEach((file) => {
+    if (extname(file) === ".json") {
+      const filePath = join(dir, file);
+      const content = fs.readFileSync(filePath, "utf8");
+      let jsonContent;
+
+      try {
+        jsonContent = JSON.parse(content);
+      } catch (error) {
+        console.error(`Error parsing file ${file}:`, error);
+        return;
+      }
+
+      if (!Array.isArray(jsonContent)) {
+        const wrappedContent = [jsonContent];
+        fs.writeFileSync(filePath, JSON.stringify(wrappedContent, null, 2));
+      }
+    }
+  });
+};
diff --git a/src/ast-builder/getters.ts b/src/ast-builder/getters.ts
@@ -121,7 +121,7 @@ export const getAstsFromSources = (astDir: string, root: string) => {
   const sources: Sources = {};
   const fullSources: FullSources = {};
   astSources.forEach(astSourceFullPath => {
-    const astContent: SourceUnit[] = [require(astSourceFullPath) as SourceUnit];
+    const astContent: SourceUnit[] = require(astSourceFullPath) as SourceUnit[];
     const withNormalPathAsts = renameAbsolutePaths(root, astContent);
     const astContractName = getContractName(astSourceFullPath)!;
     const mainAst = getMainAst(