-- LSP settings
vim.g.coq_settings = { auto_start = 'shut-up' }
local coq = require "coq"

-- Function to get root directory
local function get_root_dir(fname, patterns)
  local dir = vim.fs.dirname(fname)
  for _, pattern in ipairs(patterns) do
    local match = vim.fn.globpath(dir, pattern)
    if match ~= "" then
      return dir
    end
  end
  return nil
end

-- Function to start LSP
local function start_lsp(server_name, cmd, patterns)
  return function(args)
    local root_dir = get_root_dir(vim.api.nvim_buf_get_name(args.buf), patterns)
    if root_dir then
      vim.lsp.start(coq.lsp_ensure_capabilities({
        name = server_name,
        cmd = cmd,
        root_dir = root_dir,
      }))
    end
  end
end

-- Create autocommands for different file types
vim.api.nvim_create_autocmd('FileType', {
  pattern = 'python',
  callback = start_lsp('pyright', {'pyright-langserver', '--stdio'}, {'setup.py', 'pyproject.toml'})
})

vim.api.nvim_create_autocmd('FileType', {
  pattern = {'javascript', 'typescript'},
  callback = start_lsp('tsserver', {'typescript-language-server', '--stdio'}, {'package.json', 'tsconfig.json'})
})

vim.api.nvim_create_autocmd('FileType', {
  pattern = 'go',
  callback = start_lsp('gopls', {'gopls'}, {'go.mod'})
})

-- Enable signature completion
vim.api.nvim_create_autocmd("LspAttach", {
  callback = function(args)
    local bufnr = args.buf
    local client = vim.lsp.get_client_by_id(args.data.client_id)
    if vim.tbl_contains({ 'null-ls' }, client.name) then  -- blacklist lsp
      return
    end
    require("lsp_signature").on_attach({
      -- ... setup options here ...
    }, bufnr)
  end,
})

-- Function to format the current buffer on save
local function lsp_format_on_save(bufnr)
  vim.api.nvim_create_autocmd("BufWritePre", {
    buffer = bufnr,
    callback = function()
      vim.lsp.buf.format({ async = false })
    end,
  })
end

-- Create an autocmd for LspAttach event
vim.api.nvim_create_autocmd("LspAttach", {
  callback = function(args)
    local bufnr = args.buf
    -- Set up the buffer-local autocmd for formatting on save
    lsp_format_on_save(bufnr)
  end,
})