From 43dc4dcacb95df670c79e8e0457e893d6add4a67 Mon Sep 17 00:00:00 2001
From: eightpigs <eightpigs@outlook.com>
Date: Thu, 16 Nov 2023 23:51:33 +0800
Subject: [PATCH] Fix content-type cannot be extracted from headers

---
 lua/rest-nvim/curl/init.lua | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/lua/rest-nvim/curl/init.lua b/lua/rest-nvim/curl/init.lua
index 7529eae1..36b032b4 100644
--- a/lua/rest-nvim/curl/init.lua
+++ b/lua/rest-nvim/curl/init.lua
@@ -54,6 +54,19 @@ local function create_error_handler(opts)
   end
 end
 
+local function parse_headers(headers)
+  local parsed = {}
+  for _, header in ipairs(headers) do
+    if header ~= "" then
+      local key, value = header:match("([^:]+):%s*(.*)")
+      if key then
+        parsed[key] = value or ""
+      end
+    end
+  end
+  return parsed
+end
+
 -- get_or_create_buf checks if there is already a buffer with the rest run results
 -- and if the buffer does not exists, then create a new one
 M.get_or_create_buf = function()
@@ -103,6 +116,8 @@ local function create_callback(curl_cmd, opts)
       return
     end
     local res_bufnr = M.get_or_create_buf()
+    local header_lines = res.headers
+    res.headers = parse_headers(res.headers)
     local content_type = res.headers[utils.key(res.headers,'content-type')]
     if content_type then
       content_type = content_type:match("application/([-a-z]+)") or content_type:match("text/(%l+)")
@@ -154,9 +169,9 @@ local function create_callback(curl_cmd, opts)
       vim.api.nvim_buf_set_lines(
         res_bufnr,
         line_count + 1,
-        line_count + 1 + #res.headers,
+        line_count + 1 + #header_lines,
         false,
-        res.headers
+        header_lines
       )
     end