import XMonad

import Control.Monad.Trans.Class
import Control.Monad.Reader
import XMonad.Hooks.ManageDocks (docks)
import System.Directory (getHomeDirectory)
import System.FilePath ((</>))
import XMonad.Hooks.EwmhDesktops (ewmh)
import XMonad.Hooks.ManageHelpers (isFullscreen, doFullFloat)
import XMonad.Layout.Fullscreen (fullscreenEventHook)
import System.Environment (setEnv)
import Data.Monoid
import qualified Data.Map as Map

import Internal.Swallow
import Internal.Windows
import Internal.XMobarLog
import Internal.Keys
import Internal.Layout
import Internal.Logger
import Internal.DMenu (menuCommandString)
import Internal.RebindKeys
import Internal.KeysM
import Internal.ButtonMasks

import qualified XMonad as X
import qualified XMonad.StackSet as W

main = do

  -- Execute some commands.
  homeDir <- getHomeDirectory
  let fp = homeDir </> ".xmonad" </> "startup"

  setEnv "SUDO_ASKPASS" "/usr/bin/ssh-askpass"
  setEnv "ROFI" menuCommandString

  xmobar <- spawnXMobar

  (=<<) X.xmonad $
    applyKeys $ withButtonMasks $ ewmh $ docks $  def
       { terminal    = "alacritty"
       , modMask     = mod3Mask
       , borderWidth = 2
       , keys = const mempty
       , focusedBorderColor = "#ff6c00"
       , normalBorderColor = "#404040"
       , layoutHook = myLayout
       , startupHook = spawn fp
       , manageHook = composeAll [
           isFullscreen --> doFullFloat
         , className =? "Tilda" --> doFloat
         , className =? "yakuake" --> doFloat
         , className =? "MPlayer" --> doFloat
         , title =? "Event Tester" --> doFloat
         , title =? "Floating Term" --> doCenterFloat
         , title =? "Notes" --> doCenterFloat
         , title =? "xmessage" --> doCenterFloat
         , title =? "gxmessage" --> doCenterFloat
         , title =? "Volume Control" --> doCenterFloat
         , className =? "mpv" --> doFloat
         , className =? "gnubby_ssh_prompt" --> doFloat
         ]
         -- This config uses dynamic workspaces, but I have to seed XMonad
         -- with something. However, this configuration only supports 36
         -- monitors on boot. If you need more than 36 monitors, you'll have to
         -- configure those ones after starting XMonad.
       , workspaces = map return (['0'..'9'] ++ ['a'..'z'])
       , handleEventHook =
           composeAll [
             fullscreenEventHook,
             remapHook,
             swallowHook]
       , focusFollowsMouse = False
       , clickJustFocuses = False
       , logHook = xMobarLogHook xmobar
       }


doCenterFloat :: ManageHook
doCenterFloat =
  ask >>= \w -> doF . W.float w . centerRect . snd =<< liftX (floatLocation w)

centerRect :: W.RationalRect -> W.RationalRect
centerRect (W.RationalRect x y w h) = W.RationalRect ((1 - w) / 2) ((1 - h) / 2) w h


windowHooks :: WindowHook -> XConfig l -> XConfig l
windowHooks (Query readerT) config = do

  config {
    startupHook = do
      forAllWindows $ \w -> runReaderT readerT w
      startupHook config,

    manageHook = mappend (Query readerT >> return (Endo id)) (manageHook config)
  }