;; Which-key provides a pop-up meny for some key mappings.
(module config.which-key {autoload {util config.util}})

(def- setup {:plugins {:marks true
                       :registers true
                       :spelling {:enabled true :suggestions 20}
                       :presets {:operators false
                                 :motions false
                                 :text_objects false
                                 :windows true
                                 :nav true
                                 :z true
                                 :g true}}
             :icons {:breadcrumb "»" :separator "" :group "+"}
             :popup_mappings {:scroll_down :<c-d> :scroll_up :<c-u>}
             :window {:border :rounded
                      :position :bottom
                      :margin [1 0 1 0]
                      :padding [2 2 2 2]
                      :winblend 0}
             :layout {:height {:min 4 :max 25}
                      :width {:min 20 :max 50}
                      :spacing 3
                      :align :left}
             :ignore_missing true
             :hidden [:<silent> :<cmd> :<Cmd> :<CR> :call :lua "^:" "^ "]
             :show_help true
             :triggers :auto
             :disable {:filetypes [:netrw]}
             :triggers_blacklist {:i [:j :k] :v [:j :k]}})

(def- mopts {:mode :n
             :prefix :m
             :buffer nil
             :silent true
             :noremap true
             :nowait true})

(def- mmappings
      {:a [:<cmd>HarpoonAdd<cr> :Harpoon]
       :c [:<cmd>DiffviewClose<cr> :DiffviewClose]
       :d [:<cmd>DiffviewFileHistory<cr> :DiffviewFileHistory]
       :f [(fn []
             (util.telescope-builtin :find_files
                                     {:theme :get_dropdown :previewer false}))
           "Find files"]
       :g ["<cmd>Telescope live_grep theme=dropdown<cr>" "Find text"]
       :n [:<cmd>Neogit<cr> :Neogit]
       :p [:<cmd>SaveSession<cr> "Save Session"]
       :q ["<cmd>Gitsigns diffthis HEAD<cr>" "Gitsigns diff"]
       :r [:<cmd>HarpoonUI<cr> "Harpoon UI"]
       :s [:<cmd>HarpoonPrev<cr> "Harpoon Prev"]
       :t [:<cmd>HarpoonNext<cr> "Harpoon Next"]
       :v ["<cmd>lua vim.lsp.buf.rename()<cr>" :Rename]
       :x ["<cmd>DiffviewOpen -uno<cr>" :DiffviewOpen]})

(def- nopts {:mode :n
             :prefix :<leader>
             :buffer nil
             :silent true
             :noremap true
             :nowait true})

(def- find
      {:name :Find
       :C ["<cmd>Telescope commands theme=dropdown<cr>" :Commands]
       :H ["<cmd>Telescope highlights<cr>" :Highlights]
       :R ["<cmd>Telescope registers theme=dropdown<cr>" :Registers]
       :S ["<cmd>Telescope grep_string theme=dropdown<cr>" "Find String"]
       :b ["<cmd>Telescope git_branches theme=dropdown<cr>" "Checkout branch"]
       :c ["<cmd>Telescope colorscheme theme=dropdown<cr>" :Colorscheme]
       :f [(fn []
             (util.telescope-builtin :find_files
                                     {:theme :get_dropdown :previewer false}))
           "Find files"]
       :h ["<cmd>Telescope help_tags theme=dropdown<cr>" :Help]
       :k ["<cmd>Telescope keymaps theme=dropdown<cr>" :Keymaps]
       :l ["<cmd>Telescope resume theme=dropdown<cr>" "Last Search"]
       :p ["<cmd>Telescope projects theme=dropdown<cr>" "Find project"]
       :r ["<cmd>Telescope oldfiles theme=dropdown<cr>" "Recent File"]
       :s [:<cmd>SearchSession<cr> "Find Session"]
       :t ["<cmd>Telescope live_grep theme=dropdown<cr>" "Find text"]})

(def- diagnostics {:name :Diagnostics
                   :d [:<cmd>TroubleToggle<cr> :Trouble]
                   :D ["<cmd>Telescope diagnostics theme=dropdown<cr>"
                       "Telescope diagnostics"]})

(def- git {:name :Git
           :R ["<cmd>lua require 'gitsigns'.reset_buffer()<cr>" "Reset Buffer"]
           :b ["<cmd>Telescope git_branches theme=dropdown<cr>"
               "Checkout branch"]
           :c ["<cmd>Telescope git_commits theme=dropdown<cr>"
               "Checkout commit"]
           :d ["<cmd>Gitsigns diffthis HEAD<cr>" :Diff]
           :j ["<cmd>lua require 'gitsigns'.next_hunk()<cr>" "Next Hunk"]
           :k ["<cmd>lua require 'gitsigns'.prev_hunk()<cr>" "Prev Hunk"]
           :l ["<cmd>lua require 'gitsigns'.blame_line()<cr>" :Blame]
           :o ["<cmd>Telescope git_status theme=dropdown<cr>"
               "Open changed file"]
           :p ["<cmd>lua require 'gitsigns'.preview_hunk()<cr>" "Preview Hunk"]
           :r ["<cmd>lua require 'gitsigns'.reset_hunk()<cr>" "Reset Hunk"]
           :s ["<cmd>lua require 'gitsigns'.stage_hunk()<cr>" "Stage Hunk"]
           :u ["<cmd>lua require 'gitsigns'.undo_stage_hunk()<cr>"
               "Undo Stage Hunk"]})

(def- lsp
      {:name :LSP
       :S ["<cmd>Telescope lsp_dynamic_workspace_symbols<cr>"
           "Workspace Symbols"]
       :a ["<cmd>lua vim.lsp.buf.code_action()<cr>" "Code Action"]
       :f ["<cmd>lua vim.lsp.buf.format { async = true }<cr>" :Format]
       :i [:<cmd>LspInfo<cr> :Info]
       :l ["<cmd>lua vim.lsp.codelens.run()<cr>" "CodeLens Action"]
       :m [:<cmd>Mason<cr> :Mason]
       :r ["<cmd>lua vim.lsp.buf.rename()<cr>" :Rename]
       :s ["<cmd>Telescope lsp_document_symbols<cr>" "Document Symbols"]})

(def- lazy {:name :Lazy
            :C ["<cmd>Lazy check<cr>" :Check]
            :c ["<cmd>Lazy clean<cr>" :Clean]
            :d ["<cmd>Lazy debug<cr>" :Debug]
            :h ["<cmd>Lazy home<cr>" :Home]
            :i ["<cmd>Lazy install<cr>" :Install]
            :l ["<cmd>Lazy log<cr>" :Log]
            :p ["<cmd>Lazy profile<cr>" :Profile]
            :s ["<cmd>Lazy sync<cr>" :Sync]
            :u ["<cmd>Lazy update<cr>" :Update]})

(def- replace {:name :Replace
               :f [:<cmd>ReplaceInBuf<cr> "Replace Buffer"]
               :r [:<cmd>Replace<cr> :Replace]
               :w [:<cmd>ReplaceWord<cr> "Replace Word"]})

(def- treesitter
      {:name :Treesitter :p [:<cmd>TSPlaygroundToggle<cr> :Playground]})

(def- nmappings {:a ["<cmd>Telescope lsp_document_symbols theme=dropdown<cr>"
                     "Document Symbols"]
                 :T treesitter
                 :<BS> [:<cmd>BufDel<CR> "Close Buffer"]
                 :d diagnostics
                 :f find
                 :g git
                 :h [:<cmd>ColorizerToggle<cr> :Colorizer]
                 :l lsp
                 :n [:<cmd>CommentNormal<CR> :Comment]
                 :r replace
                 :t [:<cmd>ToggleTerm<cr> :Terminal]
                 :u [:<cmd>UndotreeToggle<cr> :Undotree]
                 :y lazy
                 :z [:<cmd>ZenMode<cr> "Zen Mode"]})

(def- vopts {:mode :v
             :prefix :<leader>
             :buffer nil
             :silent true
             :noremap true
             :nowait true})

(def- vmappings {:n [:<esc><cmd>CommentVisual<cr> :Comment]})

(def- gopts {:mode :n
             :prefix :g
             :buffer nil
             :silent true
             :noremap true
             :nowait true})

(def- gmappings {:a [:<cmd>OrgAgendaPrompt<cr> "Open agenda prompt"]
                 :c [:<cmd>OrgCapturePrompt<cr> "Open capture prompt"]
                 :m ["<cmd>Telescope orgmode search_headings theme=dropdown<cr>"
                     "Search Headings"]})

(let [which-key (util.prequire :which-key)]
  (which-key.setup setup)
  (which-key.register mmappings mopts)
  (which-key.register nmappings nopts)
  (which-key.register vmappings vopts)
  (which-key.register gmappings gopts))