1c235de97d
The problem is essentially that if we try to parse a token tree using a CodeMap different from the one the tree was originally parsed with, spans become nonsense. Since CodeMaps can't be cloned, we're basically forced to use the original ParseSess for additional parsing. Ideally, rustfmt would be a bit more clever and figure out how to parse macro arguments based on the definition of the macro itself, rather than just guessing that a particular token sequence looks like an expression, but this is good enough for now. Fixes #538.
52 lines
1.8 KiB
Rust
52 lines
1.8 KiB
Rust
// 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 Indent;
|
|
use config::Config;
|
|
|
|
pub trait Rewrite {
|
|
/// Rewrite self into offset and width.
|
|
/// `offset` is the indentation of the first line. The next lines
|
|
/// should begin with a least `offset` spaces (except backwards
|
|
/// indentation). The first line should not begin with indentation.
|
|
/// `width` is the maximum number of characters on the last line
|
|
/// (excluding offset). The width of other lines is not limited by
|
|
/// `width`.
|
|
fn rewrite(&self, context: &RewriteContext, width: usize, offset: Indent) -> Option<String>;
|
|
}
|
|
|
|
pub struct RewriteContext<'a> {
|
|
pub parse_session: &'a ParseSess,
|
|
pub codemap: &'a CodeMap,
|
|
pub config: &'a Config,
|
|
// Indentation due to nesting of blocks.
|
|
pub block_indent: Indent,
|
|
}
|
|
|
|
impl<'a> RewriteContext<'a> {
|
|
pub fn nested_context(&self) -> RewriteContext<'a> {
|
|
RewriteContext {
|
|
parse_session: self.parse_session,
|
|
codemap: self.codemap,
|
|
config: self.config,
|
|
block_indent: self.block_indent.block_indent(self.config),
|
|
}
|
|
}
|
|
|
|
pub fn snippet(&self, span: Span) -> String {
|
|
self.codemap.span_to_snippet(span).unwrap()
|
|
}
|
|
}
|