local fn = vim.fn
local keymap = vim.keymap

local utils = require("utils")

local custom_attach = function(client, bufnr)
  -- Enable completion triggered by <c-x><c-o>
  vim.api.nvim_buf_set_option(bufnr, 'omnifunc', 'v:lua.vim.lsp.omnifunc')

	-- Mappings.
	local map = function(mode, l, r, opts)
		opts = opts or {}
		opts.silent = true
    opts.noremap = true
		opts.buffer = bufnr
		keymap.set(mode, l, r, opts)
	end
--map("n", "gd", "<Cmd>Lspsaga lsp_finder<CR>") -- Press "o" to open the reference location
--map("n", "gp", "<Cmd>Lspsaga peek_definition<CR>")
--	--map("n", "gd", vim.lsp.buf.definition, { desc = "go to definition" })
  map("n", "<C-]>", vim.lsp.buf.definition)
--	map("n", "K", vim.lsp.buf.hover)
--	map("n", "<C-k>", vim.lsp.buf.signature_help)
--	map("n", "<leader>rn", vim.lsp.buf.rename, { desc = "varialble rename" })
--	map("n", "gr", vim.lsp.buf.references, { desc = "show references" })
--	map("n", "[d", vim.diagnostic.goto_prev, { desc = "previous diagnostic" })
--	map("n", "]d", vim.diagnostic.goto_next, { desc = "next diagnostic" })
	map("n", "<leader>q", function()
		vim.diagnostic.setqflist({ open = true })
	end, { desc = "put diagnostic to qf" })
--	--map.('n', '<space>q', vim.diagnostic.setloclist)
--	map("n", "ga", vim.lsp.buf.code_action, { desc = "LSP code action" })
--	map("n", "<leader>wa", vim.lsp.buf.add_workspace_folder, { desc = "add workspace folder" })
--	map("n", "<leader>wr", vim.lsp.buf.remove_workspace_folder, { desc = "remove workspace folder" })
--	map("n", "<leader>wl", function()
--		print(vim.inspect(vim.lsp.buf.list_workspace_folders()))
--	end, { desc = "list workspace folder" })
--	map("n", "gs", "vim.lsp.buf.document_symbol()<cr>")
--	map("n", "gw", "vim.lsp.buf.workspace_symbol()<cr>", { desc = "list workspace folder" })
--	--map("n", "gs", ":lua vim.lsp.buf.document_symbol()<cr>")
--	map("n", "gt", ":lua vim.lsp.buf.type_definition()<cr>")
--	map("n", "gD", ":lua vim.lsp.buf.declaration()<cr>") -- most lsp servers don't implement textDocument/Declaration, so gD is useless for now.
--	map("n", "gi", ":lua vim.lsp.buf.implementation()<cr>")
	map("n", "go", ":lua vim.diagnostic.open_float()<cr>")
--	map("n", "gk", "<Cmd>Lspsaga diagnostic_jump_prev<CR>")
--	map("n", "gj", "<Cmd>Lspsaga diagnostic_jump_next<CR>")

vim.g.diagnostics_visible = true
function _G.toggle_diagnostics()
  if vim.g.diagnostics_visible then
    vim.g.diagnostics_visible = false
    vim.diagnostic.disable()
  else
    vim.g.diagnostics_visible = true
    vim.diagnostic.enable()
  end
end
map('n', '<Leader>m', ':call v:lua.toggle_diagnostics()<CR>')
--vim.g.diagnostics_active = true
--function _G.toggle_diagnostics()
--  if vim.g.diagnostics_active then
--    vim.g.diagnostics_active = false
--    vim.lsp.diagnostic.clear(0)
--      vim.cmd([[exe "normal ii\<Esc>x"]])
--    vim.lsp.handlers["textDocument/publishDiagnostics"] = function() end
--  else
--    vim.g.diagnostics_active = true
--    vim.lsp.handlers["textDocument/publishDiagnostics"] = vim.lsp.with(
--      vim.lsp.diagnostic.on_publish_diagnostics, {
--        virtual_text = true,
--        signs = true,
--        underline = true,
--        update_in_insert = false,
--      }
--    )
--  end
--end
--
--map("n", "<leader>i", ":call v:lua.toggle_diagnostics()<CR>")


	-- Set some key bindings conditional on server capabilities
	if client.server_capabilities.documentFormattingProvider then
		map("n", "<space>f", vim.lsp.buf.format, { desc = "format code" })
	end

	-- add rust specific keymappings
	if client.name == "rust_analyzer" then
		map("n", "<leader>rr", "<cmd>RustRunnables<CR>")
		map("n", "<leader>ra", "<cmd>RustHoverAction<CR>")
	end

--For diagnostics for specific cursor position
--vim.api.nvim_create_autocmd("CursorHold", {
--  buffer = bufnr,
--  callback = function()
--    local opts = {
--      focusable = false,
--      close_events = { "BufLeave", "CursorMoved", "InsertEnter", "FocusLost" },
--      border = 'rounded',
--      source = 'always',
--      prefix = ' ',
--      scope = 'cursor',
--    }
--    vim.diagnostic.open_float(nil, opts)
--  end
--})
	-- Diagnostic position
--	vim.api.nvim_create_autocmd("CursorHold", {
--		buffer = bufnr,
--		callback = function()
--			local float_opts = {
--				focusable = false,
--				close_events = { "BufLeave", "CursorMoved", "InsertEnter", "FocusLost" },
--				border = "rounded",
--				source = "always", -- show source in diagnostic popup window
--				prefix = " ",
--			}
--
--			if not vim.b.diagnostics_pos then
--				vim.b.diagnostics_pos = { nil, nil }
--			end
--
--			local cursor_pos = vim.api.nvim_win_get_cursor(0)
--			if
--				(cursor_pos[1] ~= vim.b.diagnostics_pos[1] or cursor_pos[2] ~= vim.b.diagnostics_pos[2])
--				and #vim.diagnostic.get() > 0
--			then
--				vim.diagnostic.open_float(nil, float_opts)
--			end
--
--			vim.b.diagnostics_pos = cursor_pos
--		end,
--	})

	-- The below command will highlight the current variable and its usages in the buffer.
	if client.server_capabilities.documentHighlightProvider then
		vim.cmd([[
      hi! link LspReferenceRead Visual
      hi! link LspReferenceText Visual
      hi! link LspReferenceWrite Visual
      augroup lsp_document_highlight
        autocmd! * <buffer>
        autocmd CursorHold <buffer> lua vim.lsp.buf.document_highlight()
        autocmd CursorMoved <buffer> lua vim.lsp.buf.clear_references()
      augroup END
    ]])
	end

	if vim.g.logging_level == "debug" then
		local msg = string.format("Language server %s started!", client.name)
		vim.notify(msg, vim.log.levels.DEBUG, { title = "Server?" })
	end
end

local capabilities = vim.lsp.protocol.make_client_capabilities()
capabilities = require("cmp_nvim_lsp").default_capabilities(capabilities)
capabilities.textDocument.completion.completionItem.snippetSupport = true
capabilities.offsetEncoding = { "utf-16" }

local lspconfig = require("lspconfig")

if utils.executable("pylsp") then
	lspconfig.pylsp.setup({
		settings = {
			pylsp = {
				plugins = {
					pylint = { enabled = true, executable = "pylint" },
					pyflakes = { enabled = false },
					pycodestyle = { enabled = false },
					jedi_completion = { fuzzy = true },
					pyls_isort = { enabled = true },
					pylsp_mypy = { enabled = true },
				},
			},
		},
		flags = {
			debounce_text_changes = 200,
		},
		capabilities = capabilities,
	})
else
	vim.notify("pylsp not found!", vim.log.levels.WARN, { title = "Server?" })
end

-- if utils.executable('pyright') then
--   lspconfig.pyright.setup{
--     on_attach = custom_attach,
--     capabilities = capabilities
--   }
-- else
--   vim.notify("pyright not found!", vim.log.levels.WARN, {title = 'Server?'})
-- end

if utils.executable("clangd") then
	lspconfig.clangd.setup({
		on_attach = custom_attach,
		capabilities = capabilities,
		filetypes = { "c", "cpp", "cc" },
		flags = {
			debounce_text_changes = 500,
		},
	})
else
	vim.notify("clangd not found!", vim.log.levels.WARN, { title = "Server?" })
end

-- set up vim-language-server
if utils.executable("vim-language-server") then
	lspconfig.vimls.setup({
		on_attach = custom_attach,
		flags = {
			debounce_text_changes = 500,
		},
		capabilities = capabilities,
	})
else
	vim.notify("vim-language-server not found!", vim.log.levels.WARN, { title = "Server?" })
end

-- set up bash-language-server
if utils.executable("bash-language-server") then
	lspconfig.bashls.setup({
		on_attach = custom_attach,
		capabilities = capabilities,
	})
end

if utils.executable("lua-language-server") then
	lspconfig.sumneko_lua.setup({
		on_attach = custom_attach,
		settings = {
			Lua = {
				runtime = {
					-- Tell the language server which version of Lua you're using (most likely LuaJIT in the case of Neovim)
					version = "LuaJIT",
				},
				diagnostics = {
					-- Get the language server to recognize the `vim` global
					globals = { "vim" },
				},
				workspace = {
					-- Make the server aware of Neovim runtime files,
					library = {
						fn.stdpath("data") .. "/site/pack/packer/opt/emmylua-nvim",
						fn.stdpath("config"),
					},
					maxPreload = 2000,
					preloadFileSize = 50000,
				},
			},
		},
		capabilities = capabilities,
	})
end


if utils.executable("rust-language-server") then
require("lspconfig").rust_analyzer.setup{
    cmd = { "rustup", "run", "nightly", "rust-analyzer" },
    on_attach = custom_attach,
  	flags = {
			debounce_text_changes = 500,
		},
	--[[
    settings = {
        rust = {
            unstable_features = true,
            build_on_save = false,
            all_features = true,
        },
    }
    --]]
}
end

--vim.diagnostic.config({
--  virtual_text = false,
--	underline = true,
--})
vim.diagnostic.config({
    underline = false,
    signs = true,
    virtual_text = false,
    float = {
        show_header = true,
        source = 'if_many',
        border = 'rounded',
        focusable = false,
    },
    update_in_insert = false, -- default to false
    severity_sort = false, -- default to false
})
-- Show line diagnostics automatically in hover window
vim.o.updatetime = 250
vim.cmd [[autocmd CursorHold,CursorHoldI * lua vim.diagnostic.open_float(nil, {focus=false})]]

--local diagnostics_active = true
--local toggle_diagnostics = function()
--  diagnostics_active = not diagnostics_active
--  if diagnostics_active then
--    vim.o.updatetime = 250
--    vim.cmd [[autocmd CursorHold,CursorHoldI * lua vim.diagnostic.open_float(nil, {focus=false})]]
--    --vim.diagnostic.open_float(nil, {focus=false})
--  else
--    --vim.diagnostic.hide()
--    vim.diagnostic.disable()
--  end
--end
--vim.keymap.set("n", "<leader>a", toggle_diagnostics)

-- Toogle diagnostics
local diagnostics_active = true
local toggle_diagnostics = function()
  diagnostics_active = not diagnostics_active
  if diagnostics_active then
    vim.api.nvim_echo({ { "Show diagnostics" } }, false, {})
    vim.diagnostic.enable()
  else
    vim.api.nvim_echo({ { "Disable diagnostics" } }, false, {})
    vim.diagnostic.disable()
  end
end
vim.keymap.set("n", "<leader>a", toggle_diagnostics)

--vim.cmd [[autocmd CursorHold * lua vim.diagnostic.open_float(0, {scope="cursor", close_events = {"CursorMoved", "CursorMovedI", "BufHidden", "InsertCharPre", "WinLeave"}})]]

--function LspDiagnosticsFocus()
--    vim.api.nvim_command('set eventignore=WinLeave')
--    vim.api.nvim_command('autocmd CursorMoved <buffer> ++once set eventignore=""')
--    vim.diagnostic.open_float(nil, {focusable = false, scope = 'line', close_events = {"CursorMoved", "CursorMovedI", "BufHidden", "InsertCharPre", "WinLeave"}})
--end
--vim.api.nvim_set_keymap('', '<Leader>a', '<Cmd>lua LspDiagnosticsFocus()<CR>', {silent = true})

--local diagnostics_active = true
--map('n', '<leader>a', function()
--  diagnostics_active = not diagnostics_active
--  if diagnostics_active then
--    vim.diagnostic.show()
--  else
--    vim.diagnostic.hide()
--  end
--end)

-- Global config for diagnostic
--vim.diagnostic.config({
--	underline = false,
--	virtual_text = false,
--	signs = true,
--	severity_sort = true,
--	float = {
--    focusable = true, --
--    style = "minimal", --
--		--border = "rounded",
--    border = "shadow",
--		source = "always",
--		header = "",
--		prefix = "",
--	},
--})

vim.lsp.handlers["textDocument/publishDiagnostics"] = vim.lsp.with(vim.lsp.diagnostic.on_publish_diagnostics, {
	underline = true,
	virtual_text = false,
	signs = true,
	update_in_insert = false,
})

--vim.lsp.buf.definition
--vim.lsp.handlers["textDocument/hover"] = vim.lsp.with(vim.lsp.handlers.hover, { border = "rounded" })

vim.lsp.handlers["textDocument/hover"] = vim.lsp.with(vim.lsp.handlers.hover, { border = "rounded" })

vim.lsp.handlers["textDocument/signatureHelp"] = vim.lsp.with(vim.lsp.handlers.signature_help, { border = "rounded" })

--local signs = { Error = " ", Warn = " ", Info = "􀅴 ", Hint = " " }
--local signs = { Error = "✘", Warn = "▲", Info = "🛈 ", Hint = "⚑" }
local signs = { Error = "✘", Warn = "▲", Info = "􀅳", Hint = "⚑" }
for type, icon in pairs(signs) do
	local hl = "DiagnosticSign" .. type
	vim.fn.sign_define(hl, { text = icon, texthl = hl, numhl = "" })
end


---- Location information about the last message printed. The format is
---- `(did print, buffer number, line number)`.
--local last_echo = { false, -1, -1 }
--
---- The timer used for displaying a diagnostic in the commandline.
--local echo_timer = nil
--
---- The timer after which to display a diagnostic in the commandline.
--local echo_timeout = 250
--
---- The highlight group to use for warning messages.
--local warning_hlgroup = 'WarningMsg'
--
---- The highlight group to use for error messages.
--local error_hlgroup = 'ErrorMsg'
--
---- If the first diagnostic line has fewer than this many characters, also add
---- the second line to it.
--local short_line_limit = 20
--
---- Shows the current line's diagnostics in a floating window.
--function show_line_diagnostics()
--  vim
--    .lsp
--    .diagnostic
--    .show_line_diagnostics({ severity_limit = 'Warning' }, vim.fn.bufnr(''))
--end
--
---- Prints the first diagnostic for the current line.
--function echo_diagnostic()
--  if echo_timer then
--    echo_timer:stop()
--  end
--
--  echo_timer = vim.defer_fn(
--    function()
--      local line = vim.fn.line('.') - 1
--      local bufnr = vim.api.nvim_win_get_buf(0)
--
--      if last_echo[1] and last_echo[2] == bufnr and last_echo[3] == line then
--        return
--      end
--
--      local diags = vim
--        .lsp
--        .diagnostic
--        .get_line_diagnostics(bufnr, line, { severity_limit = 'Warning' })
--
--      if #diags == 0 then
--        -- If we previously echo'd a message, clear it out by echoing an empty
--        -- message.
--        if last_echo[1] then
--          last_echo = { false, -1, -1 }
--
--          vim.api.nvim_command('echo ""')
--        end
--
--        return
--      end
--
--      last_echo = { true, bufnr, line }
--
--      local diag = diags[1]
--      local width = vim.api.nvim_get_option('columns') - 15
--      local lines = vim.split(diag.message, "\n")
--      local message = lines[1]
--      local trimmed = false
--
--      if #lines > 1 and #message <= short_line_limit then
--        message = message .. ' ' .. lines[2]
--      end
--
--      if width > 0 and #message >= width then
--        message = message:sub(1, width) .. '...'
--      end
--
--      local kind = 'warning'
--      local hlgroup = warning_hlgroup
--
--      if diag.severity == vim.lsp.protocol.DiagnosticSeverity.Error then
--        kind = 'error'
--        hlgroup = error_hlgroup
--      end
--
--      local chunks = {
--        { kind .. ': ', hlgroup },
--        { message }
--      }
--
--      vim.api.nvim_echo(chunks, false, {})
--    end,
--    echo_timeout
--  )
--end
--vim.cmd([[
--  autocmd CursorMoved * :lua echo_diagnostic()
--]])
-- Highlight line number instead of having icons in sign column

-- See the properties of the signs with sign list.

--vim.cmd [[
--  highlight! DiagnosticLineNrError guibg=#51202A guifg=#FF0000 gui=bold
--  highlight! DiagnosticLineNrWarn guibg=#51412A guifg=#FFA500 gui=bold
--  highlight! DiagnosticLineNrInfo guibg=#1E535D guifg=#00FFFF gui=bold
--  highlight! DiagnosticLineNrHint guibg=#1E205D guifg=#0000FF gui=bold
--
--  sign define DiagnosticSignError text= texthl=DiagnosticSignError linehl= numhl=DiagnosticLineNrError
--  sign define DiagnosticSignWarn text= texthl=DiagnosticSignWarn linehl= numhl=DiagnosticLineNrWarn
--  sign define DiagnosticSignInfo text= texthl=DiagnosticSignInfo linehl= numhl=DiagnosticLineNrInfo
--  sign define DiagnosticSignHint text= texthl=DiagnosticSignHint linehl= numhl=DiagnosticLineNrHint
--]]

-- Highlight symbol under cursor

-- Add the following to your on_attach (this allows checking server capabilities to avoid calling invalid commands.

--if client.resolved_capabilities.document_highlight then
--  vim.cmd [[
--    hi! LspReferenceRead cterm=bold ctermbg=red guibg=LightYellow
--    hi! LspReferenceText cterm=bold ctermbg=red guibg=LightYellow
--    hi! LspReferenceWrite cterm=bold ctermbg=red guibg=LightYellow
--  ]]
--  vim.api.nvim_create_augroup('lsp_document_highlight', {
--    clear = false
--  })
--  vim.api.nvim_clear_autocmds({
--    buffer = bufnr,
--    group = 'lsp_document_highlight',
--  })
--  vim.api.nvim_create_autocmd({ 'CursorHold', 'CursorHoldI' }, {
--    group = 'lsp_document_highlight',
--    buffer = bufnr,
--    callback = vim.lsp.buf.document_highlight,
--  })
--  vim.api.nvim_create_autocmd('CursorMoved', {
--    group = 'lsp_document_highlight',
--    buffer = bufnr,
--    callback = vim.lsp.buf.clear_references,
--  })
--end