refactor(backend): replace remove_header_lines with process_diff

Author: rachartier

lua/tiny-code-action/backend/delta.lua

@@ -32,7 +32,7 @@ function M.get_diff(bufnr, old_lines, new_lines, opts)
   }):sync()
 
   DeltaBackend:cleanup_files(old_file, new_file)
-  return DeltaBackend:remove_header_lines(diff, opts.backend_opts.delta.header_lines_to_remove)
+  return DeltaBackend:process_diff(diff, opts.backend_opts.delta.header_lines_to_remove)
 end
 
 function M.is_diff_content(lines)

lua/tiny-code-action/backend/diffsofancy.lua

@@ -23,7 +23,7 @@ function M.get_diff(bufnr, old_lines, new_lines, opts)
   }):sync()
 
   DiffSoFancyBackend:cleanup_files(old_file, new_file)
-  return DiffSoFancyBackend:remove_header_lines(
+  return DiffSoFancyBackend:process_diff(
     fancy_output,
     opts.backend_opts.diffsofancy.header_lines_to_remove
   )

lua/tiny-code-action/backend/difftastic.lua

@@ -28,7 +28,7 @@ function M.get_diff(bufnr, old_lines, new_lines, opts)
   }):sync()
 
   DifftasticBackend:cleanup_files(old_file, new_file)
-  return DifftasticBackend:remove_header_lines(
+  return DifftasticBackend:process_diff(
     diff,
     opts.backend_opts.difftastic.header_lines_to_remove
   )

lua/tiny-code-action/backend/vim.lua

@@ -1,10 +1,15 @@
 local M = {}
 
 function M.get_diff(bufnr, old_lines, new_lines, opts)
-  return vim.diff(table.concat(old_lines, "\n"), table.concat(new_lines, "\n"), {
+  local diff = vim.diff(table.concat(old_lines, "\n"), table.concat(new_lines, "\n"), {
     algorithm = "minimal",
     ctxlen = 3,
   })
+  if type(diff) == "string" then
+    diff = vim.split(diff, "\n")
+  end
+  local BaseBackendFiles = require("tiny-code-action.base.backend").BaseBackendFiles
+  return BaseBackendFiles:process_diff(diff, 0)
 end
 
 function M.is_diff_content(lines)

lua/tiny-code-action/base/backend.lua

@@ -39,6 +39,14 @@ function BaseBackendFiles:remove_header_lines(diff_output, header_lines_to_remov
   return diff_output
 end
 
+--- Removes all occurrences of "$0" (cursor placeholder) from each line of the diff output.
+---@param diff_output table: Table of diff lines
+---@return table: Table of diff lines with "$0" removed
+function BaseBackendFiles:process_diff(diff_output, header_lines_to_remove)
+  local cleaned = self:remove_header_lines(diff_output, header_lines_to_remove)
+  return cleaned
+end
+
 function BaseBackendFiles:is_diff_content(lines)
   return require("tiny-code-action.terminal").is_diff_content(lines)
 end