charm/src/observe.luau
2024-03-29 11:47:46 -07:00

35 lines
744 B
Lua

local effect = require(script.Parent.effect)
local types = require(script.Parent.types)
type Molecule<T> = types.Molecule<T>
local function observe<K, V>(molecule: Molecule<{ [K]: V }>, factory: (value: V, key: K) -> () -> ()): () -> ()
local connections: { [K]: () -> () } = {}
local unsubscribe = effect(function()
local state = molecule()
for key, disconnect in connections do
if state[key] == nil then
connections[key] = nil
disconnect()
end
end
for key, value in state do
if not connections[key] then
connections[key] = factory(value, key)
end
end
end)
local function cleanup()
unsubscribe()
for _, disconnect in connections do
disconnect()
end
end
return cleanup
end
return observe