GitHub Viewer
"""Conservative component compatibility for Fast Refresh."""
from __future__ import annotations
import ast
import inspect
import textwrap
from typing import Any, Callable
def hook_signature(function: Callable[..., Any], _seen: frozenset[int] = frozenset()) -> tuple[str, ...] | None:
"""Record hook order and binding names before the source file changes.
Binding names distinguish inserting or moving hooks of the same kind.
Uninspectable functions remount instead of risking slot corruption.
"""
if id(function) in _seen:
return None
_seen = _seen | {id(function)}
try:
tree = ast.parse(textwrap.dedent(inspect.getsource(function)))
except (OSError, TypeError, SyntaxError):
return None
result = []
parents = {id(child): node for node in ast.walk(tree) for child in ast.iter_child_nodes(node)}
for node in ast.walk(tree):
if not isinstance(node, ast.Call):
continue
name = (
node.func.id
if isinstance(node.func, ast.Name)
else node.func.attr if isinstance(node.func, ast.Attribute) else ""
)
if not name.startswith("use_"):
continue
owner = parents.get(id(node))
binding = (
ast.dump(owner.targets[0])
if isinstance(owner, ast.Assign)
else ast.dump(owner.target) if isinstance(owner, ast.AnnAssign) else ""
)
signature = name + ":" + binding
target = None
if isinstance(node.func, ast.Name):
target = function.__globals__.get(node.func.id)
elif isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name):
owner = function.__globals__.get(node.func.value.id)
target = getattr(owner, node.func.attr, None)
if inspect.isfunction(target) and not target.__module__.startswith("pythonnative."):
nested = hook_signature(target, _seen)
if nested is None:
return None
signature += repr(nested)
result.append((node.lineno, node.col_offset, signature))
return tuple(value for _, _, value in sorted(result))