improve(src): use AST way to find unwrapped functions