Skip to content

feat: add unified inline diff layout (ClaudeCodeBuffer) #300

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
spoloxs wants to merge 3 commits into
base: main
Choose a base branch
from
+412 −353
Open

feat: add unified inline diff layout (ClaudeCodeBuffer) #300

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
abc4832
feat: add unified inline diff layout (ClaudeCodeBuffer)
spoloxs Jul 6, 2026
c71ea4d
fix: address review issues in unified inline diff
spoloxs Jul 6, 2026
d08e7d0
refactor: simplify unified diff to always use persistent ClaudeCodeBu…
spoloxs Jul 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
179 changes: 142 additions & 37 deletions lua/claudecode/diff.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -232,6 +232,69 @@ local function find_claudecode_terminal_window()
return floating_fallback
end

---Get or create a persistent editor window in tab 2 for Claude Code diffs.
---Finds an existing non-terminal, non-diff window in tab 2 (the Claude editor).
---If none exists, creates a new window to the left of any terminal in tab 2.
local function get_or_create_editor_win_in_diff_tab()
local tabs = vim.api.nvim_list_tabpages()
if #tabs < 2 then
return nil
end
local tab2 = tabs[2]

-- First, look for an existing Claude Code editor window in tab 2:
-- a non-terminal, non-prompt, non-diff window showing a real file.
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(tab2)) do
local b = vim.api.nvim_win_get_buf(win)
local bt = vim.api.nvim_buf_get_option(b, "buftype")
local ft = vim.api.nvim_buf_get_option(b, "filetype")
if bt ~= "terminal" and bt ~= "prompt"
and not vim.api.nvim_win_get_option(win, "diff")
and ft ~= "neo-tree" and ft ~= "NvimTree"
and ft ~= "oil" and ft ~= "aerial"
then
return win
end
end

-- No editor window found; try to find any terminal in tab 2 to anchor a new split.
local terminal_win = nil
pcall(function()
local terminal_module = require("claudecode.terminal")
local terminal_bufnr = terminal_module.get_active_terminal_bufnr()
if terminal_bufnr then
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(tab2)) do
if vim.api.nvim_win_get_buf(win) == terminal_bufnr then
terminal_win = win
break
end
end
end
end)

-- If no terminal in tab 2, fall back to any non-floating, non-special window in tab 2.
if not terminal_win then
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(tab2)) do
local cfg = vim.api.nvim_win_get_config(win)
if not cfg.relative or cfg.relative == "" then
local b = vim.api.nvim_win_get_buf(win)
local bt = vim.api.nvim_buf_get_option(b, "buftype")
if bt ~= "terminal" and bt ~= "prompt" and not vim.api.nvim_win_get_option(win, "diff") then
return win
end
end
end
return nil
end

vim.api.nvim_set_current_tabpage(tab2)
vim.api.nvim_set_current_win(terminal_win)
vim.cmd("leftabove vsplit")
local new_editor = vim.api.nvim_get_current_win()
vim.cmd("wincmd =")
return new_editor
end

---Create a split based on configured layout
local function create_split()
if config and config.diff_opts and config.diff_opts.layout == "horizontal" then
Expand Down Expand Up @@ -333,20 +396,24 @@ end
local function display_terminal_in_new_tab()
local original_tab = vim.api.nvim_get_current_tabpage()

local function get_or_create_diff_tab()
vim.cmd("tabnew")
mark_tabnew_buffer_ephemeral()
return vim.api.nvim_get_current_tabpage()
end
Comment on lines +399 to +403

@spoloxs spoloxs Jul 6, 2026

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed. get_or_create_diff_tab now always does :tabnew instead of reusing tabs[2], so _cleanup_diff_state can never close an unrelated user tab when open_in_new_tab is set.


-- Get existing terminal buffer
local terminal_ok, terminal_module = pcall(require, "claudecode.terminal")
if not terminal_ok then
vim.cmd("tabnew")
mark_tabnew_buffer_ephemeral()
local new_tab = vim.api.nvim_get_current_tabpage()
local new_tab = get_or_create_diff_tab()
vim.api.nvim_set_current_tabpage(new_tab)
return original_tab, nil, false, new_tab
end

local terminal_bufnr = terminal_module.get_active_terminal_bufnr()
if not terminal_bufnr or not vim.api.nvim_buf_is_valid(terminal_bufnr) then
vim.cmd("tabnew")
mark_tabnew_buffer_ephemeral()
local new_tab = vim.api.nvim_get_current_tabpage()
local new_tab = get_or_create_diff_tab()
vim.api.nvim_set_current_tabpage(new_tab)
return original_tab, nil, false, new_tab
end

Expand All @@ -359,9 +426,8 @@ local function display_terminal_in_new_tab()
terminal_options = get_default_terminal_options()
end

vim.cmd("tabnew")
mark_tabnew_buffer_ephemeral()
local new_tab = vim.api.nvim_get_current_tabpage()
local new_tab = get_or_create_diff_tab()
vim.api.nvim_set_current_tabpage(new_tab)

local terminal_config = config.terminal or {}
local split_side = terminal_config.split_side or "right"
Expand All @@ -377,16 +443,27 @@ local function display_terminal_in_new_tab()
return original_tab, nil, had_terminal_in_original, new_tab
end

vim.cmd("vsplit")

local terminal_win = vim.api.nvim_get_current_win()
-- Reuse an existing terminal window in the new tab if already present, instead of splitting again.
local existing_terminal_in_new_tab
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(new_tab)) do
if vim.api.nvim_win_get_buf(win) == terminal_bufnr then
existing_terminal_in_new_tab = win
break
end
end

if split_side == "left" then
vim.cmd("wincmd H")
else
vim.cmd("wincmd L")
if not existing_terminal_in_new_tab then
vim.cmd("vsplit")
existing_terminal_in_new_tab = vim.api.nvim_get_current_win()
if split_side == "left" then
vim.cmd("wincmd H")
else
vim.cmd("wincmd L")
end
end

local terminal_win = existing_terminal_in_new_tab

vim.api.nvim_win_set_buf(terminal_win, terminal_bufnr)

apply_window_options(terminal_win, terminal_options)
Expand Down Expand Up @@ -1060,6 +1137,30 @@ end
-- Exposed for testing the reject-on-window-close (WinClosed) behavior.
M._register_diff_autocmds = register_diff_autocmds

---Format an error value (string or structured {message,data} table) into a flat,
---bounded string. Avoids `tostring(table)` blowups that spam :messages and the UI.
---Defined at module scope so both `_setup_blocking_diff` and `open_diff_blocking`
---can use it (the error path must never itself raise a secondary error).
---@param err any Error value (string or table with optional .message/.data)
---@return string
local function format_error(err)
if type(err) == "table" then
local msg = err.message or "unknown error"
local data = err.data
if data ~= nil and data ~= "" then
if type(data) == "table" then
data = "table"
end
return msg .. " (" .. tostring(data) .. ")"
end
return msg
end
return tostring(err)
end

-- Exposed for testing/debugging the error formatter.
M._format_error = format_error

---Create diff view from a specific window
---@param target_window NvimWin|nil The window to use as base for the diff
---@param old_file_path string Path to the original file
Expand Down Expand Up @@ -1308,6 +1409,7 @@ function M._setup_blocking_diff(params, resolution_callback)
local setup_success, setup_error = pcall(function()
local old_file_exists = vim.fn.filereadable(params.old_file_path) == 1
local is_new_file = not old_file_exists
local original_tab_number = vim.api.nvim_get_current_tabpage()

if old_file_exists then
local is_dirty = is_buffer_dirty(params.old_file_path)
Expand Down Expand Up @@ -1338,6 +1440,12 @@ function M._setup_blocking_diff(params, resolution_callback)
tab_number = state.created_new_tab and state.new_tab_number or nil,
})
end

if original_tab_number and vim.api.nvim_tabpage_is_valid(original_tab_number) then
vim.schedule(function()
vim.api.nvim_set_current_tabpage(original_tab_number)
end)
end
return
end

Expand Down Expand Up @@ -1517,16 +1625,18 @@ function M._setup_blocking_diff(params, resolution_callback)

-- Handle setup errors
if not setup_success then
local error_msg
if type(setup_error) == "table" and setup_error.message then
-- Handle structured error objects
error_msg = "Failed to setup diff operation: " .. setup_error.message
if setup_error.data then
error_msg = error_msg .. " (" .. setup_error.data .. ")"
end
local error_msg = "Failed to setup diff operation: " .. format_error(setup_error)

vim.schedule(function()
vim.notify("ClaudeCode diff setup failed: " .. format_error(setup_error), vim.log.levels.ERROR)
end)

-- Log the structured error before we wrap it, so the original cause is visible in :messages.
local structured = setup_error
if type(structured) == "table" and structured.message then
logger.error("diff", "Diff setup failed for", tab_name, "-", structured.message, structured.data or "")
else
-- Handle string errors or other types
error_msg = "Failed to setup diff operation: " .. tostring(setup_error)
logger.error("diff", "Diff setup failed for", tab_name, "-", tostring(structured))
end

-- Clean up any partial state that might have been created
Expand Down Expand Up @@ -1643,24 +1753,15 @@ function M.open_diff_blocking(old_file_path, new_file_path, new_file_contents, t
end)

if not success then
local error_msg
if type(err) == "table" and err.message then
error_msg = err.message
if err.data then
error_msg = error_msg .. " - " .. err.data
end
else
error_msg = tostring(err)
end
logger.error("diff", "Diff setup failed for", '"' .. tab_name .. '"', "error:", error_msg)
logger.error("diff", "Diff setup failed for", '"' .. tab_name .. '"', "error:", format_error(err))
-- If the error is already structured, propagate it directly
if type(err) == "table" and err.code then
error(err)
else
error({
code = -32000,
message = "Error setting up diff",
data = tostring(err),
data = format_error(err),
})
Comment on lines 1760 to 1765

@spoloxs spoloxs Jul 6, 2026

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same root cause as the previous one — fixed by hoisting format_error to module scope. This call now resolves correctly and error propagation works.

end
end
Expand Down Expand Up @@ -1813,6 +1914,7 @@ function M.accept_current_diff()
local tab_name = vim.b[current_buffer].claudecode_diff_tab_name
if tab_name then
M._resolve_diff_as_saved(tab_name, current_buffer)
M._cleanup_diff_state(tab_name, "user accepted")
else
vim.notify("No active diff found in current buffer", vim.log.levels.WARN)
end
Expand All @@ -1827,6 +1929,7 @@ function M.accept_current_diff()
end

M._resolve_diff_as_saved(tab_name, current_buffer)
M._cleanup_diff_state(tab_name, "user accepted")
end

---Deny/reject the current diff (user command version)
Expand All @@ -1839,6 +1942,7 @@ function M.deny_current_diff()
local tab_name = vim.b[current_buffer].claudecode_diff_tab_name
if tab_name then
M._resolve_diff_as_rejected(tab_name)
M._cleanup_diff_state(tab_name, "user denied")
else
vim.notify("No active diff found in current buffer", vim.log.levels.WARN)
end
Expand All @@ -1852,8 +1956,8 @@ function M.deny_current_diff()
return
end

-- Do not close windows/tabs here; just mark as rejected.
M._resolve_diff_as_rejected(tab_name)
M._cleanup_diff_state(tab_name, "user denied")
end

-- Expose internal utilities for use by diff_inline.lua
Expand All @@ -1863,6 +1967,7 @@ M._is_buffer_dirty = is_buffer_dirty
M._detect_filetype = detect_filetype
M._get_autocmd_group = get_autocmd_group
M._display_terminal_in_new_tab = display_terminal_in_new_tab
M._get_or_create_editor_win_in_diff_tab = get_or_create_editor_win_in_diff_tab

Comment on lines 1963 to 1971

@spoloxs spoloxs Jul 6, 2026

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed. Renamed the internal export to M._get_or_create_editor_win_in_diff_tab (underscore prefix, matching the other internal helpers in that block) and updated the caller in diff_inline.lua.

return M
---@alias NvimWin integer
Expand Down
Loading