add tinytemplate

bartlomieju · Oct 27, 2023 · 0678a11 · 0678a11
1 parent f355a95
commit 0678a11
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -37,6 +37,7 @@ wasm-bindgen-futures = { version = "=0.4.37", optional = true }
 html-escape = { version = "0.2.13", optional = true }
 comrak = { version = "0.19.0", optional = true }
 indexmap = { version = "2.0.2", optional = true }
+tinytemplate = "1.2.1"
 
 [dev-dependencies]
 anyhow = { version = "1.0.58" }

diff --git a/examples/ddoc/main.rs b/examples/ddoc/main.rs
@@ -118,7 +118,7 @@ fn generate_docs_directory(
     // TODO: don't hardcode the path
     base_url: "/generated_docs/".to_string(),
   };
-  let html = deno_doc::html::generate(ctx.clone(), doc_nodes);
+  let html = deno_doc::html::generate(ctx.clone(), doc_nodes)?;
 
   // TODO: don't hardcode the path
   let base_path = format!(

diff --git a/src/html/mod.rs b/src/html/mod.rs
@@ -1,5 +1,6 @@
 use indexmap::IndexMap;
 use std::collections::HashMap;
+use tinytemplate::TinyTemplate;
 
 use crate::DocNodeKind;
 
@@ -24,6 +25,18 @@ pub const SEARCH_INDEX_FILENAME: &str = "search_index.js";
 pub const SEARCH_JS: &str = include_str!("./search.js");
 pub const SEARCH_FILENAME: &str = "search.js";
 
+const HTML_TEMPLATE: &str = r#"
+<html>
+<head>
+  <link rel="stylesheet" href="./styles.css">
+</head>
+<body>
+  {content}
+</body>
+<script type="module" src="./search_index.js" defer></script>
+<script type="module" src="./search.js" defer></script>
+</html>"#;
+
 // TODO(bartlomieju): reference STYLESHEET_FILENAME below
 const HTML_HEAD: &str = r#"
 <html>
@@ -98,22 +111,32 @@ pub fn generate_search_index(
 pub fn generate(
   ctx: GenerateCtx,
   doc_nodes: &[crate::DocNode],
-) -> HashMap<String, String> {
+) -> Result<HashMap<String, String>, anyhow::Error> {
+  let mut tt = TinyTemplate::new();
+  tt.set_default_formatter(&tinytemplate::format_unescaped);
+  tt.add_template("html", HTML_TEMPLATE)?;
+
   let mut files = HashMap::new();
 
   // FIXME(bartlomieju): functions can have duplicates because of overloads
   let partitions = namespace::partition_nodes_by_kind(doc_nodes);
   let name_partitions = partition_nodes_by_name(doc_nodes);
 
   let sidepanel = render_sidepanel(&ctx, &partitions);
+  let index_content = render_index(&ctx, &sidepanel, partitions);
   files.insert(
     "index".to_string(),
-    render_index(&ctx, &sidepanel, partitions),
+    tt.render(
+      "html",
+      &serde_json::json!({
+        "content": index_content
+      }),
+    )?,
   );
 
   generate_pages(name_partitions, &mut files, &ctx, &sidepanel, None);
 
-  files
+  Ok(files)
 }
 
 fn generate_pages(