feat: allow await in IIFEs for code and pg

testinggospels · Aug 4, 2023 · ceefc9b · ceefc9b
1 parent 61da2f5
commit ceefc9b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/src/handlebar/CodeHelper.ts b/src/handlebar/CodeHelper.ts
@@ -15,12 +15,12 @@ export class CodeHelper {
    * @returns {void}
    */
   register = () => {
-    this.Handlebars.registerHelper("code", (context: any) => {
+    this.Handlebars.registerHelper("code", async (context: any) => {
       /* eslint-disable no-unused-vars */
       const request: express.Request = context.data.root.request;
       const logger = context.data.root.logger;
       /* eslint-disable no-unused-vars */
-      const code = eval(context.fn(this));
+      const code = await eval(context.fn(this));
       code["CamouflageResponseType"] = "code";
       return JSON.stringify(code);
     });

diff --git a/src/handlebar/PgHelper.ts b/src/handlebar/PgHelper.ts
@@ -34,7 +34,7 @@ export class PgHelper {
                 const result = await pool.query(query)
                 log.debug(`[POSTGRES] Query response: ${JSON.stringify(result.rows)}`)
                 const fn = await context.fn(this)
-                const code = eval(fn);
+                const code = await eval(fn);
                 code["CamouflageResponseType"] = "code";
                 return JSON.stringify(code);
             } catch (err) {

diff --git a/src/parser/HttpParser.ts b/src/parser/HttpParser.ts
@@ -183,6 +183,7 @@ export class HttpParser {
           responseBody = responseBody.replace(/}}}/, "}} }");
           const template = Handlebars.compile(responseBody);
           try {
+            responseBody = responseBody.split("<br/>").join("\n").trim()
             const codeResponse = JSON.parse(
               responseBody.replace(/&quot;/g, '"')
             );