Initial commit

home/apps/nvim/lua/center/init.lua

@@ -0,0 +1,92 @@
+local M = {}
+
+M.cfg = {
+	skip_filetypes = {},
+	enabled = true,
+}
+
+M.setup = function(ctx)
+	if ctx == nil then
+		return
+	end
+
+	M.cfg.skip_filetypes = ctx.skip_filetypes or {}
+	if type(ctx.enabled) == "boolean" then
+		M.cfg.enabled = ctx.enabled
+	end
+end
+
+local function must_skip_file(skip_filetypes, current_type)
+	if skip_filetypes == nil then
+		return false
+	end
+
+	for _, value in ipairs(skip_filetypes) do
+		if value == current_type then
+			return true
+		end
+	end
+
+	return false
+end
+
+local function stay_centered(ctx)
+	if not ctx.cfg.enabled then
+		return
+	end
+	if must_skip_file(ctx.cfg.skip_filetypes, vim.bo.filetype) then
+		return
+	end
+
+	local line = vim.api.nvim_win_get_cursor(0)[1]
+	if line ~= vim.b.last_line then
+		vim.cmd("norm! zz")
+		-- vim.b.last_line = line
+		-- if ctx.mode == "insert" then
+		-- 	local column = vim.fn.getcurpos()[5]
+		-- 	vim.fn.cursor({ line, column })
+		-- end
+	end
+end
+
+local add_group = vim.api.nvim_create_augroup
+local group = add_group("StayCentered", { clear = true })
+
+local add_command = vim.api.nvim_create_autocmd
+add_command("CursorMovedI", {
+	group = group,
+	callback = function()
+		stay_centered({ mode = "insert", cfg = M.cfg })
+	end,
+})
+add_command("CursorMoved", {
+	group = group,
+	callback = function()
+		stay_centered({ mode = "other", cfg = M.cfg })
+	end,
+})
+add_command("BufEnter", {
+	group = group,
+	callback = function()
+		stay_centered({ mode = "other", cfg = M.cfg })
+	end,
+})
+
+M.enable = function()
+	M.cfg.enabled = true
+	stay_centered({ mode = "other", cfg = M.cfg })
+end
+
+M.disable = function()
+	M.cfg.enabled = false
+end
+
+M.toggle = function()
+	if M.cfg.enabled then
+		M.disable()
+	else
+		M.enable()
+	end
+end
+
+return M