@latticexyz/react 2.0.0-alpha.1.254 → 2.0.0-alpha.1.255

package/dist/index.js

@@ -1,2 +1,2 @@
-import{defineQuery as c,getComponentValue as p,Has as d,isComponentUpdate as C}from"@latticexyz/recs";import{useEffect as S,useState as x}from"react";function z(e,t,n){let[s,o]=x(t!=null?p(e,t):void 0);return S(()=>{if(o(t!=null?p(e,t):void 0),t==null)return;let u=c([d(e)],{runOnInit:!1}).update$.subscribe(r=>{if(C(r,e)&&r.entity===t){let[m]=r.value;o(m)}});return()=>u.unsubscribe()},[e,t]),s??n}import{useEffect as T,useState as g}from"react";var G=e=>{let[t,n]=g(e.get());return T(()=>{let s=e.observe_(()=>n(e.get()));return()=>s()},[e]),t};import{defineQuery as v}from"@latticexyz/recs";import{useEffect as E,useMemo as R,useState as h}from"react";import{useEffect as y,useState as V}from"react";import O from"fast-deep-equal";var a=e=>{let[t,n]=V(e);return y(()=>{O(e,t)||n(e)},[e]),t};import M from"fast-deep-equal";import{distinctUntilChanged as D,map as k}from"rxjs";function ne(e,t){let n=t?.updateOnValueChange??!0,s=a(e),o=R(()=>v(s,{runOnInit:!0}),[s]),[i,u]=h([...o.matching]);return E(()=>{u([...o.matching]);let r=o.update$.pipe(k(()=>[...o.matching]));n||(r=r.pipe(D((f,b)=>M(f,b))));let m=r.subscribe(f=>u(f));return()=>m.unsubscribe()},[o,n]),i}import{useEffect as w,useState as F}from"react";function re(e,t){let[n,s]=F(t);return w(()=>{let o=e.subscribe(s);return()=>o.unsubscribe()},[e]),n}import{useEffect as q,useState as Q}from"react";function l(e,t){let[n,s]=Q([]),o=a(t);return q(()=>(s(e.scan(t)),e.subscribe(()=>{s(e.scan(t))},t)),[o]),n}function pe(e,t){let{namespace:n,table:s,key:o}=t;return l(e,{namespace:n,table:s,key:{eq:o}})[0]}export{z as useComponentValue,G as useDeprecatedComputedValue,ne as useEntityQuery,re as useObservableValue,pe as useRow,l as useRows};
+import{defineQuery as d,getComponentValue as p,Has as C,isComponentUpdate as S}from"@latticexyz/recs";import{useEffect as T,useState as x}from"react";function G(e,t,s){let[o,n]=x(t!=null?p(e,t):void 0);return T(()=>{if(n(t!=null?p(e,t):void 0),t==null)return;let a=d([C(e)],{runOnInit:!1}).update$.subscribe(u=>{if(S(u,e)&&u.entity===t){let[f]=u.value;n(f)}});return()=>a.unsubscribe()},[e,t]),o??s}import{useEffect as g,useState as y}from"react";var N=e=>{let[t,s]=y(e.get());return g(()=>{let o=e.observe_(()=>s(e.get()));return()=>o()},[e]),t};import{defineQuery as E}from"@latticexyz/recs";import{useEffect as R,useMemo as h,useState as M}from"react";import{useEffect as V,useState as O}from"react";import v from"fast-deep-equal";var i=e=>{let[t,s]=O(e);return V(()=>{v(e,t)||s(e)},[e]),t};import D from"fast-deep-equal";import{distinctUntilChanged as k,map as w}from"rxjs";function ue(e,t){let s=t?.updateOnValueChange??!0,o=i(e),n=h(()=>E(o,{runOnInit:!0}),[o]),[r,a]=M([...n.matching]);return R(()=>{a([...n.matching]);let u=n.update$.pipe(w(()=>[...n.matching]));s||(u=u.pipe(k((m,b)=>D(m,b))));let f=u.subscribe(m=>a(m));return()=>f.unsubscribe()},[n,s]),r}import{useEffect as F,useState as q}from"react";function fe(e,t){let[s,o]=q(t);return F(()=>{let n=e.subscribe(o);return()=>n.unsubscribe()},[e]),s}import{useEffect as K}from"react";import{useCallback as I,useEffect as Q,useRef as U,useState as A}from"react";function l(e){let[t,s]=A(e),o=U(!1);Q(()=>(o.current=!0,()=>{o.current=!1}));let n=I((...r)=>{o.current?s(...r):console.warn("Ignoring `setState` call because component unmounted",...r)},[]);return[t,n]}function c(e,t){let[s,o]=l([]),n=i(t);return K(()=>{e.scan(n).then(o);let r=e.subscribe(()=>{e.scan(n).then(o)},n);return()=>{r.then(a=>a())}},[n,o,e]),s}function ye(e,t){let{namespace:s,table:o,key:n}=t;return c(e,{namespace:s,table:o,key:{eq:n}})[0]}export{G as useComponentValue,N as useDeprecatedComputedValue,ue as useEntityQuery,fe as useObservableValue,ye as useRow,c as useRows};
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useComponentValue.ts","../src/useDeprecatedComputedValue.ts","../src/useEntityQuery.ts","../src/utils/useDeepMemo.ts","../src/useObservableValue.ts","../src/store-cache/useRows.ts","../src/store-cache/useRow.ts"],"sourcesContent":["import {\n  Component,\n  ComponentValue,\n  defineQuery,\n  Entity,\n  getComponentValue,\n  Has,\n  isComponentUpdate,\n  Metadata,\n  Schema,\n} from \"@latticexyz/recs\";\nimport { useEffect, useState } from \"react\";\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined,\n  defaultValue: ComponentValue<S>\n): ComponentValue<S>;\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined\n): ComponentValue<S> | undefined;\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined,\n  defaultValue?: ComponentValue<S>\n) {\n  const [value, setValue] = useState(entity != null ? getComponentValue(component, entity) : undefined);\n\n  useEffect(() => {\n    // component or entity changed, update state to latest value\n    setValue(entity != null ? getComponentValue(component, entity) : undefined);\n    if (entity == null) return;\n\n    const queryResult = defineQuery([Has(component)], { runOnInit: false });\n    const subscription = queryResult.update$.subscribe((update) => {\n      if (isComponentUpdate(update, component) && update.entity === entity) {\n        const [nextValue] = update.value;\n        setValue(nextValue);\n      }\n    });\n    return () => subscription.unsubscribe();\n  }, [component, entity]);\n\n  return value ?? defaultValue;\n}\n","import { IComputedValue } from \"mobx\";\nimport { useEffect, useState } from \"react\";\n\n/** @deprecated See https://github.com/latticexyz/mud/issues/339 */\nexport const useDeprecatedComputedValue = <T>(computedValue: IComputedValue<T> & { observe_: any }) => {\n  const [value, setValue] = useState<T>(computedValue.get());\n\n  useEffect(() => {\n    const unsubscribe = computedValue.observe_(() => setValue(computedValue.get()));\n    return () => unsubscribe();\n  }, [computedValue]);\n\n  return value;\n};\n","import { defineQuery, QueryFragment } from \"@latticexyz/recs\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { useDeepMemo } from \"./utils/useDeepMemo\";\nimport isEqual from \"fast-deep-equal\";\nimport { distinctUntilChanged, map } from \"rxjs\";\n\n// This does a little more rendering than is necessary when arguments change,\n// but at least it's giving correct results now. Will optimize later!\n\n/**\n * Returns all matching entities for a given entity query,\n * and triggers a re-render as new query results come in.\n *\n * @param fragments Query fragments to match against, executed from left to right.\n * @param options.updateOnValueChange False - re-renders only on entity array changes. True (default) - also on component value changes.\n * @returns Set of entities matching the query fragments.\n */\nexport function useEntityQuery(fragments: QueryFragment[], options?: { updateOnValueChange?: boolean }) {\n  const updateOnValueChange = options?.updateOnValueChange ?? true;\n\n  const stableFragments = useDeepMemo(fragments);\n  const query = useMemo(() => defineQuery(stableFragments, { runOnInit: true }), [stableFragments]);\n  const [entities, setEntities] = useState([...query.matching]);\n\n  useEffect(() => {\n    setEntities([...query.matching]);\n    let observable = query.update$.pipe(map(() => [...query.matching]));\n    if (!updateOnValueChange) {\n      // re-render only on entity array changes\n      observable = observable.pipe(distinctUntilChanged((a, b) => isEqual(a, b)));\n    }\n    const subscription = observable.subscribe((entities) => setEntities(entities));\n    return () => subscription.unsubscribe();\n  }, [query, updateOnValueChange]);\n\n  return entities;\n}\n","import { useEffect, useState } from \"react\";\nimport isEqual from \"fast-deep-equal\";\n\nexport const useDeepMemo = <T>(currentValue: T): T => {\n  const [stableValue, setStableValue] = useState(currentValue);\n\n  useEffect(() => {\n    if (!isEqual(currentValue, stableValue)) {\n      setStableValue(currentValue);\n    }\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [currentValue]);\n\n  return stableValue;\n};\n","import { useEffect, useState } from \"react\";\nimport { Observable } from \"rxjs\";\n\nexport function useObservableValue<T>(observable: Observable<T>, defaultValue: T): T;\n\nexport function useObservableValue<T>(observable: Observable<T>): T | undefined;\n\nexport function useObservableValue<T>(observable: Observable<T>, defaultValue?: T) {\n  const [value, setValue] = useState(defaultValue);\n\n  useEffect(() => {\n    const subscription = observable.subscribe(setValue);\n    return () => subscription.unsubscribe();\n  }, [observable]);\n\n  return value;\n}\n","import { DatabaseClient, FilterOptions, ScanResult } from \"@latticexyz/store-cache\";\nimport { StoreConfig } from \"@latticexyz/store\";\nimport { useEffect, useState } from \"react\";\nimport { useDeepMemo } from \"../utils/useDeepMemo\";\n\n/**\n * Returns an array of all rows matching the provided filter\n */\nexport function useRows<C extends StoreConfig, T extends keyof C[\"tables\"] & string>(\n  storeCache: DatabaseClient<C>,\n  filter?: FilterOptions<C, T>\n) {\n  const [rows, setRows] = useState<ScanResult<C, T>>([]);\n  const filterMemo = useDeepMemo(filter);\n\n  useEffect(() => {\n    setRows(storeCache.scan(filter));\n\n    const unsubscribe = storeCache.subscribe(() => {\n      // very naive implementation for now, but easier and probably more efficient than\n      // manually looping through the `rows` array for every update event\n      setRows(storeCache.scan(filter));\n    }, filter);\n\n    return unsubscribe;\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [filterMemo]);\n\n  return rows;\n}\n","import { StoreConfig } from \"@latticexyz/store\";\nimport { DatabaseClient, Key, ScanResult } from \"@latticexyz/store-cache\";\nimport { useRows } from \"./useRows\";\n\nexport type UseRowFilterOptions<\n  C extends StoreConfig = StoreConfig,\n  T extends keyof C[\"tables\"] & string = keyof C[\"tables\"] & string\n> = {\n  namespace?: C[\"namespace\"];\n  table: T;\n  key: Key<C, T>;\n};\n\n/**\n * Returns a single row of a given table at the given key, updates when the key changes\n */\nexport function useRow<C extends StoreConfig, T extends keyof C[\"tables\"] & string>(\n  storeCache: DatabaseClient<C>,\n  filter: UseRowFilterOptions<C, T>\n): ScanResult<C, T>[0] | undefined {\n  const { namespace, table, key } = filter;\n  return useRows(storeCache, { namespace, table, key: { eq: key } })[0];\n}\n"],"mappings":"AAAA,OAGE,eAAAA,EAEA,qBAAAC,EACA,OAAAC,EACA,qBAAAC,MAGK,mBACP,OAAS,aAAAC,EAAW,YAAAC,MAAgB,QAa7B,SAASC,EACdC,EACAC,EACAC,EACA,CACA,GAAM,CAACC,EAAOC,CAAQ,EAAIN,EAASG,GAAU,KAAOP,EAAkBM,EAAWC,CAAM,EAAI,MAAS,EAEpG,OAAAJ,EAAU,IAAM,CAGd,GADAO,EAASH,GAAU,KAAOP,EAAkBM,EAAWC,CAAM,EAAI,MAAS,EACtEA,GAAU,KAAM,OAGpB,IAAMI,EADcZ,EAAY,CAACE,EAAIK,CAAS,CAAC,EAAG,CAAE,UAAW,EAAM,CAAC,EACrC,QAAQ,UAAWM,GAAW,CAC7D,GAAIV,EAAkBU,EAAQN,CAAS,GAAKM,EAAO,SAAWL,EAAQ,CACpE,GAAM,CAACM,CAAS,EAAID,EAAO,MAC3BF,EAASG,CAAS,EAEtB,CAAC,EACD,MAAO,IAAMF,EAAa,YAAY,CACxC,EAAG,CAACL,EAAWC,CAAM,CAAC,EAEfE,GAASD,CAClB,CC9CA,OAAS,aAAAM,EAAW,YAAAC,MAAgB,QAG7B,IAAMC,EAAiCC,GAAyD,CACrG,GAAM,CAACC,EAAOC,CAAQ,EAAIJ,EAAYE,EAAc,IAAI,CAAC,EAEzD,OAAAH,EAAU,IAAM,CACd,IAAMM,EAAcH,EAAc,SAAS,IAAME,EAASF,EAAc,IAAI,CAAC,CAAC,EAC9E,MAAO,IAAMG,EAAY,CAC3B,EAAG,CAACH,CAAa,CAAC,EAEXC,CACT,ECbA,OAAS,eAAAG,MAAkC,mBAC3C,OAAS,aAAAC,EAAW,WAAAC,EAAS,YAAAC,MAAgB,QCD7C,OAAS,aAAAC,EAAW,YAAAC,MAAgB,QACpC,OAAOC,MAAa,kBAEb,IAAMC,EAAkBC,GAAuB,CACpD,GAAM,CAACC,EAAaC,CAAc,EAAIL,EAASG,CAAY,EAE3D,OAAAJ,EAAU,IAAM,CACTE,EAAQE,EAAcC,CAAW,GACpCC,EAAeF,CAAY,CAG/B,EAAG,CAACA,CAAY,CAAC,EAEVC,CACT,EDXA,OAAOE,MAAa,kBACpB,OAAS,wBAAAC,EAAsB,OAAAC,MAAW,OAanC,SAASC,GAAeC,EAA4BC,EAA6C,CACtG,IAAMC,EAAsBD,GAAS,qBAAuB,GAEtDE,EAAkBC,EAAYJ,CAAS,EACvCK,EAAQC,EAAQ,IAAMC,EAAYJ,EAAiB,CAAE,UAAW,EAAK,CAAC,EAAG,CAACA,CAAe,CAAC,EAC1F,CAACK,EAAUC,CAAW,EAAIC,EAAS,CAAC,GAAGL,EAAM,QAAQ,CAAC,EAE5D,OAAAM,EAAU,IAAM,CACdF,EAAY,CAAC,GAAGJ,EAAM,QAAQ,CAAC,EAC/B,IAAIO,EAAaP,EAAM,QAAQ,KAAKP,EAAI,IAAM,CAAC,GAAGO,EAAM,QAAQ,CAAC,CAAC,EAC7DH,IAEHU,EAAaA,EAAW,KAAKf,EAAqB,CAACgB,EAAG,IAAMjB,EAAQiB,EAAG,CAAC,CAAC,CAAC,GAE5E,IAAMC,EAAeF,EAAW,UAAWJ,GAAaC,EAAYD,CAAQ,CAAC,EAC7E,MAAO,IAAMM,EAAa,YAAY,CACxC,EAAG,CAACT,EAAOH,CAAmB,CAAC,EAExBM,CACT,CEpCA,OAAS,aAAAO,EAAW,YAAAC,MAAgB,QAO7B,SAASC,GAAsBC,EAA2BC,EAAkB,CACjF,GAAM,CAACC,EAAOC,CAAQ,EAAIL,EAASG,CAAY,EAE/C,OAAAJ,EAAU,IAAM,CACd,IAAMO,EAAeJ,EAAW,UAAUG,CAAQ,EAClD,MAAO,IAAMC,EAAa,YAAY,CACxC,EAAG,CAACJ,CAAU,CAAC,EAERE,CACT,CCdA,OAAS,aAAAG,EAAW,YAAAC,MAAgB,QAM7B,SAASC,EACdC,EACAC,EACA,CACA,GAAM,CAACC,EAAMC,CAAO,EAAIC,EAA2B,CAAC,CAAC,EAC/CC,EAAaC,EAAYL,CAAM,EAErC,OAAAM,EAAU,KACRJ,EAAQH,EAAW,KAAKC,CAAM,CAAC,EAEXD,EAAW,UAAU,IAAM,CAG7CG,EAAQH,EAAW,KAAKC,CAAM,CAAC,CACjC,EAAGA,CAAM,GAIR,CAACI,CAAU,CAAC,EAERH,CACT,CCbO,SAASM,GACdC,EACAC,EACiC,CACjC,GAAM,CAAE,UAAAC,EAAW,MAAAC,EAAO,IAAAC,CAAI,EAAIH,EAClC,OAAOI,EAAQL,EAAY,CAAE,UAAAE,EAAW,MAAAC,EAAO,IAAK,CAAE,GAAIC,CAAI,CAAE,CAAC,EAAE,CAAC,CACtE","names":["defineQuery","getComponentValue","Has","isComponentUpdate","useEffect","useState","useComponentValue","component","entity","defaultValue","value","setValue","subscription","update","nextValue","useEffect","useState","useDeprecatedComputedValue","computedValue","value","setValue","unsubscribe","defineQuery","useEffect","useMemo","useState","useEffect","useState","isEqual","useDeepMemo","currentValue","stableValue","setStableValue","isEqual","distinctUntilChanged","map","useEntityQuery","fragments","options","updateOnValueChange","stableFragments","useDeepMemo","query","useMemo","defineQuery","entities","setEntities","useState","useEffect","observable","a","subscription","useEffect","useState","useObservableValue","observable","defaultValue","value","setValue","subscription","useEffect","useState","useRows","storeCache","filter","rows","setRows","useState","filterMemo","useDeepMemo","useEffect","useRow","storeCache","filter","namespace","table","key","useRows"]}
+{"version":3,"sources":["../src/useComponentValue.ts","../src/useDeprecatedComputedValue.ts","../src/useEntityQuery.ts","../src/utils/useDeepMemo.ts","../src/useObservableValue.ts","../src/store-cache/useRows.ts","../src/utils/useMountedState.ts","../src/store-cache/useRow.ts"],"sourcesContent":["import {\n  Component,\n  ComponentValue,\n  defineQuery,\n  Entity,\n  getComponentValue,\n  Has,\n  isComponentUpdate,\n  Metadata,\n  Schema,\n} from \"@latticexyz/recs\";\nimport { useEffect, useState } from \"react\";\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined,\n  defaultValue: ComponentValue<S>\n): ComponentValue<S>;\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined\n): ComponentValue<S> | undefined;\n\nexport function useComponentValue<S extends Schema>(\n  component: Component<S, Metadata, undefined>,\n  entity: Entity | undefined,\n  defaultValue?: ComponentValue<S>\n) {\n  const [value, setValue] = useState(entity != null ? getComponentValue(component, entity) : undefined);\n\n  useEffect(() => {\n    // component or entity changed, update state to latest value\n    setValue(entity != null ? getComponentValue(component, entity) : undefined);\n    if (entity == null) return;\n\n    const queryResult = defineQuery([Has(component)], { runOnInit: false });\n    const subscription = queryResult.update$.subscribe((update) => {\n      if (isComponentUpdate(update, component) && update.entity === entity) {\n        const [nextValue] = update.value;\n        setValue(nextValue);\n      }\n    });\n    return () => subscription.unsubscribe();\n  }, [component, entity]);\n\n  return value ?? defaultValue;\n}\n","import { IComputedValue } from \"mobx\";\nimport { useEffect, useState } from \"react\";\n\n/** @deprecated See https://github.com/latticexyz/mud/issues/339 */\nexport const useDeprecatedComputedValue = <T>(computedValue: IComputedValue<T> & { observe_: any }) => {\n  const [value, setValue] = useState<T>(computedValue.get());\n\n  useEffect(() => {\n    const unsubscribe = computedValue.observe_(() => setValue(computedValue.get()));\n    return () => unsubscribe();\n  }, [computedValue]);\n\n  return value;\n};\n","import { defineQuery, QueryFragment } from \"@latticexyz/recs\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { useDeepMemo } from \"./utils/useDeepMemo\";\nimport isEqual from \"fast-deep-equal\";\nimport { distinctUntilChanged, map } from \"rxjs\";\n\n// This does a little more rendering than is necessary when arguments change,\n// but at least it's giving correct results now. Will optimize later!\n\n/**\n * Returns all matching entities for a given entity query,\n * and triggers a re-render as new query results come in.\n *\n * @param fragments Query fragments to match against, executed from left to right.\n * @param options.updateOnValueChange False - re-renders only on entity array changes. True (default) - also on component value changes.\n * @returns Set of entities matching the query fragments.\n */\nexport function useEntityQuery(fragments: QueryFragment[], options?: { updateOnValueChange?: boolean }) {\n  const updateOnValueChange = options?.updateOnValueChange ?? true;\n\n  const stableFragments = useDeepMemo(fragments);\n  const query = useMemo(() => defineQuery(stableFragments, { runOnInit: true }), [stableFragments]);\n  const [entities, setEntities] = useState([...query.matching]);\n\n  useEffect(() => {\n    setEntities([...query.matching]);\n    let observable = query.update$.pipe(map(() => [...query.matching]));\n    if (!updateOnValueChange) {\n      // re-render only on entity array changes\n      observable = observable.pipe(distinctUntilChanged((a, b) => isEqual(a, b)));\n    }\n    const subscription = observable.subscribe((entities) => setEntities(entities));\n    return () => subscription.unsubscribe();\n  }, [query, updateOnValueChange]);\n\n  return entities;\n}\n","import { useEffect, useState } from \"react\";\nimport isEqual from \"fast-deep-equal\";\n\nexport const useDeepMemo = <T>(currentValue: T): T => {\n  const [stableValue, setStableValue] = useState(currentValue);\n\n  useEffect(() => {\n    if (!isEqual(currentValue, stableValue)) {\n      setStableValue(currentValue);\n    }\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [currentValue]);\n\n  return stableValue;\n};\n","import { useEffect, useState } from \"react\";\nimport { Observable } from \"rxjs\";\n\nexport function useObservableValue<T>(observable: Observable<T>, defaultValue: T): T;\n\nexport function useObservableValue<T>(observable: Observable<T>): T | undefined;\n\nexport function useObservableValue<T>(observable: Observable<T>, defaultValue?: T) {\n  const [value, setValue] = useState(defaultValue);\n\n  useEffect(() => {\n    const subscription = observable.subscribe(setValue);\n    return () => subscription.unsubscribe();\n  }, [observable]);\n\n  return value;\n}\n","import { DatabaseClient, FilterOptions, ScanResult } from \"@latticexyz/store-cache\";\nimport { StoreConfig } from \"@latticexyz/store\";\nimport { useEffect, useState } from \"react\";\nimport { useDeepMemo } from \"../utils/useDeepMemo\";\nimport { useMountedState } from \"../utils/useMountedState\";\n\n/**\n * Returns an array of all rows matching the provided filter\n */\nexport function useRows<C extends StoreConfig, T extends keyof C[\"tables\"] & string>(\n  storeCache: DatabaseClient<C>,\n  filter?: FilterOptions<C, T>\n) {\n  const [rows, setRows] = useMountedState<ScanResult<C, T>>([]);\n  const filterMemo = useDeepMemo(filter);\n\n  useEffect(() => {\n    storeCache.scan(filterMemo).then(setRows);\n\n    const unsubscribePromise = storeCache.subscribe(() => {\n      // very naive implementation for now, but easier and probably more efficient than\n      // manually looping through the `rows` array for every update event\n      storeCache.scan(filterMemo).then(setRows);\n    }, filterMemo);\n\n    return () => {\n      unsubscribePromise.then((unsubscribe) => unsubscribe());\n    };\n  }, [filterMemo, setRows, storeCache]);\n\n  return rows;\n}\n","import { Dispatch, SetStateAction, useCallback, useEffect, useRef, useState } from \"react\";\n\n/**\n * Use in place of useState when the component may be unmounted before the state is updated.\n */\nexport function useMountedState<T>(initialState: T | (() => T)): [T, Dispatch<SetStateAction<T>>] {\n  const [state, setState] = useState<T>(initialState);\n  const mountedRef = useRef(false);\n  useEffect(() => {\n    mountedRef.current = true;\n    return () => {\n      mountedRef.current = false;\n    };\n  });\n  const mountedSetState: typeof setState = useCallback((...args) => {\n    if (mountedRef.current) {\n      setState(...args);\n    } else {\n      console.warn(\"Ignoring `setState` call because component unmounted\", ...args);\n    }\n  }, []);\n  return [state, mountedSetState];\n}\n","import { StoreConfig } from \"@latticexyz/store\";\nimport { DatabaseClient, Key, ScanResult } from \"@latticexyz/store-cache\";\nimport { useRows } from \"./useRows\";\n\nexport type UseRowFilterOptions<\n  C extends StoreConfig = StoreConfig,\n  T extends keyof C[\"tables\"] & string = keyof C[\"tables\"] & string\n> = {\n  namespace?: C[\"namespace\"];\n  table: T;\n  key: Key<C, T>;\n};\n\n/**\n * Returns a single row of a given table at the given key, updates when the key changes\n */\nexport function useRow<C extends StoreConfig, T extends keyof C[\"tables\"] & string>(\n  storeCache: DatabaseClient<C>,\n  filter: UseRowFilterOptions<C, T>\n): ScanResult<C, T>[0] | undefined {\n  const { namespace, table, key } = filter;\n  return useRows(storeCache, { namespace, table, key: { eq: key } })[0];\n}\n"],"mappings":"AAAA,OAGE,eAAAA,EAEA,qBAAAC,EACA,OAAAC,EACA,qBAAAC,MAGK,mBACP,OAAS,aAAAC,EAAW,YAAAC,MAAgB,QAa7B,SAASC,EACdC,EACAC,EACAC,EACA,CACA,GAAM,CAACC,EAAOC,CAAQ,EAAIN,EAASG,GAAU,KAAOP,EAAkBM,EAAWC,CAAM,EAAI,MAAS,EAEpG,OAAAJ,EAAU,IAAM,CAGd,GADAO,EAASH,GAAU,KAAOP,EAAkBM,EAAWC,CAAM,EAAI,MAAS,EACtEA,GAAU,KAAM,OAGpB,IAAMI,EADcZ,EAAY,CAACE,EAAIK,CAAS,CAAC,EAAG,CAAE,UAAW,EAAM,CAAC,EACrC,QAAQ,UAAWM,GAAW,CAC7D,GAAIV,EAAkBU,EAAQN,CAAS,GAAKM,EAAO,SAAWL,EAAQ,CACpE,GAAM,CAACM,CAAS,EAAID,EAAO,MAC3BF,EAASG,CAAS,EAEtB,CAAC,EACD,MAAO,IAAMF,EAAa,YAAY,CACxC,EAAG,CAACL,EAAWC,CAAM,CAAC,EAEfE,GAASD,CAClB,CC9CA,OAAS,aAAAM,EAAW,YAAAC,MAAgB,QAG7B,IAAMC,EAAiCC,GAAyD,CACrG,GAAM,CAACC,EAAOC,CAAQ,EAAIJ,EAAYE,EAAc,IAAI,CAAC,EAEzD,OAAAH,EAAU,IAAM,CACd,IAAMM,EAAcH,EAAc,SAAS,IAAME,EAASF,EAAc,IAAI,CAAC,CAAC,EAC9E,MAAO,IAAMG,EAAY,CAC3B,EAAG,CAACH,CAAa,CAAC,EAEXC,CACT,ECbA,OAAS,eAAAG,MAAkC,mBAC3C,OAAS,aAAAC,EAAW,WAAAC,EAAS,YAAAC,MAAgB,QCD7C,OAAS,aAAAC,EAAW,YAAAC,MAAgB,QACpC,OAAOC,MAAa,kBAEb,IAAMC,EAAkBC,GAAuB,CACpD,GAAM,CAACC,EAAaC,CAAc,EAAIL,EAASG,CAAY,EAE3D,OAAAJ,EAAU,IAAM,CACTE,EAAQE,EAAcC,CAAW,GACpCC,EAAeF,CAAY,CAG/B,EAAG,CAACA,CAAY,CAAC,EAEVC,CACT,EDXA,OAAOE,MAAa,kBACpB,OAAS,wBAAAC,EAAsB,OAAAC,MAAW,OAanC,SAASC,GAAeC,EAA4BC,EAA6C,CACtG,IAAMC,EAAsBD,GAAS,qBAAuB,GAEtDE,EAAkBC,EAAYJ,CAAS,EACvCK,EAAQC,EAAQ,IAAMC,EAAYJ,EAAiB,CAAE,UAAW,EAAK,CAAC,EAAG,CAACA,CAAe,CAAC,EAC1F,CAACK,EAAUC,CAAW,EAAIC,EAAS,CAAC,GAAGL,EAAM,QAAQ,CAAC,EAE5D,OAAAM,EAAU,IAAM,CACdF,EAAY,CAAC,GAAGJ,EAAM,QAAQ,CAAC,EAC/B,IAAIO,EAAaP,EAAM,QAAQ,KAAKP,EAAI,IAAM,CAAC,GAAGO,EAAM,QAAQ,CAAC,CAAC,EAC7DH,IAEHU,EAAaA,EAAW,KAAKf,EAAqB,CAACgB,EAAG,IAAMjB,EAAQiB,EAAG,CAAC,CAAC,CAAC,GAE5E,IAAMC,EAAeF,EAAW,UAAWJ,GAAaC,EAAYD,CAAQ,CAAC,EAC7E,MAAO,IAAMM,EAAa,YAAY,CACxC,EAAG,CAACT,EAAOH,CAAmB,CAAC,EAExBM,CACT,CEpCA,OAAS,aAAAO,EAAW,YAAAC,MAAgB,QAO7B,SAASC,GAAsBC,EAA2BC,EAAkB,CACjF,GAAM,CAACC,EAAOC,CAAQ,EAAIL,EAASG,CAAY,EAE/C,OAAAJ,EAAU,IAAM,CACd,IAAMO,EAAeJ,EAAW,UAAUG,CAAQ,EAClD,MAAO,IAAMC,EAAa,YAAY,CACxC,EAAG,CAACJ,CAAU,CAAC,EAERE,CACT,CCdA,OAAS,aAAAG,MAA2B,QCFpC,OAAmC,eAAAC,EAAa,aAAAC,EAAW,UAAAC,EAAQ,YAAAC,MAAgB,QAK5E,SAASC,EAAmBC,EAA+D,CAChG,GAAM,CAACC,EAAOC,CAAQ,EAAIJ,EAAYE,CAAY,EAC5CG,EAAaN,EAAO,EAAK,EAC/BD,EAAU,KACRO,EAAW,QAAU,GACd,IAAM,CACXA,EAAW,QAAU,EACvB,EACD,EACD,IAAMC,EAAmCT,EAAY,IAAIU,IAAS,CAC5DF,EAAW,QACbD,EAAS,GAAGG,CAAI,EAEhB,QAAQ,KAAK,uDAAwD,GAAGA,CAAI,CAEhF,EAAG,CAAC,CAAC,EACL,MAAO,CAACJ,EAAOG,CAAe,CAChC,CDbO,SAASE,EACdC,EACAC,EACA,CACA,GAAM,CAACC,EAAMC,CAAO,EAAIC,EAAkC,CAAC,CAAC,EACtDC,EAAaC,EAAYL,CAAM,EAErC,OAAAM,EAAU,IAAM,CACdP,EAAW,KAAKK,CAAU,EAAE,KAAKF,CAAO,EAExC,IAAMK,EAAqBR,EAAW,UAAU,IAAM,CAGpDA,EAAW,KAAKK,CAAU,EAAE,KAAKF,CAAO,CAC1C,EAAGE,CAAU,EAEb,MAAO,IAAM,CACXG,EAAmB,KAAMC,GAAgBA,EAAY,CAAC,CACxD,CACF,EAAG,CAACJ,EAAYF,EAASH,CAAU,CAAC,EAE7BE,CACT,CEfO,SAASQ,GACdC,EACAC,EACiC,CACjC,GAAM,CAAE,UAAAC,EAAW,MAAAC,EAAO,IAAAC,CAAI,EAAIH,EAClC,OAAOI,EAAQL,EAAY,CAAE,UAAAE,EAAW,MAAAC,EAAO,IAAK,CAAE,GAAIC,CAAI,CAAE,CAAC,EAAE,CAAC,CACtE","names":["defineQuery","getComponentValue","Has","isComponentUpdate","useEffect","useState","useComponentValue","component","entity","defaultValue","value","setValue","subscription","update","nextValue","useEffect","useState","useDeprecatedComputedValue","computedValue","value","setValue","unsubscribe","defineQuery","useEffect","useMemo","useState","useEffect","useState","isEqual","useDeepMemo","currentValue","stableValue","setStableValue","isEqual","distinctUntilChanged","map","useEntityQuery","fragments","options","updateOnValueChange","stableFragments","useDeepMemo","query","useMemo","defineQuery","entities","setEntities","useState","useEffect","observable","a","subscription","useEffect","useState","useObservableValue","observable","defaultValue","value","setValue","subscription","useEffect","useCallback","useEffect","useRef","useState","useMountedState","initialState","state","setState","mountedRef","mountedSetState","args","useRows","storeCache","filter","rows","setRows","useMountedState","filterMemo","useDeepMemo","useEffect","unsubscribePromise","unsubscribe","useRow","storeCache","filter","namespace","table","key","useRows"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@latticexyz/react",
-  "version": "2.0.0-alpha.1.254+ddd17653",
+  "version": "2.0.0-alpha.1.255+2094bf35",
   "description": "React tools for MUD client.",
   "repository": {
     "type": "git",
@@ -23,9 +23,9 @@
     "test": "tsc --noEmit && vitest --run"
   },
   "dependencies": {
-    "@latticexyz/recs": "2.0.0-alpha.1.254+ddd17653",
-    "@latticexyz/store": "2.0.0-alpha.1.254+ddd17653",
-    "@latticexyz/store-cache": "2.0.0-alpha.1.254+ddd17653",
+    "@latticexyz/recs": "2.0.0-alpha.1.255+2094bf35",
+    "@latticexyz/store": "2.0.0-alpha.1.255+2094bf35",
+    "@latticexyz/store-cache": "2.0.0-alpha.1.255+2094bf35",
     "fast-deep-equal": "^3.1.3",
     "mobx": "^6.7.0",
     "react": "^18.2.0",
@@ -44,5 +44,5 @@
     "vite": "^4.3.6",
     "vitest": "0.31.4"
   },
-  "gitHead": "ddd176536fc3080c5d6928183c48fd66a3acd24c"
+  "gitHead": "2094bf35e0df6a69b893582ccb4e4e0d72fa9372"
 }

package/src/store-cache/useRow.test.ts

@@ -20,7 +20,7 @@ describe("useRow", () => {
     client = createDatabaseClient(db, config);
   });
 
-  it("should return the row of the position table with the specified key", () => {
+  it("should return the row of the position table with the specified key", async () => {
     const { result } = renderHook(() => useRow(client, { table: "Position", key: { key: "0x01" } }));
     expect(result.current).toBe(undefined);
 
@@ -38,10 +38,14 @@ describe("useRow", () => {
       { key: { first: "0x03", second: 1 }, value: { value: 4 } },
     ];
 
-    act(() => {
+    await act(async () => {
       // Set values in the tables
-      for (const update of positionUpdates) client.tables.Position.set(update.key, update.value);
-      for (const update of multiKeyUpdates) client.tables.MultiKey.set(update.key, update.value);
+      for (const update of positionUpdates) {
+        await client.tables.Position.set(update.key, update.value);
+      }
+      for (const update of multiKeyUpdates) {
+        await client.tables.MultiKey.set(update.key, update.value);
+      }
     });
 
     expect(result.current).toEqual({
@@ -51,20 +55,22 @@ describe("useRow", () => {
       table: "Position",
     });
 
-    act(() => {
-      for (const update of positionUpdates.slice(0, 3)) client.tables.Position.remove(update.key);
+    await act(async () => {
+      for (const update of positionUpdates.slice(0, 3)) {
+        await client.tables.Position.remove(update.key);
+      }
     });
 
     expect(result.current).toBe(undefined);
   });
 
-  it("should re-render only when the position value of the specified key changes", () => {
+  it("should re-render only when the position value of the specified key changes", async () => {
     const { result } = renderHook(() => useRow(client, { table: "Position", key: { key: "0x00" } }));
-    expect(result.all.length).toBe(2);
+    expect(result.all.length).toBe(1);
 
     // Update the position table
-    act(() => {
-      client.tables.Position.set({ key: "0x00" }, { x: 1, y: 2 });
+    await act(async () => {
+      await client.tables.Position.set({ key: "0x00" }, { x: 1, y: 2 });
     });
     expect(result.all.length).toBe(3);
     expect(result.current).toEqual({
@@ -75,8 +81,8 @@ describe("useRow", () => {
     });
 
     // Update an unrelated table
-    act(() => {
-      client.tables.MultiKey.set({ first: "0x03", second: 1 }, { value: 4 });
+    await act(async () => {
+      await client.tables.MultiKey.set({ first: "0x03", second: 1 }, { value: 4 });
     });
     expect(result.all.length).toBe(3);
     expect(result.current).toEqual({
@@ -87,8 +93,8 @@ describe("useRow", () => {
     });
 
     // Update the position table
-    act(() => {
-      client.tables.Position.set({ key: "0x00" }, { x: 2, y: 2 });
+    await act(async () => {
+      await client.tables.Position.set({ key: "0x00" }, { x: 2, y: 2 });
     });
     expect(result.all.length).toBe(4);
     expect(result.current).toEqual({
@@ -99,8 +105,8 @@ describe("useRow", () => {
     });
 
     // Update an unrelated key
-    act(() => {
-      client.tables.Position.set({ key: "0x01" }, { x: 2, y: 2 });
+    await act(async () => {
+      await client.tables.Position.set({ key: "0x01" }, { x: 2, y: 2 });
     });
     expect(result.all.length).toBe(4);
     expect(result.current).toEqual({
@@ -111,21 +117,21 @@ describe("useRow", () => {
     });
 
     // Update the position table
-    act(() => {
-      client.tables.Position.remove({ key: "0x00" });
+    await act(async () => {
+      await client.tables.Position.remove({ key: "0x00" });
     });
     expect(result.all.length).toBe(5);
     expect(result.current).toEqual(undefined);
   });
 
-  it("should re-render when the filter changes", () => {
-    const { result, rerender } = renderHook(({ filter }) => useRow(client, filter), {
+  it("should re-render when the filter changes", async () => {
+    const { result, rerender, waitForNextUpdate } = renderHook(({ filter }) => useRow(client, filter), {
       initialProps: {
         filter: { table: "Position", key: { key: "0x01" } } as UseRowFilterOptions<typeof config>,
       },
     });
 
-    expect(result.all.length).toBe(2);
+    expect(result.all.length).toBe(1);
     expect(result.current).toBe(undefined);
 
     const positionUpdates: KeyValue<typeof config, "Position">[] = [
@@ -142,10 +148,14 @@ describe("useRow", () => {
       { key: { first: "0x03", second: 1 }, value: { value: 4 } },
     ];
 
-    act(() => {
+    await act(async () => {
       // Set values in the tables
-      for (const update of positionUpdates) client.tables.Position.set(update.key, update.value);
-      for (const update of multiKeyUpdates) client.tables.MultiKey.set(update.key, update.value);
+      for (const update of positionUpdates) {
+        await client.tables.Position.set(update.key, update.value);
+      }
+      for (const update of multiKeyUpdates) {
+        await client.tables.MultiKey.set(update.key, update.value);
+      }
     });
 
     expect(result.all.length).toBe(3);
@@ -158,6 +168,7 @@ describe("useRow", () => {
 
     // Change the filter
     rerender({ filter: { table: "Position", key: { key: "0x02" } } });
+    await waitForNextUpdate();
 
     // Expect hook to rerender three times:
     // 1. New prop, everything else changes the same
@@ -173,6 +184,7 @@ describe("useRow", () => {
 
     // Change the filter
     rerender({ filter: { table: "MultiKey", key: { first: "0x00", second: 4 } } });
+    await waitForNextUpdate();
 
     expect(result.all.length).toBe(9);
     expect(result.current).toEqual({

package/src/store-cache/useRows.test.ts

@@ -20,7 +20,7 @@ describe("useRows", () => {
     client = createDatabaseClient(db, config);
   });
 
-  it("should return all rows of the position table", () => {
+  it("should return all rows of the position table", async () => {
     const { result } = renderHook(() => useRows(client, { table: "Position" }));
     expect(result.current.length).toBe(0);
 
@@ -38,10 +38,14 @@ describe("useRows", () => {
       { key: { first: "0x03", second: 1 }, value: { value: 4 } },
     ];
 
-    act(() => {
+    await act(async () => {
       // Set values in the tables
-      for (const update of positionUpdates) client.tables.Position.set(update.key, update.value);
-      for (const update of multiKeyUpdates) client.tables.MultiKey.set(update.key, update.value);
+      for (const update of positionUpdates) {
+        await client.tables.Position.set(update.key, update.value);
+      }
+      for (const update of multiKeyUpdates) {
+        await client.tables.MultiKey.set(update.key, update.value);
+      }
     });
 
     expect(result.current.length).toBe(positionUpdates.length);
@@ -49,8 +53,10 @@ describe("useRows", () => {
       ...positionUpdates.map((row) => ({ ...row, namespace: config["namespace"], table: "Position" })),
     ]);
 
-    act(() => {
-      for (const update of positionUpdates.slice(0, 3)) client.tables.Position.remove(update.key);
+    await act(async () => {
+      for (const update of positionUpdates.slice(0, 3)) {
+        await client.tables.Position.remove(update.key);
+      }
     });
 
     expect(result.current.length).toBe(1);
@@ -59,13 +65,13 @@ describe("useRows", () => {
     ]);
   });
 
-  it("should re-render only when the position table changes", () => {
+  it("should re-render only when the position table changes", async () => {
     const { result } = renderHook(() => useRows(client, { namespace: config["namespace"], table: "Position" }));
-    expect(result.all.length).toBe(2);
+    expect(result.all.length).toBe(1);
 
     // Update the position table
-    act(() => {
-      client.tables.Position.set({ key: "0x00" }, { x: 1, y: 2 });
+    await act(async () => {
+      await client.tables.Position.set({ key: "0x00" }, { x: 1, y: 2 });
     });
     expect(result.all.length).toBe(3);
     expect(result.current).toEqual([
@@ -73,8 +79,8 @@ describe("useRows", () => {
     ]);
 
     // Update an unrelated table
-    act(() => {
-      client.tables.MultiKey.set({ first: "0x03", second: 1 }, { value: 4 });
+    await act(async () => {
+      await client.tables.MultiKey.set({ first: "0x03", second: 1 }, { value: 4 });
     });
     expect(result.all.length).toBe(3);
     expect(result.current).toEqual([
@@ -82,8 +88,8 @@ describe("useRows", () => {
     ]);
 
     // Update the position table
-    act(() => {
-      client.tables.Position.set({ key: "0x00" }, { x: 2, y: 2 });
+    await act(async () => {
+      await client.tables.Position.set({ key: "0x00" }, { x: 2, y: 2 });
     });
     expect(result.all.length).toBe(4);
     expect(result.current).toEqual([
@@ -91,8 +97,8 @@ describe("useRows", () => {
     ]);
 
     // Update an unrelated table
-    act(() => {
-      client.tables.MultiKey.remove({ first: "0x03", second: 1 });
+    await act(async () => {
+      await client.tables.MultiKey.remove({ first: "0x03", second: 1 });
     });
     expect(result.all.length).toBe(4);
     expect(result.current).toEqual([
@@ -100,19 +106,19 @@ describe("useRows", () => {
     ]);
 
     // Update the position table
-    act(() => {
-      client.tables.Position.remove({ key: "0x00" });
+    await act(async () => {
+      await client.tables.Position.remove({ key: "0x00" });
     });
     expect(result.all.length).toBe(5);
     expect(result.current).toEqual([]);
   });
 
-  it("should re-render when the filter changes", () => {
-    const { result, rerender } = renderHook(({ filter }) => useRows(client, filter), {
+  it("should re-render when the filter changes", async () => {
+    const { result, rerender, waitForNextUpdate } = renderHook(({ filter }) => useRows(client, filter), {
       initialProps: { filter: { table: "Position" as keyof (typeof config)["tables"] } },
     });
 
-    expect(result.all.length).toBe(2);
+    expect(result.all.length).toBe(1);
     expect(result.current.length).toBe(0);
 
     const positionUpdates: KeyValue<typeof config, "Position">[] = [
@@ -129,13 +135,17 @@ describe("useRows", () => {
       { key: { first: "0x03", second: 1 }, value: { value: 4 } },
     ];
 
-    act(() => {
+    await act(async () => {
       // Set values in the tables
-      for (const update of positionUpdates) client.tables.Position.set(update.key, update.value);
-      for (const update of multiKeyUpdates) client.tables.MultiKey.set(update.key, update.value);
+      for (const update of positionUpdates) {
+        await client.tables.Position.set(update.key, update.value);
+      }
+      for (const update of multiKeyUpdates) {
+        await client.tables.MultiKey.set(update.key, update.value);
+      }
     });
 
-    expect(result.all.length).toBe(3);
+    expect(result.all.length).toBe(6);
     expect(result.current.length).toBe(positionUpdates.length);
     expect(result.current).toEqual([
       ...positionUpdates.map((row) => ({ ...row, namespace: config["namespace"], table: "Position" })),
@@ -143,12 +153,13 @@ describe("useRows", () => {
 
     // Change the filter
     rerender({ filter: { table: "MultiKey" } });
+    await waitForNextUpdate();
 
     // Expect hook to rerender three times:
     // 1. New prop, everything else changes the same
     // 2. `filterMemo` is updated by `useDeepMemo` because of the new prop
     // 3. `useEffect` runs because of the new `filterMemo`, scan is executed, new rows are returned
-    expect(result.all.length).toBe(6);
+    expect(result.all.length).toBe(9);
     expect(result.current.length).toBe(multiKeyUpdates.length);
     expect(result.current).toEqual([
       ...multiKeyUpdates.map((row) => ({ ...row, namespace: config["namespace"], table: "MultiKey" })),

package/src/store-cache/useRows.ts

@@ -2,6 +2,7 @@ import { DatabaseClient, FilterOptions, ScanResult } from "@latticexyz/store-cac
 import { StoreConfig } from "@latticexyz/store";
 import { useEffect, useState } from "react";
 import { useDeepMemo } from "../utils/useDeepMemo";
+import { useMountedState } from "../utils/useMountedState";
 
 /**
  * Returns an array of all rows matching the provided filter
@@ -10,21 +11,22 @@ export function useRows<C extends StoreConfig, T extends keyof C["tables"] & str
   storeCache: DatabaseClient<C>,
   filter?: FilterOptions<C, T>
 ) {
-  const [rows, setRows] = useState<ScanResult<C, T>>([]);
+  const [rows, setRows] = useMountedState<ScanResult<C, T>>([]);
   const filterMemo = useDeepMemo(filter);
 
   useEffect(() => {
-    setRows(storeCache.scan(filter));
+    storeCache.scan(filterMemo).then(setRows);
 
-    const unsubscribe = storeCache.subscribe(() => {
+    const unsubscribePromise = storeCache.subscribe(() => {
       // very naive implementation for now, but easier and probably more efficient than
       // manually looping through the `rows` array for every update event
-      setRows(storeCache.scan(filter));
-    }, filter);
+      storeCache.scan(filterMemo).then(setRows);
+    }, filterMemo);
 
-    return unsubscribe;
-    // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [filterMemo]);
+    return () => {
+      unsubscribePromise.then((unsubscribe) => unsubscribe());
+    };
+  }, [filterMemo, setRows, storeCache]);
 
   return rows;
 }

package/src/utils/useMountedState.ts

@@ -0,0 +1,23 @@
+import { Dispatch, SetStateAction, useCallback, useEffect, useRef, useState } from "react";
+
+/**
+ * Use in place of useState when the component may be unmounted before the state is updated.
+ */
+export function useMountedState<T>(initialState: T | (() => T)): [T, Dispatch<SetStateAction<T>>] {
+  const [state, setState] = useState<T>(initialState);
+  const mountedRef = useRef(false);
+  useEffect(() => {
+    mountedRef.current = true;
+    return () => {
+      mountedRef.current = false;
+    };
+  });
+  const mountedSetState: typeof setState = useCallback((...args) => {
+    if (mountedRef.current) {
+      setState(...args);
+    } else {
+      console.warn("Ignoring `setState` call because component unmounted", ...args);
+    }
+  }, []);
+  return [state, mountedSetState];
+}