dotfiles/common/nvim.nix

{
  pkgs,
  config,
  inputs,
  device,
  ...
}: {
  imports = [inputs.nixneovim.nixosModules.default];
  programs.nixneovim = {
    enable = true;
    extraPlugins = with pkgs.vimPlugins; [
      # neorg
      neorg
      neorg-telescope

      # Wut
      ChatGPT-nvim

      # UI and UX
      comfortable-motion
      vim-abolish
      telescope-nvim
      telescope-ui-select-nvim
      telescope-fzf-native-nvim
      telescope-file-browser-nvim
      telescope-dap-nvim
      octo-nvim

      # Treesitter stuff
      outline-nvim

      # lsp stuff
      nvim-cmp
      cmp-buffer
      cmp-path
      cmp-cmdline
      cmp-nvim-lsp
      cmp-nvim-lua
      cmp-dap
      cmp_luasnip
      cmp-tmux
      cmp-treesitter
      cmp-git
      luasnip
      fidget-nvim
      copilot-lua
      lsp-zero-nvim
      trouble-nvim
      crates-nvim
      sqls-nvim
      rustaceanvim

      # No more postman
      rest-nvim

      # UI
      noice-nvim
      nvim-web-devicons

      # Utils
      FTerm-nvim
      plenary-nvim
      nix-develop-nvim

      pkgs.tree-sitter-grammars.tree-sitter-just

      # Testing
      neotest
      # neotest-rust
    ];
    options = {
      shell = "sh";
      number = true;
      relativenumber = true;
      tabstop = 4;
      softtabstop = 4;
      shiftwidth = 4;
      expandtab = true;
      hidden = true;
      smartcase = true;
      termguicolors = true;
      signcolumn = "yes";
      wrap = true;
      completeopt = "menu,menuone,popup,noselect";
      undodir = "${config.xdg.cacheHome}/undodir";
      undofile = true;
    };

    globals = {
      mapleader = " ";
    };
    plugins = {
      lspconfig = {
        enable = true;
        servers = {
          gopls.enable = true;
          nil = {
            enable = true;
            extraConfig =
              # lua
              ''
                settings = {
                  ['nil'] = {
                    formatting = {
                      -- command = { "$\{pkgs.nixpkgs-fmt}/bin/nixpkgs-fmt" },
                      -- command = { "$\{pkgs.nixfmt}/bin/nixfmt" },
                      command = { "${pkgs.alejandra}/bin/alejandra" },
                    },
                    nix = {
                      flake = {
                          autoArchive = true,
                      },
                    },
                  },
                },
              '';
          };
          clangd.enable = true;
          lua-language-server.enable = true;
          jsonls.enable = true;
          html.enable = true;
          # pylyzer.enable = true;
          sqls = {
            enable = true;
            onAttachExtra =
              # lua
              ''
                require('sqls').on_attach(client, bufnr)
              '';
          };
          # rust-analyzer.enable = true;
        };
        extraLua.pre =
          # lua
          ''
            local lsp_zero = require'lsp-zero'
            local lspconfig = require 'lspconfig'
            lsp_zero.on_attach(function(client, bufnr)
              lsp_zero.default_keymaps({buffer = bufnr})
                if client.server_capabilities.inlayHintProvider then
                    vim.lsp.inlay_hint.enable(bufnr, true)
                end
            end)
          '';
        # extraLua.post = ''
        #   vim.lsp.inlay_hint.enable(bufnr, true)
        # '';
      };

      oil.enable = true;
      nvim-dap.enable = true;
      nvim-dap-ui.enable = true;
      todo-comments.enable = true;
      lualine.enable = true;
      commentary.enable = true;
      surround.enable = true;
      which-key.enable = true;
      ufo.enable = true;
      fugitive.enable = true;
      markdown-preview = {
        enable = true;
        autoStart = true;
      };
      treesitter-context.enable = true;
      ts-context-commentstring.enable = true;

      treesitter = {
        enable = true;
        indent = true;
        folding = true;
        refactor = {
          smartRename = {
            enable = true;
          };
        };
        grammars = with pkgs.tree-sitter-grammars; [
          tree-sitter-just
          tree-sitter-norg-meta
        ];
        installAllGrammars = true;
      };

      mini = {
        enable = true;
        ai.enable = true;
        # pairs.enable = true;
        # cursorword.enable = true;
        starter.enable = true;
      };
    };
    colorschemes = {
      catppuccin = {
        enable = true;
        flavour = "mocha";
      };
    };
    mappings = {
      normal = {
        "<leader>ff" = "require'telescope.builtin'.find_files";
        "<leader>c" = "[[<cmd>ChatGPT<cr>]]";
        "<leader>fb" = "require'telescope'.extensions.file_browser.file_browser";
        "<leader>gg" = "require'telescope.builtin'.live_grep";
        "<leader>;" = "require'telescope.builtin'.buffers";
        "<leader>o" = "[[<cmd>TroubleToggle<cr>]]";
        "<leader>ee" = "require'rest-nvim'.run";
        "<leader>ec" = "function() require'rest-nvim'.run(true) end";
        "<leader>\\\"" = ''[["+]]'';
        "vff" = "[[<cmd>vertical Gdiffsplit<cr>]]";
        "<C-k>" = "vim.lsp.buf.definition";
        "gi" = "require'telescope.builtin'.lsp_implementations";
        "gh" = "[[<cmd>Octo actions<cr>]]";
        "<leader>a" = "vim.lsp.buf.code_action";
        "F" = "function() vim.lsp.buf.format({ async = true }) end";
        "<leader><leader>" = "'<c-^>'";
        "<leader>q" = "[[<cmd>bw<cr>]]";
        "<leader>n" = "[[<cmd>bnext<cr>]]";
        "<leader>p" = "[[<cmd>bprev<cr>]]";
        "<C-w>\\\"" = "[[<cmd>split<cr>]]";
        "<C-w>%" = "[[<cmd>vsplit<cr>]]";

        "<leader>bb" = "require'dap'.toggle_breakpoint";
        "<leader>du" = "require'dapui'.toggle";
        "<leader>dr" = "[[<cmd>RustLsp debuggables<cr>]]";

        "<C-l>" = "[[<cmd>Outline<cr>]]";
        "<C-\\\\>" = "require('FTerm').toggle";
      };
      terminal = {
        "<C-\\\\>" = "require('FTerm').toggle";
      };
      insert = {
        "<C-\\\\>" = "require('FTerm').toggle";
      };
    };

    extraConfigLua = let
      codelldb =
        if device.isLinux
        then pkgs.vscode-extensions.vadimcn.vscode-lldb.adapter
        else null;
      liblldb =
        if device.isLinux
        then "${codelldb}/lldb/lib/liblldb.so"
        # else if device.isMac then
        #   "${codelldb}/lldb/lib/liblldb.dylib"
        else null;
    in
      /*
      lua
      */
      ''
        require'neotest'.setup({
          adapters = {
              -- require('neotest-rust') {
              --     args = { "--no-capture" },
              -- }
              require('rustaceanvim.neotest'),
          }
        })

        require('rest-nvim').setup()
        require('telescope').setup {
            defaults = {
                initial_mode = 'insert',
            },
            extensions = {
                fzf = {
                    fuzzy = true,                   -- false will only do exact matching
                    override_generic_sorter = true, -- override the generic sorter
                    override_file_sorter = true,    -- override the file sorter
                    case_mode = "smart_case",       -- or "ignore_case" or "respect_case"
                }
            }
        }

        require('telescope').load_extension("ui-select")
        require('telescope').load_extension("dap")
        require('telescope').load_extension("fzf")
        require('telescope').load_extension("file_browser")
        require('telescope').load_extension("rest")
        require('telescope').load_extension("neorg")

        vim.g.rustaceanvim = {
            tools = {
                enable_clippy = false,
            },
            server = {
                capabilities = require 'lsp-zero'.get_capabilities(),
                on_attach = function(client, bufnr)
                    if client.server_capabilities.inlayHintProvider then
                        vim.lsp.inlay_hint.enable(bufnr, true)
                    end
                end,
                settings = function(project_root)
                  local ra = require('rustaceanvim.config.server')
                    return ra.load_rust_analyzer_settings(project_root, {
                      settings_file_pattern = 'rust-analyzer.json'
                    })
                end
            },
            dap = {
                autoload_configurations = false,
                ${pkgs.lib.optionalString device.isLinux ''
          adapter = require'rustaceanvim.config'.get_codelldb_adapter("${codelldb}/bin/codelldb", "${liblldb}")
        ''}
            },
        }

        require("copilot").setup({
            suggestion = {
                enabled = true,
                auto_trigger = true,
                keymap = {
                    accept = "<C-l>",
                }
            },
            panel = { enabled = true },
        })

        require 'fidget'.setup()
        -- =======================================================================
        -- nvim-cmp
        -- =======================================================================
        local cmp = require("cmp")
        cmp.setup({
            view = {
                entries = { name = 'custom', selection_order = 'near_cursor' }
            },
            snippet = {
                expand = function(args)
                    require('luasnip').lsp_expand(args.body)
                end
            },
            window = {
                completion = cmp.config.window.bordered(),
                documentation = cmp.config.window.bordered(),
            },
            sources = cmp.config.sources({
                { name = "copilot", },
                { name = 'buffer' },
                { name = 'nvim_lsp' },
                { name = 'luasnip' },
                { name = 'treesitter' },
                { name = 'path' },
                { name = 'git' },
                { name = 'tmux' }
            }),
            mapping = cmp.mapping.preset.insert({
                ['<CR>'] = cmp.mapping.confirm(),
                ['<C-y>'] = cmp.mapping.complete(),
                -- ['<C-Space>'] = cmp.mapping.complete(),
                ['<C-n>'] = cmp.config.next,
                ['<C-p>'] = cmp.config.prev,
            })
        })

        cmp.setup.cmdline({ '/', '?' }, {
            mapping = cmp.mapping.preset.cmdline {
                -- ['<C-n>'] = cmp.config.disable,
                -- ['<C-p>'] = cmp.config.disable,
            },
            sources = {
                { name = 'buffer' }
            }
        })
        cmp.setup.cmdline(':', {
            mapping = cmp.mapping.preset.cmdline {
                -- ['<C-n>'] = cmp.config.disable,
                -- ['<C-p>'] = cmp.config.disable,
            },
            -- mapping = cmp.mapping.preset.cmdline(),
            sources = cmp.config.sources({
                { name = 'path' }
            }, {
                { name = 'cmdline' }
            })
        })
        cmp.setup.filetype('gitcommit', {
            sources = cmp.config.sources({
                { name = 'git' }, -- You can specify the `git` source if [you were installed it](https://github.com/petertriho/cmp-git).
            }, {
                { name = 'buffer' },
            })
        })
        require('crates').setup()
        require('outline').setup()
        require("noice").setup({
            lsp = {
                -- override markdown rendering so that **cmp** and other plugins use **Treesitter**
                override = {
                    ["vim.lsp.util.convert_input_to_markdown_lines"] = true,
                    ["vim.lsp.util.stylize_markdown"] = true,
                    ["cmp.entry.get_documentation"] = true, -- requires hrsh7th/nvim-cmp
                },
            },
            -- you can enable a preset for easier configuration
            presets = {
                bottom_search = false,         -- use a classic bottom cmdline for search
                command_palette = true,       -- position the cmdline and popupmenu together
                long_message_to_split = true, -- long messages will be sent to a split
                inc_rename = false,           -- enables an input dialog for inc-rename.nvim
                lsp_doc_border = true,        -- add a border to hover docs and signature help
            },
        })

        require 'FTerm'.setup({
            border     = 'double',
            dimensions = {
                height = 0.95,
                width = 0.95,
            },
            cmd        = "fish",
            blend      = 10,
        })

        require('neorg').setup({
            load = {
                ["core.defaults"] = {},
                ["core.completion"] = { config = { engine = "nvim-cmp", name = "[Norg]" } },
                ["core.concealer"] = {
                    config = { icon_preset = "diamond" }
                },
                ["core.keybinds"] = {
                   -- https://github.com/nvim-neorg/neorg/blob/main/lua/neorg/modules/core/keybinds/keybinds.lua
                   config = {
                     default_keybinds = true,
                     neorg_leader = "<leader>n",
                   },
                },
                ["core.dirman"] = {
                    config = {
                        workspaces = {
                            default = "~/Nextcloud/Notes",
                            Work = "~/Nextcloud/Work",
                        }
                    }
                }
            }
        })
        vim.api.nvim_create_autocmd({"BufEnter", "BufWinEnter"}, {
          pattern = {"*.norg"},
          command = "set conceallevel=3"
        })

        require('chatgpt').setup({
            api_key_cmd = "rbw get platform.openai.com",
        })

        require"octo".setup({
          use_local_fs = false,
          enable_builtin = false,
          default_remote = {"upstream", "origin"};
        })
      '';
    package = pkgs.neovim-nightly;
  };
}