--
-- Author  : Dennis Eriksen <d@ennis.no>
-- File    : rust.lua
-- Created : 2024-02-15
--

local M = {
  -- Add Rust & related to treesitter
  {
    "nvim-treesitter/nvim-treesitter",
    opts = function(_, opts)
      if type(opts.ensure_installed) == "table" then
        vim.list_extend(opts.ensure_installed, { "ron", "rust", "toml" })
      end
    end,
  },
}

-- return if rustc isn't available
if vim.fn.executable("cargo") ~= 1 then
  return M
end

  -- if rust-analyzer is available
if vim.fn.executable("cargo") == 1 then
  table.insert(M, {
    -- Extend auto completion
    {
      "hrsh7th/nvim-cmp",
      dependencies = {
        {
          "Saecki/crates.nvim",
          event = { "BufRead Cargo.toml" },
          opts = {
            completion = {
              cmp = { enabled = true },
            },
          },
        },
      },
      opts = function(_, opts)
        opts.sources = opts.sources or {}
        table.insert(opts.sources, { name = "crates" })
      end,
    },
  })
end

-- if rust-analyzer is available
if vim.fn.executable("rustfmt") == 1 then
  table.insert(M, {
    -- Formatting
    { -- https://github.com/stevearc/conform.nvim
      "stevearc/conform.nvim",
      opts = {
        formatters_by_ft = {
          rust = { "rustfmt" },
        },
      },
    },
  })
end

-- if rust-analyzer is available
if vim.fn.executable("rust-analyzer") == 1 then
  table.insert(M, {
    { -- https://github.com/neovim/nvim-lspconfig
      "neovim/nvim-lspconfig",
      opts = {
        servers = {
          -- Ensure mason installs the server
          rust_analyzer = {
            keys = {
              { "K", "<cmd>RustHoverActions<cr>", desc = "Hover Actions (Rust)" },
              { "<leader>cR", "<cmd>RustCodeAction<cr>", desc = "Code Action (Rust)" },
              { "<leader>dr", "<cmd>RustDebuggables<cr>", desc = "Run Debuggables (Rust)" },
            },
            settings = {
              ["rust-analyzer"] = {
                cargo = {
                  allFeatures = true,
                  loadOutDirsFromCheck = true,
                  runBuildScripts = true,
                },
                -- Add clippy lints for Rust.
                checkOnSave = {
                  allFeatures = true,
                  command = "clippy",
                  extraArgs = { "--no-deps" },
                },
                procMacro = {
                  enable = true,
                  ignored = {
                    ["async-trait"] = { "async_trait" },
                    ["napi-derive"] = { "napi" },
                    ["async-recursion"] = { "async_recursion" },
                  },
                },
              },
            },
          },
          taplo = {
            keys = {
              {
                "K",
                function()
                  if vim.fn.expand("%:t") == "Cargo.toml" and require("crates").popup_available() then
                    require("crates").show_popup()
                  else
                    vim.lsp.buf.hover()
                  end
                end,
                desc = "Show Crate Documentation",
              },
            },
          },
        },
      },
    },
    {
      "mrcjkb/rustaceanvim",
      version = "^4", -- Recommended
      ft = { "rust" },
    },
  })
end

return M