refactor(nvim): Move flatten_table_to_string to vectorcode.utils

Author: Zhe Yu

lua/vectorcode/init.lua

@@ -1,6 +1,7 @@
 local M = {}
 
 local vc_config = require("vectorcode.config")
+local utils = require("vectorcode.utils")
 local logger = vc_config.logger
 local get_config = vc_config.get_user_config
 local notify_opts = vc_config.notify_opts
@@ -191,8 +192,8 @@ function M.check(check_item, stdout_cb)
     return_code = code
     if type(stdout_cb) == "function" then
       stdout_cb({
-        stdout = table.concat(vim.iter(result or {}):flatten(math.huge):totable()),
-        stderr = table.concat(vim.iter(error or {}):flatten(math.huge):totable()),
+        stdout = utils.flatten_table_to_string(result),
+        stderr = utils.flatten_table_to_string(error),
         code = code,
         signal = signal,
       })

lua/vectorcode/integrations/codecompanion/common.lua

@@ -13,27 +13,14 @@ return {
   ---@param t table|string|nil
   ---@return string
   flatten_table_to_string = function(t)
-    if t == nil then
-      return ""
-    end
-    if type(t) == "string" then
-      return t
-    end
-
-    -- Handle empty tables or tables with empty strings
-    local flattened = vim
-      .iter(t)
-      :flatten(math.huge)
-      :filter(function(item)
-        return type(item) == "string" and vim.trim(item) ~= ""
-      end)
-      :totable()
-
-    if #flattened == 0 then
-      return "Unknown error occurred"
-    end
-
-    return table.concat(flattened, "\n")
+    vim.deprecate(
+      "vectorcode.integrations.codecompanion.common.flatten_table_to_string",
+      "vectorcode.utils.flatten_table_to_string",
+      "1.0.0",
+      "vectorcode",
+      true
+    )
+    return require("vectorcode.utils").flatten_table_to_string(t)
   end,
 
   ---@param use_lsp boolean

lua/vectorcode/integrations/codecompanion/files_ls_tool.lua

@@ -2,6 +2,7 @@
 
 local cc_common = require("vectorcode.integrations.codecompanion.common")
 local vc_config = require("vectorcode.config")
+local utils = require("vectorcode.utils")
 
 local default_opts = {
   use_lsp = vc_config.get_user_config().async_backend == "lsp",
@@ -43,7 +44,7 @@ return function(opts)
             cb({ status = "success", data = result })
           else
             if type(error) == "table" then
-              error = cc_common.flatten_table_to_string(error)
+              error = utils.flatten_table_to_string(error)
             end
             cb({
               status = "error",

lua/vectorcode/integrations/codecompanion/ls_tool.lua

@@ -2,6 +2,7 @@
 
 local cc_common = require("vectorcode.integrations.codecompanion.common")
 local vc_config = require("vectorcode.config")
+local utils = require("vectorcode.utils")
 local logger = vc_config.logger
 
 ---@type VectorCode.CodeCompanion.LsToolOpts
@@ -45,7 +46,7 @@ return function(opts)
             cb({ status = "success", data = result })
           else
             if type(error) == "table" then
-              error = cc_common.flatten_table_to_string(error)
+              error = utils.flatten_table_to_string(error)
             end
             cb({
               status = "error",

lua/vectorcode/integrations/codecompanion/prompts/init.lua

@@ -132,7 +132,7 @@ Here's my question:
                 vc_config.notify_opts
               )
             elseif err ~= nil then
-              err = cc_common.flatten_table_to_string(err)
+              err = utils.flatten_table_to_string(err)
               if err ~= "" then
                 vim.schedule_wrap(vim.notify)(
                   err,

lua/vectorcode/integrations/codecompanion/query_tool.lua

@@ -5,6 +5,7 @@ local cc_config = require("codecompanion.config").config
 local cc_schema = require("codecompanion.schema")
 local http_client = require("codecompanion.http")
 local vc_config = require("vectorcode.config")
+local utils = require("vectorcode.utils")
 local check_cli_wrap = vc_config.check_cli_wrap
 local logger = vc_config.logger
 
@@ -463,7 +464,7 @@ return check_cli_wrap(function(opts)
         )
 
         job_runner.run_async(args, function(result, error, code)
-          local err_string = cc_common.flatten_table_to_string(error)
+          local err_string = utils.flatten_table_to_string(error)
 
           if
             result ~= nil
@@ -603,7 +604,7 @@ DO NOT MODIFY UNLESS INSTRUCTED BY THE USER, OR A PREVIOUS QUERY RETURNED NO RES
             vim.inspect(stderr)
           )
         )
-        stderr = cc_common.flatten_table_to_string(stderr)
+        stderr = utils.flatten_table_to_string(stderr)
         if string.find(stderr, "InvalidCollectionException") then
           if cmd.project_root then
             tools.chat:add_tool_output(

lua/vectorcode/integrations/codecompanion/vectorise_tool.lua

@@ -2,6 +2,7 @@
 
 local cc_common = require("vectorcode.integrations.codecompanion.common")
 local vc_config = require("vectorcode.config")
+local utils = require("vectorcode.utils")
 local logger = vc_config.logger
 
 ---@alias VectoriseToolArgs { paths: string[], project_root: string? }
@@ -129,7 +130,7 @@ The value should be one of the following:
             vim.inspect(stderr)
           )
         )
-        stderr = cc_common.flatten_table_to_string(stderr)
+        stderr = utils.flatten_table_to_string(stderr)
         tools.chat:add_tool_output(
           self,
           string.format("**VectorCode `vectorise` Tool: %s", stderr)

lua/vectorcode/utils.lua

@@ -173,4 +173,30 @@ function M.is_directory(f)
   return stats and (stats.type == "directory") or false
 end
 
+---@param t table|string|nil
+---@return string
+M.flatten_table_to_string = function(t)
+  if t == nil then
+    return ""
+  end
+  if type(t) == "string" then
+    return t
+  end
+
+  -- Handle empty tables or tables with empty strings
+  local flattened = vim
+    .iter(t)
+    :flatten(math.huge)
+    :filter(function(item)
+      return type(item) == "string" and vim.trim(item) ~= ""
+    end)
+    :totable()
+
+  if #flattened == 0 then
+    return "Unknown error occurred"
+  end
+
+  return table.concat(flattened, "\n")
+end
+
 return M