feat: moving some code over to using the new post processing system

erunion · erunion · commit 93cdba310053 · 2022-07-16T18:46:58.000-07:00
diff --git a/src/targets/node/request/client.ts b/src/targets/node/request/client.ts
@@ -27,7 +27,7 @@ export const request: Client = {
     };
 
     let includeFS = false;
-    const { push, blank, join, unshift } = new CodeBuilder({ indent: opts.indent });
+    const { push, blank, join, unshift, addPostProcessor } = new CodeBuilder({ indent: opts.indent });
 
     push("const request = require('request');");
     blank();
@@ -89,6 +89,8 @@ export const request: Client = {
 
           reqOpts.formData[param.name] = attachment;
         });
+
+        addPostProcessor(code => code.replace(/'fs\.createReadStream\((.*)\)'/, "fs.createReadStream('$1')"));
         break;
 
       default:
@@ -107,6 +109,7 @@ export const request: Client = {
         push(`jar.setCookie(request.cookie('${encodeURIComponent(name)}=${encodeURIComponent(value)}'), '${url}');`);
       });
       blank();
+      addPostProcessor(code => code.replace(/'JAR'/, 'jar'));
     }
 
     if (includeFS) {
@@ -122,8 +125,6 @@ export const request: Client = {
     push('console.log(body);', 1);
     push('});');
 
-    return join()
-      .replace("'JAR'", 'jar')
-      .replace(/'fs\.createReadStream\((.*)\)'/, "fs.createReadStream('$1')");
+    return join();
   },
 };
diff --git a/src/targets/python/requests/client.ts b/src/targets/python/requests/client.ts
@@ -31,7 +31,7 @@ export const requests: Client<RequestsOptions> = {
       ...options,
     };
     // Start snippet
-    const { push, blank, join } = new CodeBuilder({ indent: opts.indent });
+    const { push, blank, join, addPostProcessor } = new CodeBuilder({ indent: opts.indent });
 
     // Import requests
     push('import requests');
@@ -94,6 +94,9 @@ export const requests: Client<RequestsOptions> = {
             hasPayload = true;
           }
         }
+
+        // The `open()` call must be a literal in the code snippet.
+        addPostProcessor(code => code.replace(/"open\('(.+)', 'rb'\)"/g, 'open("$1", "rb")'));
         break;
 
       default: {
@@ -167,10 +170,6 @@ export const requests: Client<RequestsOptions> = {
     // Print response
     push('print(response.text)');
 
-    return (
-      join()
-        // The `open()` call must be a literal in the code snippet.
-        .replace(/"open\('(.+)', 'rb'\)"/g, 'open("$1", "rb")')
-    );
+    return join();
   },
 };
