-// 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 syntax::codemap::{CodeMap, Span};
-use syntax::parse::ParseSess;
+use std::cell::{Cell, RefCell};
+use std::rc::Rc;
+
+use rustc_ast::ptr;
+use rustc_span::Span;
-use config::{Config, IndentStyle};
-use shape::Shape;
+use crate::config::{Config, IndentStyle};
+use crate::parse::session::ParseSess;
+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> {
+ (**self).rewrite(context, shape)
+ }
}
#[derive(Clone)]
-pub struct RewriteContext<'a> {
- pub parse_session: &'a ParseSess,
- pub codemap: &'a CodeMap,
- pub config: &'a Config,
- pub inside_macro: bool,
+pub(crate) struct RewriteContext<'a> {
+ pub(crate) parse_sess: &'a ParseSess,
+ pub(crate) config: &'a Config,
+ pub(crate) inside_macro: Rc<Cell<bool>>,
// Force block indent style even if we are using visual indent style.
- pub use_block: bool,
- // When `format_if_else_cond_comment` is true, unindent the comment on top
+ pub(crate) use_block: Cell<bool>,
+ // When `is_if_else_block` is true, unindent the comment on top
// of the `else` or `else if`.
- pub is_if_else_block: bool,
+ pub(crate) is_if_else_block: Cell<bool>,
// When rewriting chain, veto going multi line except the last element
- pub force_one_line_chain: bool,
+ pub(crate) force_one_line_chain: Cell<bool>,
+ pub(crate) snippet_provider: &'a SnippetProvider,
+ // Used for `format_snippet`
+ pub(crate) macro_rewrite_failure: Cell<bool>,
+ pub(crate) is_macro_def: 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<Cell<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) -> String {
- self.codemap.span_to_snippet(span).unwrap()
+ pub(crate) fn snippet(&self, span: Span) -> &str {
+ self.snippet_provider.span_to_snippet(span).unwrap()
+ }
+
+ /// 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.get()
+ }
+
+ pub(crate) fn budget(&self, used_width: usize) -> usize {
+ self.config.max_width().saturating_sub(used_width)
+ }
+
+ pub(crate) fn inside_macro(&self) -> bool {
+ self.inside_macro.get()
+ }
+
+ 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(),
+ }
}
- /// Return true if we should use block indent style for rewriting function call.
- pub fn use_block_indent(&self) -> bool {
- self.config.fn_call_indent() == IndentStyle::Block || self.use_block
+ pub(crate) fn leave_macro(&self) {
+ self.inside_macro.replace(false);
}
- pub fn budget(&self, used_width: usize) -> usize {
- self.config.max_width().checked_sub(used_width).unwrap_or(0)
+ pub(crate) fn is_if_else_block(&self) -> bool {
+ self.is_if_else_block.get()
}
}