1 use crate::generated_code;
2 use rustc::session::Session;
3 use rustc_parse::lexer::{self, StringReader};
5 use syntax::token::{self, TokenKind};
8 pub struct SpanUtils<'a> {
12 impl<'a> SpanUtils<'a> {
13 pub fn new(sess: &'a Session) -> SpanUtils<'a> {
17 pub fn make_filename_string(&self, file: &SourceFile) -> String {
19 FileName::Real(path) if !file.name_was_remapped => {
20 if path.is_absolute() {
24 .map_prefix(path.clone())
29 self.sess.working_dir.0.join(&path).display().to_string()
32 // If the file name is already remapped, we assume the user
33 // configured it the way they wanted to, so use that directly
34 filename => filename.to_string(),
38 pub fn snippet(&self, span: Span) -> String {
39 match self.sess.source_map().span_to_snippet(span) {
41 Err(_) => String::new(),
45 pub fn retokenise_span(&self, span: Span) -> StringReader<'a> {
46 lexer::StringReader::retokenize(&self.sess.parse_sess, span)
49 pub fn sub_span_of_token(&self, span: Span, tok: TokenKind) -> Option<Span> {
50 let mut toks = self.retokenise_span(span);
52 let next = toks.next_token();
53 if next == token::Eof {
57 return Some(next.span);
62 // // Return the name for a macro definition (identifier after first `!`)
63 // pub fn span_for_macro_def_name(&self, span: Span) -> Option<Span> {
64 // let mut toks = self.retokenise_span(span);
66 // let ts = toks.real_token();
67 // if ts == token::Eof {
70 // if ts == token::Not {
71 // let ts = toks.real_token();
72 // if ts.kind.is_ident() {
73 // return Some(ts.sp);
81 // // Return the name for a macro use (identifier before first `!`).
82 // pub fn span_for_macro_use_name(&self, span:Span) -> Option<Span> {
83 // let mut toks = self.retokenise_span(span);
84 // let mut prev = toks.real_token();
86 // if prev == token::Eof {
89 // let ts = toks.real_token();
90 // if ts == token::Not {
91 // if prev.kind.is_ident() {
92 // return Some(prev.sp);
101 /// Return true if the span is generated code, and
102 /// it is not a subspan of the root callsite.
104 /// Used to filter out spans of minimal value,
105 /// such as references to macro internal variables.
106 pub fn filter_generated(&self, span: Span) -> bool {
107 if generated_code(span) {
111 //If the span comes from a fake source_file, filter it.
112 !self.sess.source_map().lookup_char_pos(span.lo()).file.is_real_file()
116 macro_rules! filter {
117 ($util: expr, $parent: expr) => {
118 if $util.filter_generated($parent) {