feat(parser): add support for parsing captures in Hurl output

jellydn · Oct 23, 2024 · 70bae26 · 70bae26
1 parent 22cfc6f
commit 70bae26
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/src/hurl-parser.ts b/src/hurl-parser.ts
@@ -9,6 +9,7 @@ interface ParsedEntry {
 	};
 	curlCommand?: string;
 	timings?: Record<string, string>;
+	captures?: Record<string, string>;
 }
 
 export interface ParsedHurlOutput {
@@ -56,6 +57,7 @@ export function parseHurlOutput(
 	let currentEntry: ParsedEntry | null = null;
 	let isResponseHeader = false;
 	let isTimings = false;
+	let isCaptures = false;
 
 	for (const line of lines) {
 		if (line.startsWith("* Executing entry")) {
@@ -72,9 +74,11 @@ export function parseHurlOutput(
 					body: "",
 				},
 				timings: {},
+				captures: {},
 			};
 			isResponseHeader = false;
 			isTimings = false;
+			isCaptures = false;
 		} else if (line.startsWith("* Request:")) {
 			const match = line.match(/\* Request:\s*\* (\w+) (.*)/);
 			if (match && currentEntry) {
@@ -131,6 +135,18 @@ export function parseHurlOutput(
 			if (currentEntry?.timings) {
 				currentEntry.timings = formatTimings(currentEntry.timings);
 			}
+		} else if (line.startsWith("* Captures:")) {
+			isCaptures = true;
+			if (currentEntry && !currentEntry.captures) {
+				currentEntry.captures = {};
+			}
+		} else if (isCaptures && line.trim() !== "") {
+			const [key, value] = line.slice(2).split(":").map((s) => s.trim());
+			if (currentEntry?.captures && key && value) {
+				currentEntry.captures[key] = value;
+			}
+		} else if (isCaptures && line.trim() === "") {
+			isCaptures = false;
 		}
 	}
 

diff --git a/test/hurl-parser.test.ts b/test/hurl-parser.test.ts
@@ -50,11 +50,39 @@ describe("parseHurlOutput", () => {
 				connect: "2.00 ms",
 				total: "10.00 ms",
 			},
+			captures: {},
 		});
 	});
 
 	it("should handle empty input", () => {
 		const result = parseHurlOutput("", "");
 		expect(result.entries).toHaveLength(0);
 	});
+
+	it("should parse captures", () => {
+		const stderr = `
+* ------------------------------------------------------------------------------
+* Executing entry 1
+* Request:
+* GET https://example.com/api
+* Response:
+< HTTP/1.1 200 OK
+< Content-Type: application/json
+* Response body:
+{"id": "12345", "name": "Example"}
+* Captures:
+* id: 12345
+* name: Example
+* ------------------------------------------------------------------------------
+`;
+		const stdout = '{"id": "12345", "name": "Example"}';
+
+		const result = parseHurlOutput(stderr, stdout);
+
+		expect(result.entries).toHaveLength(1);
+		expect(result.entries[0].captures).toEqual({
+			id: '12345',
+			name: 'Example'
+		});
+	});
 });