From bf59119bf370ed77513db2c56f8ef3fabf04285c Mon Sep 17 00:00:00 2001 From: Richard Feldman Date: Wed, 11 May 2022 08:56:18 -0400 Subject: [PATCH] Allow custom indentation in write_impl_tags --- bindgen/src/bindgen_rs.rs | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/bindgen/src/bindgen_rs.rs b/bindgen/src/bindgen_rs.rs index 3607c800c9..d60b3fe782 100644 --- a/bindgen/src/bindgen_rs.rs +++ b/bindgen/src/bindgen_rs.rs @@ -248,6 +248,7 @@ fn write_tag_union( )?; write_impl_tags( + 3, tags.iter(), &discriminant_name, buf, @@ -289,6 +290,7 @@ fn write_impl_tags< I: IntoIterator)>, F: Fn(&str, Option) -> String, >( + indentations: usize, tags: I, discriminant_name: &str, buf: &mut String, @@ -297,11 +299,11 @@ fn write_impl_tags< for (tag_name, opt_payload_id) in tags { let branch_str = to_branch_str(tag_name, *opt_payload_id); - writeln!( - buf, - "{}{}{}{}::{} => {}", - INDENT, INDENT, INDENT, discriminant_name, tag_name, branch_str - )?; + for _ in 0..indentations { + buf.write_str(INDENT)?; + } + + writeln!(buf, "{}::{} => {}", discriminant_name, tag_name, branch_str)?; } Ok(())