Fix round-trip parse->fmt->parse for dbg stmts with more than one arg

This commit is contained in:
Joshua Warner 2024-12-01 11:25:57 -08:00
parent cfd83ffcdf
commit 912db1b76b
No known key found for this signature in database
GPG key ID: 89AD497003F93FDD
25 changed files with 244 additions and 66 deletions

View file

@ -687,8 +687,13 @@ impl IterTokens for Loc<Expr<'_>> {
.chain(e2.iter_tokens(arena))
.collect_in(arena),
Expr::Dbg => onetoken(Token::Keyword, region, arena),
Expr::DbgStmt(e1, e2) => (e1.iter_tokens(arena).into_iter())
.chain(e2.iter_tokens(arena))
Expr::DbgStmt {
first,
extra_args,
continuation,
} => (first.iter_tokens(arena).into_iter())
.chain(extra_args.iter_tokens(arena))
.chain(continuation.iter_tokens(arena))
.collect_in(arena),
Expr::LowLevelDbg(_, e1, e2) => (e1.iter_tokens(arena).into_iter())
.chain(e2.iter_tokens(arena))