From 70414baac08aab67de4b56a00792668bf2452b96 Mon Sep 17 00:00:00 2001
From: Minhyeok Park <pmh_only@pmh.codes>
Date: Mon, 31 Mar 2025 19:14:51 +0900
Subject: [PATCH] feat: add iwre transformer

---
 src/Transforms/IWRTransform.ts | 58 ++++++++++++++++++++++++++++++++++
 src/Transforms/Transform.ts    |  4 ++-
 2 files changed, 61 insertions(+), 1 deletion(-)
 create mode 100644 src/Transforms/IWRTransform.ts

diff --git a/src/Transforms/IWRTransform.ts b/src/Transforms/IWRTransform.ts
new file mode 100644
index 0000000..7238758
--- /dev/null
+++ b/src/Transforms/IWRTransform.ts
@@ -0,0 +1,58 @@
+import { Transform } from './Transform'
+
+export const IWRETransform: Transform = {
+  name: 'iwre',
+  fn: async (v: string) => {
+    // Split the input into sections using blank lines as delimiters.
+    const sections = v
+      .split(/\n\s*\n/)
+      .map((s) => s.trim())
+      .filter(Boolean)
+    if (sections.length === 0) return ''
+
+    // Process the first section (request line and headers)
+    const headerLines = sections[0]
+      .split('\n')
+      .map((line) => line.trim())
+      .filter(Boolean)
+    if (headerLines.length === 0) return ''
+
+    // The first line should be something like "POST https://test.example.com"
+    const [method, url] = headerLines[0].split(' ')
+    const psParts: string[] = [
+      `Invoke-WebRequest -Uri "${url}" -Method ${method.toUpperCase()}`
+    ]
+
+    // Build a headers object
+    const headers: { [key: string]: string } = {}
+    for (let i = 1; i < headerLines.length; i++) {
+      const line = headerLines[i]
+      const colonIndex = line.indexOf(':')
+      if (colonIndex === -1) continue // Skip lines that aren't valid headers
+      const headerName = line.substring(0, colonIndex).trim()
+      const headerValue = line.substring(colonIndex + 1).trim()
+      headers[headerName] = headerValue
+    }
+
+    if (Object.keys(headers).length > 0) {
+      // Build the header hashtable string for PowerShell using actual newlines.
+      const headerStr = Object.entries(headers)
+        .map(([key, value]) => `    "${key}" = "${value}"`)
+        .join('\n')
+      psParts.push(`-Headers @{\n${headerStr}\n}`)
+    }
+
+    // Process the body (if any)
+    if (sections.length > 1) {
+      const body = sections.slice(1).join('\n\n').trim()
+      if (body) {
+        // Escape single quotes by doubling them.
+        const escapedBody = body.replace(/'/g, "''")
+        psParts.push(`-Body '${escapedBody}'`)
+      }
+    }
+
+    // Join parts with PowerShell's line continuation character (backtick) and an actual newline.
+    return psParts.join(' `\n')
+  }
+}
diff --git a/src/Transforms/Transform.ts b/src/Transforms/Transform.ts
index d484a20..85e51df 100644
--- a/src/Transforms/Transform.ts
+++ b/src/Transforms/Transform.ts
@@ -18,6 +18,7 @@ import {
   JSONToPythonDictTransform
 } from './PythonTransforms'
 import { CurlTransform } from './CurlTransform'
+import { IWRETransform } from './IWRTransform'
 
 export interface TransformCheckboxOption {
   type: 'CHECKBOX'
@@ -111,5 +112,6 @@ export const transforms: Transform[] = [
   JSONToPythonDictTransform,
   GzipCompressTransform,
   GzipDecompressTransform,
-  CurlTransform
+  CurlTransform,
+  IWRETransform
 ]