Support offsetting the most recent break

dtolnay · dtolnay · commit 63406ac771c0 · 2022-02-03T10:59:34.000-08:00
diff --git a/compiler/rustc_ast_pretty/src/pp.rs b/compiler/rustc_ast_pretty/src/pp.rs
@@ -314,6 +314,12 @@ impl Printer {
         }
     }
 
+    pub fn offset(&mut self, offset: isize) {
+        if let Some(BufEntry { token: Token::Break(token), .. }) = &mut self.buf.last_mut() {
+            token.offset += offset;
+        }
+    }
+
     fn check_stream(&mut self) {
         while self.right_total - self.left_total > self.space {
             if *self.scan_stack.front().unwrap() == self.buf.index_of_first() {

Original file line number	Diff line number	Diff line change
`@@ -314,6 +314,12 @@ impl Printer {`
`314`	`314`	`}`
`315`	`315`	`}`
`316`	`316`
	`317`	`+ pub fn offset(&mut self, offset: isize) {`
	`318`	`+ if let Some(BufEntry { token: Token::Break(token), .. }) = &mut self.buf.last_mut() {`
	`319`	`+ token.offset += offset;`
	`320`	`+ }`
	`321`	`+ }`
	`322`	`+`
`317`	`323`	`fn check_stream(&mut self) {`
`318`	`324`	`while self.right_total - self.left_total > self.space {`
`319`	`325`	`if *self.scan_stack.front().unwrap() == self.buf.index_of_first() {`