atom.io 0.20.1 → 0.20.2

package/internal/dist/index.cjs

@@ -1298,32 +1298,33 @@ var subscribeToRootAtoms = (selector, store) => {
 // internal/src/subscribe/subscribe-to-state.ts
 function subscribeToState(token, handleUpdate, key, store) {
   const state = withdrawOrCreate(token, store);
-  if (state === void 0) {
-    throw new Error(
-      `State "${token.key}" not found in this store. Did you forget to initialize with the "atom" or "selector" function?`
-    );
-  }
-  const unsubFunction = state.subject.subscribe(key, handleUpdate);
   store.logger.info(`\u{1F440}`, state.type, state.key, `Adding subscription "${key}"`);
-  const dependencyUnsubFunctions = state.type !== `atom` && state.type !== `mutable_atom` ? subscribeToRootAtoms(state, store) : null;
-  const unsubscribe = dependencyUnsubFunctions === null ? () => {
-    store.logger.info(
-      `\u{1F648}`,
-      state.type,
-      state.key,
-      `Removing subscription "${key}"`
-    );
-    unsubFunction();
-  } : () => {
+  const isSelector = state.type === `selector` || state.type === `readonly_selector`;
+  let dependencyUnsubFunctions = null;
+  let updateHandler = handleUpdate;
+  if (isSelector) {
+    dependencyUnsubFunctions = subscribeToRootAtoms(state, store);
+    updateHandler = (update) => {
+      if (dependencyUnsubFunctions) {
+        dependencyUnsubFunctions.length = 0;
+      }
+      dependencyUnsubFunctions = subscribeToRootAtoms(state, store);
+      handleUpdate(update);
+    };
+  }
+  const mainUnsubFunction = state.subject.subscribe(key, updateHandler);
+  const unsubscribe = () => {
     store.logger.info(
       `\u{1F648}`,
       state.type,
       state.key,
       `Removing subscription "${key}"`
     );
-    unsubFunction();
-    for (const unsubFromDependency of dependencyUnsubFunctions) {
-      unsubFromDependency();
+    mainUnsubFunction();
+    if (dependencyUnsubFunctions) {
+      for (const unsubFromDependency of dependencyUnsubFunctions) {
+        unsubFromDependency();
+      }
     }
   };
   return unsubscribe;

package/internal/dist/index.js

@@ -1020,32 +1020,33 @@ var subscribeToRootAtoms = (selector, store) => {
 // internal/src/subscribe/subscribe-to-state.ts
 function subscribeToState(token, handleUpdate, key, store) {
   const state = withdrawOrCreate(token, store);
-  if (state === void 0) {
-    throw new Error(
-      `State "${token.key}" not found in this store. Did you forget to initialize with the "atom" or "selector" function?`
-    );
-  }
-  const unsubFunction = state.subject.subscribe(key, handleUpdate);
   store.logger.info(`\u{1F440}`, state.type, state.key, `Adding subscription "${key}"`);
-  const dependencyUnsubFunctions = state.type !== `atom` && state.type !== `mutable_atom` ? subscribeToRootAtoms(state, store) : null;
-  const unsubscribe = dependencyUnsubFunctions === null ? () => {
-    store.logger.info(
-      `\u{1F648}`,
-      state.type,
-      state.key,
-      `Removing subscription "${key}"`
-    );
-    unsubFunction();
-  } : () => {
+  const isSelector = state.type === `selector` || state.type === `readonly_selector`;
+  let dependencyUnsubFunctions = null;
+  let updateHandler = handleUpdate;
+  if (isSelector) {
+    dependencyUnsubFunctions = subscribeToRootAtoms(state, store);
+    updateHandler = (update) => {
+      if (dependencyUnsubFunctions) {
+        dependencyUnsubFunctions.length = 0;
+      }
+      dependencyUnsubFunctions = subscribeToRootAtoms(state, store);
+      handleUpdate(update);
+    };
+  }
+  const mainUnsubFunction = state.subject.subscribe(key, updateHandler);
+  const unsubscribe = () => {
     store.logger.info(
       `\u{1F648}`,
       state.type,
       state.key,
       `Removing subscription "${key}"`
     );
-    unsubFunction();
-    for (const unsubFromDependency of dependencyUnsubFunctions) {
-      unsubFromDependency();
+    mainUnsubFunction();
+    if (dependencyUnsubFunctions) {
+      for (const unsubFromDependency of dependencyUnsubFunctions) {
+        unsubFromDependency();
+      }
     }
   };
   return unsubscribe;

package/internal/src/subscribe/subscribe-to-state.ts

@@ -11,41 +11,36 @@ export function subscribeToState<T>(
 	store: Store,
 ): () => void {
 	const state = withdrawOrCreate(token, store)
-	if (state === undefined) {
-		throw new Error(
-			`State "${token.key}" not found in this store. Did you forget to initialize with the "atom" or "selector" function?`,
+	store.logger.info(`👀`, state.type, state.key, `Adding subscription "${key}"`)
+	const isSelector =
+		state.type === `selector` || state.type === `readonly_selector`
+	let dependencyUnsubFunctions: (() => void)[] | null = null
+	let updateHandler: UpdateHandler<T> = handleUpdate
+	if (isSelector) {
+		dependencyUnsubFunctions = subscribeToRootAtoms(state, store)
+		updateHandler = (update) => {
+			if (dependencyUnsubFunctions) {
+				dependencyUnsubFunctions.length = 0
+			}
+			dependencyUnsubFunctions = subscribeToRootAtoms(state, store)
+			handleUpdate(update)
+		}
+	}
+	const mainUnsubFunction = state.subject.subscribe(key, updateHandler)
+	const unsubscribe = () => {
+		store.logger.info(
+			`🙈`,
+			state.type,
+			state.key,
+			`Removing subscription "${key}"`,
 		)
+		mainUnsubFunction()
+		if (dependencyUnsubFunctions) {
+			for (const unsubFromDependency of dependencyUnsubFunctions) {
+				unsubFromDependency()
+			}
+		}
 	}
-	const unsubFunction = state.subject.subscribe(key, handleUpdate)
-	store.logger.info(`👀`, state.type, state.key, `Adding subscription "${key}"`)
-	const dependencyUnsubFunctions =
-		state.type !== `atom` && state.type !== `mutable_atom`
-			? subscribeToRootAtoms(state, store)
-			: null
-
-	const unsubscribe =
-		dependencyUnsubFunctions === null
-			? () => {
-					store.logger.info(
-						`🙈`,
-						state.type,
-						state.key,
-						`Removing subscription "${key}"`,
-					)
-					unsubFunction()
-				}
-			: () => {
-					store.logger.info(
-						`🙈`,
-						state.type,
-						state.key,
-						`Removing subscription "${key}"`,
-					)
-					unsubFunction()
-					for (const unsubFromDependency of dependencyUnsubFunctions) {
-						unsubFromDependency()
-					}
-				}
 
 	return unsubscribe
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "atom.io",
-  "version": "0.20.1",
+  "version": "0.20.2",
   "description": "Composable and testable reactive data library.",
   "homepage": "https://atom.io.fyi",
   "sideEffects": false,
@@ -48,13 +48,13 @@
     }
   },
   "devDependencies": {
-    "@testing-library/react": "15.0.4",
+    "@testing-library/react": "15.0.5",
     "@types/eslint": "npm:@types/eslint@8.56.10",
     "@types/eslint-v9": "npm:@types/eslint@8.56.10",
     "@types/estree": "1.0.5",
     "@types/http-proxy": "1.17.14",
     "@types/npmlog": "7.0.0",
-    "@types/react": "18.2.79",
+    "@types/react": "18.3.0",
     "@types/tmp": "0.2.6",
     "@typescript-eslint/parser": "7.7.1",
     "@typescript-eslint/rule-tester": "7.7.1",
@@ -71,8 +71,8 @@
     "npmlog": "7.0.1",
     "postgres": "3.4.4",
     "preact": "10.20.2",
-    "react": "18.2.0",
-    "react-dom": "18.2.0",
+    "react": "18.3.0",
+    "react-dom": "18.3.0",
     "react-router-dom": "6.23.0",
     "socket.io": "4.7.5",
     "socket.io-client": "4.7.5",