-- See https://xmonad.org/TUTORIAL.html
import XMonad

import XMonad.Actions.CycleWS (toggleWS)
import XMonad.Actions.RotSlaves

import XMonad.Hooks.DynamicLog
import XMonad.Hooks.EwmhDesktops
import XMonad.Hooks.ManageDocks
import XMonad.Hooks.StatusBar
import XMonad.Hooks.StatusBar.PP

import qualified XMonad.StackSet as W

import XMonad.Util.EZConfig (additionalKeysP)
import XMonad.Util.Loggers

import XMonad.Layout.Magnifier
import XMonad.Layout.ThreeColumns

main :: IO ()
main = xmonad
    . ewmhFullscreen 
    . ewmh 
    . withEasySB (statusBarProp "xmobar" (pure myXmobarPP)) defToggleStrutsKey
    $ myConfig

myConfig = def
    { terminal = "kitty tmux new"
    , modMask = mod1Mask
    , borderWidth = 3
    , workspaces = myWorkspaces
    , layoutHook = myLayout
    }
   `additionalKeysP`
   -- See below for help defining keymaps
   -- https://hackage.haskell.org/package/xmonad-contrib-0.17.1/docs/XMonad-Util-EZConfig.html#v:mkKeymap
   ([
   -- launch programs
     ("M-'", spawn "qutebrowser")
   -- launch a terminal _without_ a new tmux session
   , ("M-C-<Return>", spawn "kitty")
   -- screenshot
   , ("M-s", spawn "flameshot gui")
   -- xrandr commands for when (dis)connecting from external monitor
   -- I have temporarily given up on using autorandr fo rthis
   , ("M-x", spawn "xrandr --output DP-1 --auto --output eDP-1 --off") -- external
   , ("M-c", spawn "xrandr --output eDP-1 --auto --output DP-1 --off") -- laptop only
   -- Control monitor brightness
   , ("<XF86MonBrightnessUp>", spawn "light -A 10")
   , ("<XF86MonBrightnessDown>", spawn "light -U 10")
   -- cycle windows within a workspace
   , ("M-a", rotAllUp)
   , ("M-f", rotAllDown)
   -- switch to previous workspace
   , ("M-;", toggleWS)
   -- Warn (disable shutting down xmonad since we can do that in other ways from a terminal...
   , ("M-S-q", spawn "kitty --hold echo M-S-q quits XMonad\\! You probably meant to use M-S-c to close the current window.")
   ]
   ++
   -- access additional workspaces
   [("M-" ++ w, windows $ W.greedyView w) | w <- addlWorkspaces]
   ++
   [("M-S-" ++ w, windows $ W.shift w) | w <- addlWorkspaces]
   )

myLayout = Full ||| threeCol ||| tiled ||| Mirror tiled
  where
    --threeCol = magnifiercz' 1.3 $ ThreeColMid nmaster delta ratio
    threeCol = ThreeColMid nmaster delta ratio
    tiled    = Tall nmaster delta ratio
    nmaster  = 1      -- Default number of windows in the master pane
    ratio    = 1/2    -- Default proportion of screen occupied by master pane
    delta    = 3/100  -- Percent of screen to increment by when resizing panes

myXmobarPP :: PP
myXmobarPP = def
    { ppSep             = magenta " • "
    , ppTitleSanitize   = xmobarStrip
    , ppCurrent         = wrap " " "" . xmobarBorder "Top" "#8be9fd" 2
    , ppHidden          = lightGreen . wrap " " ""
    , ppHiddenNoWindows = gray . wrap " " ""
    , ppUrgent          = red . wrap (yellow "!") (yellow "!")
    , ppOrder           = \[ws, l, _, wins] -> [ws, l, wins]
    , ppExtras          = [logTitles formatFocused formatUnfocused]
    }
  where
    formatFocused   = wrap (white    "[") (white    "]") . magenta . ppWindow
    formatUnfocused = wrap (gray "[") (gray "]") . blue    . ppWindow

    -- | Windows should have *some* title, which should not not exceed a
    -- sane length.
    ppWindow :: String -> String
    ppWindow = xmobarRaw . (\w -> if null w then "untitled" else w) . shorten 30

    blue, gray, magenta, lightGreen, red, white, yellow :: String -> String
    magenta  = xmobarColor "#ff79c6" ""
    lightGreen = xmobarColor "#bbffbb" ""
    blue     = xmobarColor "#bdbdf9" ""
    white    = xmobarColor "#f8f8f2" ""
    yellow   = xmobarColor "#f1fa8c" ""
    red      = xmobarColor "#ff5555" ""
    gray = xmobarColor "#888888" ""

addlWorkspaces :: [String]
addlWorkspaces = ["0", "-", "=", "i"]

myWorkspaces :: [String]
myWorkspaces = ["1", "2", "3", "4", "5", "6", "7", "8", "9"] ++ addlWorkspaces