async/await transpiller, wip

wpdas · Mar 29, 2024 · d23d4cf · d23d4cf
1 parent 96c92b9
commit d23d4cf
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/lib/actions/transformSchemaToWidget.js b/lib/actions/transformSchemaToWidget.js
@@ -41,6 +41,7 @@ const transformImports = require("../parsers/transformImports");
 const analyzeFunctionSignature = require("../parsers/analyzeFunctionSignature");
 const removeFunctionParams = require("../parsers/removeFunctionParams");
 const checkSyntaxError = require("../parsers/checkSyntaxError");
+const transformAsyncAwait = require("../parsers/transformAsyncAwait");
 
 /**
  * @param {string} content
@@ -861,6 +862,14 @@ const transformSchemaToWidget = (fileSchemas, additionalFileSchemas) => {
     );
   });
 
+  // Faz transformação de async/await para o formato promisify
+  fileSchemas.forEach((fileSchema, fileSchemaIndex) => {
+    fileSchema.finalFileBundle = transformAsyncAwait(
+      fileSchema.finalFileBundle,
+    );
+    fileSchemas[fileSchemaIndex] = fileSchema;
+  });
+
   return fileSchemas;
 };
 

diff --git a/lib/parsers/transformAsyncAwait.js b/lib/parsers/transformAsyncAwait.js
@@ -0,0 +1,36 @@
+/**
+ * Experimental: Gera uma estrutura entendível pelo Near VM a partir de um await
+ * @param {string} code
+ * @returns
+ */
+function transformAsyncAwait(code) {
+  // Primeiro, remove 'async ' de todas as funções
+  let transformedCode = code.replace(/async /g, "");
+
+  // Regex ajustada para capturar chamadas de função que abrangem múltiplas linhas
+  const pattern = /const (\w+) = await (.*?);/gs;
+
+  // Função para gerar o novo bloco de código substituto
+  function replaceWithPromisify(match, varName, asyncCall) {
+    // Removendo espaços em branco extras e quebras de linha do início e do fim do asyncCall
+    asyncCall = asyncCall.replace(/^\s+|\s+$/g, "");
+    return `let ${varName} = state.${varName};
+  if (!${varName}) {
+    // 1 - usa o promisify
+    promisify(
+      () => ${asyncCall},
+      (data) => State.update({ ${varName}: data }),
+      () => State.update({ ${varName}: null }),
+    );
+
+    return "";
+  }`;
+  }
+
+  // Realizando a substituição no código transformado
+  transformedCode = transformedCode.replace(pattern, replaceWithPromisify);
+
+  return transformedCode;
+}
+
+module.exports = transformAsyncAwait;