2 use std::io::{self, Write};
5 use syntax::source_map::SourceMap;
7 use crate::config::FileName;
8 use crate::emitter::{self, Emitter};
11 use crate::config::Config;
13 use crate::create_emitter;
15 use crate::formatting::FileRecord;
17 // Append a newline to the end of each file.
18 pub(crate) fn append_newline(s: &mut String) {
23 pub(crate) fn write_all_files<T>(
24 source_file: &[FileRecord],
27 ) -> Result<(), io::Error>
31 let mut emitter = create_emitter(config);
33 emitter.emit_header(out)?;
34 for &(ref filename, ref text) in source_file {
35 write_file(None, filename, text, out, &mut *emitter)?;
37 emitter.emit_footer(out)?;
42 pub(crate) fn write_file<T>(
43 source_map: Option<&SourceMap>,
47 emitter: &mut dyn Emitter,
48 ) -> Result<emitter::EmitterResult, io::Error>
52 fn ensure_real_path(filename: &FileName) -> &Path {
54 FileName::Real(ref path) => path,
55 _ => panic!("cannot format `{}` and emit to files", filename),
59 impl From<&FileName> for syntax_pos::FileName {
60 fn from(filename: &FileName) -> syntax_pos::FileName {
62 FileName::Real(path) => syntax_pos::FileName::Real(path.to_owned()),
63 FileName::Stdin => syntax_pos::FileName::Custom("stdin".to_owned()),
68 // If parse session is around (cfg(not(test))) then try getting source from
69 // there instead of hitting the file system. This also supports getting
70 // original text for `FileName::Stdin`.
71 let original_text = source_map
72 .and_then(|x| x.get_source_file(&filename.into()))
73 .and_then(|x| x.src.as_ref().map(ToString::to_string));
74 let original_text = match original_text {
76 None => fs::read_to_string(ensure_real_path(filename))?,
79 let formatted_file = emitter::FormattedFile {
81 original_text: &original_text,
85 emitter.emit_formatted_file(out, formatted_file)