Record and Loop

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

-- Record and Loop: the key starts recording, then starts the loop, then
-- stops everything.

local cfg = UI.Schema({
  key = UI.Keybind("F6", { label = "Record, loop, stop" }),
})

local mode, loop = "idle", nil

-- The take also holds the presses of the record key itself. Drop them and
-- carry their delays forward so the rest of the timing stays exact; the
-- returned tail is the pause before the final key press.
local function withoutKey(take, key)
  local kept, carry = {}, 0
  for _, action in take do
    if action.code == key then
      carry += action.delay or 0
    else
      local copy = table.clone(action)
      copy.delay = (copy.delay or 0) + carry
      carry = 0
      table.insert(kept, copy)
    end
  end
  return kept, carry
end

Bind(cfg.key, function()
  if mode == "idle" then
    Macro.Record()
    mode = "recording"
    UI.Notify("Recording", "info")
  elseif mode == "recording" then
    local route, tail = withoutKey(Macro.Finish(), cfg.key)
    if #route == 0 then
      mode = "idle"
      UI.Notify("Nothing was recorded", "warning")
      return
    end
    loop = Run(function()
      while true do
        Macro.Play(route):Wait()
        Sleep(tail)
      end
    end)
    mode = "looping"
    UI.Notify("Looping", "success")
  else
    if loop then
      loop:Cancel()
      loop = nil
    end
    Macro.StopAll()
    mode = "idle"
    UI.Notify("Stopped", "info")
  end
end)

function OnStop()
  if loop then
    loop:Cancel()
  end
  if mode == "recording" then
    Macro.Finish()
  end
  Macro.StopAll()
end