feat(startup): ✨moved main file into startup/

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