(import-macros {: colorscheme
                : highlight
                : highlight-link
                : highlight-get-field} :../macros/cmd)

(let [mod {}
      o vim.opt
      g vim.g
      dracula (require :dracula)
      dracula-colors (dracula.colors)]
  (fn swap-bg-with-fg [fg group]
    (highlight group {: fg :bg (highlight-get-field group :fg)}))

  (set mod.colors dracula-colors)

  (fn mod.setup []
    (set g.dracula_show_end_of_buffer false)
    (set g.dracula_italic_comment true)
    (set o.background :dark)
    (colorscheme :dracula)
    ;; LSP
    (highlight-link :LspCodeLens :Comment)
    (highlight-link :LspInlayHint :Comment)
    ;; Tabline
    (highlight :TabLine {:fg mod.colors.comment :bg mod.colors.black})
    (highlight :TabLineSel {:fg mod.colors.white :bg mod.colors.bg})
    (highlight :TabLineFill {:bg mod.colors.black})
    ;; vim-matchup
    (highlight :MatchParen {:fg mod.colors.orange :bold true :italic true})
    (highlight :MatchWord {:italic true})
    (highlight-link :MatchupVirtualText :Comment)
    ;; vim-illuminate
    (highlight :IlluminatedWordText {:bg mod.colors.selection})
    (highlight :IlluminatedWordRead {:bg mod.colors.selection})
    (highlight :IlluminatedWordWrite {:bg mod.colors.selection :underline true})
    ;; rainbow-delimiters
    (highlight :RainbowDelimiterRed {:fg mod.colors.red})
    (highlight :RainbowDelimiterYellow {:fg mod.colors.yellow})
    (highlight :RainbowDelimiterBlue {:fg mod.colors.purple})
    (highlight :RainbowDelimiterOrange {:fg mod.colors.orange})
    (highlight :RainbowDelimiterGreen {:fg mod.colors.green})
    (highlight :RainbowDelimiterViolet {:fg mod.colors.pink})
    (highlight :RainbowDelimiterCyan {:fg mod.colors.cyan})
    ;; cmp
    (highlight :PmenuSel {:fg mod.colors.white :bg "#363848"})
    (highlight :CmpCursorLine {:bg "#363848"})
    (highlight :CmpItemAbbr {:fg mod.colors.white :bg :NONE})
    (highlight :CmpItemAbbrMatch {:fg mod.colors.cyan :bg :NONE :bold true})
    (highlight :CmpItemAbbrMatchFuzzy
               {:fg mod.colors.cyan :bg :NONE :bold true})
    (highlight :CmpItemMenu {:fg mod.colors.purple :bg :NONE :italic true})
    (let [cmp-groups [:CmpItemKindField
                      :CmpItemKindProperty
                      :CmpItemKindEvent
                      :CmpItemKindText
                      :CmpItemKindEnum
                      :CmpItemKindKeyword
                      :CmpItemKindConstant
                      :CmpItemKindConstructor
                      :CmpItemKindReference
                      :CmpItemKindFunction
                      :CmpItemKindStruct
                      :CmpItemKindClass
                      :CmpItemKindModule
                      :CmpItemKindOperator
                      :CmpItemKindVariable
                      :CmpItemKindUnit
                      :CmpItemKindSnippet
                      :CmpItemKindFolder
                      :CmpItemKindMethod
                      :CmpItemKindValue
                      :CmpItemKindEnumMember
                      :CmpItemKindInterface
                      :CmpItemKindColor
                      :CmpItemKindTypeParameter]]
      (each [_ group (pairs cmp-groups)]
        (swap-bg-with-fg mod.colors.black group)))
    (highlight :CmpItemKindFile {:fg mod.colors.black :bg mod.colors.white})
    (highlight :CmpItemKindFolder {:fg mod.colors.black :bg mod.colors.white})
    (highlight :CmpItemKindTabNine
               {:fg mod.colors.black :bg mod.colors.bright_magenta}))

  mod)