[tablify] / src / lib.rs

use std::borrow::ToOwned;
use std::collections::HashMap;
use std::fmt::Write;
use std::io::BufRead;
use std::iter::Iterator;

pub struct Config {}

const HEADER: &str = r#"<!DOCTYPE html>
<html>
<head>
  <meta charset="utf-8">
  <meta name="viewport" content="width=device-width, initial-scale=1">
  <style>
    td { text-align: center; }
    /* h/t https://wabain.github.io/2019/10/13/css-rotated-table-header.html */
    th, td { white-space: nowrap; }
    th { text-align: left; font-weight: normal; }
    table { border-collapse: collapse }
    tr.key > th { height: 10em; vertical-align: bottom; line-height: 1 }
    tr.key > th > div { width: 1em; }
    tr.key > th > div > div { width: 5em; transform-origin: bottom left; transform: translateX(1em) rotate(-65deg) }
    td { border: thin solid gray; }
    td.yes { border: thin solid gray; background-color: #ddd; }
    /* h/t https://stackoverflow.com/questions/5687035/css-bolding-some-text-without-changing-its-containers-size/46452396#46452396 */
    .highlight { text-shadow: -0.06ex 0 black, 0.06ex 0 black; }
  </style>
  <script>
    function highlight(id)       { const e = document.getElementById(id); if (e) { e.classList.add(   "highlight"); } }
    function clear_highlight(id) { const e = document.getElementById(id); if (e) { e.classList.remove("highlight"); } }
    function h2(a, b)  {       highlight(a);       highlight(b); }
    function ch2(a, b) { clear_highlight(a); clear_highlight(b); }
  </script>
</head>
<body>
  <table>
    <tbody>
"#;
const FOOTER: &str = "    </tbody>
  </table>
</body>
</html>";

#[derive(PartialEq, Eq, Debug)]
pub struct HTML(String);
impl HTML {
    fn escape(value: &str) -> HTML {
        let mut escaped: String = String::new();
        for c in value.chars() {
            match c {
                '>' => escaped.push_str("&gt;"),
                '<' => escaped.push_str("&lt;"),
                '\'' => escaped.push_str("&#39;"),
                '"' => escaped.push_str("&quot;"),
                '&' => escaped.push_str("&amp;"),
                ok_c => escaped.push(ok_c),
            }
        }
        HTML(escaped)
    }
}
impl From<&str> for HTML {
    fn from(value: &str) -> HTML {
        HTML(String::from(value))
    }
}
impl FromIterator<HTML> for HTML {
    fn from_iter<T>(iter: T) -> HTML
    where
        T: IntoIterator<Item = HTML>,
    {
        HTML(iter.into_iter().map(|html| html.0).collect::<String>())
    }
}
impl std::fmt::Display for HTML {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

#[derive(Debug, PartialEq, Eq)]
enum InputLine<'a> {
    Blank,
    RowHeader(&'a str),
    Entry(&'a str, Option<&'a str>),
}
impl<'a> From<&'a str> for InputLine<'a> {
    fn from(value: &'a str) -> InputLine<'a> {
        let trimmed = value.trim_end();
        if trimmed.is_empty() {
            InputLine::Blank
        } else if !trimmed.starts_with(' ') {
            InputLine::RowHeader(value.trim())
        } else {
            match value.split_once(':') {
                None => InputLine::Entry(value.trim(), None),
                Some((col, instance)) => InputLine::Entry(col.trim(), Some(instance.trim())),
            }
        }
    }
}

#[derive(Debug, PartialEq, Eq)]
struct Row {
    label: String,
    entries: HashMap<String, Vec<Option<String>>>,
}

struct Reader<Input: Iterator<Item = Result<String, std::io::Error>>> {
    input: std::iter::Enumerate<Input>,
    row: Option<Row>,
}
impl<Input: Iterator<Item = Result<String, std::io::Error>>> Reader<Input> {
    fn new(input: Input) -> Self {
        Self {
            input: input.enumerate(),
            row: None,
        }
    }
}
impl<Input: Iterator<Item = Result<String, std::io::Error>>> Iterator for Reader<Input> {
    type Item = Result<Row, std::io::Error>;
    fn next(&mut self) -> Option<Self::Item> {
        loop {
            match self.input.next() {
                None => return Ok(std::mem::take(&mut self.row)).transpose(),
                Some((_, Err(e))) => return Some(Err(e)),
                Some((n, Ok(line))) => match InputLine::from(line.as_ref()) {
                    InputLine::Blank if self.row.is_some() => {
                        return Ok(std::mem::take(&mut self.row)).transpose()
                    }
                    InputLine::Blank => {}
                    InputLine::Entry(col, instance) => match &mut self.row {
                        None => {
                            return Some(Err(std::io::Error::other(format!(
                                "{}: Entry with no header",
                                n + 1
                            ))))
                        }
                        Some(ref mut row) => {
                            row.entries
                                .entry(col.to_owned())
                                .and_modify(|is| is.push(instance.map(ToOwned::to_owned)))
                                .or_insert_with(|| vec![instance.map(ToOwned::to_owned)]);
                        }
                    },
                    InputLine::RowHeader(row) => {
                        let prev = std::mem::take(&mut self.row);
                        self.row = Some(Row {
                            label: row.to_owned(),
                            entries: HashMap::new(),
                        });
                        if prev.is_some() {
                            return Ok(prev).transpose();
                        }
                    }
                },
            }
        }
    }
}

fn read_rows(input: impl std::io::Read) -> impl Iterator<Item = Result<Row, std::io::Error>> {
    Reader::new(std::io::BufReader::new(input).lines())
}

fn column_counts(rows: &[Row]) -> Vec<(usize, String)> {
    let mut counts: Vec<_> = rows
        .iter()
        .flat_map(|r| r.entries.keys())
        .fold(HashMap::new(), |mut cs, col| {
            cs.entry(col.to_owned())
                .and_modify(|n| *n += 1)
                .or_insert(1);
            cs
        })
        .into_iter()
        .map(|(col, n)| (n, col))
        .collect();
    counts.sort_unstable_by(|(an, acol), (bn, bcol)| bn.cmp(an).then(acol.cmp(bcol)));
    counts
}
fn column_order(rows: &[Row]) -> Vec<String> {
    column_counts(rows)
        .into_iter()
        .map(|(_, col)| col)
        .collect()
}

fn render_one_instance(instance: &Option<String>) -> HTML {
    match instance {
        None => HTML::from("✓"),
        Some(instance) => HTML::escape(instance.as_ref()),
    }
}

fn render_instances(instances: &[Option<String>]) -> HTML {
    let all_empty = instances.iter().all(Option::is_none);
    if all_empty && instances.len() == 1 {
        HTML::from("")
    } else if all_empty {
        HTML(format!("{}", instances.len()))
    } else {
        HTML(
            instances
                .iter()
                .map(render_one_instance)
                .map(|html| html.0) // Waiting for slice_concat_trait to stabilize
                .collect::<Vec<_>>()
                .join(" "),
        )
    }
}

fn render_cell(col: &str, row: &mut Row) -> HTML {
    let row_label = HTML::escape(row.label.as_ref());
    let col_label = HTML::escape(col);
    let instances: Option<&Vec<Option<String>>> = row.entries.get(col);
    let class = HTML::from(if instances.is_none() { "" } else { "yes" });
    let contents = match instances {
        None => HTML::from(""),
        Some(is) => render_instances(is),
    };
    row.entries.remove(col);
    HTML(format!(
        r#"<td class="{class}" onmouseover="h2('{row_label}','{col_label}')" onmouseout="ch2('{row_label}','{col_label}')">{contents}</td>"#
    ))
}

fn render_row(columns: &[String], row: &mut Row) -> HTML {
    let row_label = HTML::escape(row.label.as_ref());
    let cells = columns
        .iter()
        .map(|col| render_cell(col, row))
        .collect::<HTML>();
    HTML(format!(
        "<tr><th id=\"{row_label}\">{row_label}</th>{cells}</tr>\n"
    ))
}

fn render_column_headers(columns: &[String]) -> HTML {
    HTML(
        String::from(r#"<tr class="key"><th></th>"#)
            + &columns.iter().fold(String::new(), |mut acc, col| {
                let col_header = HTML::escape(col.as_ref());
                write!(
                    &mut acc,
                    r#"<th id="{col_header}"><div><div>{col_header}</div></div></th>"#
                )
                .unwrap();
                acc
            })
            + "</tr>\n",
    )
}

/// # Errors
///
/// Will return `Err` if
///   * there's an i/o error while reading `input`
///   * the log has invalid syntax:
///     * an indented line with no preceding non-indented line
pub fn tablify(config: &Config, input: impl std::io::Read) -> Result<HTML, std::io::Error> {
    let rows = read_rows(input).collect::<Result<Vec<_>, _>>()?;
    let columns = column_order(&rows);
    Ok(HTML(format!(
        "{HEADER}{}{}{FOOTER}",
        render_column_headers(&columns),
        rows.into_iter()
            .map(|mut r| render_row(&columns, &mut r))
            .collect::<HTML>()
    )))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_parse_line() {
        assert_eq!(InputLine::from(""), InputLine::Blank);
        assert_eq!(InputLine::from(" "), InputLine::Blank);
        assert_eq!(InputLine::from("foo"), InputLine::RowHeader("foo"));
        assert_eq!(InputLine::from("foo "), InputLine::RowHeader("foo"));
        assert_eq!(InputLine::from(" foo"), InputLine::Entry("foo", None));
        assert_eq!(
            InputLine::from(" foo:bar"),
            InputLine::Entry("foo", Some("bar"))
        );
        assert_eq!(
            InputLine::from(" foo: bar"),
            InputLine::Entry("foo", Some("bar"))
        );
        assert_eq!(
            InputLine::from(" foo: bar "),
            InputLine::Entry("foo", Some("bar"))
        );
        assert_eq!(
            InputLine::from(" foo:  bar "),
            InputLine::Entry("foo", Some("bar"))
        );
        assert_eq!(
            InputLine::from(" foo :  bar "),
            InputLine::Entry("foo", Some("bar"))
        );
    }

    #[test]
    fn test_read_rows() {
        assert_eq!(
            read_rows(&b"foo"[..]).flatten().collect::<Vec<_>>(),
            vec![Row {
                label: "foo".to_owned(),
                entries: HashMap::new(),
            }]
        );
        assert_eq!(
            read_rows(&b"bar"[..]).flatten().collect::<Vec<_>>(),
            vec![Row {
                label: "bar".to_owned(),
                entries: HashMap::new(),
            }]
        );
        assert_eq!(
            read_rows(&b"foo\nbar\n"[..]).flatten().collect::<Vec<_>>(),
            vec![
                Row {
                    label: "foo".to_owned(),
                    entries: HashMap::new(),
                },
                Row {
                    label: "bar".to_owned(),
                    entries: HashMap::new(),
                }
            ]
        );
        assert_eq!(
            read_rows(&b"foo\n bar\n"[..]).flatten().collect::<Vec<_>>(),
            vec![Row {
                label: "foo".to_owned(),
                entries: HashMap::from([("bar".to_owned(), vec![None])]),
            }]
        );
        assert_eq!(
            read_rows(&b"foo\n bar\n baz\n"[..])
                .flatten()
                .collect::<Vec<_>>(),
            vec![Row {
                label: "foo".to_owned(),
                entries: HashMap::from([
                    ("bar".to_owned(), vec![None]),
                    ("baz".to_owned(), vec![None])
                ]),
            }]
        );
        assert_eq!(
            read_rows(&b"foo\n\nbar\n"[..])
                .flatten()
                .collect::<Vec<_>>(),
            vec![
                Row {
                    label: "foo".to_owned(),
                    entries: HashMap::new(),
                },
                Row {
                    label: "bar".to_owned(),
                    entries: HashMap::new(),
                }
            ]
        );
        assert_eq!(
            read_rows(&b"foo\n \nbar\n"[..])
                .flatten()
                .collect::<Vec<_>>(),
            vec![
                Row {
                    label: "foo".to_owned(),
                    entries: HashMap::new(),
                },
                Row {
                    label: "bar".to_owned(),
                    entries: HashMap::new(),
                }
            ]
        );
        assert_eq!(
            read_rows(&b"foo  \n bar  \n"[..])
                .flatten()
                .collect::<Vec<_>>(),
            vec![Row {
                label: "foo".to_owned(),
                entries: HashMap::from([("bar".to_owned(), vec![None])]),
            }]
        );

        let bad = read_rows(&b" foo"[..]).next().unwrap();
        assert!(bad.is_err());
        assert!(format!("{bad:?}").contains("1: Entry with no header"));

        let bad2 = read_rows(&b"foo\n\n bar"[..]).nth(1).unwrap();
        assert!(bad2.is_err());
        assert!(format!("{bad2:?}").contains("3: Entry with no header"));
    }

    #[test]
    fn test_column_counts() {
        assert_eq!(
            column_counts(
                &read_rows(&b"foo\n bar\n baz\n"[..])
                    .collect::<Result<Vec<_>, _>>()
                    .unwrap()
            ),
            vec![(1, String::from("bar")), (1, String::from("baz"))]
        );
        assert_eq!(
            column_counts(
                &read_rows(&b"foo\n bar\n baz\nquux\n baz"[..])
                    .collect::<Result<Vec<_>, _>>()
                    .unwrap()
            ),
            vec![(2, String::from("baz")), (1, String::from("bar"))]
        );
        assert_eq!(
            column_counts(
                &read_rows(&b"foo\n bar\n bar\n baz\n bar\nquux\n baz"[..])
                    .collect::<Result<Vec<_>, _>>()
                    .unwrap()
            ),
            vec![(2, String::from("baz")), (1, String::from("bar"))]
        );
        assert_eq!(
            column_counts(
                &read_rows(&b"foo\n bar: 1\n bar: 2\n baz\n bar\nquux\n baz"[..])
                    .collect::<Result<Vec<_>, _>>()
                    .unwrap()
            ),
            vec![(2, String::from("baz")), (1, String::from("bar"))]
        );
    }

    #[test]
    fn test_render_cell() {
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::new(),
                }
            ),
            HTML::from(
                r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("bar".to_owned(), vec![None])]),
                }
            ),
            HTML::from(
                r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("foo".to_owned(), vec![None])]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("foo".to_owned(), vec![None, None])]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">2</td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([(
                        "foo".to_owned(),
                        vec![Some("5".to_owned()), Some("10".to_owned())]
                    )]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 10</td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("foo".to_owned(), vec![Some("5".to_owned()), None])]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 ✓</td>"#
            )
        );
        assert_eq!(
            render_cell(
                "heart",
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("heart".to_owned(), vec![Some("<3".to_owned())])]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('nope','heart')" onmouseout="ch2('nope','heart')">&lt;3</td>"#
            )
        );
        assert_eq!(
            render_cell(
                "foo",
                &mut Row {
                    label: "bob's".to_owned(),
                    entries: HashMap::from([("foo".to_owned(), vec![None])]),
                }
            ),
            HTML::from(
                r#"<td class="yes" onmouseover="h2('bob&#39;s','foo')" onmouseout="ch2('bob&#39;s','foo')"></td>"#
            )
        );
        let mut r = Row {
            label: "nope".to_owned(),
            entries: HashMap::from([
                ("foo".to_owned(), vec![None]),
                ("baz".to_owned(), vec![None]),
            ]),
        };
        assert_eq!(r.entries.len(), 2);
        render_cell("foo", &mut r);
        assert_eq!(r.entries.len(), 1);
        render_cell("bar", &mut r);
        assert_eq!(r.entries.len(), 1);
        render_cell("baz", &mut r);
        assert_eq!(r.entries.len(), 0);
    }

    #[test]
    fn test_render_row() {
        assert_eq!(
            render_row(
                &["foo".to_owned()],
                &mut Row {
                    label: "nope".to_owned(),
                    entries: HashMap::from([("bar".to_owned(), vec![None])]),
                }
            ),
            HTML::from(
                r#"<tr><th id="nope">nope</th><td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td></tr>
"#
            )
        );
    }
}
Commit	Line	Data
88a08162 SW	1	use std::borrow::ToOwned;
88a08162 SW	2	use std::collections::HashMap;
7067975b	3	use std::fmt::Write;
9dfa98b7	4	use std::io::BufRead;
75bb888a SW	5	use std::iter::Iterator;
75bb888a SW	6
28cf4fa2 SW	7	pub struct Config {}
28cf4fa2 SW	8
5ffe8e3a	9	const HEADER: &str = r#"<!DOCTYPE html>
cc2378d5 SW	10	<html>
cc2378d5 SW	11	<head>
5ffe8e3a SW	12	<meta charset="utf-8">
5ffe8e3a SW	13	<meta name="viewport" content="width=device-width, initial-scale=1">
cc2378d5	14	<style>
b8b365ce	15	td { text-align: center; }
cc2378d5 SW	16	/* h/t https://wabain.github.io/2019/10/13/css-rotated-table-header.html */
	17	th, td { white-space: nowrap; }
	18	th { text-align: left; font-weight: normal; }
	19	table { border-collapse: collapse }
3bc643e9	20	tr.key > th { height: 10em; vertical-align: bottom; line-height: 1 }
cc2378d5 SW	21	tr.key > th > div { width: 1em; }
	22	tr.key > th > div > div { width: 5em; transform-origin: bottom left; transform: translateX(1em) rotate(-65deg) }
	23	td { border: thin solid gray; }
1dda21e6	24	td.yes { border: thin solid gray; background-color: #ddd; }
cc2378d5 SW	25	/* h/t https://stackoverflow.com/questions/5687035/css-bolding-some-text-without-changing-its-containers-size/46452396#46452396 */
cc2378d5 SW	26	.highlight { text-shadow: -0.06ex 0 black, 0.06ex 0 black; }
cc2378d5 SW	27	</style>
cc2378d5 SW	28	<script>
5ffe8e3a SW	29	function highlight(id) { const e = document.getElementById(id); if (e) { e.classList.add( "highlight"); } }
5ffe8e3a SW	30	function clear_highlight(id) { const e = document.getElementById(id); if (e) { e.classList.remove("highlight"); } }
cc2378d5 SW	31	function h2(a, b) { highlight(a); highlight(b); }
	32	function ch2(a, b) { clear_highlight(a); clear_highlight(b); }
	33	</script>
	34	</head>
	35	<body>
	36	<table>
76638ea1	37	<tbody>
5ffe8e3a	38	"#;
cc2378d5 SW	39	const FOOTER: &str = " </tbody>
	40	</table>
	41	</body>
	42	</html>";
	43
70436f23 SW	44	#[derive(PartialEq, Eq, Debug)]
	45	pub struct HTML(String);
	46	impl HTML {
	47	fn escape(value: &str) -> HTML {
	48	let mut escaped: String = String::new();
	49	for c in value.chars() {
	50	match c {
	51	'>' => escaped.push_str(">"),
	52	'<' => escaped.push_str("<"),
	53	'\'' => escaped.push_str("'"),
	54	'"' => escaped.push_str("""),
	55	'&' => escaped.push_str("&"),
	56	ok_c => escaped.push(ok_c),
	57	}
	58	}
	59	HTML(escaped)
	60	}
	61	}
	62	impl From<&str> for HTML {
	63	fn from(value: &str) -> HTML {
	64	HTML(String::from(value))
	65	}
	66	}
	67	impl FromIterator<HTML> for HTML {
	68	fn from_iter<T>(iter: T) -> HTML
	69	where
	70	T: IntoIterator<Item = HTML>,
	71	{
	72	HTML(iter.into_iter().map(\|html\| html.0).collect::<String>())
	73	}
	74	}
	75	impl std::fmt::Display for HTML {
	76	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
	77	write!(f, "{}", self.0)
	78	}
	79	}
	80
88a08162 SW	81	#[derive(Debug, PartialEq, Eq)]
	82	enum InputLine<'a> {
	83	Blank,
	84	RowHeader(&'a str),
	85	Entry(&'a str, Option<&'a str>),
e8657dff	86	}
88a08162 SW	87	impl<'a> From<&'a str> for InputLine<'a> {
	88	fn from(value: &'a str) -> InputLine<'a> {
	89	let trimmed = value.trim_end();
	90	if trimmed.is_empty() {
	91	InputLine::Blank
	92	} else if !trimmed.starts_with(' ') {
	93	InputLine::RowHeader(value.trim())
	94	} else {
	95	match value.split_once(':') {
	96	None => InputLine::Entry(value.trim(), None),
	97	Some((col, instance)) => InputLine::Entry(col.trim(), Some(instance.trim())),
	98	}
e8657dff SW	99	}
	100	}
	101	}
14e9852b	102
75bb888a	103	#[derive(Debug, PartialEq, Eq)]
88a08162 SW	104	struct Row {
	105	label: String,
	106	entries: HashMap<String, Vec<Option<String>>>,
75bb888a SW	107	}
75bb888a SW	108
88a08162	109	struct Reader<Input: Iterator<Item = Result<String, std::io::Error>>> {
8110b492	110	input: std::iter::Enumerate<Input>,
88a08162	111	row: Option<Row>,
201b9ef3	112	}
88a08162	113	impl<Input: Iterator<Item = Result<String, std::io::Error>>> Reader<Input> {
201b9ef3	114	fn new(input: Input) -> Self {
8110b492 SW	115	Self {
	116	input: input.enumerate(),
	117	row: None,
	118	}
201b9ef3 SW	119	}
201b9ef3 SW	120	}
88a08162 SW	121	impl<Input: Iterator<Item = Result<String, std::io::Error>>> Iterator for Reader<Input> {
88a08162 SW	122	type Item = Result<Row, std::io::Error>;
201b9ef3 SW	123	fn next(&mut self) -> Option<Self::Item> {
201b9ef3 SW	124	loop {
8bf0d5b1	125	match self.input.next() {
201b9ef3	126	None => return Ok(std::mem::take(&mut self.row)).transpose(),
8110b492	127	Some((_, Err(e))) => return Some(Err(e)),
88a08162 SW	128	Some((n, Ok(line))) => match InputLine::from(line.as_ref()) {
	129	InputLine::Blank if self.row.is_some() => {
	130	return Ok(std::mem::take(&mut self.row)).transpose()
8110b492	131	}
88a08162 SW	132	InputLine::Blank => {}
	133	InputLine::Entry(col, instance) => match &mut self.row {
	134	None => {
	135	return Some(Err(std::io::Error::other(format!(
	136	"{}: Entry with no header",
	137	n + 1
	138	))))
	139	}
	140	Some(ref mut row) => {
	141	row.entries
	142	.entry(col.to_owned())
	143	.and_modify(\|is\| is.push(instance.map(ToOwned::to_owned)))
	144	.or_insert_with(\|\| vec![instance.map(ToOwned::to_owned)]);
	145	}
	146	},
	147	InputLine::RowHeader(row) => {
	148	let prev = std::mem::take(&mut self.row);
	149	self.row = Some(Row {
	150	label: row.to_owned(),
	151	entries: HashMap::new(),
	152	});
	153	if prev.is_some() {
	154	return Ok(prev).transpose();
	155	}
201b9ef3	156	}
88a08162	157	},
201b9ef3 SW	158	}
	159	}
	160	}
	161	}
	162
88a08162	163	fn read_rows(input: impl std::io::Read) -> impl Iterator<Item = Result<Row, std::io::Error>> {
201b9ef3	164	Reader::new(std::io::BufReader::new(input).lines())
75bb888a SW	165	}
75bb888a SW	166
88a08162	167	fn column_counts(rows: &[Row]) -> Vec<(usize, String)> {
58b5f36d SW	168	let mut counts: Vec<_> = rows
58b5f36d SW	169	.iter()
88a08162	170	.flat_map(\|r\| r.entries.keys())
b8907770	171	.fold(HashMap::new(), \|mut cs, col\| {
88a08162	172	cs.entry(col.to_owned())
58b5f36d	173	.and_modify(\|n\| *n += 1)
f272e502	174	.or_insert(1);
58b5f36d	175	cs
f272e502	176	})
58b5f36d SW	177	.into_iter()
	178	.map(\|(col, n)\| (n, col))
	179	.collect();
38d1167a	180	counts.sort_unstable_by(\|(an, acol), (bn, bcol)\| bn.cmp(an).then(acol.cmp(bcol)));
58b5f36d	181	counts
f272e502	182	}
88a08162	183	fn column_order(rows: &[Row]) -> Vec<String> {
d22b2e05 SW	184	column_counts(rows)
	185	.into_iter()
	186	.map(\|(_, col)\| col)
	187	.collect()
	188	}
f272e502	189
58c0a717	190	fn render_one_instance(instance: &Option<String>) -> HTML {
88a08162	191	match instance {
70436f23 SW	192	None => HTML::from("✓"),
70436f23 SW	193	Some(instance) => HTML::escape(instance.as_ref()),
de408c29 SW	194	}
	195	}
	196
f915bc90 SW	197	fn render_instances(instances: &[Option<String>]) -> HTML {
	198	let all_empty = instances.iter().all(Option::is_none);
	199	if all_empty && instances.len() == 1 {
70436f23	200	HTML::from("")
de408c29	201	} else if all_empty {
f915bc90	202	HTML(format!("{}", instances.len()))
de408c29	203	} else {
70436f23	204	HTML(
88a08162	205	instances
70436f23	206	.iter()
58c0a717	207	.map(render_one_instance)
70436f23 SW	208	.map(\|html\| html.0) // Waiting for slice_concat_trait to stabilize
	209	.collect::<Vec<_>>()
	210	.join(" "),
	211	)
f915bc90 SW	212	}
	213	}
	214
	215	fn render_cell(col: &str, row: &mut Row) -> HTML {
	216	let row_label = HTML::escape(row.label.as_ref());
	217	let col_label = HTML::escape(col);
	218	let instances: Option<&Vec<Option<String>>> = row.entries.get(col);
	219	let class = HTML::from(if instances.is_none() { "" } else { "yes" });
	220	let contents = match instances {
	221	None => HTML::from(""),
	222	Some(is) => render_instances(is),
de408c29	223	};
d9bfcf4d	224	row.entries.remove(col);
5ffe8e3a SW	225	HTML(format!(
	226	r#"<td class="{class}" onmouseover="h2('{row_label}','{col_label}')" onmouseout="ch2('{row_label}','{col_label}')">{contents}</td>"#
	227	))
de408c29 SW	228	}
de408c29 SW	229
d9bfcf4d	230	fn render_row(columns: &[String], row: &mut Row) -> HTML {
70436f23	231	let row_label = HTML::escape(row.label.as_ref());
74bd4cd1 SW	232	let cells = columns
	233	.iter()
	234	.map(\|col\| render_cell(col, row))
	235	.collect::<HTML>();
70436f23	236	HTML(format!(
74bd4cd1	237	"<tr><th id=\"{row_label}\">{row_label}</th>{cells}</tr>\n"
70436f23	238	))
de408c29 SW	239	}
de408c29 SW	240
70436f23 SW	241	fn render_column_headers(columns: &[String]) -> HTML {
70436f23 SW	242	HTML(
5ffe8e3a	243	String::from(r#"<tr class="key"><th></th>"#)
70436f23 SW	244	+ &columns.iter().fold(String::new(), \|mut acc, col\| {
	245	let col_header = HTML::escape(col.as_ref());
	246	write!(
	247	&mut acc,
5ffe8e3a	248	r#"<th id="{col_header}"><div><div>{col_header}</div></div></th>"#
70436f23 SW	249	)
	250	.unwrap();
	251	acc
	252	})
	253	+ "</tr>\n",
	254	)
76638ea1 SW	255	}
76638ea1 SW	256
4b99fb70 SW	257	/// # Errors
	258	///
	259	/// Will return `Err` if
	260	/// * there's an i/o error while reading `input`
	261	/// * the log has invalid syntax:
	262	/// * an indented line with no preceding non-indented line
28cf4fa2	263	pub fn tablify(config: &Config, input: impl std::io::Read) -> Result<HTML, std::io::Error> {
4b99fb70	264	let rows = read_rows(input).collect::<Result<Vec<_>, _>>()?;
de408c29	265	let columns = column_order(&rows);
70436f23 SW	266	Ok(HTML(format!(
	267	"{HEADER}{}{}{FOOTER}",
	268	render_column_headers(&columns),
	269	rows.into_iter()
d9bfcf4d	270	.map(\|mut r\| render_row(&columns, &mut r))
70436f23 SW	271	.collect::<HTML>()
70436f23 SW	272	)))
ece97615	273	}
75bb888a SW	274
	275	#[cfg(test)]
	276	mod tests {
	277	use super::*;
	278
b8907770	279	#[test]
88a08162 SW	280	fn test_parse_line() {
	281	assert_eq!(InputLine::from(""), InputLine::Blank);
	282	assert_eq!(InputLine::from(" "), InputLine::Blank);
	283	assert_eq!(InputLine::from("foo"), InputLine::RowHeader("foo"));
	284	assert_eq!(InputLine::from("foo "), InputLine::RowHeader("foo"));
	285	assert_eq!(InputLine::from(" foo"), InputLine::Entry("foo", None));
b8907770	286	assert_eq!(
88a08162 SW	287	InputLine::from(" foo:bar"),
88a08162 SW	288	InputLine::Entry("foo", Some("bar"))
b8907770 SW	289	);
b8907770 SW	290	assert_eq!(
88a08162 SW	291	InputLine::from(" foo: bar"),
88a08162 SW	292	InputLine::Entry("foo", Some("bar"))
b8907770	293	);
0d999bc3	294	assert_eq!(
88a08162 SW	295	InputLine::from(" foo: bar "),
	296	InputLine::Entry("foo", Some("bar"))
	297	);
	298	assert_eq!(
	299	InputLine::from(" foo: bar "),
	300	InputLine::Entry("foo", Some("bar"))
	301	);
	302	assert_eq!(
	303	InputLine::from(" foo : bar "),
	304	InputLine::Entry("foo", Some("bar"))
0d999bc3	305	);
b8907770 SW	306	}
b8907770 SW	307
75bb888a SW	308	#[test]
	309	fn test_read_rows() {
	310	assert_eq!(
201b9ef3	311	read_rows(&b"foo"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	312	vec![Row {
	313	label: "foo".to_owned(),
	314	entries: HashMap::new(),
75bb888a SW	315	}]
75bb888a SW	316	);
9dfa98b7	317	assert_eq!(
201b9ef3	318	read_rows(&b"bar"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	319	vec![Row {
	320	label: "bar".to_owned(),
	321	entries: HashMap::new(),
9dfa98b7 SW	322	}]
9dfa98b7 SW	323	);
2aa9ef94	324	assert_eq!(
201b9ef3	325	read_rows(&b"foo\nbar\n"[..]).flatten().collect::<Vec<_>>(),
2aa9ef94	326	vec![
88a08162 SW	327	Row {
	328	label: "foo".to_owned(),
	329	entries: HashMap::new(),
2aa9ef94	330	},
88a08162 SW	331	Row {
	332	label: "bar".to_owned(),
	333	entries: HashMap::new(),
2aa9ef94 SW	334	}
	335	]
	336	);
201b9ef3 SW	337	assert_eq!(
201b9ef3 SW	338	read_rows(&b"foo\n bar\n"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	339	vec![Row {
	340	label: "foo".to_owned(),
	341	entries: HashMap::from([("bar".to_owned(), vec![None])]),
201b9ef3 SW	342	}]
	343	);
	344	assert_eq!(
	345	read_rows(&b"foo\n bar\n baz\n"[..])
	346	.flatten()
	347	.collect::<Vec<_>>(),
88a08162 SW	348	vec![Row {
	349	label: "foo".to_owned(),
	350	entries: HashMap::from([
	351	("bar".to_owned(), vec![None]),
	352	("baz".to_owned(), vec![None])
	353	]),
201b9ef3 SW	354	}]
	355	);
	356	assert_eq!(
	357	read_rows(&b"foo\n\nbar\n"[..])
	358	.flatten()
	359	.collect::<Vec<_>>(),
	360	vec![
88a08162 SW	361	Row {
	362	label: "foo".to_owned(),
	363	entries: HashMap::new(),
201b9ef3	364	},
88a08162 SW	365	Row {
	366	label: "bar".to_owned(),
	367	entries: HashMap::new(),
201b9ef3 SW	368	}
	369	]
	370	);
1f6bd845 SW	371	assert_eq!(
	372	read_rows(&b"foo\n \nbar\n"[..])
	373	.flatten()
	374	.collect::<Vec<_>>(),
	375	vec![
88a08162 SW	376	Row {
	377	label: "foo".to_owned(),
	378	entries: HashMap::new(),
1f6bd845	379	},
88a08162 SW	380	Row {
	381	label: "bar".to_owned(),
	382	entries: HashMap::new(),
1f6bd845 SW	383	}
	384	]
	385	);
	386	assert_eq!(
	387	read_rows(&b"foo \n bar \n"[..])
	388	.flatten()
	389	.collect::<Vec<_>>(),
88a08162 SW	390	vec![Row {
	391	label: "foo".to_owned(),
	392	entries: HashMap::from([("bar".to_owned(), vec![None])]),
1f6bd845 SW	393	}]
1f6bd845 SW	394	);
201b9ef3 SW	395
	396	let bad = read_rows(&b" foo"[..]).next().unwrap();
	397	assert!(bad.is_err());
8110b492	398	assert!(format!("{bad:?}").contains("1: Entry with no header"));
201b9ef3 SW	399
	400	let bad2 = read_rows(&b"foo\n\n bar"[..]).nth(1).unwrap();
	401	assert!(bad2.is_err());
8110b492	402	assert!(format!("{bad2:?}").contains("3: Entry with no header"));
75bb888a	403	}
f272e502 SW	404
	405	#[test]
	406	fn test_column_counts() {
	407	assert_eq!(
	408	column_counts(
	409	&read_rows(&b"foo\n bar\n baz\n"[..])
	410	.collect::<Result<Vec<_>, _>>()
	411	.unwrap()
	412	),
58b5f36d	413	vec![(1, String::from("bar")), (1, String::from("baz"))]
f272e502 SW	414	);
	415	assert_eq!(
	416	column_counts(
	417	&read_rows(&b"foo\n bar\n baz\nquux\n baz"[..])
	418	.collect::<Result<Vec<_>, _>>()
	419	.unwrap()
	420	),
38d1167a	421	vec![(2, String::from("baz")), (1, String::from("bar"))]
f272e502	422	);
397ef957 SW	423	assert_eq!(
	424	column_counts(
	425	&read_rows(&b"foo\n bar\n bar\n baz\n bar\nquux\n baz"[..])
	426	.collect::<Result<Vec<_>, _>>()
	427	.unwrap()
	428	),
38d1167a	429	vec![(2, String::from("baz")), (1, String::from("bar"))]
397ef957	430	);
b8907770 SW	431	assert_eq!(
	432	column_counts(
	433	&read_rows(&b"foo\n bar: 1\n bar: 2\n baz\n bar\nquux\n baz"[..])
	434	.collect::<Result<Vec<_>, _>>()
	435	.unwrap()
	436	),
38d1167a	437	vec![(2, String::from("baz")), (1, String::from("bar"))]
b8907770	438	);
f272e502	439	}
de408c29 SW	440
	441	#[test]
	442	fn test_render_cell() {
	443	assert_eq!(
	444	render_cell(
	445	"foo",
d9bfcf4d	446	&mut Row {
88a08162 SW	447	label: "nope".to_owned(),
88a08162 SW	448	entries: HashMap::new(),
de408c29 SW	449	}
de408c29 SW	450	),
5ffe8e3a SW	451	HTML::from(
	452	r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	453	)
de408c29 SW	454	);
	455	assert_eq!(
	456	render_cell(
	457	"foo",
d9bfcf4d	458	&mut Row {
88a08162 SW	459	label: "nope".to_owned(),
88a08162 SW	460	entries: HashMap::from([("bar".to_owned(), vec![None])]),
de408c29 SW	461	}
de408c29 SW	462	),
5ffe8e3a SW	463	HTML::from(
	464	r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	465	)
de408c29 SW	466	);
	467	assert_eq!(
	468	render_cell(
	469	"foo",
d9bfcf4d	470	&mut Row {
88a08162 SW	471	label: "nope".to_owned(),
88a08162 SW	472	entries: HashMap::from([("foo".to_owned(), vec![None])]),
de408c29 SW	473	}
de408c29 SW	474	),
5ffe8e3a SW	475	HTML::from(
	476	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	477	)
de408c29 SW	478	);
	479	assert_eq!(
	480	render_cell(
	481	"foo",
d9bfcf4d	482	&mut Row {
88a08162 SW	483	label: "nope".to_owned(),
88a08162 SW	484	entries: HashMap::from([("foo".to_owned(), vec![None, None])]),
de408c29 SW	485	}
de408c29 SW	486	),
5ffe8e3a SW	487	HTML::from(
	488	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">2</td>"#
	489	)
de408c29 SW	490	);
	491	assert_eq!(
	492	render_cell(
	493	"foo",
d9bfcf4d	494	&mut Row {
88a08162	495	label: "nope".to_owned(),
5ffe8e3a SW	496	entries: HashMap::from([(
	497	"foo".to_owned(),
	498	vec![Some("5".to_owned()), Some("10".to_owned())]
	499	)]),
de408c29 SW	500	}
de408c29 SW	501	),
5ffe8e3a SW	502	HTML::from(
	503	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 10</td>"#
	504	)
de408c29 SW	505	);
	506	assert_eq!(
	507	render_cell(
	508	"foo",
d9bfcf4d	509	&mut Row {
88a08162 SW	510	label: "nope".to_owned(),
88a08162 SW	511	entries: HashMap::from([("foo".to_owned(), vec![Some("5".to_owned()), None])]),
de408c29 SW	512	}
de408c29 SW	513	),
5ffe8e3a SW	514	HTML::from(
	515	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 ✓</td>"#
	516	)
70436f23 SW	517	);
	518	assert_eq!(
	519	render_cell(
	520	"heart",
d9bfcf4d	521	&mut Row {
88a08162 SW	522	label: "nope".to_owned(),
88a08162 SW	523	entries: HashMap::from([("heart".to_owned(), vec![Some("<3".to_owned())])]),
70436f23 SW	524	}
70436f23 SW	525	),
5ffe8e3a SW	526	HTML::from(
	527	r#"<td class="yes" onmouseover="h2('nope','heart')" onmouseout="ch2('nope','heart')"><3</td>"#
	528	)
70436f23 SW	529	);
	530	assert_eq!(
	531	render_cell(
	532	"foo",
d9bfcf4d	533	&mut Row {
88a08162 SW	534	label: "bob's".to_owned(),
88a08162 SW	535	entries: HashMap::from([("foo".to_owned(), vec![None])]),
70436f23 SW	536	}
70436f23 SW	537	),
5ffe8e3a SW	538	HTML::from(
	539	r#"<td class="yes" onmouseover="h2('bob's','foo')" onmouseout="ch2('bob's','foo')"></td>"#
	540	)
de408c29	541	);
d9bfcf4d SW	542	let mut r = Row {
	543	label: "nope".to_owned(),
	544	entries: HashMap::from([
	545	("foo".to_owned(), vec![None]),
	546	("baz".to_owned(), vec![None]),
	547	]),
	548	};
	549	assert_eq!(r.entries.len(), 2);
	550	render_cell("foo", &mut r);
	551	assert_eq!(r.entries.len(), 1);
	552	render_cell("bar", &mut r);
	553	assert_eq!(r.entries.len(), 1);
	554	render_cell("baz", &mut r);
	555	assert_eq!(r.entries.len(), 0);
de408c29	556	}
25fd008e SW	557
	558	#[test]
	559	fn test_render_row() {
	560	assert_eq!(
	561	render_row(
	562	&["foo".to_owned()],
	563	&mut Row {
	564	label: "nope".to_owned(),
	565	entries: HashMap::from([("bar".to_owned(), vec![None])]),
	566	}
	567	),
	568	HTML::from(
	569	r#"<tr><th id="nope">nope</th><td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td></tr>
	570	"#
	571	)
	572	);
	573	}
75bb888a	574	}