local cmp = require "cmp"
local cmp_lsp = require "cmp_nvim_lsp"
local cmp_kinds = require("assets.icons").icons.kinds

local function flatten_to_array(t)
	local res = {}
	local function _flatten(tbl)
		for _, v in ipairs(tbl) do
			if type(v) == "table" then
				_flatten(v)
			else
				table.insert(res, v)
			end
		end
	end
	_flatten(t)
	return res
end

local capabilities = vim.tbl_deep_extend(
	"force",
	{},
	vim.lsp.protocol.make_client_capabilities(),
	cmp_lsp.default_capabilities()
)

require("fidget").setup {}
require("mason").setup()
require("mason-lspconfig").setup {
	automatic_enable = true,
	ensure_installed = {
		"lua_ls",
		"rust_analyzer",
		"gopls",
	},
	handlers = {
		function(server_name) -- default handler (optional)
			require("lspconfig")[server_name].setup {
				capabilities = capabilities,
			}
		end,

		["tailwindcss"] = function()
			local lspconfig = require "lspconfig"
			lspconfig.tailwindcss.setup {
				capabilities = capabilities,
			}
		end,

		["css-lsp"] = function()
			local lspconfig = require "lspconfig"
			lspconfig.cssls.setup {
				capabilities = capabilities,
			}
		end,

		zls = function()
			local lspconfig = require "lspconfig"
			lspconfig.zls.setup {
				capabilities = capabilities,
				root_dir = lspconfig.util.root_pattern(
					".git",
					"build.zig",
					"zls.json"
				),
				settings = {
					zls = {
						enable_inlay_hints = true,
						enable_snippets = true,
						warn_style = true,
					},
				},
			}
			vim.g.zig_fmt_parse_errors = 0
			vim.g.zig_fmt_autosave = 0
		end,
		["lua_ls"] = function()
			local lspconfig = require "lspconfig"
			lspconfig.lua_ls.setup {
				capabilities = capabilities,
				settings = {
					Lua = {
						runtime = { version = "Lua 5.1" },
						diagnostics = {
							globals = {
								"bit",
								"vim",
								"it",
								"describe",
								"before_each",
								"after_each",
							},
						},
					},
				},
			}
		end,
	},
}

cmp.setup {
	preselect = "None",
	formatting = {
		fields = { "kind", "abbr" },
		format = function(entry, vim_item)
			vim_item.kind = cmp_kinds[vim_item.kind] or ""
			if entry.completion_item.detail then
				vim_item.menu = entry.completion_item.detail
			end
			return vim_item
		end,
	},
	completion = { completeopt = "menu,menuone" },
	snippet = {
		expand = function(args)
			require("luasnip").lsp_expand(args.body)
		end,
	},

	window = {
		completion = cmp.config.window.bordered(),
		documentation = cmp.config.window.bordered(),
	},

	view = {
		entries = "custom",
	},

	mapping = {
		["<C-p>"] = cmp.mapping.select_prev_item(),
		["<C-n>"] = cmp.mapping.select_next_item(),
		["<C-d>"] = cmp.mapping.scroll_docs(-4),
		["<C-f>"] = cmp.mapping.scroll_docs(4),
		["<C-Space>"] = cmp.mapping.complete(),
		["<C-e>"] = cmp.mapping.close(),
		["<CR>"] = cmp.mapping.confirm {
			behavior = cmp.ConfirmBehavior.Insert,
			select = true,
		},
		["<Tab>"] = cmp.mapping(function(fallback)
			if cmp.visible() then
				cmp.select_next_item()
			elseif require("luasnip").expand_or_jumpable() then
				require("luasnip").expand_or_jump()
			else
				fallback()
			end
		end, { "i", "s" }),
		["<S-Tab>"] = cmp.mapping(function(fallback)
			if cmp.visible() then
				cmp.select_prev_item()
			elseif require("luasnip").jumpable(-1) then
				require("luasnip").jump(-1)
			else
				fallback()
			end
		end, { "i", "s" }),
	},

	sources = cmp.config.sources {
		{ name = "path" },
		{ name = "nvim_lsp" },
		{ name = "luasnip" },
		{ name = "buffer" },
		{ name = "nvim_lua" },
	},
}

vim.diagnostic.config {
	virtual_text = false,
	virtual_lines = false,
	signs = true,
	underline = true,
	float = {
		focusable = false,
		style = "minimal",
		border = "rounded",
		source = true,
		header = "",
		prefix = "",
	},
}

local lspconfig = vim.lsp.config

lspconfig("texlab", {
	cmd = { "texlab" },
	filetypes = { "tex", "bib", "plaintex" },
	root_markers = {
		".git",
		".latexmkrc",
		"latexmkrc",
		".texlabroot",
		"texlabroot",
		"Tectonic.toml",
	},
	settings = {
		texlab = {
			rootDirectory = nil,
			build = {
				executable = "latexmk",
				args = {
					"-pdf",
					"-interaction=nonstopmode",
					"-synctex=1",
					"%f",
				},
				onSave = true,
				forwardSearchAfter = true,
			},
			forwardSearch = {
				executable = "zathura",
				args = {
					"--synctex-editor-command",
					[[ nvim-texlabconfig -file '%%%{input}' -line %%%{line} -server ]]
						.. vim.v.servername,
					"--synctex-forward",
					"%l:1:%f",
					"%p",
				},
			},
			chktex = {
				onEdit = false,
				onOpenAndSave = true,
			},
			diagnosticsDelay = 300,
			latexFormatter = "latexindent",
			latexindent = {
				["local"] = nil,
				modifyLineBreaks = false,
			},
			bibtexFormatter = "texlab",
			formatterLineLength = 80,
		},
	},
})

lspconfig("qmlls", {
	cmd = { "qmlls6" },
	filetypes = { "qml" },
	single_file_support = true,
	root_dir = function(bufnr, on_dir)
		local root = vim.fs.root(bufnr, ".git")
		on_dir(root)
	end,
})

lspconfig("jsonls", {
	settings = {
		json = {
			schemas = require("schemastore").json.schemas(),
			validate = { enable = true },
		},
	},
})

lspconfig("yamlls", {
	settings = {
		yaml = {
			schemaStore = {
				enable = false,
				url = "",
			},
			schemas = require("schemastore").yaml.schemas(),
		},
	},
})

local lspenable = vim.lsp.enable
local servers = {
	"html",
	"bashls",
	"pyright",
	"ts_ls",
	"texlab",
	"sourcekit",
	"qmlls",
	"tailwindcss",
	"systemd-lsp",
	require("mason-lspconfig").get_installed_servers(),
}

local flat_servers = flatten_to_array(servers)

for _, server in ipairs(flat_servers) do
	lspconfig(server, {
		on_attach = function(client, bufnr)
			if client.name == "typos_lsp" then
				return
			end
		end,
		capabilities = capabilities,
	})
	lspenable(server)
end