84 lines
2.3 KiB
Lua
84 lines
2.3 KiB
Lua
local M = {}
|
|
|
|
local sidebar_filetypes = {
|
|
['neo-tree'] = true,
|
|
['snacks_layout_box'] = true,
|
|
['snacks_terminal'] = true,
|
|
['toggleterm'] = true,
|
|
['opencode'] = true,
|
|
}
|
|
|
|
function M.focus_main_window()
|
|
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
local buf = vim.api.nvim_win_get_buf(win)
|
|
local ft = vim.bo[buf].filetype
|
|
local bt = vim.bo[buf].buftype
|
|
if not sidebar_filetypes[ft] and bt == '' then
|
|
vim.api.nvim_set_current_win(win)
|
|
return true
|
|
end
|
|
end
|
|
|
|
return false
|
|
end
|
|
|
|
local function find_window_by_filetype(filetype)
|
|
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
local buf = vim.api.nvim_win_get_buf(win)
|
|
if vim.bo[buf].filetype == filetype then return win end
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
local function ensure_neotree_window()
|
|
local neotree_win = find_window_by_filetype 'neo-tree'
|
|
if neotree_win then return neotree_win end
|
|
|
|
pcall(
|
|
function()
|
|
require('neo-tree.command').execute {
|
|
action = 'show',
|
|
source = 'filesystem',
|
|
position = 'left',
|
|
reveal = true,
|
|
}
|
|
end
|
|
)
|
|
|
|
return find_window_by_filetype 'neo-tree'
|
|
end
|
|
|
|
local function ensure_opencode_window()
|
|
local opencode_win = find_window_by_filetype 'opencode'
|
|
if opencode_win then return opencode_win end
|
|
|
|
pcall(function() require('opencode').toggle() end)
|
|
|
|
vim.wait(1000, function() return find_window_by_filetype 'opencode' ~= nil end, 50)
|
|
return find_window_by_filetype 'opencode'
|
|
end
|
|
|
|
function M.reset_ide_layout()
|
|
local neotree_win = ensure_neotree_window()
|
|
local opencode_win = ensure_opencode_window()
|
|
|
|
local total_columns = vim.o.columns
|
|
local left_width = math.max(20, math.floor(total_columns * 0.10))
|
|
local right_width = math.max(40, math.floor(total_columns * 0.25))
|
|
local min_main_width = 40
|
|
|
|
if left_width + right_width + min_main_width > total_columns then
|
|
local available_sidebar_width = math.max(total_columns - min_main_width, 0)
|
|
left_width = math.floor(available_sidebar_width * (10 / 35))
|
|
right_width = available_sidebar_width - left_width
|
|
end
|
|
|
|
if neotree_win and vim.api.nvim_win_is_valid(neotree_win) then vim.api.nvim_win_set_width(neotree_win, left_width) end
|
|
if opencode_win and vim.api.nvim_win_is_valid(opencode_win) then vim.api.nvim_win_set_width(opencode_win, right_width) end
|
|
|
|
M.focus_main_window()
|
|
end
|
|
|
|
return M
|