----------------------------------------------------------------------------------------------
---IMPORTS
----------------------------------------------------------------------------------------------
    -- Base
import XMonad
import XMonad.Config.Desktop
import Data.Maybe (isJust)
import System.IO (hPutStrLn)
import System.Exit (exitSuccess)
import qualified XMonad.StackSet as W

    -- Utilities
import XMonad.Util.Loggers
import XMonad.Util.EZConfig (additionalKeysP, additionalMouseBindings)  
import XMonad.Util.NamedScratchpad (NamedScratchpad(NS), namedScratchpadManageHook, namedScratchpadAction, customFloating)
import XMonad.Util.Run (safeSpawn, unsafeSpawn, runInTerm, spawnPipe)
import XMonad.Util.SpawnOnce

    -- Hooks
import XMonad.Hooks.DynamicLog (dynamicLogWithPP, defaultPP, wrap, pad, xmobarPP, xmobarColor, shorten, PP(..))
import XMonad.Hooks.ManageDocks (avoidStruts, docksStartupHook, manageDocks, ToggleStruts(..))
import XMonad.Hooks.ManageHelpers (isFullscreen, isDialog,  doFullFloat, doCenterFloat) 
import XMonad.Hooks.Place (placeHook, withGaps, smart)
import XMonad.Hooks.InsertPosition
import XMonad.Hooks.FloatNext (floatNextHook, toggleFloatNext, toggleFloatAllNew)
import XMonad.Hooks.SetWMName
import XMonad.Hooks.EwmhDesktops   -- required for xcomposite in obs to work

    -- Actions
import XMonad.Actions.Minimize (minimizeWindow)
import XMonad.Actions.Promote
import XMonad.Actions.RotSlaves (rotSlavesDown, rotAllDown)
import XMonad.Actions.CopyWindow (kill1, copyToAll, killAllOtherCopies, runOrCopy)
import XMonad.Actions.WindowGo (runOrRaise, raiseMaybe)
import XMonad.Actions.WithAll (sinkAll, killAll)
import XMonad.Actions.CycleWS (moveTo, shiftTo, WSType(..), shiftNextScreen, shiftPrevScreen) 
import XMonad.Actions.GridSelect (GSConfig(..), goToSelected, bringSelected, colorRangeFromClassName, buildDefaultGSConfig)
import XMonad.Actions.DynamicWorkspaces (addWorkspacePrompt, removeEmptyWorkspace)
import XMonad.Actions.Warp (warpToWindow, banishScreen, Corner(LowerRight))
import XMonad.Actions.MouseResize
import qualified XMonad.Actions.ConstrainedResize as Sqr

    -- Layouts modifiers
import XMonad.Layout.PerWorkspace (onWorkspace) 
import XMonad.Layout.Renamed (renamed, Rename(CutWordsLeft, Replace))
import XMonad.Layout.WorkspaceDir 
import XMonad.Layout.Spacing (spacing) 
import XMonad.Layout.Minimize
import XMonad.Layout.Maximize
import XMonad.Layout.NoBorders
import XMonad.Layout.BoringWindows (boringWindows)
import XMonad.Layout.LimitWindows (limitWindows, increaseLimit, decreaseLimit)
import XMonad.Layout.WindowArranger (windowArrange, WindowArrangerMsg(..))
import XMonad.Layout.Reflect (reflectVert, reflectHoriz, REFLECTX(..), REFLECTY(..))
import XMonad.Layout.MultiToggle (mkToggle, single, EOT(EOT), Toggle(..), (??))
import XMonad.Layout.MultiToggle.Instances (StdTransformers(NBFULL, MIRROR, NOBORDERS))
import qualified XMonad.Layout.ToggleLayouts as T (toggleLayouts, ToggleLayout(Toggle))

    -- Layouts
import XMonad.Layout.GridVariants (Grid(Grid))
import XMonad.Layout.SimplestFloat
import XMonad.Layout.OneBig
import XMonad.Layout.ThreeColumns
import XMonad.Layout.ResizableTile
import XMonad.Layout.ZoomRow (zoomRow, zoomIn, zoomOut, zoomReset, ZoomMessage(ZoomFullToggle))
import XMonad.Layout.IM (withIM, Property(Role))

    -- Prompts
import XMonad.Prompt (defaultXPConfig, XPConfig(..), XPPosition(Top), Direction1D(..))

----------------------------------------------------------------------------------------------
---CONFIG
----------------------------------------------------------------------------------------------
myModMask       = mod4Mask  -- Sets modkey to super/windows key
myTerminal      = "urxvtc" -- Sets default terminal
myTextEditor    = "editor"  -- Sets default text editor
myBorderWidth   = 3         -- Sets border width for windows
windowCount 	= gets $ Just . show . length . W.integrate' . W.stack . W.workspace . W.current . windowset

main = do
    xmproc0 <- spawnPipe "xmobar -x 0 /home/dt/.config/xmobar/xmobarrc0"  -- Launches xmobar
    xmproc1 <- spawnPipe "xmobar -x 1 /home/dt/.config/xmobar/xmobarrc1"  -- Launches xmobar
    xmproc2 <- spawnPipe "xmobar -x 2 /home/dt/.config/xmobar/xmobarrc2"  -- Launches xmobar 
    xmonad $ ewmh desktopConfig
        { manageHook = ( isFullscreen --> doFullFloat ) <+> manageHook defaultConfig <+> manageDocks
        , logHook = dynamicLogWithPP xmobarPP
                        { ppOutput = \x -> hPutStrLn xmproc0 x  >> hPutStrLn xmproc1 x  >> hPutStrLn xmproc2 x
                        , ppCurrent = xmobarColor "#859900" "" . wrap "[" "]"  -- Current workspace in xmobar
                        , ppVisible = xmobarColor "#859900" "" . wrap "" ""    -- Visible but not current workspace
                        , ppHidden = xmobarColor "#D33682" ""                  -- Hidden workspaces in xmobar
                        , ppHiddenNoWindows = xmobarColor "#D33682" ""         -- Hidden workspaces (no windows)
                        , ppTitle = xmobarColor "#859900" "" . shorten 80      -- Title of active window in xmobar
                        , ppSep =  "<fc=#839496> : </fc>"                     -- Separators in xmobar
                        , ppUrgent = xmobarColor "#C45500" "" . wrap "!" "!"   -- Urgent workspace
                        , ppExtras  = [windowCount]							   -- # of windows current workspace
                        , ppOrder  = \(ws:l:t:exs) -> [ws,l]++exs++[t]
                        }
        , modMask            = myModMask
        , terminal           = myTerminal
        , startupHook        = myStartupHook
        , layoutHook         = myLayoutHook 
        , workspaces         = myWorkspaces
        , borderWidth        = myBorderWidth 
        , normalBorderColor  = "#151515"
        , focusedBorderColor = "#76581E"
        } `additionalKeysP`         myKeys 

----------------------------------------------------------------------------------------------
---AUTOSTART
----------------------------------------------------------------------------------------------
myStartupHook = do
          spawnOnce "urxvtd &" 
          spawnOnce "nitrogen --restore &" 
          spawnOnce "compton --config /home/dt/.config/compton/compton.conf &" 
          setWMName "LG3D"
          --spawnOnce "/home/dt/.xmonad/xmonad.start"                                    -- Sets our wallpaper
          --spawnOnce "compton -cCGfF -o 1.00 -O 0.100 -I 0.100 -t 0 -l 0 -r 3 -D2 -m 0.88 --opacity-rule 75:'class_g *=\"XTerm\"' &"  	-- Enables compositing

----------------------------------------------------------------------------------------------
---KEYBINDINGS
----------------------------------------------------------------------------------------------
myKeys =
    -- Xmonad
        [ ("M-C-r",             spawn "xmonad --recompile")      -- Recompiles xmonad
        , ("M-S-r",             spawn "xmonad --restart")        -- Restarts xmonad
        , ("M-S-q",             io exitSuccess)                  -- Quits xmonad
    
    -- Windows
        , ("M-r",               refresh)                         -- Refresh
        , ("M-S-c",             kill1)                           -- Kill the currently focused client
        , ("M-S-a",             killAll)                         -- Kill all the windows on the current workspace
        
        , ("M-<Delete>",        withFocused $ windows . W.sink)
        , ("M-S-<Delete>",      sinkAll)                         -- Pushes all floating windows on current workspace back into tiling
        , ("M-m",               windows W.focusMaster)           -- Move focus to the master window
        , ("M-j",               windows W.focusDown)             -- Move focus to the next window
        , ("M-k",               windows W.focusUp)               -- Move focus to the prev window
        , ("M-S-m",             windows W.swapMaster)            -- Swap the focused window and the master window
        , ("M-S-j",             windows W.swapDown)              -- Swap the focused window with the next window
        , ("M-S-k",             windows W.swapUp)                -- Swap the focused window with the prev window
        , ("M-<Backspace>",     promote)                         -- Moves focused window to master window. All others maintain order
        , ("M1-S-<Tab>",        rotSlavesDown)                   -- Rotate all windows except the master and keep the focus in place
        , ("M1-C-<Tab>",        rotAllDown)                      -- Rotate all the windows in the current stack

        , ("M-*",               withFocused minimizeWindow)
        --, ("M-S-*",             sendMessage RestoreNextMinimizedWin)
        , ("M-!",               withFocused (sendMessage . maximizeRestore))
        , ("M-$",               toggleFloatNext)
        , ("M-S-$",             toggleFloatAllNew)
        , ("M-S-s",             windows copyToAll) 
        , ("M-C-s",             killAllOtherCopies) 
        
        , ("M-C-M1-<Up>",       sendMessage Arrange)
        , ("M-C-M1-<Down>",     sendMessage DeArrange)
        , ("M-<Up>",            sendMessage (MoveUp 10))         --  Move focused window to up
        , ("M-<Down>",          sendMessage (MoveDown 10))       --  Move focused window to down
        , ("M-<Right>",         sendMessage (MoveRight 10))      --  Move focused window to right
        , ("M-<Left>",          sendMessage (MoveLeft 10))       --  Move focused window to left
        , ("M-S-<Up>",          sendMessage (IncreaseUp 10))     --  Increase size of focused window up
        , ("M-S-<Down>",        sendMessage (IncreaseDown 10))   --  Increase size of focused window down
        , ("M-S-<Right>",       sendMessage (IncreaseRight 10))  --  Increase size of focused window right
        , ("M-S-<Left>",        sendMessage (IncreaseLeft 10))   --  Increase size of focused window left
        , ("M-C-<Up>",          sendMessage (DecreaseUp 10))     --  Decrease size of focused window up
        , ("M-C-<Down>",        sendMessage (DecreaseDown 10))   --  Decrease size of focused window down
        , ("M-C-<Right>",       sendMessage (DecreaseRight 10))  --  Decrease size of focused window right
        , ("M-C-<Left>",        sendMessage (DecreaseLeft 10))   --  Decrease size of focused window left

    -- Layouts
        , ("M-S-<Space>",       sendMessage ToggleStruts)
        , ("M-d",               asks (XMonad.layoutHook . config) >>= setLayout)
        , ("M-<KP_Enter>",      sendMessage NextLayout)
        , ("M-S-f",             sendMessage (T.Toggle "floatz"))
        , ("M-S-g",             sendMessage (T.Toggle "gimp"))
        , ("M-S-x",             sendMessage $ Toggle REFLECTX)
        , ("M-S-y",             sendMessage $ Toggle REFLECTY)
        , ("M-S-m",             sendMessage $ Toggle MIRROR)
        , ("M-S-b",             sendMessage $ Toggle NOBORDERS)
        , ("M-S-d",             sendMessage (Toggle NBFULL) >> sendMessage ToggleStruts)
        , ("M-<KP_Multiply>",   sendMessage (IncMasterN 1))      -- Increase the number of clients in the master pane
        , ("M-<KP_Divide>",     sendMessage (IncMasterN (-1)))   -- Decrease the number of clients in the master pane
        , ("M-S-<KP_Multiply>", increaseLimit)                   -- Increase the number of windows that can be shown
        , ("M-S-<KP_Divide>",   decreaseLimit)                   -- Decrease the number of windows that can be shown

        , ("M-h",               sendMessage Shrink)
        , ("M-l",               sendMessage Expand)
        , ("M-S-;",             sendMessage zoomReset)
        , ("M-;",               sendMessage ZoomFullToggle)

    -- Workspaces
        , ("M-<KP_Add>",        moveTo Next nonNSP)                         -- Go to next workspace
        , ("M-<KP_Subtract>",   moveTo Prev nonNSP)                         -- Go to previous workspace
        , ("M-S-<KP_Add>",      shiftTo Next nonNSP >> moveTo Next nonNSP)  -- Shifts focused window to next workspace
        , ("M-S-<KP_Subtract>", shiftTo Prev nonNSP >> moveTo Prev nonNSP)  -- Shifts focused window to previous workspace


    -- Main Run Apps
        , ("M-<Return>",        spawn myTerminal)
        , ("M-<KP_Insert>",     spawn "rofi -show run")
        
    -- Command Line Apps  (MOD + KEYPAD 1-9)
        , ("M-<KP_End>",        spawn (myTerminal ++ " -e lynx -cfg=~/.lynx.cfg -lss=~/.lynx.lss http://www.distrowatch.com")) -- Keypad 1
        , ("M-<KP_Down>",       spawn (myTerminal ++ " -e sh ./scripts/googler-script.sh"))  -- Keypad 2
        , ("M-<KP_Page_Down>",  spawn (myTerminal ++ " -e newsboat"))                -- Keypad 3
        , ("M-<KP_Left>",       spawn (myTerminal ++ " -e rtv"))                     -- Keypad 4
        , ("M-<KP_Begin>",      spawn (myTerminal ++ " -e neomutt"))                 -- Keypad 5
        , ("M-<KP_Right>",      spawn (myTerminal ++ " -e twitch-curses"))           -- Keypad 6
        , ("M-<KP_Home>",       spawn (myTerminal ++ " -e sh ./scripts/haxor-news.sh"))        -- Keypad 7
        , ("M-<KP_Up>",         spawn (myTerminal ++ " -e rainbowstream"))           -- Keypad 8
        , ("M-<KP_Page_Up>",    spawn (myTerminal ++ " -e sh ./scripts/tig-script.sh"))      -- Keypad 9
        
    -- Command Line Apps  (MOD + SHIFT + KEYPAD 1-9)
        , ("M-S-<KP_End>",        spawn (myTerminal ++ " -e ranger"))                -- Keypad 1
        , ("M-S-<KP_Down>",       spawn (myTerminal ++ " -e irssi"))                  -- Keypad 2
        , ("M-S-<KP_Page_Down>",  spawn (myTerminal ++ " -e mocp"))                  -- Keypad 3
        , ("M-S-<KP_Left>",       spawn (myTerminal ++ " -e ''"))                 -- Keypad 4
        , ("M-S-<KP_Begin>",      spawn (myTerminal ++ " -e rtorrent"))              -- Keypad 5
        , ("M-S-<KP_Right>",      spawn (myTerminal ++ " -e youtube-viewer"))        -- Keypad 6
        , ("M-S-<KP_Home>",       spawn (myTerminal ++ " -e alsamixer"))             -- Keypad 7
        , ("M-S-<KP_Up>",         spawn (myTerminal ++ " -e calcurse"))              -- Keypad 8
        , ("M-S-<KP_Page_Up>",    spawn (myTerminal ++ " -e vim /home/dt/.xmonad/xmonad.hs")) -- Keypad 9
        
    -- Command Line Apps  (MOD + CTRL + KEYPAD 1-9)
        , ("M-C-<KP_End>",        spawn (myTerminal ++ " -e htop")) -- Keypad 1
        , ("M-C-<KP_Down>",       spawn (myTerminal ++ " -e glances"))  -- Keypad 2
        , ("M-C-<KP_Page_Down>",  spawn (myTerminal ++ " -e nmon"))                -- Keypad 3
        , ("M-C-<KP_Left>",       spawn (myTerminal ++ " -e httping -KY --draw-phase localhost"))                     -- Keypad 4
        , ("M-C-<KP_Begin>",      spawn (myTerminal ++ " -e s-tui"))                 -- Keypad 5
        , ("M-C-<KP_Right>",      spawn (myTerminal ++ " -e ''"))           -- Keypad 6
        , ("M-C-<KP_Home>",       spawn (myTerminal ++ " -e ''"))        -- Keypad 7
        , ("M-C-<KP_Up>",         spawn (myTerminal ++ " -e ''"))           -- Keypad 8
        , ("M-C-<KP_Page_Up>",    spawn (myTerminal ++ " -e wopr report.xml"))      -- Keypad 9
        
    -- GUI Apps
        , ("M-w",               spawn "firefox")
        , ("M-f",               spawn "pcmanfm")
        , ("M-g",               runOrRaise "geany" (resource =? "geany"))

    -- Multimedia Keys
        , ("<XF86AudioPlay>",   spawn "cmus toggle")
        , ("<XF86AudioPrev>",   spawn "cmus prev")
        , ("<XF86AudioNext>",   spawn "cmus next")
        -- , ("<XF86AudioMute>",   spawn "amixer set Master toggle")  -- Bug prevents it from toggling correctly in 12.04.
        , ("<XF86AudioLowerVolume>", spawn "amixer set Master 5%- unmute")
        , ("<XF86AudioRaiseVolume>", spawn "amixer set Master 5%+ unmute")
        , ("<XF86HomePage>",    spawn "firefox")
        , ("<XF86Search>",      safeSpawn "firefox" ["https://www.google.com/"])
        , ("<XF86Mail>",        runOrRaise "thunderbird" (resource =? "thunderbird"))
        , ("<XF86Calculator>",  runOrRaise "gcalctool" (resource =? "gcalctool"))
        , ("<XF86Eject>",       spawn "toggleeject")
        , ("<Print>",           spawn "scrotd 0")
        ] where nonNSP          = WSIs (return (\ws -> W.tag ws /= "NSP"))
                nonEmptyNonNSP  = WSIs (return (\ws -> isJust (W.stack ws) && W.tag ws /= "NSP"))
                

----------------------------------------------------------------------------------------------
---WORKSPACES
----------------------------------------------------------------------------------------------

xmobarEscape = concatMap doubleLts
  where
        doubleLts '<' = "<<"
        doubleLts x   = [x]
        
myWorkspaces :: [String]   
myWorkspaces = clickable . (map xmobarEscape) 
               $ [" dev", "www", "sys", "doc", "vbox", "chat", "media", "gfx"]
  where                                                                      
        clickable l = [ "<action=xdotool key super+" ++ show (n) ++ ">" ++ ws ++ "</action>" |
                      (i,ws) <- zip [1..8] l,                                        
                      let n = i ] 

myManageHook = placeHook (withGaps (20,12,12,12) (smart (0.5,0.5))) <+> insertPosition End Newer <+> floatNextHook <+>
        (composeAll . concat $
        [ [ resource  =? r --> doF (W.view "main" . W.shift "main")   | r <- myTermApps    ]
        , [ resource  =? r --> doF (W.view "web" . W.shift "web")   | r <- myWebApps     ]
        , [ resource  =? r --> doF (W.view "media" . W.shift "media") | r <- myMediaApps   ]
        , [ resource  =? r --> doF (W.view "syst" . W.shift "syst")   | r <- mySystApps    ]
        , [ resource  =? r --> doFloat                            | r <- myFloatApps   ]
        , [ className =? c --> ask >>= doF . W.sink               | c <- myUnfloatApps ]
        ]) <+> manageHook defaultConfig
        where
            myTermApps    = ["termite", "xterm", "htop", "irssi"]
            myWebApps     = ["firefox", "thunderbird"]
            myMediaApps   = ["vlc", "ncmpcpp", "weechat", "mplayer", "gimp"]
            mySystApps    = ["ranger", "pcmanfm", "geany", "nitrogen"]

            myFloatApps   = ["file-roller", "nitrogen"]
            myUnfloatApps = ["gimp"]


----------------------------------------------------------------------------------------------
---LAYOUTS
----------------------------------------------------------------------------------------------
myLayoutHook = avoidStruts $ mouseResize $ windowArrange $ T.toggleLayouts floatz $ 
               mkToggle (NBFULL ?? NOBORDERS ?? EOT) $ renamed [CutWordsLeft 4] $ maximize $
               minimize $ boringWindows $ spacing 0 $
               myDevLayout $
               myWebLayout $
               mySysLayout $
               myDocLayout $
               myVboxLayout $
               myChatLayout $
               myMediaLayout $
               myGfxLayout $
               myDefaultLayout

myDefaultLayout = grid ||| threeCol ||| threeRow ||| oneBig ||| noBorders monocle ||| space ||| floatz
myDevLayout = onWorkspace (myWorkspaces !! 0) $ grid  ||| threeCol ||| threeRow ||| oneBig ||| floatz
myWebLayout = onWorkspace (myWorkspaces !! 1) $ noBorders monocle ||| oneBig ||| threeCol ||| threeRow ||| grid
mySysLayout = onWorkspace (myWorkspaces !! 2) $ oneBig ||| threeCol ||| threeRow ||| grid
myDocLayout = onWorkspace (myWorkspaces !! 3) $ oneBig ||| threeCol ||| threeRow ||| grid
myVboxLayout = onWorkspace (myWorkspaces !! 4) $ noBorders monocle ||| oneBig ||| threeCol ||| threeRow ||| grid
myChatLayout = onWorkspace (myWorkspaces !! 5) $ grid  ||| threeCol ||| threeRow ||| oneBig ||| floatz
myMediaLayout = onWorkspace (myWorkspaces !! 6) $ noBorders monocle ||| oneBig ||| space ||| threeRow
myGfxLayout = onWorkspace (myWorkspaces !! 7) $ T.toggleLayouts gimp $ noBorders monocle ||| floatz ||| grid

grid            = renamed [Replace "grid"]         $ limitWindows 12 $ spacing 4 $ mkToggle (single MIRROR) $ Grid (16/10)
threeCol        = renamed [Replace "threeCol"]     $ limitWindows 3  $ ThreeCol 1 (3/100) (1/2) 
threeRow        = renamed [Replace "threeRow"]     $ limitWindows 3  $ Mirror $ mkToggle (single MIRROR) zoomRow
oneBig          = renamed [Replace "oneBig"]       $ limitWindows 6  $ Mirror $ mkToggle (single MIRROR) $ mkToggle (single REFLECTX) $ mkToggle (single REFLECTY) $ OneBig (5/9) (8/12)
monocle         = renamed [Replace "monocle"]      $ limitWindows 20   Full
space           = renamed [Replace "space"]        $ limitWindows 4  $ spacing 36 $ Mirror $ mkToggle (single MIRROR) $ mkToggle (single REFLECTX) $ mkToggle (single REFLECTY) $ OneBig (2/3) (2/3)
floatz          = renamed [Replace "floatz"]       $ limitWindows 20   simplestFloat
gimp            = renamed [Replace "gimp"]         $ limitWindows 5  $ withIM 0.11 (Role "gimp-toolbox") $ reflectHoriz $ withIM 0.15 (Role "gimp-dock") Full

