Is there a FoldAll trait or equivalent?

[alshdavid]

I am trying to traverse the following source code

(async function () {
	const foo = await import('./a.js')
})();

I can use a VisitAll visitor to access the body of the function

struct Walker {}

impl VisitAll for Walker {
	fn visit_module(&mut self,n: &Module) {
		n.visit_all_children_with(self);
	}

	fn visit_call_expr(&mut self, node: &CallExpr) {
		match &node.callee {
			Callee::Import(import) => {
				dbg!(&import);
			},
			_ => {}
		}
	}
}

// Parse, etc
module.visit_all_with(&mut Walker{});

However there doesn't seem to be an equivalent using Fold. I have tried:

impl Fold for Walker {
	fn fold_module(&mut self, module: Module) -> Module {
		return module.fold_children_with(self);
	}

	fn fold_call_expr(&mut self, call_expr: CallExpr) -> CallExpr {
		dbg!(&call_expr);
		return call_expr;
	}
}

// parse module, etc
module.fold_with(&mut Walker{})

But this does not run on the function block. Any suggestions on how best to deeply mutate the tree?

Following resources like these: #501

[alshdavid]

Looks like I have to trigger it from each individual fold call like

fn fold_call_expr(&mut self, call_expr: CallExpr) -> CallExpr {
	let call_expr = module.fold_children_with(self);
	dbg!(&call_expr);
	return call_expr;
}