Refactor post

lua/fittencode/client/fitten_client.lua

@@ -38,7 +38,9 @@ function M:login(username, password, on_success, on_error)
     password = password,
   }
   Promise:new(function(resolve, reject)
-    self.rest:post(URL.LOGIN, data, function(response)
+    self.rest:post(URL.LOGIN, {
+      'Content-Type: application/json',
+    }, data, function(response)
       resolve(response)
     end, function()
       schedule(on_error)
@@ -77,7 +79,9 @@ function M:generate_one_stage(api_key, params, on_success, on_error)
   local data = params
 
   Promise:new(function(resolve, _)
-    self.rest:post(url, data, function(response)
+    self.rest:post(url, {
+      'Content-Type: application/json',
+    }, data, function(response)
       resolve(response)
     end, function()
       schedule(on_error, NetworkError:new())

lua/fittencode/rest/backend/curl/init.lua

@@ -66,7 +66,7 @@ function M:get(url, headers, data, on_success, on_error)
   _spawn(args, on_success, on_error)
 end
 
-local function post_largedata(url, encoded_data, on_success, on_error)
+local function post_largedata(url, headers, encoded_data, on_success, on_error)
   Promise:new(function(resolve, reject)
     FS.write_temp_file(encoded_data, function(_, path)
       resolve(path)
@@ -77,38 +77,42 @@ local function post_largedata(url, encoded_data, on_success, on_error)
     local args = {
       '-X',
       'POST',
-      '-H',
-      'Content-Type: application/json',
       '-d',
       '@' .. path,
       url,
     }
+    for _, v in ipairs(headers) do
+      table.insert(args, '-H')
+      table.insert(args, v)
+    end
     vim.list_extend(args, DEFAULT_ARGS)
     _spawn(args, on_success, on_error, function()
       FS.delete(path)
     end)
   end)
 end
 
-function M:post(url, data, on_success, on_error)
+function M:post(url, headers, data, on_success, on_error)
   local success, encoded_data = pcall(fn.json_encode, data)
   if not success then
     Log.error('Failed to encode data: {}', data)
     schedule(on_error)
     return
   end
   if #encoded_data > 200 then
-    return post_largedata(url, encoded_data, on_success, on_error)
+    return post_largedata(url, headers, encoded_data, on_success, on_error)
   end
   local args = {
     '-X',
     'POST',
-    '-H',
-    'Content-Type: application/json',
     '-d',
     encoded_data,
     url,
   }
+  for _, v in ipairs(headers) do
+    table.insert(args, '-H')
+    table.insert(args, v)
+  end
   vim.list_extend(args, DEFAULT_ARGS)
   _spawn(args, on_success, on_error)
 end

lua/fittencode/rest/rest.lua

@@ -18,7 +18,7 @@ end
 function M:get(url, headers, data, on_success, on_error)
 end
 
-function M:post(url, data, on_success, on_error)
+function M:post(url, headers, data, on_success, on_error)
 end
 
 return M