import XMonad
import XMonad.Actions.CycleWS
import XMonad.Hooks.DynamicLog
import XMonad.Hooks.ManageDocks
import XMonad.Hooks.ManageHelpers
import XMonad.Hooks.SetWMName
import XMonad.Util.Run(spawnPipe)
import XMonad.Util.EZConfig
import XMonad.Util.Scratchpad
import XMonad.Config.Desktop
import XMonad.Config.Azerty
import XMonad.Layout.Fullscreen
import XMonad.Layout.NoBorders
import XMonad.Layout.Maximize
import XMonad.Layout.Tabbed
import XMonad.Layout.Minimize
import XMonad.Layout.BoringWindows
import XMonad.Layout.PerWorkspace
import XMonad.Prompt
import XMonad.Prompt.Ssh
import System.IO
import qualified Data.Map as M
import qualified XMonad.StackSet as W
import qualified Data.List as L

baseConfig = desktopConfig
    { modMask = mod5Mask -- AltGr
    , startupHook = startupHook desktopConfig >> setWMName "LG3D"
    , terminal = "mate-terminal.wrapper"
    , workspaces = ["1:web", "2:com", "3:dev"] ++ map show [4..7] ++ ["8:media", "9:div"]
    , layoutHook = smartBorders $ fullscreenFull $ avoidStruts $ boringWindows $ minimize $ maximize $ onWorkspace "8:media" Full $ onWorkspaces ["2:com", "9:div"] simpleTabbed $ (Tall 1 0.03 0.5 ||| Full ||| simpleTabbed )
    , handleEventHook = handleEventHook desktopConfig <+> fullscreenEventHook
    }
    `removeKeys`
    [(m .|. modMask baseConfig , k)
        | (k) <- [xK_1 .. xK_9]
        , (m) <- [(0), (shiftMask)]]
--    `removeKeys`
--    [(modMask baseConfig .|. shiftMask, xK_q)]


baseXPConfig = defaultXPConfig {
    position = Top
}

workspaceManageHook :: ManageHook
workspaceManageHook = composeAll [
     className =? "Firefox" --> doShift "1:web"
    ,className =? "Thunderbird" --> doShift "2:com"
    ,className =? "Gajim" --> doShift "2:com"
    ,className =? "jetbrains-phpstorm" --> doShift "3:dev"
    ,className =? "jetbrains-pycharm" --> doShift "3:dev"
    ,className =? "jetbrains-clion" --> doShift "3:dev"
    ,className =? "code" --> doShift "3:dev"
    ,className =? "qtcreator" --> doShift "3:dev"
    ,className =? "KeePass2" --> doShift "9:div"
    ,className =? "Xmessage" --> doFloat
    ,className =? "Clementine" --> doShift "8:media"
    ,className =? "vlc" --> doShift "8:media"
    ]

xmobarAction :: String  -- action
             -> String -- output string
             -> String
xmobarAction action = wrap t "</action>"
    where t = concat ["<action=`", action, "`>"]

xmobarSwitchWs :: String -> String
xmobarSwitchWs wsName = xmobarAction action wsName
    where action = concat ["xdotool key ISO_Level3_Shift+F", workspace]
          workspace = takeWhile (/=':') wsName -- Takes all chars until ':'

main = do
    xmobar_proc <- spawnPipe "xmobar"
    xmonad $ baseConfig
        { manageHook = (scratchpadManageHook $ W.RationalRect 0.0 0.0 1.0 0.5) <+> workspaceManageHook <+> fullscreenManageHook <+> manageDocks
        , logHook =  dynamicLogWithPP xmobarPP
                        { ppOutput = hPutStrLn xmobar_proc
                        , ppHidden = xmobarSwitchWs
                        , ppTitle = xmobarColor "green" "" . shorten 90
                        , ppSort = fmap(.scratchpadFilterOutWorkspace) (ppSort xmobarPP)
                        , ppOrder = \(ws:_:t:_) -> [ ws, t]
                        }
        }
        `additionalKeys`
        [ ((modMask baseConfig .|. controlMask, xK_l), spawn "xscreensaver-command -lock")
        , ((modMask baseConfig .|. controlMask, xK_s), spawn "systemctl hibernate")
       --  , ((modMask baseConfig, xK_r), spawn "~/.xmonad/action-manager/command.sh redshift ~/.xmonad/actioncontrol")
       -- , ((modMask baseConfig, xK_e), spawn "caja")
        , ((0, xK_Print), spawn "scrot")
        , ((mod1Mask, xK_Print), spawn "scrot -s")
        -- Normal Alt-Tab behavior
        , ((mod1Mask,               xK_Tab   ), windows W.focusDown) -- %! Move focus to the next window
        , ((mod1Mask .|. shiftMask, xK_Tab   ), windows W.focusUp  ) -- %! Move focus to the previous window
        -- XMonad.Actions.CycleWS
        , ((modMask baseConfig, xK_Right), nextWS)
        , ((modMask baseConfig, xK_Left), prevWS)
        , ((modMask baseConfig .|. shiftMask, xK_Right), shiftToNext >> nextWS)
        , ((modMask baseConfig .|. shiftMask, xK_Left), shiftToPrev >> prevWS)
        -- XMonad.Layout.Maximize
        , ((modMask baseConfig, xK_F11), withFocused (sendMessage . maximizeRestore))
        -- XMonad.Prompt.Ssh
        , ((modMask baseConfig, xK_s), sshPrompt baseXPConfig)
        -- XMonad.Layout.Minimize
        , ((modMask baseConfig, xK_c), withFocused minimizeWindow)
        , ((modMask baseConfig, xK_g), sendMessage RestoreNextMinimizedWin)
        -- XMonad.Layout.BoringWindows
        , ((modMask baseConfig, xK_j), focusUp)
        , ((modMask baseConfig, xK_k), focusDown)
        , ((modMask baseConfig, xK_m), focusMaster)
        -- XMonad.Util.Scratchpad
        , ((modMask baseConfig, xK_F12), scratchpadSpawnActionCustom "mate-terminal --disable-factory --name scratchpad")
        -- Azerty support
        , ((modMask baseConfig, xK_semicolon), sendMessage (IncMasterN (-1)))
        -- mute button
        , ((0, 0x1008FF12), spawn "~/.xmonad/action-manager/command.sh mt ~/.xmonad/actioncontrol")
        -- volumeup button
        , ((0, 0x1008FF13), spawn "~/.xmonad/action-manager/command.sh + ~/.xmonad/actioncontrol")
        -- volumedown button
        , ((0, 0x1008FF11), spawn "~/.xmonad/action-manager/command.sh - ~/.xmonad/actioncontrol")
        -- Media buttons
        , ((0, 0x1008ff14), spawn "clementine --play-pause")
        , ((0, 0x1008ff15), spawn "clementine --stop")
        , ((0, 0x1008ff16), spawn "clementine --restart-or-previous")
        , ((0, 0x1008ff17), spawn "clementine --next")
	-- Brightness buttons
	, ((0, 0x1008ff02), spawn "~/.xmonad/brightness-adjust +")
	, ((0, 0x1008ff03), spawn "~/.xmonad/brightness-adjust -")
        ]
        `additionalKeys`
        --
        -- mod-ctrl-[F1..F9], Switch to workspace N
        -- mod-ctrl-shift-[F1..F9], Move client to workspace N
        --
        [((m .|. modMask baseConfig , k), windows $ f i)
            | (i, k) <- zip (workspaces baseConfig) [xK_F1 .. xK_F9]
            , (f, m) <- [(W.greedyView, 0), (W.shift, shiftMask)]]