local lspconfig = require 'lspconfig'
local treesitter = require 'nvim-treesitter.configs'
local treesitter_context = require 'treesitter-context'
local cmp = require 'cmp'
local luasnip = require 'luasnip'
local cmp_lsp = require 'cmp_nvim_lsp'
local rust_tools = require 'rust-tools'
local notify = require "notify"

local function autocmd(args)
	local event = args[1]
	local group = args[2]
	local callback = args[3]

	vim.api.nvim_create_autocmd(event, {
		group = group,
		buffer = args[4],
		callback = function()
			callback()
		end,
		once = args.once,
	})
end

local function on_attach(client, buffer)
	local augroup_highlight = vim.api.nvim_create_augroup("custom-lsp-references", { clear = true })
	local autocmd_clear = vim.api.nvim_clear_autocmds

	local opts = { buffer = buffer, remap = false }

	vim.bo[buffer].omnifunc = 'v:lua.vim.lsp.omnifunc'


	-- Enable completion triggered by <c-x><c-o>
	vim.bo[buffer].omnifunc = 'v:lua.vim.lsp.omnifunc'

	vim.keymap.set('n', '<leader>gD', vim.lsp.buf.declaration, opts)
	vim.keymap.set('n', '<leader>gd', vim.lsp.buf.definition, opts)
	vim.keymap.set('n', '<leader>K', vim.lsp.buf.hover, opts)
	vim.keymap.set('n', '<leader>gi', vim.lsp.buf.implementation, opts)
	vim.keymap.set('n', '<C-k>', vim.lsp.buf.signature_help, opts)
	vim.keymap.set('n', '<leader>wa', vim.lsp.buf.add_workspace_folder, opts)
	vim.keymap.set('n', '<leader>wr', vim.lsp.buf.remove_workspace_folder, opts)
	vim.keymap.set('n', '<leader>wl', function() print(vim.inspect(vim.lsp.buf.list_workspace_folders())) end, opts)
	vim.keymap.set('n', '<leader>D', vim.lsp.buf.type_definition, opts)
	vim.keymap.set('n', '<leader>rn', vim.lsp.buf.rename, opts)
	vim.keymap.set({ 'n', 'v' }, '<leader>ca', vim.lsp.buf.code_action, opts)
	vim.keymap.set('n', 'gr', vim.lsp.buf.references, opts)
	vim.keymap.set('n', '<leader>f', function()
		notify("Formatting!")
		vim.lsp.buf.format({ async = false })
	end, opts)

	-- vim.cmd [[autocmd BufWritePre * lua vim.lsp.buf.format({ async = false })]]

	if client.server_capabilities.documentHighlightProvider then
		autocmd_clear { group = augroup_highlight, buffer = buffer }
		autocmd { "CursorHold", augroup_highlight, vim.lsp.buf.document_highlight, buffer }
		autocmd { "CursorMoved", augroup_highlight, vim.lsp.buf.clear_references, buffer }
	end
end

local function init()
	rust_tools.setup {
		server = {
			settings = {
				['rust-analyzer'] = {
					cargo = {
						buildScripts = {
							enable = true,
						},
					},
					diagnostics = {
						enable = false,
					},
					files = {
						excludeDirs = { ".direnv", ".git" },
						watcherExclude = { ".direnv", ".git" },
					},
				},
			},
			on_attach = on_attach,
		},
	}



	local language_servers = {
		astro = {},
		clangd = {},
		cssls = {},
		dagger = {},
		diagnosticls = {
			filetypes = { "python" },
			init_options = {
				filetypes = {
					python = "black"
				},
				formatFiletypes = {
					python = { "black" }
				},
				formatters = {
					black = {
						command = "black",
						args = { "--quiet", "-" },
						rootPatterns = { "pyproject.toml" },
					},
				},
			}
		},
		elixirls = {
			cmd = { "elixir-ls" },
		},
		gopls = {
			settings = {
				gopls = {
					gofumpt = true,
				},
			},
		},
		html = {
			filetypes = { "html", "htmldjango" },
		},
		jsonls = {},
		lua_ls = {
			settings = {
				Lua = {
					diagnostics = {
						globals = { "vim" },
					},
					runtime = {
						version = 'LuaJIT',
					},
					telemetry = {
						enable = false,
					},
					workspace = {
						library = vim.api.nvim_get_runtime_file("", true),
					},
				}


			}

		},
		nil_ls = {
			settings = {
				['nil'] = {
					formatting = { command = { 'alejandra' } },
				}

			}
		},
		pyright = {},
		tsserver = {},
		tailwindcss = {
			filetypes = { "templ", "astro", "typescriptreact", "react", "html", "heex" },
			settings = {
				tailwindCSS = {
					includeLanguages = {
						templ = "html",
					},
				},
			},
		},
		yamlls = {
			settings = {
				yaml = {
					keyOrdering = false,
				},
			},
		},
		zls = {},
	}

	local capabilities = cmp_lsp.default_capabilities()

	-- Initialize servers
	for server, server_config in pairs(language_servers) do
		local config = { on_attach = on_attach, capabilities = capabilities }

		if server_config then
			for k, v in pairs(server_config) do
				config[k] = v
			end
		end

		lspconfig[server].setup(config)
	end

	vim.keymap.set('n', '<leader>e', vim.diagnostic.open_float)
	vim.keymap.set('n', '[d', vim.diagnostic.goto_prev)
	vim.keymap.set('n', ']d', vim.diagnostic.goto_next)
	vim.keymap.set('n', '<leader>q', vim.diagnostic.setloclist)


	treesitter.setup {
		auto_install = false,
		ignore_install = {},
		ensure_installed = {},
		highlight = { enable = true },
		indent = { enable = false },
		modules = {},
		rainbow = { enable = true },
		sync_install = false,
	}

	treesitter_context.setup()


	cmp.setup({
		snippet = {
			-- REQUIRED - you must specify a snippet engine
			expand = function(args)
				luasnip.lsp_expand(args.body) -- For `luasnip` users.
			end,
		},
		window = {
			completion = cmp.config.window.bordered(),
			documentation = cmp.config.window.bordered(),
		},
		mapping = cmp.mapping.preset.insert({
			['<C-b>'] = cmp.mapping.scroll_docs(-4),
			['<C-f>'] = cmp.mapping.scroll_docs(4),
			['<C-Space>'] = cmp.mapping.complete(),
			['<C-e>'] = cmp.mapping.abort(),
			['<CR>'] = cmp.mapping.confirm({ select = true }), -- Accept currently selected item. Set `select` to `false` to only confirm explicitly selected items.
		}),
		sources = cmp.config.sources({
			{ name = 'nvim_lsp' },
			{ name = 'luasnip' }, -- For luasnip users.
		}, {
			{ name = 'buffer' },
		})
	})

	vim.g.vimtex_view_method = "zathura"
end


return {
	init = init,
}