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
             ]