Update rustc-ap-* crates to 581.0.0 (#3783)

[rust.git] / src / rewrite.rs

diff --git a/src/rewrite.rs b/src/rewrite.rs
index 9d3733a279ce33de449c6dc022a42cfb99645d15..19b749dc8e1b76fe4d2cc9b52f169ddaa0c01625 100644 (file)
--- a/src/rewrite.rs
+++ b/src/rewrite.rs
@@ -1,16 +1,7 @@
-// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
-// file at the top-level directory of this distribution and at
-// http://rust-lang.org/COPYRIGHT.
-//
-// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
-// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
-// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
-// option. This file may not be copied, modified, or distributed
-// except according to those terms.
-
 // A generic trait to abstract the rewriting of an element (of the AST).
 
 use std::cell::RefCell;
+use std::rc::Rc;
 
 use syntax::parse::ParseSess;
 use syntax::ptr;
@@ -18,58 +9,90 @@
 
 use crate::config::{Config, IndentStyle};
 use crate::shape::Shape;
+use crate::skip::SkipContext;
 use crate::visitor::SnippetProvider;
 use crate::FormatReport;
 
-pub trait Rewrite {
+pub(crate) trait Rewrite {
     /// Rewrite self into shape.
-    fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String>;
+    fn rewrite(&self, context: &RewriteContext<'_>, shape: Shape) -> Option<String>;
 }
 
 impl<T: Rewrite> Rewrite for ptr::P<T> {
-    fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
+    fn rewrite(&self, context: &RewriteContext<'_>, shape: Shape) -> Option<String> {
         (**self).rewrite(context, shape)
     }
 }
 
 #[derive(Clone)]
-pub struct RewriteContext<'a> {
-    pub parse_session: &'a ParseSess,
-    pub source_map: &'a SourceMap,
-    pub config: &'a Config,
-    pub inside_macro: RefCell<bool>,
+pub(crate) struct RewriteContext<'a> {
+    pub(crate) parse_session: &'a ParseSess,
+    pub(crate) source_map: &'a SourceMap,
+    pub(crate) config: &'a Config,
+    pub(crate) inside_macro: Rc<RefCell<bool>>,
     // Force block indent style even if we are using visual indent style.
-    pub use_block: RefCell<bool>,
+    pub(crate) use_block: RefCell<bool>,
     // When `is_if_else_block` is true, unindent the comment on top
     // of the `else` or `else if`.
-    pub is_if_else_block: RefCell<bool>,
+    pub(crate) is_if_else_block: RefCell<bool>,
     // When rewriting chain, veto going multi line except the last element
-    pub force_one_line_chain: RefCell<bool>,
-    pub snippet_provider: &'a SnippetProvider<'a>,
+    pub(crate) force_one_line_chain: RefCell<bool>,
+    pub(crate) snippet_provider: &'a SnippetProvider<'a>,
     // Used for `format_snippet`
     pub(crate) macro_rewrite_failure: RefCell<bool>,
     pub(crate) report: FormatReport,
+    pub(crate) skip_context: SkipContext,
+    pub(crate) skipped_range: Rc<RefCell<Vec<(usize, usize)>>>,
+}
+
+pub(crate) struct InsideMacroGuard {
+    is_nested_macro_context: bool,
+    inside_macro_ref: Rc<RefCell<bool>>,
+}
+
+impl InsideMacroGuard {
+    pub(crate) fn is_nested(&self) -> bool {
+        self.is_nested_macro_context
+    }
+}
+
+impl Drop for InsideMacroGuard {
+    fn drop(&mut self) {
+        self.inside_macro_ref.replace(self.is_nested_macro_context);
+    }
 }
 
 impl<'a> RewriteContext<'a> {
-    pub fn snippet(&self, span: Span) -> &str {
+    pub(crate) fn snippet(&self, span: Span) -> &str {
         self.snippet_provider.span_to_snippet(span).unwrap()
     }
 
-    /// Return true if we should use block indent style for rewriting function call.
-    pub fn use_block_indent(&self) -> bool {
+    /// Returns `true` if we should use block indent style for rewriting function call.
+    pub(crate) fn use_block_indent(&self) -> bool {
         self.config.indent_style() == IndentStyle::Block || *self.use_block.borrow()
     }
 
-    pub fn budget(&self, used_width: usize) -> usize {
+    pub(crate) fn budget(&self, used_width: usize) -> usize {
         self.config.max_width().saturating_sub(used_width)
     }
 
-    pub fn inside_macro(&self) -> bool {
+    pub(crate) fn inside_macro(&self) -> bool {
         *self.inside_macro.borrow()
     }
 
-    pub fn is_if_else_block(&self) -> bool {
+    pub(crate) fn enter_macro(&self) -> InsideMacroGuard {
+        let is_nested_macro_context = self.inside_macro.replace(true);
+        InsideMacroGuard {
+            is_nested_macro_context,
+            inside_macro_ref: self.inside_macro.clone(),
+        }
+    }
+
+    pub(crate) fn leave_macro(&self) {
+        self.inside_macro.replace(false);
+    }
+
+    pub(crate) fn is_if_else_block(&self) -> bool {
         *self.is_if_else_block.borrow()
     }
 }