diff options
| -rw-r--r-- | lua/user/mods.lua | 120 |
1 files changed, 96 insertions, 24 deletions
diff --git a/lua/user/mods.lua b/lua/user/mods.lua index 63b3ac0..f9efd5a 100644 --- a/lua/user/mods.lua +++ b/lua/user/mods.lua @@ -612,37 +612,109 @@ end -------------------------------------------------- --- Function to create or toggle a scratch buffer -function M.Scratch() - local scratch_dir = vim.fn.expand('~/notes/private') - local scratch_date = os.date('%Y-%m-%d') - local scratch_file = 'scratch-' .. scratch_date .. '.md' - local scratch_buf = vim.fn.bufnr(scratch_file) - local bufinfo = vim.fn.getbufinfo(scratch_buf) - +---- Function to create or toggle a scratch buffer +local scratch_buf = nil -- Store the scratch buffer globally +local scratch_win = nil -- Store the scratch window globally +local scratch_date = os.date('%Y-%m-%d') +local scratch_dir = vim.fn.expand('~/notes/private') +local scratch_file = 'scratch-' .. scratch_date .. '.md' + +function M.Scratch(Split_direction) -- Check if the directory exists, and create it if it doesn't if vim.fn.isdirectory(scratch_dir) == 0 then vim.fn.mkdir(scratch_dir, 'p') end - if scratch_buf == -1 then - -- If the buffer doesn't exist, create it - vim.cmd('vsplit ' .. scratch_dir .. '/' .. scratch_file) - if vim.fn.empty(vim.fn.glob(scratch_dir .. '/' .. scratch_file)) == 1 then - vim.cmd(':normal i# Quick Notes - ' .. scratch_date) - vim.cmd(':normal o') - vim.cmd(':normal 28a-') - vim.cmd(':normal o') - vim.cmd(':w') - --vim.cmd(':startinsert') + + -- Determine the window type based on Split_direction + local current_window_type = 'float' + if Split_direction == 'float' then + current_window_type = 'float' + elseif Split_direction == 'vertical' then + current_window_type = 'vertical' + elseif Split_direction == 'horizontal' then + current_window_type = 'horizontal' + end + + local file_path = scratch_dir .. '/' .. scratch_file + + if scratch_win and vim.api.nvim_win_is_valid(scratch_win) then + -- Window exists, save buffer to file and close it + WriteScratchBufferToFile(scratch_buf, file_path) + vim.cmd(':w!') + vim.api.nvim_win_close(scratch_win, true) + scratch_win = nil + else + if scratch_buf and vim.api.nvim_buf_is_valid(scratch_buf) then + -- Buffer exists, reuse it + OpenScratchWindow(scratch_buf, current_window_type) + else + -- Buffer doesn't exist, create it and load the file if it exists + scratch_buf = OpenScratchBuffer(file_path) + OpenScratchWindow(scratch_buf, current_window_type) end - elseif vim.fn.empty(bufinfo[1].windows) == 1 then - -- If the buffer exists but is not open, open it - vim.cmd('vsplit +buffer' .. scratch_buf) + end +end + +function WriteScratchBufferToFile(buf, file_path) + local lines = vim.api.nvim_buf_get_lines(buf, 0, -1, false) + local content = table.concat(lines, '\n') + local escaped_file_path = vim.fn.fnameescape(file_path) + + -- Write the buffer content to the file + local file = io.open(escaped_file_path, 'w') + if file then + file:write(content) + file:close() + end +end + +function OpenScratchBuffer(file_path) + local buf = vim.api.nvim_create_buf(true, false) + + -- Set the file name for the buffer + local escaped_file_path = vim.fn.fnameescape(file_path) + vim.api.nvim_buf_set_name(buf, escaped_file_path) + + -- Check if the file exists and load it if it does + if vim.fn.filereadable(file_path) == 1 then + local file_contents = vim.fn.readfile(file_path) + vim.api.nvim_buf_set_lines(buf, 0, -1, true, file_contents) else - -- If the buffer exists and is open, close it - local scratch_win = vim.fn.filter(vim.fn.getwininfo(), 'v:val.bufnr == ' .. scratch_buf)[1].winnr + -- Insert initial content + vim.api.nvim_buf_set_lines(buf, 0, -1, true, { + '# Quick Notes - ' .. scratch_date, + '--------------------------', + '', + }) + + -- Save the initial content to the file vim.cmd(':w') - vim.cmd(scratch_win .. 'wincmd c') + end + + return buf +end + +function OpenScratchWindow(buf, current_window_type) + if current_window_type == 'float' then + local opts = { + relative = 'win', + width = 120, + height = 10, + border = 'single', + row = 20, + col = 20, + } + scratch_win = vim.api.nvim_open_win(buf, true, opts) + -- Go to the last line of the buffer + vim.api.nvim_win_set_cursor(0, { vim.api.nvim_buf_line_count(buf), 1 }) + elseif current_window_type == 'vertical' then + vim.cmd('vsplit') + vim.api.nvim_win_set_buf(0, buf) + scratch_win = 0 + elseif current_window_type == 'horizontal' then + vim.cmd('split') + vim.api.nvim_win_set_buf(0, buf) + scratch_win = 0 end end |