refactor: updated index useUI function

Author: Austin1serb

packages/core/src/index.js

@@ -1,42 +1,36 @@
 import { useCallback } from 'react';
 
-// Convert kebab-case to camelCase for dataset API
-// "theme-secondary" -> "themeSecondary"
-function kebabToCamelCase(str) {
-	return str.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase());
-}
-
 function useUI(key, initialValue) {
-	const parseValue = stringValue => {
-		if (stringValue === null || stringValue === undefined) return initialValue;
-
-		if (typeof initialValue === 'boolean') {
-			return stringValue === 'true';
-		} else if (typeof initialValue === 'number') {
-			return Number(stringValue);
-		} else {
-			return stringValue;
-		}
-	};
-
+	//setValue(valueOrUpdater)
 	const setValue = useCallback(
 		valueOrUpdater => {
-			if (typeof window === 'undefined' || typeof document === 'undefined') return;
-			const datasetKey = kebabToCamelCase(key);
-			const getCurrentValue = () => parseValue(document.body.dataset[datasetKey]);
-
+			if (typeof window === 'undefined') return;
+			// Convert kebab-case to camelCase for dataset API
+			// "theme-secondary" -> "themeSecondary"
+			const camelKey = key.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase());
 			let newValue;
 			if (typeof valueOrUpdater === 'function') {
-				newValue = valueOrUpdater(getCurrentValue());
+				const parse = v => {
+					if (!v) return initialValue;
+					switch (typeof initialValue) {
+						case 'boolean':
+							return v === 'true';
+						case 'number': {
+							const n = Number(v);
+							return isNaN(n) ? initialValue : n;
+						}
+						default:
+							return v;
+					}
+				};
+				newValue = valueOrUpdater(parse(document.body.dataset[camelKey]));
 			} else {
 				newValue = valueOrUpdater;
 			}
-			document.body.dataset[datasetKey] = String(newValue);
+			document.body.dataset[camelKey] = String(newValue);
 		},
 		[key]
 	);
-
-	// The actual current value lives in the DOM!
 	return [initialValue, setValue];
 }