use alloc::borrow::ToOwned;
use alloc::boxed::Box;
use alloc::format;
use alloc::string::{String, ToString};
use alloc::vec;
use alloc::vec::Vec;
use super::*;
fn filename(sm: &SourceMap, path: &str) -> FileName {
FileName::Real(sm.path_mapping().to_real_filename(sm.working_dir(), PathBuf::from(path)))
}
fn init_source_map() -> SourceMap {
let sm = SourceMap::new(FilePathMapping::empty());
sm.new_source_file(filename(&sm, "blork.rs"), "first line.\nsecond line".to_string());
sm.new_source_file(filename(&sm, "empty.rs"), String::new());
sm.new_source_file(filename(&sm, "blork2.rs"), "first line.\nsecond line".to_string());
sm
}
impl SourceMap {
fn merge_spans(&self, sp_lhs: Span, sp_rhs: Span) -> Option<Span> {
if !sp_lhs.eq_ctxt(sp_rhs) {
return None;
}
let lhs_end = match self.lookup_line(sp_lhs.hi()) {
Ok(x) => x,
Err(_) => return None,
};
let rhs_begin = match self.lookup_line(sp_rhs.lo()) {
Ok(x) => x,
Err(_) => return None,
};
if lhs_end.line != rhs_begin.line {
return None;
}
if (sp_lhs.lo() <= sp_rhs.lo()) && (sp_lhs.hi() <= sp_rhs.lo()) {
Some(sp_lhs.to(sp_rhs))
} else {
None
}
}
fn bytepos_to_file_charpos(&self, bpos: BytePos) -> CharPos {
let idx = self.lookup_source_file_idx(bpos);
let sf = &(*self.files.borrow().source_files)[idx];
let bpos = sf.relative_position(bpos);
sf.bytepos_to_file_charpos(bpos)
}
}
#[test]
fn t3() {
let sm = init_source_map();
let srcfbp1 = sm.lookup_byte_offset(BytePos(23));
assert_eq!(srcfbp1.sf.name, filename(&sm, "blork.rs"));
assert_eq!(srcfbp1.pos, BytePos(23));
let srcfbp1 = sm.lookup_byte_offset(BytePos(24));
assert_eq!(srcfbp1.sf.name, filename(&sm, "empty.rs"));
assert_eq!(srcfbp1.pos, BytePos(0));
let srcfbp2 = sm.lookup_byte_offset(BytePos(25));
assert_eq!(srcfbp2.sf.name, filename(&sm, "blork2.rs"));
assert_eq!(srcfbp2.pos, BytePos(0));
}
#[test]
fn t4() {
let sm = init_source_map();
let cp1 = sm.bytepos_to_file_charpos(BytePos(22));
assert_eq!(cp1, CharPos(22));
let cp2 = sm.bytepos_to_file_charpos(BytePos(25));
assert_eq!(cp2, CharPos(0));
}
#[test]
fn t5() {
let sm = init_source_map();
let loc1 = sm.lookup_char_pos(BytePos(22));
assert_eq!(loc1.file.name, filename(&sm, "blork.rs"));
assert_eq!(loc1.line, 2);
assert_eq!(loc1.col, CharPos(10));
let loc2 = sm.lookup_char_pos(BytePos(25));
assert_eq!(loc2.file.name, filename(&sm, "blork2.rs"));
assert_eq!(loc2.line, 1);
assert_eq!(loc2.col, CharPos(0));
}
fn init_source_map_mbc() -> SourceMap {
let sm = SourceMap::new(FilePathMapping::empty());
sm.new_source_file(filename(&sm, "blork.rs"), "fir€st €€€€ line.\nsecond line".to_string());
sm.new_source_file(filename(&sm, "blork2.rs"), "first line€€.\n€ second line".to_string());
sm
}
#[test]
fn t6() {
let sm = init_source_map_mbc();
let cp1 = sm.bytepos_to_file_charpos(BytePos(3));
assert_eq!(cp1, CharPos(3));
let cp2 = sm.bytepos_to_file_charpos(BytePos(6));
assert_eq!(cp2, CharPos(4));
let cp3 = sm.bytepos_to_file_charpos(BytePos(56));
assert_eq!(cp3, CharPos(12));
let cp4 = sm.bytepos_to_file_charpos(BytePos(61));
assert_eq!(cp4, CharPos(15));
}
#[test]
fn t7() {
let sm = init_source_map();
let span = Span::with_root_ctxt(BytePos(12), BytePos(23));
let file_lines = sm.span_to_lines(span).unwrap();
assert_eq!(file_lines.file.name, filename(&sm, "blork.rs"));
assert_eq!(file_lines.lines.len(), 1);
assert_eq!(file_lines.lines[0].line_index, 1);
}
fn span_from_selection(input: &str, selection: &str) -> Span {
assert_eq!(input.len(), selection.len());
let left_index = selection.find('~').unwrap() as u32;
let right_index = selection.rfind('~').map_or(left_index, |x| x as u32);
Span::with_root_ctxt(BytePos(left_index), BytePos(right_index + 1))
}
#[test]
fn span_to_snippet_and_lines_spanning_multiple_lines() {
let sm = SourceMap::new(FilePathMapping::empty());
let inputtext = "aaaaa\nbbbbBB\nCCC\nDDDDDddddd\neee\n";
let selection = " \n ~~\n~~~\n~~~~~ \n \n";
sm.new_source_file(filename(&sm, "blork.rs"), inputtext.to_string());
let span = span_from_selection(inputtext, selection);
assert_eq!(&sm.span_to_snippet(span).unwrap(), "BB\nCCC\nDDDDD");
let lines = sm.span_to_lines(span).unwrap();
let expected = vec![
LineInfo { line_index: 1, start_col: CharPos(4), end_col: CharPos(6) },
LineInfo { line_index: 2, start_col: CharPos(0), end_col: CharPos(3) },
LineInfo { line_index: 3, start_col: CharPos(0), end_col: CharPos(5) },
];
assert_eq!(lines.lines, expected);
}
#[test]
fn t8() {
let sm = init_source_map();
let span = Span::with_root_ctxt(BytePos(12), BytePos(23));
let snippet = sm.span_to_snippet(span);
assert_eq!(snippet, Ok("second line".to_string()));
}
#[test]
fn t9() {
let sm = init_source_map();
let span = Span::with_root_ctxt(BytePos(12), BytePos(23));
let sstr = sm.span_to_diagnostic_string(span);
assert_eq!(sstr, "blork.rs:2:1: 2:12");
}
#[test]
fn span_merging_fail() {
let sm = SourceMap::new(FilePathMapping::empty());
let inputtext = "bbbb BB\ncc CCC\n";
let selection1 = " ~~\n \n";
let selection2 = " \n ~~~\n";
sm.new_source_file(filename(&sm, "blork.rs"), inputtext.to_owned());
let span1 = span_from_selection(inputtext, selection1);
let span2 = span_from_selection(inputtext, selection2);
assert!(sm.merge_spans(span1, span2).is_none());
}
#[test]
fn t10() {
let sm = SourceMap::new(FilePathMapping::empty());
let unnormalized = "first line.\r\nsecond line";
let normalized = "first line.\nsecond line";
let src_file = sm.new_source_file(filename(&sm, "blork.rs"), unnormalized.to_string());
assert_eq!(src_file.src.as_ref().unwrap().as_ref(), normalized);
assert!(
src_file.src_hash.matches(unnormalized),
"src_hash should use the source before normalization"
);
let SourceFile {
name,
src_hash,
checksum_hash,
normalized_source_len,
unnormalized_source_len,
lines,
multibyte_chars,
normalized_pos,
stable_id,
..
} = (*src_file).clone();
let imported_src_file = sm.new_imported_source_file(
name,
src_hash,
checksum_hash,
stable_id,
normalized_source_len.to_u32(),
unnormalized_source_len,
CrateNum::ZERO,
FreezeLock::new(lines.read().clone()),
multibyte_chars,
normalized_pos,
0,
);
assert!(
imported_src_file.external_src.borrow().get_source().is_none(),
"imported source file should not have source yet"
);
imported_src_file.add_external_src(|| Some(unnormalized.to_string()));
assert_eq!(
imported_src_file.external_src.borrow().get_source().unwrap(),
normalized,
"imported source file should be normalized"
);
}
fn path(p: &str) -> PathBuf {
path_str(p).into()
}
fn path_str(p: &str) -> String {
#[cfg(not(windows))]
{
return p.into();
}
#[cfg(windows)]
{
let mut path = p.replace('/', "\\");
if let Some(rest) = path.strip_prefix('\\') {
path = ["X:\\", rest].concat();
}
path
}
}
fn map_path_prefix(mapping: &FilePathMapping, p: &str) -> String {
mapping.map_prefix(path(p)).0.to_string_lossy().to_string()
}
fn reverse_map_prefix(mapping: &FilePathMapping, p: &str) -> Option<String> {
mapping.reverse_map_prefix_heuristically(&path(p)).map(|q| q.to_string_lossy().to_string())
}
#[test]
fn path_prefix_remapping() {
{
let mapping = &FilePathMapping::new(
vec![(path("abc/def"), path("foo"))],
RemapPathScopeComponents::all(),
);
assert_eq!(map_path_prefix(mapping, "abc/def/src/main.rs"), path_str("foo/src/main.rs"));
assert_eq!(map_path_prefix(mapping, "abc/def"), path_str("foo"));
}
{
let mapping = &FilePathMapping::new(
vec![(path("abc/def"), path("/foo"))],
RemapPathScopeComponents::all(),
);
assert_eq!(map_path_prefix(mapping, "abc/def/src/main.rs"), path_str("/foo/src/main.rs"));
assert_eq!(map_path_prefix(mapping, "abc/def"), path_str("/foo"));
}
{
let mapping = &FilePathMapping::new(
vec![(path("/abc/def"), path("foo"))],
RemapPathScopeComponents::all(),
);
assert_eq!(map_path_prefix(mapping, "/abc/def/src/main.rs"), path_str("foo/src/main.rs"));
assert_eq!(map_path_prefix(mapping, "/abc/def"), path_str("foo"));
}
{
let mapping = &FilePathMapping::new(
vec![(path("/abc/def"), path("/foo"))],
RemapPathScopeComponents::all(),
);
assert_eq!(map_path_prefix(mapping, "/abc/def/src/main.rs"), path_str("/foo/src/main.rs"));
assert_eq!(map_path_prefix(mapping, "/abc/def"), path_str("/foo"));
}
}
#[test]
fn to_real_filename_with_full_scopes() {
let mapping = &FilePathMapping::new(
vec![(path("/foo"), path("FOO")), (path("/bar"), path("BAR"))],
RemapPathScopeComponents::all(),
);
let working_directory = path("/foo");
let working_directory = mapping.to_real_filename(&RealFileName::empty(), working_directory);
assert_eq!(working_directory.path(RemapPathScopeComponents::DIAGNOSTICS), path("FOO"));
assert_eq!(working_directory.path(RemapPathScopeComponents::MACRO), path("FOO"));
assert_eq!(working_directory.path(RemapPathScopeComponents::DEBUGINFO), path("FOO"));
assert_eq!(
mapping.to_real_filename(&working_directory, path("/foo/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/foo/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("FOO/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::all()
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/bar/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/bar/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/bar/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("BAR/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("BAR/src/main.rs")
},
scopes: RemapPathScopeComponents::all()
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/quux/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/quux/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/quux/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("/quux/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("/quux/src/main.rs")
},
scopes: RemapPathScopeComponents::all()
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::all()
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("./src/main.rs"),),
RealFileName {
local: Some(InnerRealFileName {
name: path("./src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("./src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::all()
}
);
assert_eq!(
mapping.to_real_filename(
&mapping.to_real_filename(&RealFileName::empty(), path("/abc")),
path("quux/src/main.rs"),
),
RealFileName {
local: Some(InnerRealFileName {
name: path("quux/src/main.rs"),
working_directory: path("/abc"),
embeddable_name: path("/abc/quux/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("quux/src/main.rs"),
working_directory: path("/abc"),
embeddable_name: path("/abc/quux/src/main.rs")
},
scopes: RemapPathScopeComponents::empty()
}
);
}
#[test]
fn to_real_filename_with_mixed_scopes() {
let mapping = &FilePathMapping::new(
vec![(path("/foo"), path("FOO")), (path("/bar"), path("BAR"))],
RemapPathScopeComponents::OBJECT,
);
let working_directory = path("/foo");
let working_directory = mapping.to_real_filename(&RealFileName::empty(), working_directory);
assert_eq!(working_directory.path(RemapPathScopeComponents::DIAGNOSTICS), path("/foo"));
assert_eq!(working_directory.path(RemapPathScopeComponents::MACRO), path("FOO"));
assert_eq!(working_directory.path(RemapPathScopeComponents::DEBUGINFO), path("FOO"));
assert_eq!(
mapping.to_real_filename(&working_directory, path("/foo/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/foo/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("FOO/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/bar/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/bar/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/bar/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("BAR/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("BAR/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/quux/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/quux/src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/quux/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("/quux/src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("/quux/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("/foo"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("FOO"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(
&mapping.to_real_filename(&RealFileName::empty(), path("/abc")),
path("quux/src/main.rs"),
),
RealFileName {
local: Some(InnerRealFileName {
name: path("quux/src/main.rs"),
working_directory: path("/abc"),
embeddable_name: path("/abc/quux/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("quux/src/main.rs"),
working_directory: path("/abc"),
embeddable_name: path("/abc/quux/src/main.rs")
},
scopes: RemapPathScopeComponents::empty()
}
);
}
#[test]
fn to_real_filename_without_remapped_cwd() {
let mapping = &FilePathMapping::new(
vec![(path("/foo"), path("FOO")), (path("/cwd/bar"), path("CWDBAR"))],
RemapPathScopeComponents::OBJECT,
);
let working_directory = path("/cwd");
let working_directory = mapping.to_real_filename(&RealFileName::empty(), working_directory);
assert_eq!(working_directory.path(RemapPathScopeComponents::DIAGNOSTICS), path("/cwd"));
assert_eq!(working_directory.path(RemapPathScopeComponents::MACRO), path("/cwd"));
assert_eq!(working_directory.path(RemapPathScopeComponents::DEBUGINFO), path("/cwd"));
assert_eq!(
mapping.to_real_filename(&working_directory, path("/foo/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/foo/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/foo/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("FOO/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("FOO/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/bar/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/bar/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/bar/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("/bar/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/bar/src/main.rs")
},
scopes: RemapPathScopeComponents::empty()
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("/cwd/bar/src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("/cwd/bar/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/cwd/bar/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("CWDBAR/src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("CWDBAR/src/main.rs")
},
scopes: RemapPathScopeComponents::OBJECT
}
);
assert_eq!(
mapping.to_real_filename(&working_directory, path("src/main.rs")),
RealFileName {
local: Some(InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/cwd/src/main.rs")
}),
maybe_remapped: InnerRealFileName {
name: path("src/main.rs"),
working_directory: path("/cwd"),
embeddable_name: path("/cwd/src/main.rs")
},
scopes: RemapPathScopeComponents::empty()
}
);
}
#[test]
fn path_prefix_remapping_reverse() {
{
let mapping = &FilePathMapping::new(
vec![(path("abc"), path("/")), (path("def"), path("."))],
RemapPathScopeComponents::all(),
);
assert_eq!(reverse_map_prefix(mapping, "/hello.rs"), None);
assert_eq!(reverse_map_prefix(mapping, "./hello.rs"), None);
}
{
let mapping = &FilePathMapping::new(
vec![(path("abc"), path("/redacted")), (path("def"), path("/redacted"))],
RemapPathScopeComponents::all(),
);
assert_eq!(reverse_map_prefix(mapping, "/redacted/hello.rs"), None);
}
{
let mapping = &FilePathMapping::new(
vec![(path("abc"), path("/redacted")), (path("def/ghi"), path("/fake/dir"))],
RemapPathScopeComponents::all(),
);
assert_eq!(
reverse_map_prefix(mapping, "/redacted/path/hello.rs"),
Some(path_str("abc/path/hello.rs"))
);
assert_eq!(
reverse_map_prefix(mapping, "/fake/dir/hello.rs"),
Some(path_str("def/ghi/hello.rs"))
);
}
}
#[test]
fn test_next_point() {
let sm = SourceMap::new(FilePathMapping::empty());
sm.new_source_file(filename(&sm, "example.rs"), "a…b".to_string());
let span = DUMMY_SP;
let span = sm.next_point(span);
assert_eq!(span.lo().0, 0);
assert_eq!(span.hi().0, 0);
let span = Span::with_root_ctxt(BytePos(0), BytePos(1));
assert_eq!(sm.span_to_snippet(span), Ok("a".to_string()));
let span = sm.next_point(span);
assert_eq!(sm.span_to_snippet(span), Ok("…".to_string()));
assert_eq!(span.lo().0, 1);
assert_eq!(span.hi().0, 4);
let span = Span::with_root_ctxt(BytePos(1), BytePos(1));
let span = sm.next_point(span);
assert_eq!(span.lo().0, 1);
assert_eq!(span.hi().0, 4);
let span = Span::with_root_ctxt(BytePos(1), BytePos(4));
let span = sm.next_point(span);
assert_eq!(span.lo().0, 4);
assert_eq!(span.hi().0, 5);
let span = Span::with_root_ctxt(BytePos(4), BytePos(5));
let span = sm.next_point(span);
assert_eq!(span.lo().0, 5);
assert_eq!(span.hi().0, 6);
assert!(sm.span_to_snippet(span).is_err());
let span = Span::with_root_ctxt(BytePos(5), BytePos(5));
let span = sm.next_point(span);
assert_eq!(span.lo().0, 5);
assert_eq!(span.hi().0, 6);
assert!(sm.span_to_snippet(span).is_err());
}
#[test]
fn test_span_followed_by_stops_at_end_of_file() {
let sm = SourceMap::new(FilePathMapping::empty());
sm.new_source_file(filename(&sm, "example.rs"), "x".to_string());
let span = Span::with_root_ctxt(BytePos(0), BytePos(1));
assert_eq!(sm.span_followed_by(span, "y"), None);
}
#[test]
fn test_span_followed_by_skips_whitespace() {
let sm = SourceMap::new(FilePathMapping::empty());
sm.new_source_file(filename(&sm, "example.rs"), "x \n yz".to_string());
let span = Span::with_root_ctxt(BytePos(0), BytePos(1));
let span = sm.span_followed_by(span, "yz").unwrap();
assert_eq!(sm.span_to_snippet(span), Ok("yz".to_string()));
}
#[cfg(target_os = "linux")]
#[test]
fn read_binary_file_handles_lying_stat() {
let cmdline = Path::new("/proc/self/cmdline");
let len = eko::file::metadata(cmdline).unwrap().len() as usize;
let real = eko::file::read(cmdline).unwrap();
assert!(len < real.len());
let bin = RealFileLoader.read_binary_file(cmdline).unwrap();
assert_eq!(&real[..], &bin[..]);
let kernel_max = Path::new("/sys/devices/system/cpu/kernel_max");
let len = eko::file::metadata(kernel_max).unwrap().len() as usize;
let real = eko::file::read(kernel_max).unwrap();
assert!(len > real.len());
let bin = RealFileLoader.read_binary_file(kernel_max).unwrap();
assert_eq!(&real[..], &bin[..]);
}