Fix: simplify check for supported hooks

src/lib/utils.js

@@ -167,17 +167,16 @@ export function isInReactHooks(node, returnHook = false) {
   return hookDef ? true : false
 }
 
-function isReactPrimitive(node) {
-  if (
-    node.type === 'Identifier' &&
-    (node.name === 'useEffect' || node.name === 'useCallback')
-  ) {
-    return true
+function isInSupportedReactPrimitives(node) {
+  const supportedPrimitives = ['useEffect', 'useCallback', 'useMemo']
+
+  if (node.type === 'Identifier') {
+    return supportedPrimitives.includes(node.name)
   }
 
   if (node.type === 'MemberExpression') {
     const flatExpr = flattenMemberExpression(node)
-    return flatExpr.endsWith('useEffect') || flatExpr.endsWith('useCallback')
+    return supportedPrimitives.some((d) => flatExpr.endsWith(d))
   }
 
   return false
@@ -191,7 +190,7 @@ export function getNearestHook(node) {
     return false
   }
 
-  if (!isReactPrimitive(parentCaller.callee)) {
+  if (!isInSupportedReactPrimitives(parentCaller.callee)) {
     return getNearestHook(parentCaller)
   }
 

tests/StateSnapshot.test.js

@@ -298,6 +298,35 @@ ruleTester.run('state-snapshot-rule', rule, {
       </>
     );
   })`,
+    `
+  function useExample2(s) {
+    const {b: {c} } = useSnapshot(s.a1);
+    const val = useMemo(()=>{
+      return s.a1.b.c
+    },[s.a1.b.c])
+  }`,
+    `
+  function useExample3(s) {
+    const {b: {c} } = useSnapshot(s.a1);
+    const val = useMemo(()=>{
+      return c
+    },[c])
+  }`,
+    `
+  const foo = proxy({ foo: 123 });
+  function useExample2(s) {
+    const val = useMemo(()=>{
+      return foo.foo
+    },[foo.foo])
+  }`,
+    `
+  const foo = proxy({ foo: 123 });
+  function useExample2(s) {
+    const val = useMemo(()=>{
+      return foo.foo
+    },[foo])
+  }
+  `,
   ],
   invalid: [
     {