ast: Use enum dispatch for tree traversal

ast: Use enum dispatch for tree traversal #7699