diff --git a/src/state.ts b/src/state.ts index 4dad65f..38865a1 100644 --- a/src/state.ts +++ b/src/state.ts @@ -28,9 +28,13 @@ export function createState(initial: T): State { subs.forEach((s) => s()); } - function sub(fn: Listener) { + function sub(fn: Listener, fire = false) { subs.push(fn); + if (fire) { + fn(); + } + return () => { const idx = subs.indexOf(fn); if (idx > -1) {