module KeyBindings (
  KeyBindings.modify
) where

import Control.Monad (void)
import Data.List (sort, isSuffixOf)
import Data.Maybe (isJust)
import Graphics.X11.Types
import System.Exit
import Text.EditDistance
import XMonad
import XMonad.Actions.CopyWindow (kill1,copy)
import XMonad.Actions.CycleWS
import XMonad.Actions.DynamicProjects
import XMonad.Core
import XMonad.Layout.ToggleLayouts
import XMonad.Operations
import XMonad.Prompt
import XMonad.Prompt.ConfirmPrompt
import XMonad.Prompt.Shell
import XMonad.Prompt.Workspace (workspacePrompt)
import XMonad.Util.EZConfig
import XMonad.Util.NamedScratchpad
import qualified Solarized as S
import qualified Scratchpad as R
import qualified XMonad.StackSet as W

-- Custom (in libs)
import Password (passPrompt)

modify :: XConfig l -> XConfig l
modify conf = conf
  { modMask = mod4Mask -- Use the "Win" key for the mod key
  }

  `additionalKeysP` -- Add some extra key bindings:
    [ ("M-S-q", confirmPrompt promptConfig "exit" (io exitSuccess))
    , ("<XF86MonBrightnessDown>", spawn "/run/current-system/sw/bin/light -U 10")
    , ("<XF86MonBrightnessUp>", spawn "/run/current-system/sw/bin/light -A 10")
    , ("<XF86AudioPlay>", spawn "/etc/profiles/per-user/e/bin/mpc toggle")
    , ("<XF86AudioNext>", spawn "/etc/profiles/per-user/e/bin/mpc next")
    , ("<XF86AudioPrev>", spawn "/etc/profiles/per-user/e/bin/mpc prev")
    , ("M-<Down>", windows W.focusDown)
    , ("M-<Esc>", sendMessage (Toggle "Full"))
    , ("M-$", sendMessage (Toggle "Full"))
    , ("M-<Left>", prevWS)
    , ("M-i", prevScreen)
    , ("M-e", nextScreen)
    , ("M-<Right>", nextWS)
    , ("M-<Tab>", toggleWS' ["NSP"])
    , ("M-<Up>", windows W.focusUp)
    , ("M-S-<Delete>", spawn "/etc/profiles/per-user/e/bin/mpc pause; /run/current-system/sw/bin/xset s activate")
    , ("M-S-<Left>", shiftToPrev >> prevWS)
    , ("M-S-<Right>", shiftToNext >> nextWS)
    , ("M-s s", spawn "/etc/profiles/per-user/e/bin/flameshot gui")
    , ("M-s t", spawn "/run/current-system/sw/bin/scrot /tmp/screen.png")
    -- Workspace and tasks
    , ("M-b", switchProjectPrompt promptConfig)
    , ("M-p c", withFocused centerWindow)
    , ("M-p m", shiftToProjectPrompt promptConfig)
    , ("M-p n", switchProjectPrompt promptConfig)
    , ("M-p r", renameProjectPrompt promptConfig)
    , ("M-p s", shellPrompt promptConfig)
    , ("M-d", passPrompt promptConfig)
    -- Scratchpads
    , ("M-p p", namedScratchpadAction R.pads "htop")
    -- Copy windows to workspaces
    , ("M-<Delete>", kill1)
    , ("M-c", workspacePrompt promptConfig (\name -> windows $ copy name))
    -- Mouse stuff
    , ("<F9>", spawn "find-cursor")
    -- , ("M-$ $", spawn "xdotool click 3")
    -- , ("M-$ M-$", spawn "xdotool click 3") -- helpfull to combine M + right click
    -- resizing the master/slave ratio
    , ("M-h", sendMessage Shrink) -- Shrink the master area
    , ("M-l", sendMessage Expand) -- Expand the master area
    -- dunstctl calls
    -- , ("C-<Space>", spawn "dunstctl close")
    , ("C-S-<Space>", spawn "dunstctl close-all")
    ]


-- Borrowed from https://www.reddit.com/r/xmonad/comments/gzq316/how_can_i_centre_a_floating_window_without/fthtx29/
centerWindow :: Window -> X ()
centerWindow win = do
    (_, W.RationalRect x y w h) <- floatLocation win
    windows $ W.float win (W.RationalRect ((1 - w) / 2) ((1 - h) / 2) w h)
    return ()

viewProject :: WorkspaceId -> X ()
viewProject id = do
  project <- lookupProject id
  case project of
    Just p -> switchProject p
    Nothing -> return ()

promptConfig = def
  { position          = Bottom
  , alwaysHighlight   = True
  , bgColor           = S.magenta
  , bgHLight          = S.base0
  , borderColor       = S.magenta
  , defaultText       = ""
  , fgColor           = S.base02
  , fgHLight          = S.base03
  , font              = "xft:Iosevka Samae:style=Regular:size=10:charwidth=6.5"
  , height            = 24
  , promptBorderWidth = 5
  }

-- Slightly taken from
-- https://mail.haskell.org/pipermail/xmonad/2010-October/010671.html
data FuzzySpawn = FuzzySpawn deriving (Read, Show)
instance XPrompt FuzzySpawn where showXPrompt _ = "RunC: "
fuzzyPrompt config = do
    cmds <- io getCommands
    let compl s
          | null s = []
          | otherwise = let weight c = levenshteinDistance defaultEditCosts s c
            in map snd $ take 20 $ sort $ map (\c -> (weight c,c)) cmds
    mkXPrompt FuzzySpawn config (return . compl) spawn

-- https://github.com/MasseR/xmonad-masser/blob/master/src/XMonad/Password.hs