Rustdoc: formatting to buffers

src/librustdoc/html/format.rs

@@ -18,6 +18,100 @@ use crate::clean::{self, PrimitiveType};
 use crate::html::item_type::ItemType;
 use crate::html::render::{self, cache, CURRENT_DEPTH};
 
+pub trait Print {
+ fn print(self, buffer: &mut Buffer);
+}
+
+impl<F> Print for F
+ where F: FnOnce(&mut Buffer),
+{
+ fn print(self, buffer: &mut Buffer) {
+ (self)(buffer)
+ }
+}
+
+impl Print for String {
+ fn print(self, buffer: &mut Buffer) {
+ buffer.write_str(&self);
+ }
+}
+
+impl Print for &'_ str {
+ fn print(self, buffer: &mut Buffer) {
+ buffer.write_str(self);
+ }
+}
+
+#[derive(Debug, Clone)]
+pub struct Buffer {
+ for_html: bool,
+ buffer: String,
+}
+
+impl Buffer {
+ crate fn empty_from(v: &Buffer) -> Buffer {
+ Buffer {
+ for_html: v.for_html,
+ buffer: String::new(),
+ }
+ }
+
+ crate fn html() -> Buffer {
+ Buffer {
+ for_html: true,
+ buffer: String::new(),
+ }
+ }
+
+ crate fn is_empty(&self) -> bool {
+ self.buffer.is_empty()
+ }
+
+ crate fn into_inner(self) -> String {
+ self.buffer
+ }
+
+ crate fn insert_str(&mut self, idx: usize, s: &str) {
+ self.buffer.insert_str(idx, s);
+ }
+
+ crate fn push_str(&mut self, s: &str) {
+ self.buffer.push_str(s);
+ }
+
+ // Intended for consumption by write! and writeln! (std::fmt) but without
+ // the fmt::Result return type imposed by fmt::Write (and avoiding the trait
+ // import).
+ crate fn write_str(&mut self, s: &str) {
+ self.buffer.push_str(s);
+ }
+
+ // Intended for consumption by write! and writeln! (std::fmt) but without
+ // the fmt::Result return type imposed by fmt::Write (and avoiding the trait
+ // import).
+ crate fn write_fmt(&mut self, v: fmt::Arguments<'_>) {
+ use fmt::Write;
+ self.buffer.write_fmt(v).unwrap();
+ }
+
+ crate fn to_display<T: Print>(mut self, t: T) -> String {
+ t.print(&mut self);
+ self.into_inner()
+ }
+
+ crate fn with_formatter<T: FnOnce(&mut fmt::Formatter<'_>) -> fmt::Result>(&mut self, t: T) {
+ self.from_display(display_fn(move |f| (t)(f)));
+ }
+
+ crate fn from_display<T: std::fmt::Display>(&mut self, t: T) {
+ if self.for_html {
+ write!(self, "{}", t);
+ } else {
+ write!(self, "{:#}", t);
+ }
+ }
+}
+
 /// Helper to render an optional visibility with a space after it (if the
 /// visibility is preset)
 #[derive(Copy, Clone)]
@@ -200,11 +294,11 @@ impl<'a> fmt::Display for WhereClause<'a> {
  }
  &clean::WherePredicate::RegionPredicate { ref lifetime, ref bounds } => {
  clause.push_str(&format!("{}: {}",
- lifetime,
- bounds.iter()
- .map(|b| b.to_string())
- .collect::<Vec<_>>()
- .join(" + ")));
+  lifetime,
+  bounds.iter()
+  .map(|b| b.to_string())
+  .collect::<Vec<_>>()
+  .join(" + ")));
  }
  &clean::WherePredicate::EqPredicate { ref lhs, ref rhs } => {
  if f.alternate() {
@@ -778,9 +872,9 @@ impl fmt::Display for clean::Impl {
 
 // The difference from above is that trait is not hyperlinked.
 pub fn fmt_impl_for_trait_page(i: &clean::Impl,
- f: &mut fmt::Formatter<'_>,
- use_absolute: bool) -> fmt::Result {
- fmt_impl(i, f, false, use_absolute)
+ f: &mut Buffer,
+ use_absolute: bool) {
+ f.with_formatter(|f| fmt_impl(i, f, false, use_absolute))
 }
 
 impl fmt::Display for clean::Arguments {

src/librustdoc/html/layout.rs

@@ -1,16 +1,21 @@
-use std::fmt;
-use std::io;
 use std::path::PathBuf;
 
 use crate::externalfiles::ExternalHtml;
 use crate::html::render::SlashChecker;
+use crate::html::format::{Buffer, Print};
 
 #[derive(Clone)]
 pub struct Layout {
  pub logo: String,
  pub favicon: String,
  pub external_html: ExternalHtml,
  pub krate: String,
+ /// The given user css file which allow to customize the generated
+ /// documentation theme.
+ pub css_file_extension: Option<PathBuf>,
+ /// If false, the `select` element to have search filtering by crates on rendered docs
+ /// won't be generated.
+ pub generate_search_filter: bool,
 }
 
 pub struct Page<'a> {
@@ -25,19 +30,15 @@ pub struct Page<'a> {
  pub static_extra_scripts: &'a [&'a str],
 }
 
-pub fn render<T: fmt::Display, S: fmt::Display>(
- dst: &mut dyn io::Write,
+pub fn render<T: Print, S: Print>(
  layout: &Layout,
  page: &Page<'_>,
- sidebar: &S,
- t: &T,
- css_file_extension: bool,
+ sidebar: S,
+ t: T,
  themes: &[PathBuf],
- generate_search_filter: bool,
-) -> io::Result<()> {
+) -> String {
  let static_root_path = page.static_root_path.unwrap_or(page.root_path);
- write!(dst,
-"<!DOCTYPE html>\
+ format!("<!DOCTYPE html>\
 <html lang=\"en\">\
 <head>\
  <meta charset=\"utf-8\">\
@@ -164,7 +165,7 @@ pub fn render<T: fmt::Display, S: fmt::Display>(
  <script defer src=\"{root_path}search-index{suffix}.js\"></script>\
 </body>\
 </html>",
- css_extension = if css_file_extension {
+ css_extension = if layout.css_file_extension.is_some() {
  format!("<link rel=\"stylesheet\" \
  type=\"text/css\" \
  href=\"{static_root_path}theme{suffix}.css\">",
@@ -173,7 +174,7 @@ pub fn render<T: fmt::Display, S: fmt::Display>(
  } else {
  String::new()
  },
- content = *t,
+ content = Buffer::html().to_display(t),
  static_root_path = static_root_path,
  root_path = page.root_path,
  css_class = page.css_class,
@@ -207,7 +208,7 @@ pub fn render<T: fmt::Display, S: fmt::Display>(
  in_header = layout.external_html.in_header,
  before_content = layout.external_html.before_content,
  after_content = layout.external_html.after_content,
- sidebar = *sidebar,
+ sidebar = Buffer::html().to_display(sidebar),
  krate = layout.krate,
  themes = themes.iter()
  .filter_map(|t| t.file_stem())
@@ -228,7 +229,7 @@ pub fn render<T: fmt::Display, S: fmt::Display>(
  root_path=page.root_path,
  extra_script=e)
  }).collect::<String>(),
- filter_crates=if generate_search_filter {
+ filter_crates=if layout.generate_search_filter {
  "<select id=\"crate-search\">\
  <option value=\"All crates\">All crates</option>\
  </select>"
@@ -238,9 +239,9 @@ pub fn render<T: fmt::Display, S: fmt::Display>(
  )
 }
 
-pub fn redirect(dst: &mut dyn io::Write, url: &str) -> io::Result<()> {
+pub fn redirect(url: &str) -> String {
  // <script> triggers a redirect before refresh, so this is fine.
- write!(dst,
+ format!(
 r##"<!DOCTYPE html>
 <html lang="en">
 <head>