{ lib, ... }:

{
  programs.nixvim = {
    keymaps =
      let
        normal =
          lib.mapAttrsToList
            (
              key: action: {
                mode = "n";
                inherit action key;
              }
            )
            {
              "<Space>" = "<NOP>";
              "<esc>" = ":noh<CR>";
              "Y" = "y$";
              "<C-c>" = ":b#<CR>";
              "<C-x>" = ":close<CR>";
              "<leader>s" = ":w<CR>";
              "<C-s>" = ":w<CR>";
              "<leader>h" = "<C-w>h";
              "<leader>l" = "<C-w>l";
              "L" = "$";
              "H" = "^";
              "<C-Up>" = ":resize -2<CR>";
              "<C-Down>" = ":resize +2<CR>";
              "<C-Left>" = ":vertical resize +2<CR>";
              "<C-Right>" = ":vertical resize -2<CR>";
              "<M-k>" = ":move-2<CR>";
              "<M-j>" = ":move+<CR>";
              # Metals (Scala) keybindings
              "<leader>ws" = "<cmd>lua require('metals').hover_worksheet()<CR>";
              "<leader>sm" = "<cmd>lua require('telescope').extensions.metals.commands()<CR>";
              "<leader>tt" = "<cmd>lua require('metals.tvp').toggle_tree_view()<CR>";
              "<leader>tr" = "<cmd>lua require('metals.tvp').reveal_in_tree()<CR>";
            };
        visual =
          lib.mapAttrsToList
            (
              key: action: {
                mode = "v";
                inherit action key;
              }
            )
            {
              # Removed gv to allow dot-repeat
              ">" = ">";
              "<" = "<";
              # TAB keeps gv for convenience
              "<TAB>" = ">gv";
              "<S-TAB>" = "<gv";
              "K" = ":m '<-2<CR>gv=gv";
              "J" = ":m '>+1<CR>gv=gv";
            };
      in
      normal ++ visual;
  };
}