From d55aa14c56a2eba30a9dc1571685f9fe72acff4d Mon Sep 17 00:00:00 2001
From: Miguel Grinberg <miguel.grinberg@gmail.com>
Date: Thu, 7 Nov 2024 17:43:34 +0000
Subject: [PATCH] parse {placeholders} in request URLs (#64)

(cherry picked from commit d81ce6bbc9221e180ae69b8df1caa5724c71a915)
---
 src/parse.ts        | 11 +++++++++++
 tests/parse.test.ts | 32 ++++++++++++++++++++++----------
 2 files changed, 33 insertions(+), 10 deletions(-)

diff --git a/src/parse.ts b/src/parse.ts
index a4c9a4f..ab98d9d 100644
--- a/src/parse.ts
+++ b/src/parse.ts
@@ -142,11 +142,22 @@ function parseCommand(source: string, options: ParseOptions) {
   skip(" ", "\n");
   const urlStart = index;
   until("{", "\n");
+  if (source[index] == "{" && source[index + 1].match(/[a-z]/i)) {
+    // this is a placeholder element inside the URL (as used in doc examples),
+    // so we continue scanning
+    index++;
+    until("{", "\n");
+  }
 
   data.url = source.slice(urlStart, index).trim();
   if (data.url[0] != "/") {
     data.url = "/" + data.url;
   }
+  data.url = data.url
+    // replaces { with %7B (braces in many doc examples are not URIencoded)
+    .replace(/{/g, "%7B")
+    // replaces } with %7D
+    .replace(/}/g, "%7D");
   const parsedUrl = new URL(`http://localhost${data.url}`);
   data.rawPath =
     parsedUrl.pathname != "/"
diff --git a/tests/parse.test.ts b/tests/parse.test.ts
index cf05c37..ce43357 100644
--- a/tests/parse.test.ts
+++ b/tests/parse.test.ts
@@ -42,6 +42,18 @@ describe("parse", () => {
     });
   });
 
+  it("parses search POST request given in a single line", async () => {
+    const req = await parseRequest(`POST /my-index/_search { "size": 5 }`);
+    expect(req).toMatchObject({
+      api: "search",
+      params: { index: "my-index" },
+      method: "POST",
+      url: "/my-index/_search",
+      path: "/my-index/_search",
+      body: { size: 5 },
+    });
+  });
+
   it("parses a complex sequence of requests", async () => {
     const reqs = await parseRequests(`PUT /customer/_doc/1?foo=bar
 {
@@ -73,11 +85,11 @@ POST /_bulk?foo=bar
 { "name": "John\\nDoe" }
 
 
-GET /customer/_doc/1
+GET /{customer}/_doc/1
 
 POST _nodes/reload_secure_settings\n{\n  "reload_secure_settings": "s3cr3t" <1>\n}
 
-GET my_index/_analyze <3>\n{\n  "field": "text",\n  "text": "The quick Brown Foxes."\n}
+GET {my_index}/_analyze <3>\n{\n  "field": "text",\n  "text": "The quick Brown Foxes."\n}
 
 POST\n_ml/anomaly_detectors/it_ops_new_logs/model_snapshots/1491852978/_update\n{\n  "description": "Snapshot 1",\n  "retain": true\n}
 `);
@@ -152,11 +164,11 @@ POST\n_ml/anomaly_detectors/it_ops_new_logs/model_snapshots/1491852978/_update\n
     });
     expect(reqs[7]).toMatchObject({
       api: "get",
-      params: { index: "customer", id: "1" },
+      params: { index: "{customer}", id: "1" },
       method: "GET",
-      url: "/customer/_doc/1",
-      path: "/customer/_doc/1",
-      rawPath: "/customer/_doc/1",
+      url: "/%7Bcustomer%7D/_doc/1",
+      path: "/{customer}/_doc/1",
+      rawPath: "/%7Bcustomer%7D/_doc/1",
     });
     expect(reqs[8]).toMatchObject({
       api: "nodes.reload_secure_settings",
@@ -169,11 +181,11 @@ POST\n_ml/anomaly_detectors/it_ops_new_logs/model_snapshots/1491852978/_update\n
     });
     expect(reqs[9]).toMatchObject({
       api: "indices.analyze",
-      params: { index: "my_index" },
+      params: { index: "{my_index}" },
       method: "GET",
-      url: "/my_index/_analyze",
-      path: "/my_index/_analyze",
-      rawPath: "/my_index/_analyze",
+      url: "/%7Bmy_index%7D/_analyze",
+      path: "/{my_index}/_analyze",
+      rawPath: "/%7Bmy_index%7D/_analyze",
       body: { field: "text", text: "The quick Brown Foxes." },
     });
     expect(reqs[10]).toMatchObject({