1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
use std::fs;
use std::io::{self, Write};
use std::path::Path;
use syntax::source_map::SourceMap;
use crate::config::FileName;
use crate::emitter::{self, Emitter};
use crate::NewlineStyle;
#[cfg(test)]
use crate::config::Config;
#[cfg(test)]
use crate::create_emitter;
#[cfg(test)]
use crate::formatting::FileRecord;
pub(crate) fn append_newline(s: &mut String) {
s.push_str("\n");
}
#[cfg(test)]
pub(crate) fn write_all_files<T>(
source_file: &[FileRecord],
out: &mut T,
config: &Config,
) -> Result<(), io::Error>
where
T: Write,
{
let mut emitter = create_emitter(config);
emitter.emit_header(out)?;
for &(ref filename, ref text) in source_file {
write_file(
None,
filename,
text,
out,
&mut *emitter,
config.newline_style(),
)?;
}
emitter.emit_footer(out)?;
Ok(())
}
pub(crate) fn write_file<T>(
source_map: Option<&SourceMap>,
filename: &FileName,
formatted_text: &str,
out: &mut T,
emitter: &mut dyn Emitter,
newline_style: NewlineStyle,
) -> Result<emitter::EmitterResult, io::Error>
where
T: Write,
{
fn ensure_real_path(filename: &FileName) -> &Path {
match *filename {
FileName::Real(ref path) => path,
_ => panic!("cannot format `{}` and emit to files", filename),
}
}
impl From<&FileName> for syntax_pos::FileName {
fn from(filename: &FileName) -> syntax_pos::FileName {
match filename {
FileName::Real(path) => syntax_pos::FileName::Real(path.to_owned()),
FileName::Stdin => syntax_pos::FileName::Custom("stdin".to_owned()),
}
}
}
let original_text = if newline_style != NewlineStyle::Auto && *filename != FileName::Stdin {
fs::read_to_string(ensure_real_path(filename))?
} else {
match source_map
.and_then(|x| x.get_source_file(&filename.into()))
.and_then(|x| x.src.as_ref().map(ToString::to_string))
{
Some(ori) => ori,
None => fs::read_to_string(ensure_real_path(filename))?,
}
};
let formatted_file = emitter::FormattedFile {
filename,
original_text: &original_text,
formatted_text,
};
emitter.emit_formatted_file(out, formatted_file)
}