{ pkgs, ... }:
# let
  # customPlugins = {
  #   vim-zettel = pkgs.vimUtils.buildVimPlugin {
  #     name = "vim-zettel";
  #     src = pkgs.fetchFromGitHub {
  #       owner = "michal-h21";
  #       repo = "vim-zettel";
  #       rev = "929d90eec62e6f693c2702d2b6f76a93f2f1689d";
  #       sha256 = "1fzKmknfVlEYwqeXgbKITbb2/PJ023iJyMz6vak3qh4=";
  #     };
  #   };
  #  };

  # Thank god I don't use python anymore
  # my-python-packages = python-packages: with python-packages; [
  #   # other python packages you want
  # ];
  # python-with-my-packages = pkgs.python3.withPackages my-python-packages;
# in

{
  home.sessionVariables.EDITOR = "nvim";
  home.shellAliases = {
    k = "cd ~/kitaab/vimwiki; vim -c :VimwikiIndex; cd $OLDPWD";
    # kitaab zettel
    kz = "cd ~/kitaab/vimwiki; vim -c :VimwikiIndex -c :ZettelNew; cd $OLDPWD";
    # kitaab get by tagged
    kt = "cd /home/anish/kitaab/vimwiki; eza *_index.wiki | fzf --preview 'bat --style=numbers --color=always --terminal-width -1 ./{}';cd $OLDPWD";
  };
  # Add %update to the highlight clause for vimwiki files
  home.file.".config/nvim/after/syntax/vimwiki.vim".text = ''
    " match with %update
    syntax match VimwikiPlaceholder /^\s*%update\ze\%(\s.*\)\?$/ nextgroup=VimwikiPlaceholderParam skipwhite
    " match with %link
    syntax match VimwikiPlaceholder /^\s*%link\ze\%(\s.*\)\?$/ nextgroup=VimwikiPlaceholderParam skipwhite
  '';

  #environment.systemPackages = with customPlugins; [ tidal ];
  programs.neovim = {
    enable = true;

    extraConfig = ''
      filetype plugin on

      " set spell
      set spelllang=en_gb
      set ignorecase
      set smartcase
      set incsearch
      " if hidden is not set, TextEdit might fail.
      set hidden
      " Some servers have issues with backup files, see #649
      set nobackup
      set nowritebackup
      " Better display for messages
      set cmdheight=2
      " You will have bad experience for diagnostic messages when it's default 4000.
      set updatetime=300
      " don't give |ins-completion-menu| messages.
      set shortmess+=c
      " always show signcolumns
      set signcolumn=yes
      " highlight row
      set cursorline

      let g:conjure#filetype#fennel = "conjure.client.fennel.stdio"
      " we only use fennel with love; no we don't...
      let g:conjure#client#fennel#stdio#command = "fennel" 
      " let g:conjure#extract#tree_sitter#enabled = true
      map <localleader>lt :ConjureLogToggle<CR>

      let g:rainbow_active = 0
      autocmd BufReadPost,BufNewFile c,clj,cljs RainbowParenthesesToggleAll

      " general commands
      nnoremap <BS> :noh<CR>
      map Q <Nop>
      nnoremap Y y$
      nnoremap <leader>n :vnew<CR>
      " center moving
      nnoremap n nzzzv
      nnoremap N Nzzzv
      nnoremap J mzJ'z
      " undo points
      inoremap , ,<c-g>u
      inoremap . .<c-g>u
      inoremap ? ?<c-g>u
      inoremap ! !<c-g>u
      inoremap ) )<c-g>u
      inoremap [ [<c-g>u
      inoremap ] ]<c-g>u
      inoremap { {<c-g>u
      inoremap ( (<c-g>u
      inoremap } }<c-g>u
      " moving text
      vnoremap J :m '>+1<CR>gv=gv
      vnoremap K :m '>-2<CR>gv=gv
      inoremap <c-j> <esc>:m .+1<CR>==
      inoremap <c-k> <esc>:m .-2<CR>==
      nnoremap <leader>j <esc>:m .+1<CR>==
      nnoremap <leader>k <esc>:m .-2<CR>==
      " Pasting 
      :map <leader>sy :!xclip -i<CR><CR>
      :vmap <leader>sy "*y
      :map <leader>sp :r!xclip -o<CR>

      " barbar
      nnoremap <silent> gT :BufferPrevious<CR>
      nnoremap <silent> gt :BufferNext<CR>
      nnoremap <silent> gj :BufferPick<CR>
      nnoremap <silent> gq :BufferClose<CR>

      " kitaab
      map <leader>cz :VimwikiIndex<CR>:ZettelNew
      map <leader>zs :ZettelSearch<CR>
      map <leader>zn :ZettelNew<CR>
      map <leader>zo :ZettelOpen<CR>
      map <leader>zf :Telescope oldfiles only_cwd=true<CR>
      map <leader>wt :VimwikiMakeTomorrowDiaryNote
      map <leader>wd :VimwikiMakeYesterdayDiaryNote
      map <leader>zm :ZenMode<CR>:setlocal nospell<CR>
      let g:vimwiki_list = [{ 'path': '~/kitaab/vimwiki', 'auto_tags': 1, 'auto_diary_index': 1, 'auto_generate_links': 0, 'nested_syntaxes': {'python': 'python', 'c++': 'cpp', 'nix': 'nix', 'bash': 'sh'} }]
      let g:sync_taskwarrior = 0
      let g:zettel_format = "%y%m%d-%H%M"


      let g:scnvim_snippet_format = "luasnip"


      " lua config 
      lua <<EOF
      vim.g.diagnostics_active = true
      function _G.toggle_diagnostics()
        if vim.g.diagnostics_active then
          vim.g.diagnostics_active = false
          vim.lsp.diagnostic.clear(0)
          vim.lsp.handlers["textDocument/publishDiagnostics"] = function() end
        else
          vim.g.diagnostics_active = true
          vim.lsp.handlers["textDocument/publishDiagnostics"] = vim.lsp.with(
            vim.lsp.diagnostic.on_publish_diagnostics, {
              virtual_text = true,
              signs = true,
              underline = true,
              update_in_insert = false,
            }
          )
        end
      end

      vim.diagnostic.config({
        signs = true,
        update_in_insert = false,
        underline = false,
        severity_sort = true,
        float = {
          focusable = false,
          style = 'minimal',
          border = 'rounded',
          source = 'always',
        },
      })

      vim.lsp.handlers['textDocument/hover'] = vim.lsp.with(
        vim.lsp.handlers.hover,
        {
          border = 'rounded',
        }
      )

      vim.lsp.handlers['textDocument/signatureHelp'] = vim.lsp.with(
        vim.lsp.handlers.signature_help,
        {
          border = 'rounded',
        }
      )

      local orig_util_open_floating_preview = vim.lsp.util.open_floating_preview
      function vim.lsp.util.open_floating_preview(contents, syntax, opts, ...)
        opts = opts or {}
        opts.border = opts.border or 'rounded'
        return orig_util_open_floating_preview(contents, syntax, opts, ...)
      end


      vim.api.nvim_set_keymap('n', '<leader>vt', ':call v:lua.toggle_diagnostics()<CR>',  {noremap = true, silent = true})

      -- autopairs
      require('nvim-autopairs').setup{}

      -- marks
      require('marks').setup{}

      -- Setup nvim-cmp.
      local has_words_before = function()
        local line, col = unpack(vim.api.nvim_win_get_cursor(0))
        return col ~= 0 and vim.api.nvim_buf_get_lines(0, line - 1, line, true)[1]:sub(col, col):match("%s") == nil
      end

      local luasnip = require("luasnip")
      local lspkind = require("lspkind")
      local cmp = require('cmp')
      local cmp_autopairs = require('nvim-autopairs.completion.cmp')
      local t = function(str)
          return vim.api.nvim_replace_termcodes(str, true, true, true)
      end
      local check_back_space = function()
          local col = vim.fn.col(".") - 1
          return col == 0 or vim.fn.getline("."):sub(col, col):match("%s") ~= nil
      end
      cmp.setup {
          formatting = {
              format = lspkind.cmp_format({
                mode = 'symbol_text', 
                maxwidth = 50, 
            })
          },
          mapping = {
              ['<C-p>'] = cmp.mapping.select_prev_item(),
              ['<C-n>'] = cmp.mapping.select_next_item(),
              ['<C-d>'] = cmp.mapping.scroll_docs(-4),
              ['<C-f>'] = cmp.mapping.scroll_docs(4),
              ['<C-Space>'] = cmp.mapping.complete(),
              ['<C-e>'] = cmp.mapping.close(),
              ['<CR>'] = cmp.mapping.confirm({
                  behavior = cmp.ConfirmBehavior.Insert,
                  select = true
              }),
              ["<Tab>"] = cmp.mapping(function(fallback)  
                if cmp.visible() then  
                    cmp.select_next_item()  
                elseif luasnip.expand_or_jumpable() then
                  luasnip.expand_or_jump()
                elseif has_words_before() then
                  cmp.complete()
                else
                    fallback()  
                end
              end, {"i", "s"}),
              ["<S-Tab>"] = cmp.mapping(function(fallback)  
                if cmp.visible() then  
                    cmp.select_prev_item()  
                elseif luasnip.jumpable(-1) then
                  luasnip.jump(-1)
                else  
                    fallback()  
                end  
              end, { "i", "s" }),
          },
          snippet = {
            expand = function(args)
              require'luasnip'.lsp_expand(args.body)
            end
          },
          sources = {
              {name = 'buffer'}, {name = 'nvim_lsp'}, 
              {name = "look"}, {name = "path"},
              {name = "calc"}, {name = "spell"},
              {name = 'cmp-conjure'}, {name = "luasnip"}, 
              {name = 'tags'}
          },
          completion = {completeopt = 'menu,menuone,noinsert'}
      }
      cmp.event:on( 'confirm_done', cmp_autopairs.on_confirm_done({  map_char = { tex = ' ' } }))
      require("luasnip.loaders.from_vscode").lazy_load()

      -- null-ls
      -- being very janky
      local null_ls = require("null-ls")
      null_ls.setup({
        sources = {
          null_ls.builtins.diagnostics.shellcheck,
          null_ls.builtins.diagnostics.proselint.with({ filetypes = { "vimwiki", "mail", }, }),
          null_ls.builtins.diagnostics.commitlint,
          null_ls.builtins.diagnostics.jsonlint,
      --     null_ls.builtins.code_actions.statix,
      --     null_ls.builtins.diagnostics.clj_kondo,
      --     null_ls.builtins.formatting.isort,
      --     null_ls.builtins.formatting.black,
        },
      })

      -- Setup lspconfig
      require("lsp-format").setup {}
      require("nvim-navic").setup {}
      local format = require("lsp-format")
      local navic = require("nvim-navic")

      -- lspsaga, code_action lightbulb is annoying
      local saga = require('lspsaga')
      saga.init_lsp_saga({
        border_style = "rounded",
        code_action_icon = "",
        code_action_num_shortcut = true,
        code_action_lightbulb = {
            enable = false,
            enable_in_insert = false,
            cache_code_action = true,
            sign = true,
            update_time = 150,
            sign_priority = 20,
            virtual_text = false,
        },
        code_action_keys = {
            quit = 'q',
            exec = '<CR>',
        },
        -- show outline
        show_outline = {
          win_position = 'right',
          --set special filetype win that outline window split.like NvimTree neotree
          -- defx, db_ui
          auto_enter = true,
          auto_preview = true,
          virt_text = '┃',
          jump_key = 'o',
          -- auto refresh when change buffer
          auto_refresh = true,
        },
      })

      capabilities = require('cmp_nvim_lsp').default_capabilities()

      -- fennel-ls is bleeding edge
      local lspconfig = require('lspconfig')
      require("lspconfig.configs")["fennel-ls"] = {
        default_config = {
          cmd = {"fennel-ls"},
          filetypes = {"fennel"},
          root_dir = function(dir) return lspconfig.util.find_git_ancestor(dir) end,
          settings = {}
        }
      }

      local servers = { 'clojure_lsp', 'terraform_lsp', 'zls', 'pyright', 'rust_analyzer', 'tsserver', 'lua_ls', 'nil_ls', 'gopls'}
      for _, lsp in ipairs(servers) do
        lspconfig[lsp].setup {
          on_attach = function(client, bufnr)
            format.on_attach(client)
            navic.attach(client, bufnr)
          end,
          capabilities = capabilities,
        }
      end

      -- fennel-ls doesn't support gps
      lspconfig['fennel-ls'].setup {
        on_attach = function(client, bufnr)
          format.on_attach(client)
        end,
        capabilities = capabilities,
      }

      -- deprecated pylsp
      -- require('lspconfig').pylsp.setup {
      --   on_attach = function(client, bufnr)
      --     format.on_attach(client)
      --     navic.attach(client, bufnr)
      --   end,
      --   capabilities = capabilities,
      --   settings = {
      --     pylsp = {
      --       configurationSources = { "flake8", "mypy", "black" },
      --       plugins = {
      --         black = {enabled = true},
      --         jedi_signature_help = {enabled = true},
      --         jedi_completion = { include_params = true },
      --         pylsp_mypy={ enabled = true },
      --         pycodestyle={
      --           enabled=true,
      --           ignore={'E501', 'E231'},
      --           maxLineLength=120,
      --         },
      --       },
      --     },
      --   },
      -- }

      require('gitsigns').setup { sign_priority=80 }

      -- treesitter
      require('nvim-treesitter.configs').setup { 
        -- parser_install_dir = "~/.local/share/nvim/site/parser/",
        -- ensure_installed={'nix', 'clojure', 'python', 'fennel', 'lua', 'html', 'css', 'regex', 'supercollider', 'beancount', 'janet'},
        highlight = { enabled = true, additional_vim_regex_highting = true, },
        textobjects = { enabled = true },
      }
      -- 'glsl' needs c++ compiler lol

      -- lualine
      local navic = require("nvim-navic")
      require('lualine').setup {
        options = { theme = 'ayu' },
        sections = {
          lualine_c = {
              'filename', { navic.get_location, condition=navic.is_available },
          },
          lualine_y = {
            function()
              local msg = 'No Active Lsp'
              local buf_ft = vim.api.nvim_buf_get_option(0, 'filetype')
              local clients = vim.lsp.get_active_clients()
              if next(clients) == nil then
                return msg
              end
              for _, client in ipairs(clients) do
                local filetypes = client.config.filetypes
                if filetypes and vim.fn.index(filetypes, buf_ft) ~= -1 then
                  return client.name
                end
              end
              return msg
            end
            },
          lualine_x = {'filetype'}
        }
      }

      -- nvim-lint
      -- require('lint').linters_by_ft = {
      --   markdown = {'vale'},
      -- }

      -- f_term
      local map = vim.api.nvim_set_keymap
      local opts = { noremap = true, silent = true }
      map('n', '<TAB>', '<CMD>lua require("FTerm").toggle()<CR>', opts)
      map('t', '<TAB>', '<C-\\><C-n><CMD>lua require("FTerm").toggle()<CR>', opts)

      -- Custom functions
      function update_date()
        local time = os.date("%Y-%m-%d %H:%M")
        local message = "%update " .. time
        local lines = vim.api.nvim_buf_get_lines(0, 0,4,false)
        for i=1,4,1 do
          if (lines[i]:len() == 0) then
            vim.api.nvim_buf_set_lines(0, i-1, i, false, {message, ""})
            return
          end
        end
        if (lines[4]:len() ~= 0) then
          vim.api.nvim_buf_set_lines(0, 3, 4, false, {message})
        end
        return
      end

      vim.api.nvim_create_autocmd(
        "BufWritePre", 
        { pattern = { "*.wiki" }, callback = update_date}
      )

      -- set spelling for text files
      vim.api.nvim_create_autocmd(
        { "BufRead", "BufNewFile" },
        { pattern = { "*.wiki", "*.txt", "*.md", "*.tex" }, 
        command = "setlocal spell" }
      )
      vim.api.nvim_create_autocmd(
         "FileType",
        { pattern = { "mail" }, 
        command = "setlocal spell" }
      )

      -- auto session
      require('auto-session').setup {
        log_level = 'info',
        auto_session_suppress_dirs = {'~/'}
      }
      vim.o.sessionoptions="blank,buffers,curdir,folds,help,tabpages,winsize,winpos,terminal"

      require("which-key").setup{}
      require('leap').set_default_keymaps()
      
      -- supercollider
      local scnvim = require 'scnvim'
      local map = scnvim.map
      local map_expr = scnvim.map_expr
      scnvim.setup {
        keymaps = {
          ['<localleader>e'] = map('editor.send_line', {'i', 'n'}),
          ['<localleader>s'] = map_expr('CmdPeriod.run', {'i', 'n'}),
          ['<localleader>t'] = {
            map('editor.send_block', {'i', 'n'}),
            map('editor.send_selection', 'x'),
          },
          ['<CR>'] = map('postwin.toggle'),
          ['<M-CR>'] = map('postwin.toggle', 'i'),
          ['<M-L>'] = map('postwin.clear', {'n', 'i'}),
          ['<C-k>'] = map('signature.show', {'n', 'i'}),
          ['<F12>'] = map('sclang.hard_stop', {'n', 'x', 'i'}),
          ['<leader>st'] = map('sclang.start'),
          ['<leader>sk'] = map('sclang.recompile'),
          ['<F1>'] = map_expr('s.boot'),
          ['<F2>'] = map_expr('s.meter'),
        }
      } 
      -- You gotta run :SCNvimGenerateAssets for this first
      -- require("luasnip").add_snippets("supercollider", require("scnvim/utils").get_snippets())

      -- zen mode
      require('zen-mode').setup{
        window = {
          backdrop = 1.0, -- shade the backdrop of the Zen window. Set to 1 to keep the same as Normal
          -- height and width can be:
          -- * an absolute number of cells when > 1
          -- * a percentage of the width / height of the editor when <= 1
          -- * a function that returns the width or the height
          width = 120, -- width of the Zen window
          height = 1, -- height of the Zen window
          -- by default, no options are changed for the Zen window
          -- uncomment any of the options below, or add other vim.wo options you want to apply
          options = {
            signcolumn = "no", -- disable signcolumn
            number = false, -- disable number column
            relativenumber = false, -- disable relative numbers
            cursorline = false, -- disable cursorline
            cursorcolumn = false, -- disable cursor column
            foldcolumn = "0", -- disable fold column
            list = false, -- disable whitespace characters
          },
        },
        plugins = {
          -- disable some global vim options (vim.o...)
          -- comment the lines to not apply the options
          options = {
            enabled = true,
            ruler = false, -- disables the ruler text in the cmd line area
            showcmd = false, -- disables the command in the last line of the screen
          },
          twilight = { enabled = true }, -- enable to start Twilight when zen mode opens
          gitsigns = { enabled = false }, -- disables git signs
          tmux = { enabled = false }, -- disables the tmux statusline
          -- this will change the font size on kitty when in zen mode
          -- to make this work, you need to set the following kitty options:
          -- - allow_remote_control socket-only
          -- - listen_on unix:/tmp/kitty
          kitty = {
            enabled = true,
            font = "+2", -- font size increment
          },
        },
      }
      require('twilight').setup{}

      -- hologram
      require('hologram').setup{
          auto_display = false -- WIP automatic markdown image display, may be prone to breaking
      }

      EOF

    '';
    extraPackages = with pkgs;  [
      # used to compile tree-sitter grammar
      # python-with-my-packages
      nodejs
      clojure-lsp
      clojure
      clj-kondo
      terraform-lsp
      gopls
      gcc
      shellcheck
      proselint
      statix
      nodePackages.typescript
      nodePackages.typescript-language-server
      luajitPackages.lua-lsp
      fennel-ls
      starpls-bin
    ];

    plugins = with pkgs.vimPlugins; [
      # ui
      lualine-nvim
      fzf-vim
      vim-devicons
      nvim-web-devicons
      telescope-nvim
      plenary-nvim
      nvim-navic
      fterm
      barbar-nvim

      # TODO
      # neotree
      neovim-ayu
      rainbow_parentheses-vim
      vim-surround
      undotree

      (nvim-treesitter.withPlugins (p: [ p.nix p.clojure p.fennel p.lua p.html p.css p.regex p.supercollider p.beancount p.markdown p.glsl p.yaml p.toml p.dockerfile p.json ]))
      nvim-treesitter-context
      auto-session
      which-key-nvim
      zen-mode-nvim
      twilight-nvim
      
      # TODO remove?
      vim-dasht

      # language
      vim-nix
      emmet-vim
      csv-vim
      direnv-vim
      conjure

      # kitaab stuff
      vimwiki
      vim-zettel
      hologram-nvim

      # lsp stuff
      # nvim-lint
      null-ls-nvim
      nvim-lspconfig
      nvim-cmp
      cmp-nvim-lsp
      cmp-treesitter
      nvim-lsp-ts-utils
      lspsaga
      cmp-conjure
      cmp-buffer
      cmp-path
      cmp-spell
      nvim-autopairs
      cmp_luasnip
      luasnip
      lsp-format
      friendly-snippets
      lspkind-nvim

      # git stuff
      vim-fugitive
      gitsigns-nvim
      fzf-checkout-vim

      # Clojure stuff
      # conjure
      vim-sexp
      vim-sexp-mappings-for-regular-people
      fennel-vim

      vim-tidal
      # experimental 
      nvim-luapad
      scnvim
      leap
      vim-beancount
      # vimtex
      # custom
      nvim-parinfer
      # vim-processing
    ];
    # withPython3 = true;
    # extraPython3Packages = pkgs: with pkgs; [ tasklib six ];
    # vimAlias = true;
  };
}