-- luacheck: read_globals hs

local wm = {
  grid = require('ext.grid'),

  -- configuration
  animationDuration = 0.0,
  gridSizes = { default = '30x20', interactive = '8x4' },
  gridTextSize = 50,
  margins = { w = 4, h = 4 }
}

function wm:init ()
  -- setup
  local bind = require('hs.hotkey').bind

  hs.window.animationDuration = self.animationDuration
  self.grid.setGrid(self.gridSizes.default)
  self.grid.setMargins(self.margins)
  self.grid.ui.textSize = self.gridTextSize

  --
  -- resize to grid
  --

  -- show interactive grid menu
  bind({"cmd", "ctrl"}, "2",
    function()
      self.grid.setGrid(self.gridSizes.interactive)
      self.grid.show(
        function()
          self.grid.setGrid(self.gridSizes.default)
        end
      )
    end
  )

  -- left half
  bind({"cmd", "ctrl"}, "J", self.adjustWindow(0, 0, 15, 20))
  -- right half
  bind({"cmd", "ctrl"}, "L", self.adjustWindow(15, 0, 15, 20))
  -- top half
  bind({"cmd", "ctrl"}, "I", self.adjustWindow(0, 0, 30, 10))
  -- bottom half
  bind({"cmd", "ctrl"}, "K", self.adjustWindow(0, 10, 30, 10))

  -- left narrow
  bind({"ctrl", "alt"}, "U", self.adjustWindow(0, 0, 12, 20))
  -- right narrow
  bind({"ctrl", "alt"}, "O", self.adjustWindow(18, 0, 12, 20))

  -- left wide
  bind({"cmd", "ctrl"}, "U", self.adjustWindow(0, 0, 18, 20))
  -- right wide
  bind({"cmd", "ctrl"}, "O", self.adjustWindow(12, 0, 18, 20))

  -- left fat
  bind({"ctrl", "alt"}, "J", self.adjustWindow(0, 0, 21, 20))
  -- right wide
  bind({"ctrl", "alt"}, "L", self.adjustWindow(9, 0, 21, 20))
  -- top fat
  bind({"ctrl", "alt"}, "I", self.adjustWindow(0, 0, 30, 14))
  -- bottom wide
  bind({"ctrl", "alt"}, "K", self.adjustWindow(0, 6, 30, 14))

  -- top left quarter
  bind({"cmd", "ctrl", "shift"}, "J", self.adjustWindow(0, 0, 15, 10))
  -- top right quarter
  bind({"cmd", "ctrl", "shift"}, "I", self.adjustWindow(15, 0, 15, 10))
  -- bottom right quarter
  bind({"cmd", "ctrl", "shift"}, "L", self.adjustWindow(15, 10, 15, 10))
  -- bottom left quarter
  bind({"cmd", "ctrl", "shift"}, "K", self.adjustWindow(0, 10, 15, 10))

  -- center super narrow
  bind({"cmd", "ctrl", "alt"}, "\\", self.adjustWindow(10, 0, 10, 20))
  -- center narrow small
  bind({"ctrl", "alt"}, "\\", self.adjustWindow(9, 0, 12, 20))
  -- center narrow
  bind({"cmd", "ctrl"}, "\\", self.adjustWindow(7, 0, 16, 20))

  -- center medium small
  bind({"ctrl", "alt"}, "'", self.adjustWindow(6, 0, 18, 20))
  -- center medium
  bind({"cmd", "ctrl"}, "'", self.adjustWindow(5, 0, 20, 20))

  -- center wide small
  bind({"ctrl", "alt"}, ";", self.adjustWindow(4, 0, 22, 20))
  -- center wide
  bind({"cmd", "ctrl"}, ";", self.adjustWindow(3, 0, 24, 20))

  -- maximized
  bind({"cmd", "ctrl"}, "H", self.grid.maximizeWindow)


  --
  -- move between displays
  --

  -- move to screen to the left
  bind({"cmd", "ctrl"}, ",",
    function()
      local win = hs.window.focusedWindow()
      win:moveOneScreenWest()
      self.grid.snap(win)
    end
  )

  -- move to screen to the right
  bind({"cmd", "ctrl"}, ".",
    function()
      local win = hs.window.focusedWindow()
      win:moveOneScreenEast()
      self.grid.snap(win)
    end
  )

  -- move to screen above
  bind({"cmd", "ctrl"}, "P",
    function()
      local win = hs.window.focusedWindow()
      win:moveOneScreenNorth()
      self.grid.snap(win)
    end
  )

  -- move to screen bellow
  bind({"cmd", "ctrl"}, "N",
    function()
      local win = hs.window.focusedWindow()
      win:moveOneScreenSouth()
      self.grid.snap(win)
    end
  )
end


--
-- private methods
--

wm.adjustWindow = function (x, y, w, h)
  return function()
    wm.grid.adjustWindow(
      function(cell)
        cell.x = x
        cell.y = y
        cell.w = w
        cell.h = h
      end
    )
  end
end


-- the end
return wm