diff options
Diffstat (limited to 'lua/startup')
| -rw-r--r-- | lua/startup/init.lua | 375 |
1 files changed, 375 insertions, 0 deletions
diff --git a/lua/startup/init.lua b/lua/startup/init.lua new file mode 100644 index 0000000..19f69bd --- /dev/null +++ b/lua/startup/init.lua @@ -0,0 +1,375 @@ +local startup = {} +local ns = vim.api.nvim_create_namespace "startup" +-- tables with tables: {line, align, cursor should move on, highlight} +startup.lines = {} +startup.formatted_text = {} +startup.sections = {} +startup.section_highlights = {} +startup.open_sections = {} +startup.good_lines = {} + +local section_alignments = {} + +local current_section = "" + +local opts = { noremap = true, silent = true } +local settings = require "startup.config" + +local utils = require "startup.utils" +local spaces = utils.spaces + +function startup.open_section() + vim.api.nvim_buf_set_option(0, "modifiable", true) + local line_nr = vim.api.nvim_win_get_cursor(0)[1] + local section_name = vim.trim(vim.api.nvim_get_current_line()) + local section_align = section_alignments[section_name] + local section_highlight = startup.section_highlights[section_name] + local section_entries = startup.sections[section_name] + if section_name == "" then + return + end + if section_entries == nil then + return + end + section_entries = require("startup").align(section_entries, section_align) + for i, section in ipairs(startup.open_sections) do + if section == section_name then + vim.api.nvim_buf_set_lines( + 0, + line_nr, + line_nr + #section_entries, + false, + {} + ) + table.remove(startup.open_sections, i) + return + end + end + vim.api.nvim_buf_set_lines(0, line_nr, line_nr, false, section_entries) + for number, _ in ipairs(section_entries) do + vim.api.nvim_buf_add_highlight( + 0, + ns, + section_highlight, + line_nr + number - 1, + 0, + -1 + ) + end + table.insert(startup.open_sections, section_name) + vim.cmd [[silent! %s/\s\+$//]] -- clear trailing whitespace + vim.api.nvim_win_set_cursor(0, { line_nr, math.floor(vim.o.columns / 2) }) + vim.api.nvim_buf_set_option(0, "modifiable", false) +end + +local function create_mappings(mappings) + vim.api.nvim_buf_set_keymap( + 0, + "n", + settings.mappings.execute_command, + ":lua require'startup'.check_line()<CR>", + opts + ) + vim.api.nvim_buf_set_keymap( + 0, + "n", + settings.mappings.open_file, + "<cmd>lua require('startup').open_file()<CR>", + opts + ) + vim.api.nvim_buf_set_keymap( + 0, + "n", + settings.mappings.open_section, + "<cmd>lua require'startup'.open_section()<CR>", + opts + ) + vim.api.nvim_buf_set_keymap( + 0, + "n", + settings.mappings.open_file_split, + "<cmd>lua require('startup').open_file_vsplit()<CR>", + opts + ) + vim.api.nvim_buf_set_keymap( + 0, + "n", + settings.mappings.open_help, + "<cmd>lua require'startup.utils'.key_help()<CR>", + opts + ) + if mappings ~= {} then + for _, cmd in pairs(mappings) do + vim.api.nvim_buf_set_keymap( + 0, + "n", + cmd[2], + "<cmd>" .. cmd[1] .. "<CR>", + opts + ) + end + end +end + +function startup.new_file() + local name = vim.fn.input "Filename: > " + vim.cmd("e " .. name) +end + +local sections_with_mappings = {} + +function startup.check_line() + local line = vim.api.nvim_get_current_line() + for _, section in ipairs(sections_with_mappings) do + for name, command in pairs(settings[section].content) do + if line:match(name) then + vim.cmd(command[1]) + end + end + end +end + +function startup.open_file() + local line = vim.api.nvim_get_current_line() + local filename = line + print(filename) + vim.cmd("e " .. filename) +end + +function startup.open_file_vsplit() + local line = vim.api.nvim_get_current_line() + local filename = line + print(filename) + vim.cmd("vsplit " .. filename) +end +function startup.align(dict, alignment) + local margin_calculated = 0 + if settings[current_section].margin < 1 then + margin_calculated = vim.o.columns * settings[current_section].margin + else + margin_calculated = settings[current_section].margin + end + local aligned = {} + local max_len = utils.longest_line(dict) + if alignment == "center" then + local space_left = vim.o.columns - max_len + for _, line in ipairs(dict) do + table.insert(aligned, spaces(space_left / 2) .. line) + end + elseif alignment == "left" then + for _, line in ipairs(dict) do + table.insert(aligned, spaces(margin_calculated) .. line) + end + elseif alignment == "right" then + for _, line in ipairs(dict) do + table.insert( + aligned, + spaces(vim.o.columns - max_len - margin_calculated - 10) .. line + ) + end + end + margin_calculated = 0 + return aligned +end + +local function empty(amount) + for _ = 1, amount, 1 do + table.insert(startup.lines, { " ", "center", false, "normal" }) + end +end + +function startup.mapping_names(mappings) + local mapnames = {} + local strings = {} + for title, command in pairs(mappings) do + if settings.options.mapping_keys then + table.insert(strings, title .. command[2]) + else + table.insert(strings, title) + end + end + local length = utils.longest_line(strings) + 18 + -- local length = vim.o.columns * 0.4 + for name, cmd in pairs(mappings) do + if settings.options.mapping_keys then + local space = utils.spaces(length - #cmd[2] - #name) + table.insert(mapnames, name .. space .. cmd[2]) + else + local space = utils.spaces(length - #name) + table.insert(mapnames, name .. space) + end + end + + return mapnames +end + +function startup.display() + if vim.g.startup_nvim_displayed then + return + end + vim.g.startup_nvim_displayed = true + -- vim.schedule(function() + local padding_nr = 1 + U.set_buf_options() + local parts = settings.parts + vim.cmd[[hi link StartupFoldedSection Special]] + for _, part in ipairs(parts) do + empty(settings.options.paddings[padding_nr]) + padding_nr = padding_nr + 1 + current_section = part + local options = settings[part] + U.validate_settings(options) + if type(options.content) == "function" then + options.content = options.content() + end + if options.highlight == "" then + vim.cmd( + "highlight Startup" + .. part + .. " guifg=" + .. options.default_color + .. " guibg=" + .. settings.colors.background + ) + options.highlight = "Startup" .. part + end + if options.type == "text" then + if options.fold_section then + section_alignments[vim.trim(options.title)] = options.align + startup.sections[vim.trim(options.title)] = options.content + startup.section_highlights[vim.trim(options.title)] = options.highlight + startup.good_lines[#startup.good_lines+1] = vim.trim(options.title) + table.insert( + startup.lines, + { options.title, options.align, true, "StartupFoldedSection" } + ) + else + for _, line in ipairs(options.content) do + table.insert( + startup.lines, + { line, options.align, false, options.highlight } + ) + end + end + elseif options.type == "mapping" then + if options.fold_section then + section_alignments[vim.trim(options.title)] = options.align + for _, line in ipairs(require"startup".mapping_names(options.content)) do + startup.good_lines[#startup.good_lines + 1] = vim.trim(line) + end + startup.sections[vim.trim(options.title)] = require("startup").mapping_names( + options.content + ) + startup.section_highlights[vim.trim(options.title)] = options.highlight + startup.good_lines[#startup.good_lines+1] = vim.trim(options.title) + table.insert( + startup.lines, + { options.title, options.align, true, "StartupFoldedSection" } + ) + for _, line in ipairs(options.content) do + startup.good_lines[#startup.good_lines+1] = vim.trim(line) + end + else + for _, line in ipairs(require("startup").mapping_names(options.content)) do + startup.good_lines[#startup.good_lines+1] = vim.trim(line) + table.insert( + startup.lines, + { line, options.align, true, options.highlight } + ) + if settings.options.empty_lines_between_mappings then + empty(1) + end + end + end + table.insert(sections_with_mappings, part) + create_mappings(options.content) + elseif options.type == "oldfiles" then + local old_files + if options.oldfiles_directory then + old_files = utils.get_oldfiles_directory(options.oldfiles_amount) + else + old_files = utils.get_oldfiles(options.oldfiles_amount) + end + if options.fold_section then + section_alignments[vim.trim(options.title)] = options.align + startup.sections[vim.trim(options.title)] = old_files + startup.section_highlights[vim.trim(options.title)] = options.highlight + startup.good_lines[#startup.good_lines+1] = vim.trim(options.title) + for _, line in ipairs(old_files) do + startup.good_lines[#startup.good_lines+1] = vim.trim(line) + end + table.insert( + startup.lines, + { options.title, options.align, true, "StartupFoldedSection" } + ) + else + for _, line in ipairs(old_files) do + startup.good_lines[#startup.good_lines+1] = vim.trim(line) + table.insert( + startup.lines, + { line, options.align, true, options.highlight } + ) + end + end + end + create_mappings {} + vim.cmd(options.command) + end + -- print("startup.lines:") + -- dump(startup.lines) + if settings.folded_section_color ~= "" then + vim.cmd([[highlight StartupFoldedSection guifg=]]..settings.colors.folded_section) + end + -- current_section = "" + for _, line in ipairs(startup.lines) do + table.insert( + startup.formatted_text, + require("startup").align({ line[1] }, line[2])[1] + ) + end + vim.api.nvim_buf_set_option(0, "modifiable", true) + vim.api.nvim_buf_set_lines(0, 0, -1, true, {}) + vim.api.nvim_buf_set_lines(0, 0, -1, false, startup.formatted_text) + vim.cmd [[silent! %s/\s\+$//]] -- clear trailing whitespace + for linenr, line in ipairs(startup.lines) do + vim.api.nvim_buf_add_highlight(0, ns, line[4], linenr - 1, 0, -1) + end + vim.api.nvim_buf_set_option(0, "modifiable", false) + vim.api.nvim_win_set_cursor(0, { 1, 1 }) + vim.api.nvim_win_set_cursor(0, { + #settings.header.content + settings.options.paddings[1] + 1, + math.floor(vim.o.columns / 2), + }) + -- end) + vim.cmd [[autocmd CursorMoved * lua require"startup.utils".reposition_cursor()]] +end + +function startup.setup(update) + if vim.g.startup_nvim_loaded then + return + end + vim.g.startup_nvim_loaded = true + settings = vim.tbl_deep_extend("force", settings, update or {}) + vim.cmd [[autocmd BufRead * lua if vim.fn.argc() == 0 then require("startup").display() end]] + vim.cmd[[autocmd VimResized * lua if vim.bo.ft == "startup" then require"startup".redraw() end]] +end + +function startup.redraw() + startup.formatted_text = {} + for _, line in ipairs(startup.lines) do + table.insert( + startup.formatted_text, + require("startup").align({ line[1] }, line[2])[1] + ) + end + vim.api.nvim_buf_set_option(0, "modifiable", true) + vim.api.nvim_buf_set_lines(0, 0, -1, true, {}) + vim.api.nvim_buf_set_lines(0, 0, -1, false, startup.formatted_text) + vim.cmd [[silent! %s/\s\+$//]] -- clear trailing whitespace + for linenr, line in ipairs(startup.lines) do + vim.api.nvim_buf_add_highlight(0, ns, line[4], linenr - 1, 0, -1) + end + vim.api.nvim_buf_set_option(0, "modifiable", false) +end + +return startup |