adjust returned keys for each module

wpdas · Apr 27, 2024 · 31b1e2a · 31b1e2a
1 parent 8f580a2
commit 31b1e2a
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 4 deletions.
diff --git a/lib/parsers/extractTopLevelDeclarations.js b/lib/parsers/extractTopLevelDeclarations.js
@@ -1,3 +1,5 @@
+// NOTE: Este arquivo nao esta sendo usado, ver se tem algo útil, se nao, deletar!
+
 const parser = require("@babel/parser");
 const traverse = require("@babel/traverse").default;
 const generate = require("@babel/generator").default;

diff --git a/lib/parsers/filterReturn.js b/lib/parsers/filterReturn.js
@@ -1,6 +1,7 @@
 const parser = require("@babel/parser");
 const traverse = require("@babel/traverse").default;
 const t = require("@babel/types");
+const removeGetNameAndLastComma = require("./removeGetNameAndLastComma");
 const generate = require("@babel/generator").default;
 
 /**
@@ -75,7 +76,7 @@ function filterReturn(code, keys) {
 
   // Gera o novo código a partir do AST modificado
   const { code: newCode } = generate(ast);
-  return newCode;
+  return removeGetNameAndLastComma(newCode);
 }
 
 module.exports = filterReturn;
diff --git a/lib/parsers/removeGetNameAndLastComma.js b/lib/parsers/removeGetNameAndLastComma.js
@@ -0,0 +1,8 @@
+// Troca o "const Get = " por "", isso porque o módulo é a função diretamente
+// e remove o último ";" encontrado porque a função vai ser colocar em uma lista de objetos, ou seja,
+// vai ter um "," separando cada objeto.
+const removeGetNameAndLastComma = (code) => {
+  return code.replace("const Get = ", "").replace(/;(?=[^;]*$)/, "");
+};
+
+module.exports = removeGetNameAndLastComma;
diff --git a/lib/parsers/wrapCodeInGetFunction.js b/lib/parsers/wrapCodeInGetFunction.js
@@ -6,6 +6,7 @@ const babel = require("@babel/core");
 // const reactPreset = require("@babel/preset-react");
 const typescriptPreset = require("@babel/preset-typescript");
 const convertFunctionsToArrow = require("./convertFunctionsToArrow");
+const removeGetNameAndLastComma = require("./removeGetNameAndLastComma");
 
 /**
  * Esta função vai cobrir o arquivo/code com uma função Get e todas as declarações sendo exportadas
@@ -108,9 +109,9 @@ function wrapCodeInGetFunction(code, filePath) {
   // Troca o "const Get = " por "", isso porque o módulo é a função diretamente
   // e remove o último ";" encontrado porque a função vai ser colocar em uma lista de objetos, ou seja,
   // vai ter um "," separando cada objeto.
-  return convertFunctionsToArrow(transformedCode)
-    .replace("const Get = ", "")
-    .replace(/;(?=[^;]*$)/, "");
+  return removeGetNameAndLastComma(convertFunctionsToArrow(transformedCode));
+  // .replace("const Get = ", "")
+  // .replace(/;(?=[^;]*$)/, "");
 }
 
 module.exports = wrapCodeInGetFunction;