statewire 0.4.0 → 0.6.0

package/dist/util/recordMutations.js.map

@@ -1 +1 @@
-{"version":3,"file":"recordMutations.js","names":["sharedDraftTraps","PathKey","Path","DRAFT","Symbol","Node","base","Record","PropertyKey","copy","modified","isArray","parent","path","proxy","assigned","Set","isDraftable","value","Array","proto","Object","getPrototypeOf","prototype","nodeOf","undefined","latest","node","prepareCopy","slice","markModified","n","resolveAssigned","container","key","keys","child","finalize","result","has","recordMutations","T","mutate","draft","state","changedPaths","Error","seen","record","join","add","push","createNode","touched","prop","Proxy","get","_target","current","Number","set","next","source","hasOwn","deleteProperty","Reflect","root"],"sources":["../../src/util/recordMutations.ts"],"sourcesContent":["import { sharedDraftTraps } from \"./draftTraps\";\n\ntype PathKey = string | number;\ntype Path = readonly PathKey[];\n\nconst DRAFT = Symbol(\"statewire.draft\");\n\ntype Node = {\n  base: Record<PropertyKey, unknown>;\n  copy: Record<PropertyKey, unknown> | null;\n  modified: boolean;\n  isArray: boolean;\n  parent: Node | null;\n  path: Path;\n  proxy: unknown;\n  assigned: Set<PropertyKey> | null;\n};\n\nconst isDraftable = (value: unknown): value is object => {\n  if (typeof value !== \"object\" || value === null) return false;\n  if (Array.isArray(value)) return true;\n  const proto: unknown = Object.getPrototypeOf(value);\n  return proto === Object.prototype || proto === null;\n};\n\nconst nodeOf = (value: unknown): Node | undefined =>\n  typeof value === \"object\" && value !== null\n    ? (value as { [DRAFT]?: Node })[DRAFT]\n    : undefined;\n\nconst latest = (node: Node) => node.copy ?? node.base;\n\nconst prepareCopy = (node: Node) => {\n  node.copy ??= node.isArray\n    ? ((node.base as unknown as unknown[]).slice() as unknown as Record<\n        PropertyKey,\n        unknown\n      >)\n    : { ...node.base };\n};\n\nconst markModified = (node: Node) => {\n  for (let n: Node | null = node; n && !n.modified; n = n.parent) {\n    n.modified = true;\n  }\n};\n\nconst resolveAssigned = (container: object) => {\n  for (const key of Object.keys(container)) {\n    const value = (container as Record<string, unknown>)[key];\n    const child = nodeOf(value);\n    if (child) {\n      (container as Record<string, unknown>)[key] = finalize(child);\n    } else if (isDraftable(value)) {\n      resolveAssigned(value);\n    }\n  }\n};\n\nconst finalize = (node: Node): unknown => {\n  if (!node.modified) return node.base;\n  const result = node.copy!;\n  for (const key of Object.keys(result)) {\n    const value = result[key];\n    const child = nodeOf(value);\n    if (child) {\n      result[key] = finalize(child);\n    } else if (node.assigned?.has(key) && isDraftable(value)) {\n      resolveAssigned(value);\n    }\n  }\n  return result;\n};\n\nexport const recordMutations = <T>(\n  base: T,\n  mutate: (draft: T) => void,\n): { state: T; changedPaths: Path[] } => {\n  if (!isDraftable(base)) {\n    throw new Error(\"statewire: cannot record mutations on primitive state\");\n  }\n\n  const changedPaths: Path[] = [];\n  const seen = new Set<string>();\n  const record = (path: Path) => {\n    const key = path.join(\"\u0000\");\n    if (seen.has(key)) return;\n    seen.add(key);\n    changedPaths.push(path);\n  };\n\n  const createNode = (value: object, parent: Node | null, path: Path): Node => {\n    const node: Node = {\n      base: value as Record<PropertyKey, unknown>,\n      copy: null,\n      modified: false,\n      isArray: Array.isArray(value),\n      parent,\n      path,\n      proxy: null,\n      assigned: null,\n    };\n    const touched = (prop: PropertyKey) => {\n      if (typeof prop !== \"string\") return;\n      record(node.isArray ? node.path : [...node.path, prop]);\n    };\n    node.proxy = new Proxy(value, {\n      get: (_target, prop) => {\n        if (prop === DRAFT) return node;\n        const current = latest(node)[prop];\n        if (typeof prop === \"symbol\") return current;\n        if (nodeOf(current)) return current;\n        if (isDraftable(current) && current === node.base[prop]) {\n          prepareCopy(node);\n          const child = createNode(current, node, [\n            ...path,\n            node.isArray ? Number(prop) : prop,\n          ]);\n          node.copy![prop] = child.proxy;\n          return child.proxy;\n        }\n        return current;\n      },\n      set: (_target, prop, next) => {\n        const source = latest(node);\n        if (Object.hasOwn(source, prop) && source[prop] === next) return true;\n        prepareCopy(node);\n        markModified(node);\n        touched(prop);\n        node.copy![prop] = next;\n        (node.assigned ??= new Set()).add(prop);\n        return true;\n      },\n      deleteProperty: (_target, prop) => {\n        if (!Object.hasOwn(latest(node), prop)) return true;\n        prepareCopy(node);\n        markModified(node);\n        touched(prop);\n        delete node.copy![prop];\n        return true;\n      },\n      has: (_target, prop) => prop === DRAFT || Reflect.has(latest(node), prop),\n      ...sharedDraftTraps(() => latest(node), \"inside a mutator\"),\n    });\n    return node;\n  };\n\n  const root = createNode(base as unknown as object, null, []);\n  mutate(root.proxy as T);\n  return { state: finalize(root) as T, changedPaths };\n};\n"],"mappings":";;AAKA,MAAMG,QAAQC,OAAO,iBAAiB;AAatC,MAAMa,eAAeC,UAAoC;CACvD,IAAI,OAAOA,UAAU,YAAYA,UAAU,MAAM,OAAO;CACxD,IAAIC,MAAMR,QAAQO,KAAK,GAAG,OAAO;CACjC,MAAME,QAAiBC,OAAOC,eAAeJ,KAAK;CAClD,OAAOE,UAAUC,OAAOE,aAAaH,UAAU;AACjD;AAEA,MAAMI,UAAUN,UACd,OAAOA,UAAU,YAAYA,UAAU,OAClCA,MAA6Bf,SAC9BsB,KAAAA;AAEN,MAAMC,UAAUC,SAAeA,KAAKlB,QAAQkB,KAAKrB;AAEjD,MAAMsB,eAAeD,SAAe;CAClCA,KAAKlB,SAASkB,KAAKhB,UACbgB,KAAKrB,KAA8BuB,MAAM,IAI3C,EAAE,GAAGF,KAAKrB,KAAK;AACrB;AAEA,MAAMwB,gBAAgBH,SAAe;CACnC,KAAK,IAAII,IAAiBJ,MAAMI,KAAK,CAACA,EAAErB,UAAUqB,IAAIA,EAAEnB,QACtDmB,EAAErB,WAAW;AAEjB;AAEA,MAAMsB,mBAAmBC,cAAsB;CAC7C,KAAK,MAAMC,OAAOb,OAAOc,KAAKF,SAAS,GAAG;EACxC,MAAMf,QAASe,UAAsCC;EACrD,MAAME,QAAQZ,OAAON,KAAK;EAC1B,IAAIkB,OACF,UAAuCF,OAAOG,SAASD,KAAK;OACvD,IAAInB,YAAYC,KAAK,GAC1Bc,gBAAgBd,KAAK;CAEzB;AACF;AAEA,MAAMmB,YAAYV,SAAwB;CACxC,IAAI,CAACA,KAAKjB,UAAU,OAAOiB,KAAKrB;CAChC,MAAMgC,SAASX,KAAKlB;CACpB,KAAK,MAAMyB,OAAOb,OAAOc,KAAKG,MAAM,GAAG;EACrC,MAAMpB,QAAQoB,OAAOJ;EACrB,MAAME,QAAQZ,OAAON,KAAK;EAC1B,IAAIkB,OACFE,OAAOJ,OAAOG,SAASD,KAAK;OACvB,IAAIT,KAAKZ,UAAUwB,IAAIL,GAAG,KAAKjB,YAAYC,KAAK,GACrDc,gBAAgBd,KAAK;CAEzB;CACA,OAAOoB;AACT;AAEA,MAAaE,mBACXlC,MACAoC,WACuC;CACvC,IAAI,CAACzB,YAAYX,IAAI,GACnB,MAAM,IAAIwC,MAAM,uDAAuD;CAGzE,MAAMD,eAAuB,CAAA;CAC7B,MAAME,uBAAO,IAAI/B,IAAY;CAC7B,MAAMgC,UAAUnC,SAAe;EAC7B,MAAMqB,MAAMrB,KAAKoC,KAAK,IAAG;EACzB,IAAIF,KAAKR,IAAIL,GAAG,GAAG;EACnBa,KAAKG,IAAIhB,GAAG;EACZW,aAAaM,KAAKtC,IAAI;CACxB;CAEA,MAAMuC,cAAclC,OAAeN,QAAqBC,SAAqB;EAC3E,MAAMc,OAAa;GACjBrB,MAAMY;GACNT,MAAM;GACNC,UAAU;GACVC,SAASQ,MAAMR,QAAQO,KAAK;GAC5BN;GACAC;GACAC,OAAO;GACPC,UAAU;EACZ;EACA,MAAMsC,WAAWC,SAAsB;GACrC,IAAI,OAAOA,SAAS,UAAU;GAC9BN,OAAOrB,KAAKhB,UAAUgB,KAAKd,OAAO,CAAC,GAAGc,KAAKd,MAAMyC,IAAI,CAAC;EACxD;EACA3B,KAAKb,QAAQ,IAAIyC,MAAMrC,OAAO;GAC5BsC,MAAMC,SAASH,SAAS;IACtB,IAAIA,SAASnD,OAAO,OAAOwB;IAC3B,MAAM+B,UAAUhC,OAAOC,IAAI,CAAC,CAAC2B;IAC7B,IAAI,OAAOA,SAAS,UAAU,OAAOI;IACrC,IAAIlC,OAAOkC,OAAO,GAAG,OAAOA;IAC5B,IAAIzC,YAAYyC,OAAO,KAAKA,YAAY/B,KAAKrB,KAAKgD,OAAO;KACvD1B,YAAYD,IAAI;KAChB,MAAMS,QAAQgB,WAAWM,SAAS/B,MAAM,CACtC,GAAGd,MACHc,KAAKhB,UAAUgD,OAAOL,IAAI,IAAIA,IAAI,CACnC;KACD3B,KAAKlB,KAAM6C,QAAQlB,MAAMtB;KACzB,OAAOsB,MAAMtB;IACf;IACA,OAAO4C;GACT;GACAE,MAAMH,SAASH,MAAMO,SAAS;IAC5B,MAAMC,SAASpC,OAAOC,IAAI;IAC1B,IAAIN,OAAO0C,OAAOD,QAAQR,IAAI,KAAKQ,OAAOR,UAAUO,MAAM,OAAO;IACjEjC,YAAYD,IAAI;IAChBG,aAAaH,IAAI;IACjB0B,QAAQC,IAAI;IACZ3B,KAAKlB,KAAM6C,QAAQO;IACnB,CAAClC,KAAKZ,6BAAa,IAAIC,IAAI,EAAA,CAAGkC,IAAII,IAAI;IACtC,OAAO;GACT;GACAU,iBAAiBP,SAASH,SAAS;IACjC,IAAI,CAACjC,OAAO0C,OAAOrC,OAAOC,IAAI,GAAG2B,IAAI,GAAG,OAAO;IAC/C1B,YAAYD,IAAI;IAChBG,aAAaH,IAAI;IACjB0B,QAAQC,IAAI;IACZ,OAAO3B,KAAKlB,KAAM6C;IAClB,OAAO;GACT;GACAf,MAAMkB,SAASH,SAASA,SAASnD,SAAS8D,QAAQ1B,IAAIb,OAAOC,IAAI,GAAG2B,IAAI;GACxE,GAAGtD,uBAAuB0B,OAAOC,IAAI,GAAG,kBAAkB;EAC5D,CAAC;EACD,OAAOA;CACT;CAEA,MAAMuC,OAAOd,WAAW9C,MAA2B,MAAM,CAAA,CAAE;CAC3DoC,OAAOwB,KAAKpD,KAAU;CACtB,OAAO;EAAE8B,OAAOP,SAAS6B,IAAI;EAAQrB;CAAa;AACpD"}
+{"version":3,"file":"recordMutations.js","names":["sharedDraftTraps","PathKey","Path","DRAFT","Symbol","Node","base","Record","PropertyKey","copy","modified","isArray","parent","path","proxy","assigned","Set","isDraftable","value","Array","proto","Object","getPrototypeOf","prototype","nodeOf","undefined","latest","node","prepareCopy","slice","markModified","n","resolveAssigned","container","key","keys","child","finalize","result","has","applyOptimistic","state","TState","mutate","draft","recordMutations","T","changedPaths","Error","seen","record","join","add","push","createNode","touched","prop","Proxy","get","_target","current","Number","set","next","source","hasOwn","deleteProperty","Reflect","root"],"sources":["../../src/util/recordMutations.ts"],"sourcesContent":["import { sharedDraftTraps } from \"./draftTraps\";\n\ntype PathKey = string | number;\ntype Path = readonly PathKey[];\n\nconst DRAFT = Symbol(\"statewire.draft\");\n\ntype Node = {\n  base: Record<PropertyKey, unknown>;\n  copy: Record<PropertyKey, unknown> | null;\n  modified: boolean;\n  isArray: boolean;\n  parent: Node | null;\n  path: Path;\n  proxy: unknown;\n  assigned: Set<PropertyKey> | null;\n};\n\nconst isDraftable = (value: unknown): value is object => {\n  if (typeof value !== \"object\" || value === null) return false;\n  if (Array.isArray(value)) return true;\n  const proto: unknown = Object.getPrototypeOf(value);\n  return proto === Object.prototype || proto === null;\n};\n\nconst nodeOf = (value: unknown): Node | undefined =>\n  typeof value === \"object\" && value !== null\n    ? (value as { [DRAFT]?: Node })[DRAFT]\n    : undefined;\n\nconst latest = (node: Node) => node.copy ?? node.base;\n\nconst prepareCopy = (node: Node) => {\n  node.copy ??= node.isArray\n    ? ((node.base as unknown as unknown[]).slice() as unknown as Record<\n        PropertyKey,\n        unknown\n      >)\n    : { ...node.base };\n};\n\nconst markModified = (node: Node) => {\n  for (let n: Node | null = node; n && !n.modified; n = n.parent) {\n    n.modified = true;\n  }\n};\n\nconst resolveAssigned = (container: object) => {\n  for (const key of Object.keys(container)) {\n    const value = (container as Record<string, unknown>)[key];\n    const child = nodeOf(value);\n    if (child) {\n      (container as Record<string, unknown>)[key] = finalize(child);\n    } else if (isDraftable(value)) {\n      resolveAssigned(value);\n    }\n  }\n};\n\nconst finalize = (node: Node): unknown => {\n  if (!node.modified) return node.base;\n  const result = node.copy!;\n  for (const key of Object.keys(result)) {\n    const value = result[key];\n    const child = nodeOf(value);\n    if (child) {\n      result[key] = finalize(child);\n    } else if (node.assigned?.has(key) && isDraftable(value)) {\n      resolveAssigned(value);\n    }\n  }\n  return result;\n};\n\n/** Run a mutator against a recording draft, as the client does for `optimistic`. */\nexport const applyOptimistic = <TState>(\n  state: TState,\n  mutate: (draft: TState) => void,\n): TState => recordMutations(state, mutate).state;\n\nexport const recordMutations = <T>(\n  base: T,\n  mutate: (draft: T) => void,\n): { state: T; changedPaths: Path[] } => {\n  if (!isDraftable(base)) {\n    throw new Error(\"statewire: cannot record mutations on primitive state\");\n  }\n\n  const changedPaths: Path[] = [];\n  const seen = new Set<string>();\n  const record = (path: Path) => {\n    const key = path.join(\"\u0000\");\n    if (seen.has(key)) return;\n    seen.add(key);\n    changedPaths.push(path);\n  };\n\n  const createNode = (value: object, parent: Node | null, path: Path): Node => {\n    const node: Node = {\n      base: value as Record<PropertyKey, unknown>,\n      copy: null,\n      modified: false,\n      isArray: Array.isArray(value),\n      parent,\n      path,\n      proxy: null,\n      assigned: null,\n    };\n    const touched = (prop: PropertyKey) => {\n      if (typeof prop !== \"string\") return;\n      record(node.isArray ? node.path : [...node.path, prop]);\n    };\n    node.proxy = new Proxy(value, {\n      get: (_target, prop) => {\n        if (prop === DRAFT) return node;\n        const current = latest(node)[prop];\n        if (typeof prop === \"symbol\") return current;\n        if (nodeOf(current)) return current;\n        if (isDraftable(current) && current === node.base[prop]) {\n          prepareCopy(node);\n          const child = createNode(current, node, [\n            ...path,\n            node.isArray ? Number(prop) : prop,\n          ]);\n          node.copy![prop] = child.proxy;\n          return child.proxy;\n        }\n        return current;\n      },\n      set: (_target, prop, next) => {\n        const source = latest(node);\n        if (Object.hasOwn(source, prop) && source[prop] === next) return true;\n        prepareCopy(node);\n        markModified(node);\n        touched(prop);\n        node.copy![prop] = next;\n        (node.assigned ??= new Set()).add(prop);\n        return true;\n      },\n      deleteProperty: (_target, prop) => {\n        if (!Object.hasOwn(latest(node), prop)) return true;\n        prepareCopy(node);\n        markModified(node);\n        touched(prop);\n        delete node.copy![prop];\n        return true;\n      },\n      has: (_target, prop) => prop === DRAFT || Reflect.has(latest(node), prop),\n      ...sharedDraftTraps(() => latest(node), \"inside a mutator\"),\n    });\n    return node;\n  };\n\n  const root = createNode(base as unknown as object, null, []);\n  mutate(root.proxy as T);\n  return { state: finalize(root) as T, changedPaths };\n};\n"],"mappings":";;AAKA,MAAMG,QAAQC,OAAO,iBAAiB;AAatC,MAAMa,eAAeC,UAAoC;CACvD,IAAI,OAAOA,UAAU,YAAYA,UAAU,MAAM,OAAO;CACxD,IAAIC,MAAMR,QAAQO,KAAK,GAAG,OAAO;CACjC,MAAME,QAAiBC,OAAOC,eAAeJ,KAAK;CAClD,OAAOE,UAAUC,OAAOE,aAAaH,UAAU;AACjD;AAEA,MAAMI,UAAUN,UACd,OAAOA,UAAU,YAAYA,UAAU,OAClCA,MAA6Bf,SAC9BsB,KAAAA;AAEN,MAAMC,UAAUC,SAAeA,KAAKlB,QAAQkB,KAAKrB;AAEjD,MAAMsB,eAAeD,SAAe;CAClCA,KAAKlB,SAASkB,KAAKhB,UACbgB,KAAKrB,KAA8BuB,MAAM,IAI3C,EAAE,GAAGF,KAAKrB,KAAK;AACrB;AAEA,MAAMwB,gBAAgBH,SAAe;CACnC,KAAK,IAAII,IAAiBJ,MAAMI,KAAK,CAACA,EAAErB,UAAUqB,IAAIA,EAAEnB,QACtDmB,EAAErB,WAAW;AAEjB;AAEA,MAAMsB,mBAAmBC,cAAsB;CAC7C,KAAK,MAAMC,OAAOb,OAAOc,KAAKF,SAAS,GAAG;EACxC,MAAMf,QAASe,UAAsCC;EACrD,MAAME,QAAQZ,OAAON,KAAK;EAC1B,IAAIkB,OACF,UAAuCF,OAAOG,SAASD,KAAK;OACvD,IAAInB,YAAYC,KAAK,GAC1Bc,gBAAgBd,KAAK;CAEzB;AACF;AAEA,MAAMmB,YAAYV,SAAwB;CACxC,IAAI,CAACA,KAAKjB,UAAU,OAAOiB,KAAKrB;CAChC,MAAMgC,SAASX,KAAKlB;CACpB,KAAK,MAAMyB,OAAOb,OAAOc,KAAKG,MAAM,GAAG;EACrC,MAAMpB,QAAQoB,OAAOJ;EACrB,MAAME,QAAQZ,OAAON,KAAK;EAC1B,IAAIkB,OACFE,OAAOJ,OAAOG,SAASD,KAAK;OACvB,IAAIT,KAAKZ,UAAUwB,IAAIL,GAAG,KAAKjB,YAAYC,KAAK,GACrDc,gBAAgBd,KAAK;CAEzB;CACA,OAAOoB;AACT;;AAGA,MAAaE,mBACXC,OACAE,WACWE,gBAAgBJ,OAAOE,MAAM,CAAC,CAACF;AAE5C,MAAaI,mBACXvC,MACAqC,WACuC;CACvC,IAAI,CAAC1B,YAAYX,IAAI,GACnB,MAAM,IAAI0C,MAAM,uDAAuD;CAGzE,MAAMD,eAAuB,CAAA;CAC7B,MAAME,uBAAO,IAAIjC,IAAY;CAC7B,MAAMkC,UAAUrC,SAAe;EAC7B,MAAMqB,MAAMrB,KAAKsC,KAAK,IAAG;EACzB,IAAIF,KAAKV,IAAIL,GAAG,GAAG;EACnBe,KAAKG,IAAIlB,GAAG;EACZa,aAAaM,KAAKxC,IAAI;CACxB;CAEA,MAAMyC,cAAcpC,OAAeN,QAAqBC,SAAqB;EAC3E,MAAMc,OAAa;GACjBrB,MAAMY;GACNT,MAAM;GACNC,UAAU;GACVC,SAASQ,MAAMR,QAAQO,KAAK;GAC5BN;GACAC;GACAC,OAAO;GACPC,UAAU;EACZ;EACA,MAAMwC,WAAWC,SAAsB;GACrC,IAAI,OAAOA,SAAS,UAAU;GAC9BN,OAAOvB,KAAKhB,UAAUgB,KAAKd,OAAO,CAAC,GAAGc,KAAKd,MAAM2C,IAAI,CAAC;EACxD;EACA7B,KAAKb,QAAQ,IAAI2C,MAAMvC,OAAO;GAC5BwC,MAAMC,SAASH,SAAS;IACtB,IAAIA,SAASrD,OAAO,OAAOwB;IAC3B,MAAMiC,UAAUlC,OAAOC,IAAI,CAAC,CAAC6B;IAC7B,IAAI,OAAOA,SAAS,UAAU,OAAOI;IACrC,IAAIpC,OAAOoC,OAAO,GAAG,OAAOA;IAC5B,IAAI3C,YAAY2C,OAAO,KAAKA,YAAYjC,KAAKrB,KAAKkD,OAAO;KACvD5B,YAAYD,IAAI;KAChB,MAAMS,QAAQkB,WAAWM,SAASjC,MAAM,CACtC,GAAGd,MACHc,KAAKhB,UAAUkD,OAAOL,IAAI,IAAIA,IAAI,CACnC;KACD7B,KAAKlB,KAAM+C,QAAQpB,MAAMtB;KACzB,OAAOsB,MAAMtB;IACf;IACA,OAAO8C;GACT;GACAE,MAAMH,SAASH,MAAMO,SAAS;IAC5B,MAAMC,SAAStC,OAAOC,IAAI;IAC1B,IAAIN,OAAO4C,OAAOD,QAAQR,IAAI,KAAKQ,OAAOR,UAAUO,MAAM,OAAO;IACjEnC,YAAYD,IAAI;IAChBG,aAAaH,IAAI;IACjB4B,QAAQC,IAAI;IACZ7B,KAAKlB,KAAM+C,QAAQO;IACnB,CAACpC,KAAKZ,6BAAa,IAAIC,IAAI,EAAA,CAAGoC,IAAII,IAAI;IACtC,OAAO;GACT;GACAU,iBAAiBP,SAASH,SAAS;IACjC,IAAI,CAACnC,OAAO4C,OAAOvC,OAAOC,IAAI,GAAG6B,IAAI,GAAG,OAAO;IAC/C5B,YAAYD,IAAI;IAChBG,aAAaH,IAAI;IACjB4B,QAAQC,IAAI;IACZ,OAAO7B,KAAKlB,KAAM+C;IAClB,OAAO;GACT;GACAjB,MAAMoB,SAASH,SAASA,SAASrD,SAASgE,QAAQ5B,IAAIb,OAAOC,IAAI,GAAG6B,IAAI;GACxE,GAAGxD,uBAAuB0B,OAAOC,IAAI,GAAG,kBAAkB;EAC5D,CAAC;EACD,OAAOA;CACT;CAEA,MAAMyC,OAAOd,WAAWhD,MAA2B,MAAM,CAAA,CAAE;CAC3DqC,OAAOyB,KAAKtD,KAAU;CACtB,OAAO;EAAE2B,OAAOJ,SAAS+B,IAAI;EAAQrB;CAAa;AACpD"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "statewire",
-  "version": "0.4.0",
+  "version": "0.6.0",
   "description": "State over the wire: snapshot-first delta streaming and command submission for any state and any command",
   "license": "MIT",
   "repository": {

package/src/client/connection.ts

@@ -7,7 +7,7 @@ import type { StatewireClient } from "../statewire";
 import type { StatewireHost } from "../host";
 
 export type Episode = {
-  cause: "dropped" | "evicted" | "error";
+  cause: "dropped" | "evicted" | "error" | "unauthorized";
   message?: string;
 };
 

package/src/client/core.ts

@@ -159,6 +159,7 @@ const createCore = (
       const connection = getConnection();
       const connectionChanged = connection !== notifiedConnection;
       notifiedConnection = connection;
+      if (connectionChanged) options.onConnectionChange?.(connection);
       if (stateChanged || connectionChanged) {
         for (const listener of [...listeners]) listener();
       }

package/src/client/session.ts

@@ -147,11 +147,7 @@ const rejectedSend = (
 
 type AttachState = "detached" | "attached" | "finished";
 
-export type StatewireSessionOptions = StatewireSessionStoreOptions & {
-  onCommandSessionLost?:
-    | ((command: StatewireClient.SessionLostCommand) => "replay" | void)
-    | undefined;
-};
+export type StatewireSessionOptions = StatewireSessionStoreOptions;
 
 type Session = {
   record: StatewireSessionRecord;
@@ -172,7 +168,6 @@ type Session = {
   transport: RemoteLink | null;
   waiters: (() => void)[];
   lifecycle: CommandsListener | null;
-  onCommandSessionLost: StatewireSessionOptions["onCommandSessionLost"];
   notify: () => void;
   emitEvent: (event: unknown) => void;
   send: (command: StatewireWireCommand) => Promise<unknown>;
@@ -219,7 +214,6 @@ const createSession = (options: StatewireSessionOptions): Session => {
     transport: null,
     waiters: [],
     lifecycle: null,
-    onCommandSessionLost: options.onCommandSessionLost,
     notify: () => {},
     emitEvent: () => {},
     send: (command) => sendCommand(s, command),
@@ -298,25 +292,15 @@ const settleAll = (
   }
 };
 
-// Offers every unsettled real command for replay, in seq order. Not caught:
-// a throwing callback aborts the eviction before any entry settles; the
-// attach retry evicts again and re-offers.
+// Claims every unsettled real command for replay under the rotated client id,
+// in seq order; context submissions settle and the context re-sends whole.
 const claimReplays = (s: Session): Pending[] => {
-  const handler = s.onCommandSessionLost;
-  if (!handler) return [];
-  const claimed = new Set<Pending>();
-  for (const entry of s.pending) {
-    if (!entry.inv || isResolved(entry.state)) continue;
-    const decision = handler({
-      method: entry.inv.method,
-      params: [...entry.inv.params],
-      seq: entry.seq,
-    });
-    if (decision === "replay") claimed.add(entry);
-  }
-  if (claimed.size === 0) return [];
-  s.pending = s.pending.filter((p) => !claimed.has(p));
-  return [...claimed];
+  const claimed = s.pending.filter(
+    (entry) => entry.inv !== null && !isResolved(entry.state),
+  );
+  if (claimed.length === 0) return [];
+  s.pending = s.pending.filter((p) => !claimed.includes(p));
+  return claimed;
 };
 
 const evict = (
@@ -745,7 +729,6 @@ export const useStatewireSession = (
   emit: (event: unknown) => void,
 ): Session => {
   const [s] = useState(() => createSession(options));
-  s.onCommandSessionLost = options.onCommandSessionLost;
   useEffect(() => {
     s.emitEvent = emit;
     return () => {

package/src/index.ts

@@ -3,3 +3,4 @@ export { StatewireClient, useStatewire, StatewireResource } from "./statewire";
 export { StatewireSSE } from "./transports/sse";
 export { StatewireWS } from "./transports/ws";
 export { StatewireSendError } from "./types";
+export { applyOptimistic } from "./util/recordMutations";

package/src/statewire.ts

@@ -119,6 +119,7 @@ export namespace StatewireClient {
       update: Statewire.CommandUpdate<TCommands>,
       ctx: { state: NonNullable<TState> },
     ) => void;
+    onConnectionChange?: (connection: Connection) => void;
   };
 
   export type Helpers<
@@ -140,7 +141,7 @@ export namespace StatewireClient {
     | { status: "live" }
     | {
         status: "reconnecting";
-        cause: "dropped" | "evicted" | "error";
+        cause: "dropped" | "evicted" | "error" | "unauthorized";
         attempt: number;
         nextRetryAt: number;
         lastError?: string;
@@ -202,33 +203,26 @@ export namespace StatewireClient {
         cause: "unmount" | "stream-gone" | "session-lost" | "superseded";
       };
 
-  /**
-   * A pending command offered for replay while the session is being evicted.
-   * Return "replay" to re-enqueue it under the rotated client id instead of
-   * settling it as a session-lost failure.
-   */
-  export type SessionLostCommand = {
-    method: string;
-    params: unknown[];
-    seq: number;
-  };
+  /** Static headers, or a resolver invoked per request/attach; `staleAuth` is true right after an auth rejection. */
+  export type HeadersOption =
+    | HeadersInit
+    | ((ctx: { staleAuth: boolean }) => HeadersInit | Promise<HeadersInit>);
 
   export type SSEOptions = StatewireSessionStoreOptions & {
     url: string;
-    headers?: HeadersInit | (() => HeadersInit | Promise<HeadersInit>);
+    headers?: HeadersOption;
     fetch?: typeof globalThis.fetch;
     onChange?: (connection: Connection) => void;
-    onCommandSessionLost?: (command: SessionLostCommand) => "replay" | void;
   };
 
   export type WSOptions = StatewireSessionStoreOptions & {
     url: string;
+    headers?: HeadersOption;
     protocols?: string | string[];
     query?:
       | Record<string, string>
       | (() => Record<string, string> | Promise<Record<string, string>>);
     webSocket?: new (url: string, protocols?: string | string[]) => WebSocket;
     onChange?: (connection: Connection) => void;
-    onCommandSessionLost?: (command: SessionLostCommand) => "replay" | void;
   };
 }

package/src/transports/remote.ts

@@ -20,13 +20,7 @@ export const useStatewireRemote = <
   useLink: (options: TOptions, ctx: StatewireRemoteProps) => RemoteLink,
 ): Statewire<any, any, any> => {
   const emit = useStatewireEmit<any>();
-  const s = useStatewireSession(
-    {
-      sessionStore: options.sessionStore,
-      onCommandSessionLost: options.onCommandSessionLost,
-    },
-    emit,
-  );
+  const s = useStatewireSession({ sessionStore: options.sessionStore }, emit);
   const conn = useLink(options, s.ctx);
   useEffect(() => setTransport(s, conn), [s, conn]);
 

package/src/transports/sse.ts

@@ -72,13 +72,16 @@ export type RemoteLink = {
 const baseUrl = (options: StatewireClient.SSEOptions) =>
   options.url.replace(/\/$/, "");
 
+export type AuthState = { stale: boolean };
+
 const requestHeaders = async (
   options: StatewireClient.SSEOptions,
   extra: Record<string, string>,
+  ctx: { staleAuth: boolean },
 ) => {
   const { headers } = options;
   const merged = new Headers(
-    typeof headers === "function" ? await headers() : headers,
+    typeof headers === "function" ? await headers(ctx) : headers,
   );
   for (const [name, value] of Object.entries(extra)) merged.set(name, value);
   return merged;
@@ -131,12 +134,19 @@ const consume = async (
 const attemptStream = async (
   c: Conn,
   options: StatewireClient.SSEOptions,
+  auth: AuthState,
   abort: AbortController,
 ): Promise<"stop" | "reattach" | "ended"> => {
-  const headers = await requestHeaders(options, {
-    "Statewire-Client-Id": c.ctx.clientId,
-    accept: "text/event-stream",
-  });
+  const stale = auth.stale;
+  auth.stale = false;
+  const headers = await requestHeaders(
+    options,
+    {
+      "Statewire-Client-Id": c.ctx.clientId,
+      accept: "text/event-stream",
+    },
+    { staleAuth: stale },
+  );
   const heartbeat = new AbortController();
   const headerTimer = setTimeout(
     () => heartbeat.abort(),
@@ -158,6 +168,11 @@ const attemptStream = async (
     clearTimeout(headerTimer);
   }
   if (res.status === 429 || res.status === 503) throw overloaded(res);
+  if (res.status === 401) {
+    if (stale) c.episode = { cause: "unauthorized" };
+    auth.stale = true;
+    throw new Error("unauthorized");
+  }
   if (!res.ok || !res.body) throw new Error(`HTTP ${res.status}`);
   const reader = res.body.getReader();
   try {
@@ -170,23 +185,28 @@ const attemptStream = async (
 const overloaded = (res: Response) =>
   overloadedError(parseRetryAfter(res.headers.get("Retry-After")));
 
-const sendCommand = async (
+const postCommand = async (
   c: Conn,
   options: StatewireClient.SSEOptions,
   { seq, command }: { seq: number; command: StatewireWireCommand },
-): Promise<StatewireSubmitResult> => {
-  const timeout = AbortSignal.timeout(timing.heartbeatTimeoutMs);
-  const res = await timedFetch(
+  timeout: AbortSignal,
+  stale: boolean,
+): Promise<Response> =>
+  timedFetch(
     options,
     `${baseUrl(options)}/commands`,
     {
       method: "POST",
-      headers: await requestHeaders(options, {
-        "content-type": "application/json",
-        "Statewire-Client-Id": c.ctx.clientId,
-        "Statewire-Command-Seq": String(seq),
-        ...(c.lease !== null && { "Statewire-Lease": c.lease }),
-      }),
+      headers: await requestHeaders(
+        options,
+        {
+          "content-type": "application/json",
+          "Statewire-Client-Id": c.ctx.clientId,
+          "Statewire-Command-Seq": String(seq),
+          ...(c.lease !== null && { "Statewire-Lease": c.lease }),
+        },
+        { staleAuth: stale },
+      ),
       body: JSON.stringify(encodeWireCommand(command)),
       signal: AbortSignal.any([
         c.lifetime?.signal ?? AbortSignal.abort(),
@@ -195,6 +215,18 @@ const sendCommand = async (
     },
     timeout,
   );
+
+const sendCommand = async (
+  c: Conn,
+  options: StatewireClient.SSEOptions,
+  auth: AuthState,
+  args: { seq: number; command: StatewireWireCommand },
+): Promise<StatewireSubmitResult> => {
+  const timeout = AbortSignal.timeout(timing.heartbeatTimeoutMs);
+  let res = await postCommand(c, options, args, timeout, false);
+  if (res.status === 401 && typeof options.headers === "function") {
+    res = await postCommand(c, options, args, timeout, true);
+  }
   if (res.ok) {
     let body: { res?: StatewireResponse } | null;
     try {
@@ -209,24 +241,35 @@ const sendCommand = async (
       ? { type: "receipt", response: reviveResponse(body.res) }
       : { type: "receipt" };
   }
+  if (res.status === 429 || res.status === 503) throw overloaded(res);
+  if (res.status === 401) {
+    // Fresh credentials did not clear the 401: signal auth-failed on the
+    // connection and let the send retry as a transport fault.
+    auth.stale = true;
+    c.episode = { cause: "unauthorized" };
+    c.connection.reconnect();
+    throw new Error("unauthorized");
+  }
+  // Bodies settle, statuses pace: only a statewire-shaped body carries a
+  // verdict; any bare status is middleware noise — a transport fault, retried.
+  const body = (await res.json().catch(() => null)) as {
+    error?: string;
+    rejections?: { message?: string }[];
+  } | null;
+  if (body?.error === "rejected") {
+    return {
+      type: "rejected",
+      message: body.rejections?.[0]?.message ?? `HTTP ${res.status}`,
+    };
+  }
   if (res.status === 409 || res.status === 412 || res.status === 423) {
-    // Only a body naming the same verdict as the status is statewire's own;
-    // a bare status is middleware noise — a transport fault, retried.
     const verdict =
       res.status === 409
         ? ("seq-gap" as const)
         : res.status === 412
           ? ("unknown-client" as const)
           : ("stale-lease" as const);
-    const body = (await res.json().catch(() => null)) as {
-      error?: string;
-    } | null;
     if (body?.error === verdict) return { type: verdict };
-    throw new Error(`HTTP ${res.status}`);
-  }
-  if (res.status === 429 || res.status === 503) throw overloaded(res);
-  if (res.status >= 400 && res.status < 500) {
-    return { type: "rejected", message: `HTTP ${res.status}` };
   }
   throw new Error(`HTTP ${res.status}`);
 };
@@ -237,18 +280,19 @@ const useSseLink = (
 ): RemoteLink => {
   const latest = useRef(options);
   latest.current = options;
+  const [auth] = useState((): AuthState => ({ stale: false }));
   const c = useStatewireConn({
     url: options.url,
     onChange: options.onChange,
     ctx,
-    attemptOnce: (c, abort) => attemptStream(c, options, abort),
+    attemptOnce: (c, abort) => attemptStream(c, options, auth, abort),
   });
   const [link] = useState(
     (): RemoteLink => ({
       get connection() {
         return c.connection;
       },
-      send: (args) => sendCommand(c, latest.current, args),
+      send: (args) => sendCommand(c, latest.current, auth, args),
     }),
   );
   return link;

package/src/transports/testing.ts

@@ -1,6 +1,5 @@
 import type { ResourceElement } from "@assistant-ui/tap";
 import type { StatewireEnvelope } from "../client/connection";
-import { recordMutations } from "../util/recordMutations";
 import { StatewireSSE } from "./sse";
 import type { Statewire } from "../types";
 import { openSSEStream, sseFrame } from "./fake-sse";
@@ -11,11 +10,7 @@ export * from "./fixtures";
 /** Let queued microtasks and op batches drain through the statewire client. */
 export const tick = () => new Promise<void>((r) => setTimeout(r, 0));
 
-/** Run a mutator against a recording draft, as the client does for `optimistic`. */
-export const applyOptimistic = <TState>(
-  state: TState,
-  mutate: (draft: TState) => void,
-): TState => recordMutations(state, mutate).state;
+export { applyOptimistic } from "../util/recordMutations";
 
 export type MockTransport<TState> = {
   /** Pass as the `transport:` option of useStatewire / Harness. */

package/src/transports/ws.ts

@@ -29,11 +29,13 @@ type PendingCommand = {
 type WsChannel = {
   socket: WebSocket | null;
   pending: PendingCommand[];
+  authStale: boolean;
 };
 
 const OPEN = 1;
 
 export const WS_SUBPROTOCOL = "statewire.v1";
+export const UNAUTHORIZED_CLOSE = 4401;
 
 const socketUrl = async (
   options: StatewireClient.WSOptions,
@@ -48,6 +50,23 @@ const socketUrl = async (
   return `${base}/ws?${params.toString()}`;
 };
 
+const headersFrame = async (
+  options: StatewireClient.WSOptions,
+  stale: boolean,
+): Promise<string | null> => {
+  const { headers } = options;
+  if (headers === undefined) return null;
+  const resolved = new Headers(
+    typeof headers === "function"
+      ? await headers({ staleAuth: stale })
+      : headers,
+  );
+  return JSON.stringify({
+    type: "headers",
+    headers: Object.fromEntries(resolved),
+  });
+};
+
 const settleOverloaded = (
   w: WsChannel,
   { seq, retryAfterMs }: { seq?: number; retryAfterMs?: number },
@@ -81,6 +100,8 @@ const runSocket = (
   c: Conn,
   w: WsChannel,
   ws: WebSocket,
+  frame: string | null,
+  staleAuth: boolean,
   abort: AbortController,
 ): Promise<"stop" | "reattach"> =>
   new Promise((resolve, reject) => {
@@ -98,6 +119,7 @@ const runSocket = (
     arm();
     abort.signal.addEventListener("abort", () => ws.close());
     ws.onerror = () => {};
+    if (frame !== null) ws.onopen = () => ws.send(frame);
     const stale = () => Boolean(outcome) || closed || abort.signal.aborted;
     const fail = (error: unknown) => {
       clearTimeout(timer);
@@ -141,12 +163,16 @@ const runSocket = (
         fail(error);
       }
     };
-    ws.onclose = () => {
+    ws.onclose = (event?: { code?: number }) => {
       closed = true;
       clearTimeout(timer);
       if (w.socket === ws) failPending(w, new Error("network"));
       if (outcome) resolve(outcome);
-      else reject(new Error(timedOut ? "timeout" : "network"));
+      else if (event?.code === UNAUTHORIZED_CLOSE) {
+        if (staleAuth) c.episode = { cause: "unauthorized" };
+        w.authStale = true;
+        reject(new Error("unauthorized"));
+      } else reject(new Error(timedOut ? "timeout" : "network"));
     };
   });
 
@@ -159,6 +185,9 @@ const attemptSocket = async (
   let ws: WebSocket | null = null;
   try {
     const url = await socketUrl(options, c.ctx.clientId);
+    const stale = w.authStale;
+    w.authStale = false;
+    const frame = await headersFrame(options, stale);
     abort.signal.throwIfAborted();
     const { protocols, webSocket } = options;
     const WS = webSocket ?? globalThis.WebSocket;
@@ -171,7 +200,7 @@ const attemptSocket = async (
           : protocols;
     ws = new WS(url, [WS_SUBPROTOCOL, ...extra]);
     w.socket = ws;
-    return await runSocket(c, w, ws, abort);
+    return await runSocket(c, w, ws, frame, stale, abort);
   } finally {
     if (w.socket === ws) w.socket = null;
   }
@@ -219,7 +248,9 @@ const useWsLink = (
   options: StatewireClient.WSOptions,
   ctx: StatewireRemoteProps,
 ): RemoteLink => {
-  const [w] = useState((): WsChannel => ({ socket: null, pending: [] }));
+  const [w] = useState(
+    (): WsChannel => ({ socket: null, pending: [], authStale: false }),
+  );
   const c = useStatewireConn({
     url: options.url,
     onChange: options.onChange,