import XMonad
import XMonad.Config.Desktop
import XMonad.Hooks.DynamicLog
import XMonad.Hooks.ManageDocks
import XMonad.Util.Run(spawnPipe, hPutStrLn, runProcessWithInput)
import XMonad.Util.EZConfig(additionalKeys)

-- For shutdown commands and keys
import Data.Map(fromList)
import XMonad.Prompt
import XMonad.Prompt.XMonad
import System.Exit(ExitCode(ExitSuccess), exitWith)

-- Brightness and audio keys
import Graphics.X11.ExtraTypes.XF86

main = do
  xmproc <- spawnPipe "xmobar"
  xmonad $ docks defaultConfig
        { manageHook = manageDocks <+> manageHook defaultConfig
        , layoutHook = avoidStruts  $  layoutHook defaultConfig
	, logHook = dynamicLogWithPP xmobarPP
		{ ppOutput = hPutStrLn xmproc
		, ppTitle = xmobarColor "green" "" . shorten 50
		}
        , startupHook = startup
        , terminal    = "gnome-terminal"
	, modMask     = mod4Mask
	} `additionalKeys`
        [
          -- scrot
          ((controlMask, xK_Print), spawn "sleep 0.2; scrot -s")
        , ((0, xK_Print), spawn "scrot")

        -- rofi
        , ((mod4Mask, xK_p ), spawn "rofi -show run")

        -- shutdown
        , ((mod4Mask .|. shiftMask, xK_q),
           xmonadPrompt defaultXPConfig
           { promptKeymap = fromList
             [ ((0, xK_r), do
                   spawn "emacsclient -e '(kill emacs)'"
                   spawn "systemctl reboot")
             , ((0 , xK_s), do
                   spawn "emacsclient -e '(kill emacs)'"
                   spawn "sudo poweroff")
             , ((0, xK_e), do
                   spawn "emacsclient -e '(kill emacs)'"
                   io $ exitWith ExitSuccess)
             , ((0, xK_l),  do
                   spawn "xscreensaver-command -lock"
                   quit)
             , ((0, xK_z), do
                   spawn "xscreensaver-command -lock"
                   spawn "systemctl suspend"
                   quit)
             , ((0, xK_Escape), quit)
             ]
           , defaultText = "(r) Reboot, (s) Shutdown, (e) Exit, (l) Lock, (z) Sleep"
           })
        -- pulseaudio


        -- brightness
        , ((0, xF86XK_MonBrightnessUp),
           let
              returnValM = fmap init $ runProcessWithInput "xbacklight" [] ""
           in do
             currentBrightness <- returnValM
             if (read currentBrightness :: Double) == 0 then
               spawn "xbacklight -set 2"
             else
               spawn "xbacklight -inc 5")

        , ((0, xF86XK_MonBrightnessDown), spawn "xbacklight -dec 5")
           -- Another option for brightness configuration: never let it reach 0.
           {-|
           let
              returnValM = fmap init $ runProcessWithInput "xbacklight" [] ""
           in do
             currentBrightness <- returnValM
             if (read currentBrightness :: Double) - 5 >= 0 then
               spawn "xbacklight -dec 5"
             else return ())
           -}
        ]

startup :: X ()
startup = do
  spawn "feh --bg-scale ~/Owncloud/Backgrounds/Xmbindings.png"
  spawn "owncloud"
  spawn "trayer --transparent true --alpha 0 --tint 0x00000000 --SetDockType true --expand true --edge top --align right --width 15 --height 18"
  spawn "xfce4-clipman"
  spawn "xbacklight -set 12"