local alert = require('hs.alert')
local app = require('hs.application')
local eventtap = require('hs.eventtap')
local geometry = require('hs.geometry')
local hotkey = require('hs.hotkey')
local layout = require('hs.layout')
local screen = require('hs.screen')
local timer = require('hs.timer')
local win = require('hs.window')

-- Custom variables
--------------------

local hyper = { 'cmd', 'alt', 'shift', 'ctrl' }
local alt = { 'alt' }
local altShift = { 'alt', 'shift' }
local laptopMonitor = "Built-in Retina Display"
local mainMonitor = "DELL U3415W"
-- local mainMonitor = "DELL U3419W"

-- Custom positions and layouts where which apps can be on the screen
--------------------

local screenPositions = {
  left = geometry.rect(0, 0, 0.5, 1),
  leftTop = { x = 0, y = 0, w = 0.5, h = 0.5 },
  leftBottom = { x= 0, y = 0.5, w = 0.5, h = 0.5 },
  right = geometry.rect(0.5, 0, 0.5, 1),
  rightTop = { x = 0.5, y = 0, w = 0.5, h = 0.5 },
  rightBottom = { x= 0.5, y = 0.5, w = 0.5, h = 0.5 },
  top = geometry.rect(0, 0, 1, 0.5),
  bottom = geometry.rect(0, 0.5, 1, 0.5),
  center = geometry.rect(0.2, 0.2, 0.6, 0.6),
  max = geometry.rect(0.1, 0.1, 0.8, 0.8),
  full = geometry.rect(0, 0, 1, 1),
}

local layoutDouble = {
  { "Calendar", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Firefox", nil, mainMonitor, screenPositions.left, nil, nil },
  { "ForkLift", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Mail", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Messages", nil, mainMonitor, screenPositions.rightTop, nil, nil },
  { "Microsoft Teams (work or school)", nil, mainMonitor, screenPositions.rightBottom, nil, nil },
  { "Reminders", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Signal", nil, mainMonitor, screenPositions.rightBottom, nil, nil },
  { "Strongbox", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "TIDAL", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Telegram", nil, mainMonitor, screenPositions.rightTop, nil, nil },
  { "WezTerm", nil, mainMonitor, screenPositions.right, nil, nil },
}

local layoutSingle = {
  { "Calendar", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Firefox", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "ForkLift", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Mail", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Messages", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Microsoft Teams (work or school)", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Reminders", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Signal", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Strongbox", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "TIDAL", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "Telegram", nil, laptopMonitor, screenPositions.full, nil, nil },
  { "WezTerm", nil, laptopMonitor, screenPositions.full, nil, nil },
}

local appsToLaunch = {
  "Calendar",
  "Firefox",
  "ForkLift",
  "Mail",
  "Messages",
  "Microsoft Teams (work or school)",
  "Signal",
  "Strongbox",
  "TIDAL",
  "WezTerm",
}

-- Local helper functions
--------------------

local function tablelength(T)
  local count = 0
  for _ in pairs(T) do count = count + 1 end
  return count
end

local function launchApps()
  for i, appName in ipairs(appsToLaunch) do
    app.launchOrFocus(appName)
  end
end

local function moveMouse()
  local pt = hs.geometry.rectMidPoint(win.focusedWindow():frame())
  hs.mouse.absolutePosition(pt)
end

local function applyPosition(pos)
  local w = win.focusedWindow()
  local screenName = w:screen():name()
  local tempPos = screenPositions[pos]
  local tempLayout = { { app.frontmostApplication(), w, screenName, tempPos, nil, nil } }
  layout.apply(tempLayout)
end

-- Window management and general config
--------------------

win.animationDuration = 0
app.enableSpotlightForNameSearches(true)

-- Keybindings
--------------------

-- Applying main two layouts
hotkey.bind(hyper, 'q', function() layout.apply(layoutSingle) end)
hotkey.bind(hyper, 'w', function()
  if(tablelength(screen.allScreens()) == 2)
  then
    layout.apply(layoutDouble)
  else
    layout.apply(layoutSingle)
  end
end)

-- Direct app navigation
hotkey.bind(hyper, 'a', function() app.launchOrFocus('WezTerm') end)
hotkey.bind(hyper, 's', function() app.launchOrFocus('Firefox') end)
hotkey.bind(hyper, 'd', function() app.launchOrFocus('ForkLift') end)
hotkey.bind(hyper, 'g', function() launchApps() end)

-- Moving windows around / navigating windows
hotkey.bind(hyper, '[', function() win.focusedWindow():moveOneScreenNorth(); moveMouse() end)
hotkey.bind(hyper, ']', function() win.focusedWindow():moveOneScreenSouth(); moveMouse() end)
hotkey.bind(hyper, 'z', function() applyPosition('full'); moveMouse() end)
hotkey.bind(hyper, 'x', function() applyPosition('max'); moveMouse() end)
hotkey.bind(hyper, 'c', function() applyPosition('center'); moveMouse() end)
hotkey.bind(hyper, 'h', function() applyPosition('left'); moveMouse() end)
hotkey.bind(hyper, 'u', function() applyPosition('leftTop'); moveMouse() end)
hotkey.bind(hyper, 'n', function() applyPosition('leftBottom'); moveMouse() end)
hotkey.bind(hyper, 'l', function() applyPosition('right'); moveMouse() end)
hotkey.bind(hyper, 'i', function() applyPosition('rightTop'); moveMouse() end)
hotkey.bind(hyper, 'm', function() applyPosition('rightBottom'); moveMouse() end)
hotkey.bind(hyper, 'k', function() applyPosition('top'); moveMouse() end)
hotkey.bind(hyper, 'j', function() applyPosition('bottom'); moveMouse() end)

-- map hyper + number to the corresponding fn-key, since the touchbar
-- kinda sucks, and karabiner-elements is breaking fn-function to show keys
hotkey.bind(hyper, '1', function() eventtap.keyStroke({}, 'F1') end)
hotkey.bind(hyper, '2', function() eventtap.keyStroke({}, 'F2') end)
hotkey.bind(hyper, '3', function() eventtap.keyStroke({}, 'F3') end)
hotkey.bind(hyper, '4', function() eventtap.keyStroke({}, 'F4') end)
hotkey.bind(hyper, '5', function() eventtap.keyStroke({}, 'F5') end)
hotkey.bind(hyper, '6', function() eventtap.keyStroke({}, 'F6') end)
hotkey.bind(hyper, '7', function() eventtap.keyStroke({}, 'F7') end)
hotkey.bind(hyper, '8', function() eventtap.keyStroke({}, 'F8') end)
hotkey.bind(hyper, '9', function() eventtap.keyStroke({}, 'F9') end)
hotkey.bind(hyper, '0', function() eventtap.keyStroke({}, 'F10') end)
hotkey.bind(hyper, '-', function() eventtap.keyStroke({}, 'F11') end)
hotkey.bind(hyper, '=', function() eventtap.keyStroke({}, 'F12') end)

-- macos keyboard input source switching is still broken AF, so we map
-- umlauts and others manually with hammerspoon, instead of using US-german-layout
-- dunno why I had to add a space beföre ´äÄ, but this way it works
hotkey.bind(alt, 'a', function() eventtap.keyStrokes(' ä') end)
hotkey.bind(alt, 'o', function() eventtap.keyStrokes('ö') end)
hotkey.bind(alt, 's', function() eventtap.keyStrokes('ß') end)
hotkey.bind(alt, 'u', function() eventtap.keyStrokes('ü') end)
hotkey.bind(altShift, 'a', function() eventtap.keyStrokes(' Ä') end)
hotkey.bind(altShift, 'o', function() eventtap.keyStrokes('Ö') end)
hotkey.bind(altShift, 'u', function() eventtap.keyStrokes('Ü') end)