@exodus/atoms 5.5.0 → 5.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exodus/atoms",
-  "version": "5.5.0",
+  "version": "5.6.0",
   "main": "src/index.js",
   "description": "Abstraction for encapsulating a piece of data behind a simple unified interface: get, set, observe",
   "author": "Exodus Movement Inc.",
@@ -36,5 +36,5 @@
     "eslint": "^8.44.0",
     "jest": "^29.1.2"
   },
-  "gitHead": "42aecad05fd3135cde0287a37ab4e2af232ee66d"
+  "gitHead": "aec5804ed663dec0767599dde6208d8487ab07f7"
 }

package/src/enhancers/optimistic-notifier.js

@@ -0,0 +1,60 @@
+import makeConcurrent from 'make-concurrent'
+import createInMemoryAtom from '../factories/memory'
+
+const optimisticNotifier = (atom) => {
+  const memoryAtom = createInMemoryAtom()
+
+  const set = makeConcurrent(async (newValue) => {
+    let previous
+    await memoryAtom.set((previousValue) => {
+      previous = previousValue
+      return newValue
+    })
+
+    atom
+      .set(newValue)
+      // only revert in case the value hasn't changed in the meanwhile
+      .catch(() => memoryAtom.set((value) => (value === newValue ? previous : value)))
+  })
+
+  let unsubscribeSource
+  let subscribers = 0
+
+  const maybeUnsubscribeSource = () => {
+    subscribers -= 1
+    if (subscribers > 0) return
+    unsubscribeSource()
+    unsubscribeSource = undefined
+  }
+
+  const observe = (callback) => {
+    if (subscribers === 0) {
+      unsubscribeSource = atom.observe(set)
+    }
+
+    subscribers += 1
+    const unsubscribe = memoryAtom.observe(callback)
+
+    return () => {
+      unsubscribe()
+      maybeUnsubscribeSource()
+    }
+  }
+
+  const get = async () => {
+    if (subscribers === 0) {
+      return atom.get()
+    }
+
+    return memoryAtom.get()
+  }
+
+  return {
+    ...memoryAtom,
+    set,
+    get,
+    observe,
+  }
+}
+
+export default optimisticNotifier