use syntax::codemap::{CodeMap, Span};
use syntax::parse::ParseSess;
use config::{Config, IndentStyle};
use shape::Shape;
use visitor::SnippetProvider;
use std::cell::RefCell;
pub trait Rewrite {
fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String>;
}
#[derive(Clone)]
pub struct RewriteContext<'a> {
pub parse_session: &'a ParseSess,
pub codemap: &'a CodeMap,
pub config: &'a Config,
pub inside_macro: RefCell<bool>,
pub use_block: RefCell<bool>,
pub is_if_else_block: RefCell<bool>,
pub force_one_line_chain: RefCell<bool>,
pub snippet_provider: &'a SnippetProvider<'a>,
}
impl<'a> RewriteContext<'a> {
pub fn snippet(&self, span: Span) -> &str {
self.snippet_provider.span_to_snippet(span).unwrap()
}
pub fn use_block_indent(&self) -> bool {
self.config.indent_style() == IndentStyle::Block || *self.use_block.borrow()
}
pub fn budget(&self, used_width: usize) -> usize {
self.config.max_width().checked_sub(used_width).unwrap_or(0)
}
pub fn inside_macro(&self) -> bool {
*self.inside_macro.borrow()
}
pub fn is_if_else_block(&self) -> bool {
*self.is_if_else_block.borrow()
}
}