lew/guestbook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

configurable separator, .env.example

Browse source
This commit is contained in:
Lewis Wynne 2026-04-09 16:58:42 +01:00
parent d751eb62a9
commit ef6a190549
5 changed files with 39 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/config.rs
View file

@ -15,6 +15,7 @@ pub struct Config {
pub max_website_length: usize,
pub open_registration: bool,
pub template: Option<String>,
pub separator: String,
}
impl Config {
@ -57,6 +58,8 @@ impl Config {
open_registration: env::var("BOOK_OPEN_REGISTRATION")
.map(|v| v != "false")
.unwrap_or(true),
separator: env::var("BOOK_SEPARATOR")
.unwrap_or_else(|_| "------------------------------------------------------------".into()),
template: env::var("BOOK_TEMPLATE").ok().map(|path| {
std::fs::read_to_string(&path)
.unwrap_or_else(|e| panic!("failed to read template {path}: {e}"))

28
src/render.rs
View file

@ -35,23 +35,23 @@ entries
</html>
"#;
pub fn render_page(template: &str, title: &str, entries: &[Entry], form_html: &str) -> String {
let entries_html = render_entries(entries);
pub fn render_page(template: &str, title: &str, entries: &[Entry], form_html: &str, separator: &str) -> String {
let entries_html = render_entries(entries, separator);
template
.replace("{{title}}", title)
.replace("{{form}}", form_html)
.replace("{{entries}}", &entries_html)
}
fn render_entries(entries: &[Entry]) -> String {
fn render_entries(entries: &[Entry], separator: &str) -> String {
let mut html = String::new();
for entry in entries {
html.push_str(&render_entry(entry));
html.push_str(&render_entry(entry, separator));
}
html
}
fn render_entry(entry: &Entry) -> String {
fn render_entry(entry: &Entry, separator: &str) -> String {
let mut header = format!("{} - {}", entry.meta.date, entry.meta.name);
if !entry.meta.website.is_empty() {
header.push_str(&format!(
@ -60,7 +60,7 @@ fn render_entry(entry: &Entry) -> String {
));
}
format!(
"\n{header}\n\n{}\n\n------------------------------------------------------------\n",
"\n{header}\n\n{}\n\n{separator}\n",
entry.body
)
}
@ -99,16 +99,16 @@ mod tests {
#[test]
fn test_render_default_template() {
let html = render_page(DEFAULT_TEMPLATE, "ily.rs", &[], FORM_HTML);
let html = render_page(DEFAULT_TEMPLATE, "ily.rs", &[], FORM_HTML, "---");
assert!(html.contains("<title>ily.rs</title>"));
assert!(html.contains("action=\"/submit\""));
assert!(html.contains("<pre style=font:unset>"));
assert!(html.contains("<pre>"));
}
#[test]
fn test_render_custom_template() {
let custom = "<html>{{title}} {{form}} {{entries}}</html>";
let html = render_page(custom, "my site", &[], FORM_HTML);
let html = render_page(custom, "my site", &[], FORM_HTML, "---");
assert!(html.contains("my site"));
assert!(html.contains("action=\"/submit\""));
}
@ -116,31 +116,31 @@ mod tests {
#[test]
fn test_render_entry_no_website() {
let entry = make_entry("alice", "2026-04-09", "Hello!");
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML);
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML, "---");
assert!(html.contains("2026-04-09 - alice"));
assert!(html.contains("Hello!"));
assert!(html.contains("----"));
assert!(html.contains("---"));
}
#[test]
fn test_render_entry_with_website() {
let mut entry = make_entry("bob", "2026-04-09", "Hi!");
entry.meta.website = "https://bob.com".into();
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML);
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML, "---");
assert!(html.contains(r#"<a href="https://bob.com">"#));
}
#[test]
fn test_render_preserves_html_in_body() {
let entry = make_entry("carol", "2026-04-09", "<b>Bold</b> <script>alert(1)</script>");
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML);
let html = render_page(DEFAULT_TEMPLATE, "test", &[entry], FORM_HTML, "---");
assert!(html.contains("<b>Bold</b>"));
assert!(html.contains("<script>alert(1)</script>"));
}
#[test]
fn test_render_empty_form_when_closed() {
let html = render_page(DEFAULT_TEMPLATE, "test", &[], "");
let html = render_page(DEFAULT_TEMPLATE, "test", &[], "", "---");
assert!(!html.contains("action=\"/submit\""));
}
}

2
src/web.rs
View file

@ -44,6 +44,7 @@ async fn index(State(state): State<Arc<AppState>>) -> Html<String> {
&state.config.site_title,
&entries,
form,
&state.config.separator,
);
Html(html)
}
@ -134,6 +135,7 @@ mod tests {
max_website_length: 100,
open_registration: true,
template: None,
separator: "---".into(),
}
}