feat: graphql support (#433)

Author: boltlessengineer

lua/rest-nvim/client/curl/cli.lua

@@ -298,11 +298,9 @@ function builder.build(req, ignore_stats)
     if req.body then
         if req.body.__TYPE == "external" then
             insert(args, builder.file(req.body.data.path))
-        elseif req.body.__TYPE == "graphql" then
-            log.error("graqphql body is not supportted yet")
         elseif req.body.__TYPE == "multipart_form_data" then
             log.error("multipart-form-data body is not supportted yet")
-        elseif vim.list_contains({ "json", "xml", "raw" }, req.body.__TYPE) then
+        elseif vim.list_contains({ "json", "xml", "raw", "graphql", }, req.body.__TYPE) then
             insert(args, builder.raw_body(req.body.data))
         else
             log.error(("unkown body type: '%s'"):format(req.body.__TYPE))

lua/rest-nvim/client/curl_cli.lua

@@ -11,6 +11,7 @@ local COMPATIBLE_METHODS = {
     "CONNECT",
     "PATCH",
     "LIST",
+    "GRAPHQL",
 }
 
 ---@type rest.Client

lua/rest-nvim/parser/init.lua

@@ -136,6 +136,19 @@ function parser.parse_body(content_type, body_node, source, context)
             name = get_node_field_text(body_node, "name", source),
             path = path,
         }
+    elseif node_type == "graphql_body" then
+        body.__TYPE = "graphql"
+        local query_text = vim.treesitter.get_node_text(assert(body_node:named_child(0)), source)
+        local variables_text
+        local variables_node = body_node:named_child(1)
+        if variables_node then
+            variables_text = vim.treesitter.get_node_text(variables_node, source)
+        end
+        body.data = vim.json.encode({
+            query = query_text,
+            variables = vim.json.decode(variables_text)
+        })
+        logger.debug(body.data)
     elseif node_type == "json_body" or content_type == "application/json" then
         body.__TYPE = "json"
         body.data = vim.trim(vim.treesitter.get_node_text(body_node, source))
@@ -171,8 +184,6 @@ function parser.parse_body(content_type, body_node, source, context)
         body.__TYPE = "multipart_form_data"
         -- TODO:
         logger.error("multipart form data is not supported yet")
-    elseif node_type == "graphql_body" then
-        logger.error("graphql body is not supported yet")
     end
     return body
 end
@@ -370,6 +381,9 @@ function parser.parse(node, source, ctx)
         logger.info("no method provided, falling back to 'GET'")
         method = "GET"
     end
+    if method == "GRAPHQL" then
+        method = "POST"
+    end
     -- NOTE: url will be parsed after because in-place variables should be parsed first
     local url
 

spec/examples/graphql.http

@@ -0,0 +1,30 @@
+@token = YOUR_GITHUB_OAUTH_TOKEN
+
+### GraphQL request
+GRAPHQL https://api.github.com/graphql
+Authorization: token {{token}}
+
+query ($name: String!, $owner: String!) {
+    repository(name: $name, owner: $owner) {
+        name
+            fullName: nameWithOwner
+            description
+            diskUsage
+            forkCount
+            stargazers(first: 5) {
+                totalCount
+                nodes {
+                    login
+                    name
+                }
+            }
+        watchers {
+            totalCount
+        }
+    }
+}
+
+{
+    "name": "NativeVim",
+    "owner": "boltlessengineer"
+}

spec/parser/http_parser_spec.lua

@@ -202,6 +202,41 @@ key5 = value5
                 },
             }, parser.parse(req_node, source))
         end)
+        it("parse graphql body", function ()
+            local source = open("spec/examples/graphql.http")
+            local _, tree = utils.ts_parse_source(source)
+            local req_node = assert(tree:root():child(1))
+            local req = parser.parse(req_node, source)
+            assert(req)
+            assert.same("POST", req.method)
+            assert.same("graphql", req.body.__TYPE)
+            assert.same({
+                query = [[query ($name: String!, $owner: String!) {
+    repository(name: $name, owner: $owner) {
+        name
+            fullName: nameWithOwner
+            description
+            diskUsage
+            forkCount
+            stargazers(first: 5) {
+                totalCount
+                nodes {
+                    login
+                    name
+                }
+            }
+        watchers {
+            totalCount
+        }
+    }
+}
+]],
+                variables = {
+                    name = "NativeVim",
+                    owner = "boltlessengineer",
+                }
+            }, vim.json.decode(req.body.data))
+        end)
     end)
 
     describe("variables", function()