local null_ls = require("null-ls")

local function required_files(files)
  return function(params)
    local root = vim.fn.fnamemodify(params.bufname, ":h")

    local function check()
      for _, f in ipairs(files) do
        if not vim.loop.fs_stat(string.format('%s/%s', root, f)) then
          -- Not this directory; try the parent
          root = vim.fn.fnamemodify(root, ":h")
          if root == '/' then
            return false
          end
          return check()
        end
      end
      return true
    end

    return check()
  end
end

local jenkins_lint = {
  method = null_ls.methods.DIAGNOSTICS,
  filetypes = { "Jenkinsfile" },
  generator = null_ls.generator {
    command = "java",
    args = {'-jar', os.getenv('JENKINS_CLI'), 'declarative-linter'},
    to_stdin = true,
    check_exit_code = function(code)
      return code <= 1
    end,
    on_output = function(params, done)
      local diags = {}
      for _, line in ipairs(vim.split(params.output, "\n")) do
        local ok, _, msg, row, col = line:find('^WorkflowScript: %d+: (.+) @ line (%d+), column (%d+).')
        if ok then
          diags[#diags+1] = {
            row = row,
            col = col - 1,
            message = msg,
            severity = 1,
            source = "Jenkins",
          }
        end
      end

      return done(diags)
    end,
  }
}

local tcl_lint = {
  method = null_ls.methods.DIAGNOSTICS,
  filetypes = { "tcl" },
  generator = null_ls.generator {
    command = "make",
    args = {'tcl_lint'},
    ignore_stderr = true,
    check_exit_code = {0, 2},
    runtime_condition = required_files{'Makefile', 'nagelfar.syntax', 'nagelfar/nagelfar.tcl'},
    on_output = function(params, done)
      local diags = {}
      for _, line in ipairs(vim.split(params.output, "\n")) do
        -- tcl/fts_xprop.tcl: Line   7: W Unknown command "cfg_get"
        local ok, _, path, row, sev, msg = line:find('^([^:]+): Line%s+(%d+): (.) (.+)$')
        if ok and params.bufname == params.root ..'/'..path then
          diags[#diags+1] = {
            row = row,
            message = msg,
            severity = sev,
            source = "TCL",
          }
        end
      end

      return done(diags)
    end,
  }
}

null_ls.setup {
  sources = {
    null_ls.builtins.diagnostics.teal,
    null_ls.builtins.formatting.shfmt,
    null_ls.builtins.diagnostics.luacheck.with{
      -- This shouldn't be needed but is required
      extra_args = {
        '--config', '$XDG_CONFIG_HOME/luacheck/.luacheckrc'
      }
    },

    null_ls.builtins.diagnostics.shellcheck.with {
      extra_args = {
        '--shell', 'bash',
        '--exclude', table.concat({
          '1003', -- Want to escape a single quote? echo 'This is how it'\''s done'.
          '2016', -- Expressions don't expand in single quotes
        }, ',')
      }
    },

    null_ls.builtins.diagnostics.mypy,
    -- null_ls.builtins.diagnostics.pylint,
    null_ls.builtins.diagnostics.flake8.with{
      extra_args = function(params)
        -- params.root is set to the first parent dir with with either .git or
        -- Makefile
        if vim.loop.fs_stat(params.root..'/setup.cfg') then
          return {}
        end
        -- These ignores will override setup.cfg
        return { '--ignore', table.concat({
          'E501', -- line too long
          'E221', -- multiple space before operators
          'E201', -- whitespace before/after '['/']'
          'E202', -- whitespace before ']'
          'E272', -- multiple spaces before keyword
          'E241', -- multiple spaces after ':'
          'E231', -- missing whitespace after ':'
          'E203', -- whitespace before ':'
          'E741', -- ambiguous variable name
          'E226', -- missing whitespace around arithmetic operator
          'E305', 'E302', -- expected 2 blank lines after class
          'E251', -- unexpected spaces around keyword / parameter equals (E251)
        }, ',')}
      end
    },

    jenkins_lint,
    tcl_lint,
  },
  diagnostics_format = "#{s}: #{m} (#{c})",
  on_attach = function(_, bufnr)
    vim.api.nvim_buf_set_keymap(bufnr, 'n', '<leader>ca',
      '<cmd>lua vim.lsp.buf.code_action()<CR>', {noremap = true, silent = true})

    vim.api.nvim_buf_set_option(bufnr, 'omnifunc', 'v:lua.vim.lsp.omnifunc')
  end
}