scripts/.lune/lib/channel.luau

--- An MPSC synchronization primitive powered by Lua upvalues which retains only
--- one value at a time.

--- ## Usage
--- ```luau
--- local send, recv = watch((nil :: any) :: string)
--- task.delay(5, send, "hello, world!")
--- task.spawn(function()
---     local value = recv()
---     print("received value:", value)
--- end)
--- ```
type Watch<T> = {
    value: T?,
    receivers: { thread },
}

--- Creates a new `Watch` channel, returning its send and receive handles.
local function chan<T>(_phantom: T): ((T) -> (), () -> T?)
    local watch: Watch<T> = {
        value = nil,
        receivers = {},
    }

    local function send(value: T)
        watch.value = value

        for _, receiver in watch.receivers do
            coroutine.resume(receiver, value)
        end
    end

    local function recv(): T
        local value = watch.value
        watch.value = nil

        if value == nil then
            table.insert(watch.receivers, coroutine.running())
            return coroutine.yield()
        end

        return value :: T
    end

    return send, recv
end

return chan