11import ast
22from typing import Optional , Union , Set
33
4- from .utils import is_hook_def , is_element_def , ErrorVisitor , set_current
4+ from .utils import is_hook_def , is_component_def , ErrorVisitor , set_current
55
66
77HOOKS_WITH_DEPS = ("use_effect" , "use_callback" , "use_memo" )
@@ -11,13 +11,13 @@ class ExhaustiveDepsVisitor(ErrorVisitor):
1111 def __init__ (self ) -> None :
1212 super ().__init__ ()
1313 self ._current_function : Optional [ast .FunctionDef ] = None
14- self ._current_hook_or_element : Optional [ast .FunctionDef ] = None
14+ self ._current_hook_or_component : Optional [ast .FunctionDef ] = None
1515
1616 def visit_FunctionDef (self , node : ast .FunctionDef ) -> None :
17- if is_hook_def (node ) or is_element_def (node ):
18- with set_current (self , hook_or_element = node ):
17+ if is_hook_def (node ) or is_component_def (node ):
18+ with set_current (self , hook_or_component = node ):
1919 self .generic_visit (node )
20- elif self ._current_hook_or_element is not None :
20+ elif self ._current_hook_or_component is not None :
2121 for deco in node .decorator_list :
2222 if not isinstance (deco , ast .Call ):
2323 continue
@@ -36,15 +36,15 @@ def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
3636 for kw in deco .keywords :
3737 if kw .arg == "args" :
3838 self ._check_hook_dependency_list_is_exhaustive (
39- self ._current_hook_or_element ,
39+ self ._current_hook_or_component ,
4040 called_func_name ,
4141 node ,
4242 kw .value ,
4343 )
4444 break
4545
4646 def visit_Call (self , node : ast .Call ) -> None :
47- if self ._current_hook_or_element is None :
47+ if self ._current_hook_or_component is None :
4848 return
4949
5050 called_func = node .func
@@ -75,15 +75,15 @@ def visit_Call(self, node: ast.Call) -> None:
7575
7676 if isinstance (func , ast .Lambda ):
7777 self ._check_hook_dependency_list_is_exhaustive (
78- self ._current_hook_or_element ,
78+ self ._current_hook_or_component ,
7979 called_func_name ,
8080 func ,
8181 args ,
8282 )
8383
8484 def _check_hook_dependency_list_is_exhaustive (
8585 self ,
86- current_hook_or_element : ast .FunctionDef ,
86+ current_hook_or_component : ast .FunctionDef ,
8787 hook_name : str ,
8888 func : Union [ast .FunctionDef , ast .Lambda ],
8989 dependency_expr : Optional [ast .expr ],
@@ -98,7 +98,7 @@ def _check_hook_dependency_list_is_exhaustive(
9898 func_name = "lambda" if isinstance (func , ast .Lambda ) else func .name
9999
100100 top_level_variable_finder = _TopLevelVariableFinder ()
101- top_level_variable_finder .visit (current_hook_or_element )
101+ top_level_variable_finder .visit (current_hook_or_component )
102102 variables_defined_in_scope = top_level_variable_finder .variable_names
103103
104104 missing_name_finder = _MissingNameFinder (
0 commit comments