{-

Docs:

- EZConfig: https://hackage.haskell.org/package/xmonad-contrib-0.13/docs/XMonad-Util-EZConfig.html#g:3
- Media keys: https://hackage.haskell.org/package/X11-1.9/docs/Graphics-X11-ExtraTypes-XF86.html
- Audio control: https://xmonadhaskell.wordpress.com/2018/10/24/xmonad-audio-control/

- XMonad API: https://hackage.haskell.org/package/xmonad
- Contrib API: https://hackage.haskell.org/package/xmonad-contrib

-}

import Graphics.X11.ExtraTypes.XF86
import XMonad
import XMonad.Actions.CopyWindow
import XMonad.Config
import XMonad.Hooks.EwmhDesktops (ewmh)
import XMonad.Hooks.ManageDocks
import XMonad.Layout.BinarySpacePartition
import XMonad.Layout.Dwindle as Dwindle
import XMonad.Layout.LayoutModifier
import XMonad.Layout.NoBorders
import XMonad.Layout.ResizableTile
import XMonad.Layout.Spacing
import XMonad.Layout.Spiral
import XMonad.Layout.Tabbed
import XMonad.Layout.TwoPane
import XMonad.Util.CustomKeys (customKeys)
import XMonad.Util.EZConfig (additionalKeys)

nixBin :: String
nixBin = "/home/ben/.nix-profile/bin/"

altMask :: KeyMask
altMask = mod1Mask

-- Colors
cyan = "#2aa198" -- ^ solarized cyan
base3 = "#eee8d4" -- ^ solarized base3

insKeys :: XConfig l -> [((KeyMask, KeySym), X ())]
insKeys conf@(XConfig {modMask = modMask}) =
  [ ((modMask, xK_y), spawn $ nixBin <> "passmenu")
  , ((modMask, xK_u), spawn $ nixBin <> "rofi -sidebar-mode -show window")
  , ((modMask, xK_i), spawn $ "rofi -sidebar-mode -show run")

  , ((modMask, xK_o), spawn $ nixBin <> "dmenu_run")
  , ((modMask, xK_e), spawn "emacsclient -c")

  , ((modMask .|. altMask, xK_h), spawn "home-manager switch")

  -- sticky windows
  , ((modMask, xK_a ), windows copyToAll) -- @@ Make focused window always visible
  , ((modMask .|. shiftMask, xK_a ),  killAllOtherCopies) -- @@ Toggle window state back

  -- media/ function keys
  -- backlight
  , ((0, xK_F5), spawn "xbacklight -dec 5")
  , ((0, xK_F6), spawn "xbacklight -inc 5")
  , ((0, xF86XK_KbdBrightnessDown), spawn "xbacklight -dec 5")
  , ((0, xF86XK_KbdBrightnessUp), spawn "xbacklight -inc 5")
  -- volume controls
  , ((0, xK_F1), amixer "toggle")
  , ((0, xK_F2), amixer "2%+")
  , ((0, xK_F3), amixer "2%-")

  , ((0, xF86XK_AudioMute), amixer "toggle")
  , ((0, xF86XK_AudioLowerVolume), amixer "2%-")
  , ((0, xF86XK_AudioRaiseVolume), amixer "2%+")
  ]

amixer :: String -> X ()
amixer cmd = spawn $ "amixer -q sset Master " <> cmd

-- | Golden-ratio spiral
goldenSpiral :: SpiralWithDir a
goldenSpiral = spiral (6 / 7)

myWorkspaces :: [String]
myWorkspaces = ["1:chat", "2:emacs", "3:work"] ++ map show [4 .. 9] ++ ["0"]

addSpace :: l a -> ModifiedLayout Spacing l a
addSpace = spacingRaw
  True (Border 5 5 5 5)
  True (Border 5 5 5 5)
  True

myTabCfg = def { fontName = "xft:mononoki:size=10:ant"
               , activeBorderColor = cyan
               , inactiveBorderColor = base3
               , activeColor = base3
               , inactiveColor = base3
               }

myLayout = avoidStruts $
  noBorders (tabbed shrinkText myTabCfg) -- default tab config
  ||| tiled
  ||| Mirror tiled
  -- ||| noBorders Full
  ||| twopane
  ||| Mirror twopane
  ||| emptyBSP
  ||| goldenSpiral
  ||| Spiral L Dwindle.CW (3/2) (11/10) -- L means the non-main windows are put to the left.
  where
     -- The last parameter is fraction to multiply the slave window heights
     -- with. Useless here.
     tiled = addSpace $ ResizableTall nmaster delta ratio []
     -- In this layout the second pane will only show the focused window.
     twopane = addSpace $ TwoPane delta ratio
     -- The default number of windows in the master pane
     nmaster = 1
     -- Default proportion of screen occupied by master pane
     ratio   = 1/2
     -- Percent of screen to increment by when resizing panes
     delta   = 3/100

myConf = additionalKeys c (insKeys c)
  where c = def
          { modMask = mod4Mask -- ^ super instead of alt
          , normalBorderColor = base3
          , focusedBorderColor = cyan
          , borderWidth = 3
          , manageHook = manageDocks <+> manageHook def
          , layoutHook = myLayout
          , terminal = "/home/ben/.nix-profile/bin/xterm"
          , workspaces = myWorkspaces
          }

main :: IO ()
main = xmonad $ ewmh $ docks $ myConf