fix false possitive early return errors

Author: rmorshea

flake8_idom_hooks/rules_of_hooks.py

@@ -42,7 +42,11 @@ def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
             ):
                 self.generic_visit(node)
         else:
-            with set_current(self, function=node):
+            with set_current(
+                self,
+                function=node,
+                early_return=None,
+            ):
                 self.generic_visit(node)
 
     def visit_Call(self, node: ast.Call) -> None:

tests/cases/hook_usage.py

@@ -190,3 +190,24 @@ def example():
         return None
     # error: ROH103 hook 'use_state' used after an early return
     use_state()
+
+
+@component
+def example():
+    def closure():
+        # this return is ok since it's not in the same function
+        return None
+
+    # Ok: no early return error
+    use_state()
+
+
+@component
+def example():
+    @use_effect
+    def some_effect():
+        # this return is ok since it's not in the same function
+        return None
+
+    # Ok: no early return error
+    use_state()