src/source.luau

@@ -3,22 +3,29 @@ if not game then script = require "test/relative-string" end
 local graph = require(script.Parent.graph)
 type Node<T> = graph.Node<T>
 local create = graph.create
-local set = graph.set
+local track = graph.track
+local update = graph.update
 
 export type Source<T> = (() -> T) & ((T) -> T)
 
-local function source<T>(value: T): Source<T>
-    local node, read_node_value = create(value :: T)
+local function source<T>(initial_value: T): Source<T>
+    local node = create(initial_value)
 
     return function(...): T
-        if select("#", ...) == 0 then return read_node_value() end -- check if any args were given
+        if select("#", ...) == 0 then -- no args were given
+            track(node)
+            return node.cache
+        end
 
         local v = ... :: T
-        if node.cache == v and (type(v) ~= "table" or table.isfrozen(v)) then return v end
+        if node.cache == v and (type(v) ~= "table" or table.isfrozen(v)) then 
+            return v
+        end
 
-        set(node, v)
+        node.cache = v
+        update(node)
         return v
     end
 end
 
-return source :: (<T>(value: T) -> Source<T>) & (<T>() -> Source<T>)
+return source :: (<T>(initial_value: T) -> Source<T>) & (<T>() -> Source<T>)