import Control.Monad
 
 import Data.List
 import Data.Ratio ((%))
 
 import System.IO
 
 import XMonad hiding ( (|||) )
 import XMonad.Actions.CopyWindow(copy,copyWindow,kill1,killAllOtherCopies)
 import XMonad.Actions.DynamicWorkspaces
 import XMonad.Actions.GridSelect
 import XMonad.Actions.SpawnOn
 import XMonad.Core
 import XMonad.Hooks.DynamicLog
 import XMonad.Hooks.ManageDocks
 import XMonad.Layout.Accordion
 import XMonad.Layout.Circle
 import XMonad.Layout.Combo
 import XMonad.Layout.DragPane
 import XMonad.Layout.SubLayouts
 import XMonad.Layout.Gaps
 import XMonad.Layout.Grid
 import XMonad.Layout.IM
 import XMonad.Layout.LayoutCombinators
 import XMonad.Layout.BoringWindows hiding (Replace)
 import XMonad.Layout.NoBorders
 import XMonad.Layout.OneBig
 import XMonad.Layout.PerWorkspace
 import XMonad.Layout.Reflect
 import XMonad.Layout.Renamed
 import XMonad.Layout.ResizableTile
 import XMonad.Layout.Simplest
 import XMonad.Layout.Spiral
 import XMonad.Layout.Tabbed
 import XMonad.Layout.ThreeColumns
 import XMonad.Layout.TwoPane
 import XMonad.Layout.WindowNavigation
 import XMonad.Prompt
 import XMonad.StackSet as W
 import XMonad.StackSet hiding ( workspaces )
 import XMonad.Util.EZConfig(additionalKeys)
 import XMonad.Util.Loggers
 import XMonad.Util.Run(spawnPipe)
 
 nmaster = 1
 ratio = 13/21
 delta = 3/100
 tiled = Tall nmaster delta ratio
 myTabbed = renamed [Replace "Tabbed"] $ tabbedBottom shrinkText defaultTheme
 threeLayout = ThreeColMid nmaster delta ratio
 writingLayout = renamed [Replace "Writing"] $ gaps [(L,500), (R, 500)] Full
 
 imLayout = renamed [Replace "im"] $ withIM ratio empathyRoster Accordion where
     ratio           = 1%6
     empathyRoster   = And (ClassName "Empathy") (Role "contact_list")
 
 
 base =     mthree ||| wide ||| Full ||| Accordion ||| Circle ||| spiral (6/7)
        ||| myTabbed ||| wideAccordion ||| writingLayout ||| rows ||| grid
        ||| (TwoPane (3/100) (1/2)) ||| (OneBig (3/4) (3/4))
    where
       mthree = renamed [Replace "ThreeWide"] $ Mirror threeLayout
       wide = renamed [Replace "Wide"] $ Mirror tiled
       wideAccordion = renamed [Replace "WideAccordion"] $ Mirror Accordion
       grid = renamed [Replace "Gridding"] $ GridRatio (1/2)
       rows = renamed [Replace "WritingNew"] $ dragPane Vertical 1 0.5
 
 
 
 myLayout = avoidStruts $ smartBorders $
    onWorkspace "web" myTabbed $
    onWorkspace "terminal" tiled $
    onWorkspace "IM" imLayout $
    onWorkspace "images" (gimpLayout ||| tiled ||| threeLayout ||| base) $
    tiled ||| threeLayout ||| base
  where
    gimpLayout = renamed [Replace "gimp"] $ withIM (11/64) (Role "gimp-toolbox") $ ResizableTall 2 (1/118) (11/20) [1]
 
 chooseLayout :: GSConfig String -> X ()
 chooseLayout conf = do
    a <- gridselect conf $ makeLayoutList ["Tall", "ThreeCol", "Accordion", "Full", "Tabbed", "Spiral", "Wide", "ThreeWide", "WideAccordion", "Writing",
                                           "WritingNew", "Gridding", "TwoPane", "OneBig"]
    changeLayout a
    return ()
       where
          makeLayoutList [] = []
          makeLayoutList (l:ls) = (l,l):(makeLayoutList ls)
          changeLayout a =
             case a of
                Just x -> sendMessage $ JumpToLayout x
 
 myPP = sjanssenPP {
    ppCurrent = xmobarColor "grey" "white",
    ppHidden = xmobarColor "red" "black",
    ppHiddenNoWindows = id,
    ppTitle = xmobarColor "green" "" . shorten 110
 }
 
 myManageHook = composeAll
    [
    (role =? "gimp-toolbox" <||> role =? "gimp-image-winow") --> (ask >>= doF . W.sink)
    , className =? "X-terminal-emulator"      --> doCopy "terminal"
    , className =? "Xmessage"  --> doFloat
    , className =? "Qasmixer" --> doFloat
    , className =? "Sonata" --> doFloat
    , className =? "feh" --> viewShift "images"
    , className =? "Display.im6" --> viewShift "images"
    , manageDocks
    ]
  where role = stringProperty "WM_WINDOW_ROLE"
 
 
 doCopy :: WorkspaceId -> ManageHook
 doCopy i = doF . copyWin i =<< ask
 copyWin i a = copyWindow a i
 
 viewShift = doF . liftM2 (.) W.greedyView W.shift
 
 copyNSwitch windows target = do
   windows $ copy target
   windows $ W.greedyView target
 
 shiftNSwitch windows target = do
   windows $ shift target
   windows $ W.greedyView target
 
 main = do
    xmproc <- spawnPipe "/home/edwlan/.cabal/bin/xmobar /home/edwlan/.xmobarrc"
    xmonad $ defaultConfig
       {
          manageHook = myManageHook <+> manageSpawn <+> manageHook defaultConfig,
          layoutHook = myLayout,
          logHook = dynamicLogWithPP myPP {
             ppOutput = hPutStrLn xmproc
          },
          modMask = mod4Mask,
          focusFollowsMouse = False,
          XMonad.workspaces = ["|", "web", "terminal", "1", "2", "3", "4", "5", "6", "images", "IM"]
       } `additionalKeys` (
       [
          (((mod4Mask .|. controlMask, xK_q     ),
                spawn "if type xmonad; then xmonad --recompile && xmonad --restart; else xmessage xmonad not in \\$PATH: \"$PATH\"; fi")),
 
          ((mod4Mask .|. controlMask .|. shiftMask, xK_h ), sendMessage $ Move L),
          ((mod4Mask .|. controlMask .|. shiftMask, xK_j ), sendMessage $ Move D),
          ((mod4Mask .|. controlMask .|. shiftMask, xK_k   ), sendMessage $ Move U),
          ((mod4Mask .|. controlMask .|. shiftMask, xK_l), sendMessage $ Move R),
          ((mod4Mask .|. controlMask, xK_k), killAllOtherCopies),
          ((mod4Mask .|. controlMask, xK_m), withWorkspace defaultXPConfig (windows . shift)),
 
          ((mod4Mask, numPadKeys !! 1), windows $ W.greedyView "web" ),
          ((mod4Mask, numPadKeys !! 2), windows $ W.greedyView "terminal" ),
          ((mod4Mask, numPadKeys !! 3), windows $ W.greedyView "1" ),
          ((mod4Mask, numPadKeys !! 4), windows $ W.greedyView "2" ),
          ((mod4Mask, numPadKeys !! 5), windows $ W.greedyView "3" ),
          ((mod4Mask, numPadKeys !! 6), windows $ W.greedyView "4" ),
          ((mod4Mask, numPadKeys !! 7), windows $ W.greedyView "5" ),
          ((mod4Mask, numPadKeys !! 8), windows $ W.greedyView "6" ),
          ((mod4Mask, numPadKeys !! 9), windows $ W.greedyView "images" ),
 
          ((mod4Mask .|. shiftMask, numPadKeys !! 1), copyNSwitch windows "web" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 2), copyNSwitch windows "terminal" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 3), copyNSwitch windows "1" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 4), copyNSwitch windows "2" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 5), copyNSwitch windows "3" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 6), copyNSwitch windows "4" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 7), copyNSwitch windows "5" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 8), copyNSwitch windows "6" ),
          ((mod4Mask .|. shiftMask, numPadKeys !! 9), copyNSwitch windows "images" ),
 
          ((mod4Mask .|. controlMask, numPadKeys !! 1), shiftNSwitch windows "web" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 2), shiftNSwitch windows "terminal" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 3), shiftNSwitch windows "1" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 4), shiftNSwitch windows "2" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 5), shiftNSwitch windows "3" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 6), shiftNSwitch windows "4" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 7), shiftNSwitch windows "5" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 8), shiftNSwitch windows "6" ),
          ((mod4Mask .|. controlMask, numPadKeys !! 9), shiftNSwitch windows "images" ),
 
          ((mod4Mask .|. shiftMask, xK_1), copyNSwitch windows "web" ),
          ((mod4Mask .|. shiftMask, xK_2), copyNSwitch windows "terminal" ),
          ((mod4Mask .|. shiftMask, xK_3), copyNSwitch windows "1" ),
          ((mod4Mask .|. shiftMask, xK_4), copyNSwitch windows "2" ),
          ((mod4Mask .|. shiftMask, xK_5), copyNSwitch windows "3" ),
          ((mod4Mask .|. shiftMask, xK_6), copyNSwitch windows "4" ),
          ((mod4Mask .|. shiftMask, xK_7), copyNSwitch windows "5" ),
          ((mod4Mask .|. shiftMask, xK_8), copyNSwitch windows "6" ),
          ((mod4Mask .|. shiftMask, xK_9), copyNSwitch windows "images" ),
 
          ((mod4Mask .|. controlMask, xK_1), shiftNSwitch windows "web" ),
          ((mod4Mask .|. controlMask, xK_2), shiftNSwitch windows "terminal" ),
          ((mod4Mask .|. controlMask, xK_3), shiftNSwitch windows "1" ),
          ((mod4Mask .|. controlMask, xK_4), shiftNSwitch windows "2" ),
          ((mod4Mask .|. controlMask, xK_5), shiftNSwitch windows "3" ),
          ((mod4Mask .|. controlMask, xK_6), shiftNSwitch windows "4" ),
          ((mod4Mask .|. controlMask, xK_7), shiftNSwitch windows "5" ),
          ((mod4Mask .|. controlMask, xK_8), shiftNSwitch windows "6" ),
          ((mod4Mask .|. controlMask, xK_9), shiftNSwitch windows "images" ),
 
          ((mod4Mask, xK_1), windows $ W.greedyView "web" ),
          ((mod4Mask, xK_2), windows $ W.greedyView "terminal" ),
          ((mod4Mask, xK_3), windows $ W.greedyView "1" ),
          ((mod4Mask, xK_4), windows $ W.greedyView "2" ),
          ((mod4Mask, xK_5), windows $ W.greedyView "3" ),
          ((mod4Mask, xK_6), windows $ W.greedyView "4" ),
          ((mod4Mask, xK_7), windows $ W.greedyView "5" ),
          ((mod4Mask, xK_8), windows $ W.greedyView "6" ),
          ((mod4Mask, xK_9), windows $ W.greedyView "images" ),
          ((mod4Mask, xK_grave), windows $ W.greedyView "IM" ),
 
          ((mod4Mask .|. shiftMask, xK_BackSpace), removeWorkspace),
          ((mod4Mask .|. shiftMask, xK_k), kill1),
          ((mod4Mask .|. shiftMask, xK_m), withWorkspace defaultXPConfig (windows . copy)),
          ((mod4Mask .|. shiftMask, xK_n), addWorkspacePrompt defaultXPConfig),
          ((mod4Mask .|. shiftMask, xK_Return), spawnHere "/usr/bin/x-terminal-emulator"),
          ((mod4Mask .|. shiftMask, xK_r), renameWorkspace defaultXPConfig),
          ((mod4Mask .|. shiftMask, xK_w), gridselectWorkspace defaultGSConfig (\ws -> greedyView ws . shift ws)),
          ((mod4Mask, xK_b), sendMessage ToggleStruts),
          ((mod4Mask, xK_g), goToSelected defaultGSConfig),
          ((mod4Mask, xK_KP_Subtract), spawn "/usr/bin/zsh /home/edwlan/bin/dmenu_queueplay_mpd"),
          ((mod4Mask, xK_KP_Multiply), spawn "/usr/bin/zsh /home/edwlan/bin/dmenu_queue_mpd"),
          ((mod4Mask, xK_KP_Divide), spawn "/usr/bin/zsh /home/edwlan/bin/dmenu_play_mpd"),
          ((mod4Mask, xK_KP_Add), spawn "/usr/bin/zsh /home/edwlan/bin/dzen_mpc_status"),
          ((mod4Mask, xK_p), spawnHere "/usr/bin/dmenu_run -f"),
 
          ((mod4Mask, xK_q), ((withSelectedWindow $ windows . W.focusWindow) defaultGSConfig) >> (windows $ W.shiftMaster)),
          ((mod4Mask, xK_semicolon), chooseLayout defaultGSConfig),
          ((mod4Mask, xK_w), gridselectWorkspace defaultGSConfig (\ws -> greedyView ws))
       ] ++ zip (zip (repeat (mod4Mask)) ([xK_0])) (map (withNthWorkspace greedyView) [0..]) ++
          zip (zip (repeat (mod4Mask .|. shiftMask)) ([xK_0])) (map (withNthWorkspace copy) [0..]) 
          ++ zip (zip (repeat (mod4Mask)) (map (numPadKeys !!) ([0]))) (map (withNthWorkspace greedyView) [0..])
       )
 
 numPadKeys = [xK_KP_Insert -- 0
              , xK_KP_End,  xK_KP_Down,  xK_KP_Page_Down -- 1, 2, 3
              , xK_KP_Left, xK_KP_Begin, xK_KP_Right     -- 4, 5, 6
              , xK_KP_Home, xK_KP_Up,    xK_KP_Page_Up   -- 7, 8, 9
              ]