import System.IO
import System.Exit

import XMonad

import qualified Data.Map        as M
import qualified XMonad.StackSet as W

import XMonad.Hooks.DynamicLog
import XMonad.Hooks.ManageDocks
import XMonad.Hooks.ManageHelpers
import XMonad.Hooks.EwmhDesktops

import XMonad.Config.Desktop

import XMonad.Util.Run

import XMonad.Layout.LayoutHints
import XMonad.Layout.Maximize
import XMonad.Layout.NoBorders
import XMonad.Layout.PerWorkspace
import XMonad.Layout.Reflect
import XMonad.Layout.ResizableTile
import XMonad.Layout.SimplestFloat
import XMonad.Layout.WindowNavigation

import XMonad.Actions.Navigation2D

--------------------------------------

-- Options
-- -------
myBorderWidth        = 2
myModMask            = mod1Mask
myNormalBorderColor = "#202020"
myFocusedBorderColor = "#819a67"
myTerminal           = "termite -e /home/jake/scripts/tmux/new-session.sh"
myWorkspaces         = ["tile","float","max", "xtra"]


-- Keybinds
-- --------
myKeys conf@(XConfig {XMonad.modMask = modm}) = M.fromList $

    [ ((modm .|. shiftMask, xK_Return), spawn $ XMonad.terminal conf)
    , ((modm,               xK_p     ), spawn "dmenu_run -p 'Run:'")
    , ((modm .|. shiftMask, xK_c     ), kill)
    , ((modm,               xK_space ), sendMessage NextLayout)
    , ((modm .|. shiftMask, xK_space ), setLayout $ XMonad.layoutHook conf)
    , ((modm,               xK_Tab   ), windows W.focusDown)
    , ((modm,               xK_j     ), windowGo D False )
    , ((modm,               xK_k     ), windowGo U False )
    , ((modm,               xK_l     ), windowGo R False )
    , ((modm,               xK_h     ), windowGo L False )
    , ((modm,               xK_Return), windows W.swapMaster)
    , ((modm .|. shiftMask, xK_j     ), windowSwap D False )
    , ((modm .|. shiftMask, xK_k     ), windowSwap U False )
    , ((modm .|. shiftMask, xK_l     ), windowSwap R False )
    , ((modm .|. shiftMask, xK_h     ), windowSwap L False )
    , ((modm .|. controlMask,  xK_h     ), sendMessage Shrink)
    , ((modm .|. controlMask,  xK_l     ), sendMessage Expand)
    , ((modm,               xK_t     ), withFocused $ windows . W.sink)
    , ((modm,               xK_m     ), withFocused (sendMessage . maximizeRestore))
    , ((modm,               xK_comma ), sendMessage (IncMasterN 1))
    , ((modm,               xK_period), sendMessage (IncMasterN (-1)))
    , ((modm .|. shiftMask, xK_q     ), io (exitWith ExitSuccess))
    , ((modm .|. shiftMask, xK_r     ), restart "xmonad" True)
    , ((modm,               xK_b     ), sendMessage ToggleStruts)
    , ((modm .|. controlMask, xK_j     ), sendMessage MirrorShrink)
    , ((modm .|. controlMask, xK_k     ), sendMessage MirrorExpand)

    -- Manual Movements
    -- ------ ---------
    , ((modm .|. controlMask, xK_Right), sendMessage $ Move R)
    , ((modm .|. controlMask, xK_Left ), sendMessage $ Move L)
    , ((modm .|. controlMask, xK_Up   ), sendMessage $ Move U)
    , ((modm .|. controlMask, xK_Down ), sendMessage $ Move D)

    -- App Shortcuts
    , ((mod4Mask .|. shiftMask, xK_Return), spawn "termite")

    , ((mod4Mask,   xK_w), spawn "termite -e /home/jake/scripts/weelaunch.sh")
    , ((mod4Mask .|. shiftMask,   xK_w), spawn "termite --profile=Writing -- vim -c Goyo -c Limelight")

    , ((modm, xK_F10), spawn "/home/jake/scripts/nkjv/nkjv-menu.sh")
    , ((modm, xK_F11), spawn "/home/jake/scripts/nkjv/nkjv-menu.sh md")
    , ((modm, xK_F12), spawn "/home/jake/scripts/nkjv/nkjv-menu.sh marp")
    ]

     ++
    [((m .|. modm, k), windows $ f i)
         | (i, k) <- zip (XMonad.workspaces conf) [xK_1 .. xK_9]
        , (f, m) <- [(W.greedyView, 0), (W.shift, shiftMask)]]

-- Multi monitor stuff?
    -- ++
    --[((m .|. modm, key), screenWorkspace sc >>= flip whenJust (windows . f))
        -- | (key, sc) <- zip [xK_w, xK_e, xK_r] [0..]
        --, (f, m) <- [(W.view, 0), (W.shift, shiftMask)]]


-- Mouse
-- -----
myMouseBindings (XConfig {XMonad.modMask = modMask}) = M.fromList $

    [ ((modMask, button1), (\w -> focus w >> mouseMoveWindow w))
    , ((modMask, button2), (\w -> focus w >> windows W.swapMaster))
    , ((modMask, button3), (\w -> focus w >> mouseResizeWindow w))
    ]

myFocusFollowsMouse :: Bool
myFocusFollowsMouse = True


-- Layouts
-- -------
myLayout = smartBorders $ maximize ( lays )
  where
     tiled   = ResizableTall 1 (1/100) (132/227) []
     tiled2  = ResizableTall 1 (1/100) (1/2) []
     w1 = onWorkspace "float" simplestFloat (tiled2)
     w2 = reflectHoriz tiled
     w3 = ResizableTall 2 (1/100) (55/100) []
     lays = onWorkspace "max" Full (w1) ||| Mirror w3 ||| tiled


-- Kludges
-- -------
myManageHook = composeAll
    [ className =? "feh"            --> doFloat
    , className =? "lxappearance"   --> doFloat
    , className =? "Xmessage"       --> doFloat
    , className =? "sxiv"           --> doFloat
    , className =? "Sxiv"           --> doFloat
    , resource  =? "desktop_window" --> doIgnore
    , isFullscreen                  --> doFullFloat ]
    where unfloat = ask >>= doF . W.sink


-- Log, Startup
-- -------
myLogHook = return ()
myStartupHook = do
      spawn "$HOME/.config/polybar/launch.sh xmonad"
      -- spawn "nitrogen --restore"


-- XMobar PP
-- ------ --
-- myPP :: PP
-- myPP = defaultPP
--        { ppCurrent = xmobarColor "#7a8f65" ""
--        , ppSep = xmobarColor "#acacab" "" " // "
--        , ppLayout = xmobarColor "#658f8f" ""
--        , ppUrgent = xmobarColor "8f6f65" ""
--        , ppTitle = xmobarColor "#acacab" "" . pad }


-- Get Ready
-- --- -----
myconfig = desktopConfig {
        terminal           = myTerminal,
        focusFollowsMouse  = myFocusFollowsMouse,
        borderWidth        = myBorderWidth,
        modMask            = myModMask,
        workspaces         = myWorkspaces,
        normalBorderColor  = myNormalBorderColor,
        focusedBorderColor = myFocusedBorderColor,

        keys               = myKeys,
        mouseBindings      = myMouseBindings,

        layoutHook         = desktopLayoutModifiers $ myLayout,
        manageHook         = myManageHook
            <+> manageDocks <+> manageHook desktopConfig,
        logHook            = myLogHook,
        handleEventHook    = handleEventHook desktopConfig <+> fullscreenEventHook,
        startupHook        = myStartupHook <+> startupHook desktopConfig
    }

-- Make it rain
-- ---- -- ----
main = do
       -- xmobar <- spawnPipe "xmobar"
       xmonad $ ewmh $ myconfig
              -- { logHook = dynamicLogWithPP $ myPP { ppOutput = hPutStrLn xmobar } }