import XMonad
import XMonad.Config.Desktop
import Data.Monoid
import Data.Word
import Graphics.X11.Xlib
import Graphics.X11.Xlib.Extras
import XMonad.Actions.UpdatePointer (updatePointer)
import XMonad.Hooks.DynamicLog
import XMonad.Hooks.EwmhDesktops
import XMonad.Hooks.ManageDocks
import XMonad.Hooks.UrgencyHook
import XMonad.Layout.Accordion
import XMonad.Layout.Circle
import XMonad.Layout.GridVariants
import XMonad.Layout.LayoutCombinators hiding ( (|||) )
import XMonad.Layout.Spiral
import XMonad.Layout.Tabbed
import XMonad.Layout.ThreeColumns
import XMonad.Prompt
import XMonad.Prompt.Shell ( shellPrompt )
import XMonad.Util.Run(spawnPipe)
import XMonad.Util.EZConfig(additionalKeys)
import XMonad.Util.NamedWindows
import XMonad.Util.Run
import XMonad.Util.SpawnOnce
import System.IO

import qualified XMonad.StackSet as W

colourBackground :: String
colourBackground = "#1c1c1c"

colourForeground :: String
colourForeground = "#d0d0d0"

highlightBackground :: String
highlightBackground = "#ffaf00"

highlightForeground :: String
highlightForeground = "#1c1c1c"

-- Set xmobar as my task bar.
myWsBar :: String
myWsBar = "xmobar -v -d $HOME/.xmobarrc"

myXPConfig = def   -- Configure the prompt's appearance
  { alwaysHighlight   = True
  , bgColor           = colourBackground
  , bgHLight          = highlightBackground
  , fgColor           = colourForeground
  , fgHLight          = highlightForeground
  , font              = "xft:Open Sans:size=9"
  , position          = Top
  , promptBorderWidth = 0
  }

backgroundCmd   = "feh --bg-scale ~/Documents/Images/Posters/STEVEs.jpeg"
compositeMgr    = "xcompmgr -f -C -n -D 3"
hdmi1Off        = "xrandr --output eDP-1 --primary --output HDMI-1 --off"
musicCmd        = "mpd"
nextcloudCmd    = "nextcloud"
notificationCmd = "lxqt-notificationd"
screensaverCmd  = "xscreensaver --no-splash"
screensaverLock  = "xscreensaver-command -lock"
screenshot       = "scrot"   -- Screenshot the entire screen
screenshotWindow = "sleep 0.2; scrot -s"   -- Screenshot the selected window
settingsDaemon  = "xsettingsd"
terminalCmd     = "termonad"
trayerCmd       = "trayer --edge top --SetPartialStrut false --width 7 --transparent true --alpha 0 --tint 0x1C1C1C --height 20 --monitor primary"

main = do
    -- Make sure that HDMI is turned off by default
    spawn hdmi1Off     -- Turn off HDMI-1 before we set the modes
    wsbar <- spawnPipe myWsBar
    xmonad $ ewmh desktopConfig
        { focusFollowsMouse = False
        , terminal = terminalCmd   -- Set the default terminal
        , startupHook = do
            spawnOnce screensaverCmd    -- Launch the screen saver
            spawnOnce backgroundCmd     -- Set the background wallpaper
            spawnOnce compositeMgr      -- Launch the composite manager
            spawnOnce settingsDaemon    -- Launch the settings daemon
            spawnOnce trayerCmd         -- Launch the system tray, configured to work correctly with dual monitors
            spawnOnce notificationCmd   -- Launch the notification service
            spawnOnce musicCmd          -- Launch the music playing daemon
            spawnOnce nextcloudCmd      -- Launch the cloud service
        , manageHook      = manageDocks <+> manageHook desktopConfig
        , layoutHook      = avoidStruts $ layoutHook desktopConfig
            ||| ThreeColMid 1 (3/100) (1/2)
            ||| spiral (6/7)
            ||| Grid (16/10)
            ||| Circle
            ||| simpleTabbed
            ||| Accordion
            ||| SplitGrid T 1 2 (1/3) (16/10) (5/100)
            -- ||| (ThreeColMid 1 (3/100) (1/3) *//* Full)
        , handleEventHook = handleEventHook desktopConfig <+> docksEventHook
        , logHook         = dynamicLogWithPP xmobarPP
            { ppOutput  = hPutStrLn wsbar
            , ppCurrent = xmobarColor "#ffaf00" "" . wrap "[" "]"
            , ppVisible = xmobarColor "#5fafd7" "" . wrap "(" ")"
            , ppLayout  = xmobarColor "#5fafd7" ""
            , ppTitle   = xmobarColor "#ffaf00" "" . shorten 50
            }  >> updatePointer (0.25, 0.25) (0.25, 0.25) -- near the top-left
        , modMask = mod4Mask -- Rebind Mod to the Windows key
        --, borderWidth = 1
        } `additionalKeys`
            -- Use Xmonad's built-in launcher
            [ ((mod4Mask, xK_p), shellPrompt myXPConfig)
            -- Lock the screen
            , ((0, 0x1008ff2d), spawn screensaverLock)
            -- XF86ScreenSaver
            , ((mod4Mask .|. controlMask, xK_l), spawn screensaverLock)
            , ((controlMask, xK_Print), spawn screenshotWindow)
            , ((0, xK_Print), spawn screenshot)
            -- Turn on the eDP-1 port and set it as the primary display
            , ((mod4Mask .|. shiftMask, xK_e), spawn
                "xrandr --output eDP-1 --primary --auto"
                )
            -- Turn off the eDP-1 port
            , ((mod4Mask .|. controlMask, xK_e), spawn
                "xrandr --output eDP-1 --off"
                )
            -- Turn off the HDMI port
            , ((mod4Mask .|. controlMask, xK_h), spawn
                "xrandr --output HDMI-1 --off"
                )
            -- Turn on the HDMI-1 port and set it as the secondary display
            , ((mod4Mask .|. shiftMask, xK_h), spawn
                "xrandr --newmode 3840x2160 266.75 3840 4056 4456 5072 2160 2163 2168 2192 -hsync +vsync; xrandr --addmode HDMI-1 3840x2160; xrandr --output eDP-1 --off --output HDMI-1 --mode 3840x2160; feh --bg-scale ~/Documents/Images/Posters/STEVEs.jpeg"
                )
            , ((0 , 0x1008FF11), spawn "amixer set Master 2%-") -- XF86AudioLowerVolume
            -- , ((mod4Mask , xK_Down), spawn "amixer set Master 2%-")
            , ((0 , 0x1008FF12), spawn "amixer set Master toggle") -- XF86AudioMute
            , ((0 , 0x1008FF13), spawn "amixer set Master 2%+") -- XF86AudioRaiseVolume
            -- , ((mod4Mask , xK_Up), spawn "amixer set Master 2%+")
            , ((0 , 0x1008FF14), spawn "mpc toggle") -- Play/pause
            -- , ((mod4Mask , xK_End), spawn "mpc toggle")
            , ((mod4Mask .|. controlMask, xK_space), spawn "mpc toggle") -- Play/pause
            , ((0 , 0x1008FF15), spawn "mpc stop") -- Stop
            , ((0 , 0x1008FF16), spawn "mpc prev") -- XF86AudioPrevious
            , ((mod4Mask , xK_Left), spawn "mpc prev") -- prev
            , ((0 , 0x1008FF17), spawn "mpc next") -- XF86AudioNext
            , ((mod4Mask , xK_Right), spawn "mpc next") -- Next
            -- XF86MonBrightnessUp
            , ((0 , 0x1008ff02), spawn "xbrightness +5000")
            -- XF86MonBrightnessDown
            , ((0 , 0x1008ff03), spawn "xbrightness -5000")
            ]