init.lua

hs.application.enableSpotlightForNameSearches(true)

hs.hotkey.bind({"alt"}, "d", function() hs.reload() end)

hs.hotkey.bind({"alt","shift"}, "h", function() hs.hints.windowHints() end)
hs.hints.style = "vimperator"

function frame(fn)
  return function()
    local win = hs.window.focusedWindow()
    local f = win:frame()
    local screen = win:screen()
    local s = screen:frame()
    fn(f, s)
    win:setFrame(f)
  end
end

for i = 1,9 do
  hs.hotkey.bind(
      { "alt" }, tostring(i),
      frame(
          function(f, s)
            f.x = s.x + s.w * (10 - i) / 10
            f.w = s.w * i / 10
            f.y = s.y
            f.h = s.h
          end
      )
  )
  hs.hotkey.bind(
      { "alt", "shift" }, tostring(i),
      frame(
          function(f, s)
            f.x = s.x
            f.w = s.w * i / 10
            f.y = s.y
            f.h = s.h
          end
      )
  )
  hs.hotkey.bind(
      { "alt", "ctrl" }, tostring(i),
      frame(
          function(f, s)
            f.x = s.x
            f.w = s.w
            f.y = s.y
            f.h = s.h * i / 10
          end
      )
  )
  hs.hotkey.bind(
      { "alt", "ctrl", "shift" }, tostring(i),
      frame(
          function(f, s)
            f.x = s.x
            f.w = s.w
            f.y = s.y + s.h * (10 - i) / 10
            f.h = s.h * i / 10
          end
      )
  )
end

hs.hotkey.bind(
    {'alt'}, '0',
    frame(
        function(f, s)
          f.x = s.x
          f.w = s.w
          f.y = s.y
          f.h = s.h
        end
    )
)

M = 10
N = 10

k = hs.hotkey.modal.new('alt', 'a')
function k:entered() hs.alert 'Resize mode' end
function k:exited() hs.alert 'Exited resize mode' end
k:bind('alt', 'a', function() k:exit() end)
k:bind('', 'escape', function() k:exit() end)
k:bind('', 'J', 'Pressed J',function() print 'let the record show that J was pressed' end)

max = math.max
min = math.min

function bind(mods, keys, fn)
  for i,key in ipairs(keys) do
    k:bind(mods, key, frame(fn))
  end
end

function resizeBindings(base, far, k, sk, S)

  function extendBase(f, s)
    local cur = f[k]
    f[k] = max(s[k], f[k] - s[sk] / S)
    local increase = cur - f[k]
    f[sk] = f[sk] + increase
  end

  function shrinkBase(f, s)
    local cur = f[k]
    f[k] = min(s[k] + s[sk], f[k] + s[sk] / S)
    local decrease = f[k] - cur
    f[sk] = f[sk] - decrease
  end

  function throwBase(f, s)
    local cur = f[k]
    f[k] = s[k]
    local increase = cur - f[k]
    f[sk] = f[sk] + increase
  end

  function extendFar(f, s) f[sk] = min(s[sk] - (f[k] - s[k]), f[sk] + s[sk] / S) end
  function shrinkFar(f, s) f[sk] = max(0, f[sk] - s[sk] / S) end
  function throwFar(f, s) f[sk] = s[sk] - (f[k] - s[k]) end

  function nudgeBase(f, s) f[k] = max(s[k], f[k] - s[sk] / S) end
  function flushBase(f, s) f[k] = s[k] end

  function nudgeFar(f, s) f[k] = min(s[sk] - f[sk] + s[k], f[k] + s[sk] / S) end
  function flushFar(f, s) f[k] = s[sk] - f[sk] + s[k] end

  -- For the "base" (left or top) edge:
  -- - default: extend
  -- - shift: contract
  -- - ctrl: extend to edge
  bind('', base, extendBase)
  bind('shift', base, shrinkBase)
  bind('ctrl', base, throwBase)

  -- Same three bindings, but for the "far" edge (right or bottom)
  bind('', far, extendFar)
  bind('shift', far, shrinkFar)
  bind('ctrl', far, throwFar)

  -- cmd: "nudge" in the "base" direction
  -- cmd+ctrl: move all the way to "base" edge
  bind("cmd", base, nudgeBase)
  bind({"cmd", "ctrl"}, base, flushBase)

  -- Same two "move" bindings, but toward the "far" edge
  bind("cmd", far, nudgeFar)
  bind({"cmd", "ctrl"}, far, flushFar)
end

resizeBindings({ 'h', 'left' }, { 'l', 'right' }, 'x', 'w', M)
resizeBindings({ 'k',   'up' }, { 'j',  'down' }, 'y', 'h', N)

function map(tbl, f)
  local t = {}
  for k,v in pairs(tbl) do
    t[k] = f(v)
  end
  return t
end

function mapToArr(tbl, f)
  local t = {}
  local i = 1
  for k,v in pairs(tbl) do
    t[i] = f(k, v)
    i = i + 1
  end
  return t
end

function reduce(tbl, f)
  local t = nil
  for k,v in pairs(tbl) do
    if t == nil then
      t = v
    else
      t = f(t, v)
    end
  end
  return t
end

function foldLeft(tbl, z, f)
  local t = z
  for k,v in pairs(tbl) do
    t = f(t, v)
  end
  return t
end

function str(o)
  if type(o) == 'table' then
    return '{ '..table.concat(mapToArr(o, function(k, v) return k..': '..str(v) end), ', ')..' }'
  else
    return tostring(o)
  end
end

function bigsmall(big, sml)
  return {
  [   "Google Chrome" ] = { screen = big, shape = { x=0.2, y=0, w=0.6, h=1 } },
  [      "Superhuman" ] = { screen = big, shape = { x=0.2, y=0, w=0.6, h=1 } },
  [          "iTerm2" ] = { screen = verticalDell or big, shape = { x=0  , y=0, w=1  , h=1 } },
  [   "IntelliJ IDEA" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [            "Code" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [           "Slack" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [ "Microsoft Teams" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [          "Gitter" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [            "Mail" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [          "Safari" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [        "Calendar" ] = { screen = big, shape = { x=0.4, y=0, w=0.6, h=1 } },
  [          "Signal" ] = { screen = sml, shape = { x=0  , y=0, w=1  , h=1 } },
  [        "WhatsApp" ] = { screen = sml, shape = { x=0  , y=0, w=1  , h=1 } },
}
end

function laptopOnly(laptop)
  local right = { x=0.1, y=0, w=0.9, h=1 }
  local  full = { x=0  , y=0, w=1  , h=1 }
  return {
    [   "Google Chrome" ] = { screen = laptop, shape = right },
    [      "Superhuman" ] = { screen = laptop, shape = right },
    [          "iTerm2" ] = { screen = laptop, shape =  full },
    [   "IntelliJ IDEA" ] = { screen = laptop, shape = right },
    [            "Code" ] = { screen = laptop, shape = right },
    [           "Slack" ] = { screen = laptop, shape = right },
    [ "Microsoft Teams" ] = { screen = laptop, shape = right },
    [          "Gitter" ] = { screen = laptop, shape = right },
    [            "Mail" ] = { screen = laptop, shape = right },
    [          "Safari" ] = { screen = laptop, shape = right },
    [        "Calendar" ] = { screen = laptop, shape = right },
    [          "Signal" ] = { screen = laptop, shape =  full },
    [        "WhatsApp" ] = { screen = laptop, shape =  full },
  }
end

function layout(map)
  hs.layout.apply(
      mapToArr(
          map,
          function(k, v)
            return { k, nil, v.screen, v.shape, nil, nil }
          end
      )
  )
end

function screenstrs()
  local screens = hs.screen.allScreens()
  local names = map(screens, function(s) return s:name() end)
  print("screens: "..str(names))
  return table.concat(names, ',')
end

function msg(m)
  print(m)
  hs.alert(m)
end

function findscreen(...)
  if type(...) == 'string' then
    local screen = hs.screen.find(...)
    if screen ~= nil then
      return screen
    end
  elseif type(...) == 'array' then
    for s in ... do
      local screen = hs.screen.find(name)
      if screen ~= nil then
        return screen
      end
    end
  else
    for _, name in pairs(...) do
      print('\tchecking name: '..name)
      local screen = hs.screen.find(name:lower())
      if screen ~= nil then
        print('\t\tfound screen: '..name)
        return screen
      end
    end
  end
  msg("Couldn't find screens: "..str(...).." ("..type(...).."; available: "..screenstrs()..")")
end

hs.hotkey.bind(
    { 'cmd', 'ctrl' }, 'm',
    function()
      local laptop = findscreen({'Color LCD', 'Built%-in Retina Display'})
--       local     lg = findscreen({'LG ULTRAWIDE', 'DELL U2715H'})
      local     lg = findscreen({'LG ULTRAWIDE', 'LG HDR WFHD'})
      if laptop == nil or lg == nil then
        hs.alert('missing a screen; laptop: '..(laptop and laptop:name() or '??')..', LG: '..(lg and lg:name() or '??'))
        return nil
      end

      layout(bigsmall(lg, laptop))
    end
)

hs.hotkey.bind(
    { 'cmd', 'ctrl' }, 'l',
    function()
      local laptop = findscreen('Color LCD') or findscreen('Built-in Retina Display')
      if laptop == nil then return nil end
      layout(laptopOnly(laptop))
    end
)

function focusApp(name)
  local app = hs.application.find(name)
  if app == nil then
    hs.alert('Opening '..name)
    hs.application.open(name, 0, true)
    return
  end
  local focused = app:focusedWindow()
  if focused == nil then
    hs.alert("No focused window for "..name)
    local windows = app:allWindows()
    if #windows > 0 then
      windows[1]:focus()
    else
        hs.alert("No windows found for "..name)
      return
    end
  else
    focused:focus()
  end
end

function bindFocusApp(name, modifiers, hotkey)
  hs.hotkey.bind(modifiers, hotkey, function() focusApp(name) end)
end

appShortcuts = {
  [       "Activity Monitor" ] = { modifiers = { 'alt', 'cmd'   }, key = 'a' },
  [          "Google Chrome" ] = { modifiers = { 'alt', 'cmd'   }, key = 'c' },
  [                 "Safari" ] = { modifiers = { 'alt', 'cmd'   }, key = 's' },
  [                  "iTerm" ] = { modifiers =   'alt'           , key = 't' },
  [               "Chat GPT" ] = { modifiers =   'alt'           , key = 'g' },
  [             "Superhuman" ] = { modifiers =   'alt'           , key = 'h' },
  [ "com.jetbrains.intellij" ] = { modifiers =   'alt'           , key = 'i' },
  [          "IntelliJ IDEA" ] = { modifiers = { 'alt', 'shift' }, key = 'i' },
  [             "VIP Access" ] = { modifiers =   'alt'           , key = 'j' },
  [                  "Slack" ] = { modifiers =   'alt'           , key = 'k' },
  [                  "CLion" ] = { modifiers =   'alt'           , key = 'l' },
  [                 "Cursor" ] = { modifiers =   'alt'           , key = 'o' },
  [        "Microsoft Teams" ] = { modifiers =   'alt'           , key = 'm' },
  [                 "Signal" ] = { modifiers =   'alt'           , key = 'n' },
  [          "Final Cut Pro" ] = { modifiers =   'alt'           , key = 'p' },
  [       "Quicktime Player" ] = { modifiers =   'alt'           , key = 'q' },
  [                "RStudio" ] = { modifiers =   'alt'           , key = 'r' },
  [     "System Preferences" ] = { modifiers =   'alt'           , key = 's' },
  [                "Preview" ] = { modifiers =   'alt'           , key = 'v' },
  [               "WhatsApp" ] = { modifiers =   'alt'           , key = 'w' },
  [                "zoom.us" ] = { modifiers =   'alt'           , key = 'z' },
}

-- Handle Finder specially; open a new window if none exist
hs.hotkey.bind({ 'cmd', 'alt' }, 'f',
    function()
      local finder = hs.application.find('Finder')
      local window = finder:focusedWindow()
      if window == nil then
        print 'applescript…'
        hs.applescript([[
tell application "System Events"
  tell process "Finder"
    set frontmost to true
    click menu item "New Finder Window" of menu "File" of menu bar 1
  end tell
end tell
        ]])
      else
        window:focus()
      end
    end
)

mapToArr(
    appShortcuts,
    function(name, key)
      bindFocusApp(name, key.modifiers, key.key)
    end
)

hs.alert.show("Config loaded: "..hs.screen.mainScreen():name())