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 cmp = require("cmp")
local cmp_lsp = require("cmp_nvim_lsp")
local capabilities = vim.tbl_deep_extend(
	"force",
	{},
	vim.lsp.protocol.make_client_capabilities(),
	cmp_lsp.default_capabilities()
)

local cmp_kinds = {
	Text = '  ',
	Method = '  ',
	Function = '  ',
	Constructor = '  ',
	Field = '  ',
	Variable = '  ',
	Class = '  ',
	Interface = '  ',
	Module = '  ',
	Property = '  ',
	Unit = '  ',
	Value = '  ',
	Enum = '  ',
	Keyword = '  ',
	Snippet = '  ',
	Color = '  ',
	File = '  ',
	Reference = '  ',
	Folder = '  ',
	EnumMember = '  ',
	Constant = '  ',
	Struct = '  ',
	Event = '  ',
	Operator = '  ',
	TypeParameter = '  ',
}

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,
	},

	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({
	-- update_in_insert = true,
	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" },
})

local lspenable = vim.lsp.enable
local servers = {
	"html",
	"bashls",
	"pyright",
	"ts_ls",
	"texlab",
	"sourcekit",
	"qmlls",
	"tailwindcss",
	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)
			require("workspace-diagnostics").populate_workspace_diagnostics( client, bufnr )
		end,
	})
	lspenable(server)
end