[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_instance(instance: &Option<String>) -> HTML {
    match instance {
        None => HTML::from("✓"),
        Some(instance) => HTML::escape(instance.as_ref()),
    }
}

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 all_empty = instances
        .iter()
        .flat_map(|is| is.iter())
        .all(Option::is_none);
    let contents = if instances.is_none() || (all_empty && instances.unwrap().len() == 1) {
        HTML::from("")
    } else if all_empty {
        HTML(format!("{}", instances.unwrap().len()))
    } else {
        HTML(
            instances
                .unwrap()
                .iter()
                .map(render_instance)
                .map(|html| html.0) // Waiting for slice_concat_trait to stabilize
                .collect::<Vec<_>>()
                .join(" "),
        )
    };
    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());
    HTML(format!(
        "<tr><th id=\"{row_label}\">{row_label}</th>{}</tr>\n",
        &columns
            .iter()
            .map(|col| render_cell(col, row))
            .collect::<HTML>()
    ))
}

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);
    }
}
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
88a08162 SW	190	fn render_instance(instance: &Option<String>) -> HTML {
88a08162 SW	191	match instance {
70436f23 SW	192	None => HTML::from("✓"),
70436f23 SW	193	Some(instance) => HTML::escape(instance.as_ref()),
de408c29 SW	194	}
	195	}
	196
d9bfcf4d	197	fn render_cell(col: &str, row: &mut Row) -> HTML {
70436f23 SW	198	let row_label = HTML::escape(row.label.as_ref());
70436f23 SW	199	let col_label = HTML::escape(col);
88a08162 SW	200	let instances: Option<&Vec<Option<String>>> = row.entries.get(col);
	201	let class = HTML::from(if instances.is_none() { "" } else { "yes" });
	202	let all_empty = instances
	203	.iter()
	204	.flat_map(\|is\| is.iter())
	205	.all(Option::is_none);
	206	let contents = if instances.is_none() \|\| (all_empty && instances.unwrap().len() == 1) {
70436f23	207	HTML::from("")
de408c29	208	} else if all_empty {
88a08162	209	HTML(format!("{}", instances.unwrap().len()))
de408c29	210	} else {
70436f23	211	HTML(
88a08162 SW	212	instances
88a08162 SW	213	.unwrap()
70436f23	214	.iter()
88a08162	215	.map(render_instance)
70436f23 SW	216	.map(\|html\| html.0) // Waiting for slice_concat_trait to stabilize
	217	.collect::<Vec<_>>()
	218	.join(" "),
	219	)
de408c29	220	};
d9bfcf4d	221	row.entries.remove(col);
5ffe8e3a SW	222	HTML(format!(
	223	r#"<td class="{class}" onmouseover="h2('{row_label}','{col_label}')" onmouseout="ch2('{row_label}','{col_label}')">{contents}</td>"#
	224	))
de408c29 SW	225	}
de408c29 SW	226
d9bfcf4d	227	fn render_row(columns: &[String], row: &mut Row) -> HTML {
70436f23 SW	228	let row_label = HTML::escape(row.label.as_ref());
70436f23 SW	229	HTML(format!(
92476edc	230	"<tr><th id=\"{row_label}\">{row_label}</th>{}</tr>\n",
de408c29 SW	231	&columns
	232	.iter()
	233	.map(\|col\| render_cell(col, row))
70436f23 SW	234	.collect::<HTML>()
70436f23 SW	235	))
de408c29 SW	236	}
de408c29 SW	237
70436f23 SW	238	fn render_column_headers(columns: &[String]) -> HTML {
70436f23 SW	239	HTML(
5ffe8e3a	240	String::from(r#"<tr class="key"><th></th>"#)
70436f23 SW	241	+ &columns.iter().fold(String::new(), \|mut acc, col\| {
	242	let col_header = HTML::escape(col.as_ref());
	243	write!(
	244	&mut acc,
5ffe8e3a	245	r#"<th id="{col_header}"><div><div>{col_header}</div></div></th>"#
70436f23 SW	246	)
	247	.unwrap();
	248	acc
	249	})
	250	+ "</tr>\n",
	251	)
76638ea1 SW	252	}
76638ea1 SW	253
4b99fb70 SW	254	/// # Errors
	255	///
	256	/// Will return `Err` if
	257	/// * there's an i/o error while reading `input`
	258	/// * the log has invalid syntax:
	259	/// * an indented line with no preceding non-indented line
28cf4fa2	260	pub fn tablify(config: &Config, input: impl std::io::Read) -> Result<HTML, std::io::Error> {
4b99fb70	261	let rows = read_rows(input).collect::<Result<Vec<_>, _>>()?;
de408c29	262	let columns = column_order(&rows);
70436f23 SW	263	Ok(HTML(format!(
	264	"{HEADER}{}{}{FOOTER}",
	265	render_column_headers(&columns),
	266	rows.into_iter()
d9bfcf4d	267	.map(\|mut r\| render_row(&columns, &mut r))
70436f23 SW	268	.collect::<HTML>()
70436f23 SW	269	)))
ece97615	270	}
75bb888a SW	271
	272	#[cfg(test)]
	273	mod tests {
	274	use super::*;
	275
b8907770	276	#[test]
88a08162 SW	277	fn test_parse_line() {
	278	assert_eq!(InputLine::from(""), InputLine::Blank);
	279	assert_eq!(InputLine::from(" "), InputLine::Blank);
	280	assert_eq!(InputLine::from("foo"), InputLine::RowHeader("foo"));
	281	assert_eq!(InputLine::from("foo "), InputLine::RowHeader("foo"));
	282	assert_eq!(InputLine::from(" foo"), InputLine::Entry("foo", None));
b8907770	283	assert_eq!(
88a08162 SW	284	InputLine::from(" foo:bar"),
88a08162 SW	285	InputLine::Entry("foo", Some("bar"))
b8907770 SW	286	);
b8907770 SW	287	assert_eq!(
88a08162 SW	288	InputLine::from(" foo: bar"),
88a08162 SW	289	InputLine::Entry("foo", Some("bar"))
b8907770	290	);
0d999bc3	291	assert_eq!(
88a08162 SW	292	InputLine::from(" foo: bar "),
	293	InputLine::Entry("foo", Some("bar"))
	294	);
	295	assert_eq!(
	296	InputLine::from(" foo: bar "),
	297	InputLine::Entry("foo", Some("bar"))
	298	);
	299	assert_eq!(
	300	InputLine::from(" foo : bar "),
	301	InputLine::Entry("foo", Some("bar"))
0d999bc3	302	);
b8907770 SW	303	}
b8907770 SW	304
75bb888a SW	305	#[test]
	306	fn test_read_rows() {
	307	assert_eq!(
201b9ef3	308	read_rows(&b"foo"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	309	vec![Row {
	310	label: "foo".to_owned(),
	311	entries: HashMap::new(),
75bb888a SW	312	}]
75bb888a SW	313	);
9dfa98b7	314	assert_eq!(
201b9ef3	315	read_rows(&b"bar"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	316	vec![Row {
	317	label: "bar".to_owned(),
	318	entries: HashMap::new(),
9dfa98b7 SW	319	}]
9dfa98b7 SW	320	);
2aa9ef94	321	assert_eq!(
201b9ef3	322	read_rows(&b"foo\nbar\n"[..]).flatten().collect::<Vec<_>>(),
2aa9ef94	323	vec![
88a08162 SW	324	Row {
	325	label: "foo".to_owned(),
	326	entries: HashMap::new(),
2aa9ef94	327	},
88a08162 SW	328	Row {
	329	label: "bar".to_owned(),
	330	entries: HashMap::new(),
2aa9ef94 SW	331	}
	332	]
	333	);
201b9ef3 SW	334	assert_eq!(
201b9ef3 SW	335	read_rows(&b"foo\n bar\n"[..]).flatten().collect::<Vec<_>>(),
88a08162 SW	336	vec![Row {
	337	label: "foo".to_owned(),
	338	entries: HashMap::from([("bar".to_owned(), vec![None])]),
201b9ef3 SW	339	}]
	340	);
	341	assert_eq!(
	342	read_rows(&b"foo\n bar\n baz\n"[..])
	343	.flatten()
	344	.collect::<Vec<_>>(),
88a08162 SW	345	vec![Row {
	346	label: "foo".to_owned(),
	347	entries: HashMap::from([
	348	("bar".to_owned(), vec![None]),
	349	("baz".to_owned(), vec![None])
	350	]),
201b9ef3 SW	351	}]
	352	);
	353	assert_eq!(
	354	read_rows(&b"foo\n\nbar\n"[..])
	355	.flatten()
	356	.collect::<Vec<_>>(),
	357	vec![
88a08162 SW	358	Row {
	359	label: "foo".to_owned(),
	360	entries: HashMap::new(),
201b9ef3	361	},
88a08162 SW	362	Row {
	363	label: "bar".to_owned(),
	364	entries: HashMap::new(),
201b9ef3 SW	365	}
	366	]
	367	);
1f6bd845 SW	368	assert_eq!(
	369	read_rows(&b"foo\n \nbar\n"[..])
	370	.flatten()
	371	.collect::<Vec<_>>(),
	372	vec![
88a08162 SW	373	Row {
	374	label: "foo".to_owned(),
	375	entries: HashMap::new(),
1f6bd845	376	},
88a08162 SW	377	Row {
	378	label: "bar".to_owned(),
	379	entries: HashMap::new(),
1f6bd845 SW	380	}
	381	]
	382	);
	383	assert_eq!(
	384	read_rows(&b"foo \n bar \n"[..])
	385	.flatten()
	386	.collect::<Vec<_>>(),
88a08162 SW	387	vec![Row {
	388	label: "foo".to_owned(),
	389	entries: HashMap::from([("bar".to_owned(), vec![None])]),
1f6bd845 SW	390	}]
1f6bd845 SW	391	);
201b9ef3 SW	392
	393	let bad = read_rows(&b" foo"[..]).next().unwrap();
	394	assert!(bad.is_err());
8110b492	395	assert!(format!("{bad:?}").contains("1: Entry with no header"));
201b9ef3 SW	396
	397	let bad2 = read_rows(&b"foo\n\n bar"[..]).nth(1).unwrap();
	398	assert!(bad2.is_err());
8110b492	399	assert!(format!("{bad2:?}").contains("3: Entry with no header"));
75bb888a	400	}
f272e502 SW	401
	402	#[test]
	403	fn test_column_counts() {
	404	assert_eq!(
	405	column_counts(
	406	&read_rows(&b"foo\n bar\n baz\n"[..])
	407	.collect::<Result<Vec<_>, _>>()
	408	.unwrap()
	409	),
58b5f36d	410	vec![(1, String::from("bar")), (1, String::from("baz"))]
f272e502 SW	411	);
	412	assert_eq!(
	413	column_counts(
	414	&read_rows(&b"foo\n bar\n baz\nquux\n baz"[..])
	415	.collect::<Result<Vec<_>, _>>()
	416	.unwrap()
	417	),
38d1167a	418	vec![(2, String::from("baz")), (1, String::from("bar"))]
f272e502	419	);
397ef957 SW	420	assert_eq!(
	421	column_counts(
	422	&read_rows(&b"foo\n bar\n bar\n baz\n bar\nquux\n baz"[..])
	423	.collect::<Result<Vec<_>, _>>()
	424	.unwrap()
	425	),
38d1167a	426	vec![(2, String::from("baz")), (1, String::from("bar"))]
397ef957	427	);
b8907770 SW	428	assert_eq!(
	429	column_counts(
	430	&read_rows(&b"foo\n bar: 1\n bar: 2\n baz\n bar\nquux\n baz"[..])
	431	.collect::<Result<Vec<_>, _>>()
	432	.unwrap()
	433	),
38d1167a	434	vec![(2, String::from("baz")), (1, String::from("bar"))]
b8907770	435	);
f272e502	436	}
de408c29 SW	437
	438	#[test]
	439	fn test_render_cell() {
	440	assert_eq!(
	441	render_cell(
	442	"foo",
d9bfcf4d	443	&mut Row {
88a08162 SW	444	label: "nope".to_owned(),
88a08162 SW	445	entries: HashMap::new(),
de408c29 SW	446	}
de408c29 SW	447	),
5ffe8e3a SW	448	HTML::from(
	449	r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	450	)
de408c29 SW	451	);
	452	assert_eq!(
	453	render_cell(
	454	"foo",
d9bfcf4d	455	&mut Row {
88a08162 SW	456	label: "nope".to_owned(),
88a08162 SW	457	entries: HashMap::from([("bar".to_owned(), vec![None])]),
de408c29 SW	458	}
de408c29 SW	459	),
5ffe8e3a SW	460	HTML::from(
	461	r#"<td class="" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	462	)
de408c29 SW	463	);
	464	assert_eq!(
	465	render_cell(
	466	"foo",
d9bfcf4d	467	&mut Row {
88a08162 SW	468	label: "nope".to_owned(),
88a08162 SW	469	entries: HashMap::from([("foo".to_owned(), vec![None])]),
de408c29 SW	470	}
de408c29 SW	471	),
5ffe8e3a SW	472	HTML::from(
	473	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')"></td>"#
	474	)
de408c29 SW	475	);
	476	assert_eq!(
	477	render_cell(
	478	"foo",
d9bfcf4d	479	&mut Row {
88a08162 SW	480	label: "nope".to_owned(),
88a08162 SW	481	entries: HashMap::from([("foo".to_owned(), vec![None, None])]),
de408c29 SW	482	}
de408c29 SW	483	),
5ffe8e3a SW	484	HTML::from(
	485	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">2</td>"#
	486	)
de408c29 SW	487	);
	488	assert_eq!(
	489	render_cell(
	490	"foo",
d9bfcf4d	491	&mut Row {
88a08162	492	label: "nope".to_owned(),
5ffe8e3a SW	493	entries: HashMap::from([(
	494	"foo".to_owned(),
	495	vec![Some("5".to_owned()), Some("10".to_owned())]
	496	)]),
de408c29 SW	497	}
de408c29 SW	498	),
5ffe8e3a SW	499	HTML::from(
	500	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 10</td>"#
	501	)
de408c29 SW	502	);
	503	assert_eq!(
	504	render_cell(
	505	"foo",
d9bfcf4d	506	&mut Row {
88a08162 SW	507	label: "nope".to_owned(),
88a08162 SW	508	entries: HashMap::from([("foo".to_owned(), vec![Some("5".to_owned()), None])]),
de408c29 SW	509	}
de408c29 SW	510	),
5ffe8e3a SW	511	HTML::from(
	512	r#"<td class="yes" onmouseover="h2('nope','foo')" onmouseout="ch2('nope','foo')">5 ✓</td>"#
	513	)
70436f23 SW	514	);
	515	assert_eq!(
	516	render_cell(
	517	"heart",
d9bfcf4d	518	&mut Row {
88a08162 SW	519	label: "nope".to_owned(),
88a08162 SW	520	entries: HashMap::from([("heart".to_owned(), vec![Some("<3".to_owned())])]),
70436f23 SW	521	}
70436f23 SW	522	),
5ffe8e3a SW	523	HTML::from(
	524	r#"<td class="yes" onmouseover="h2('nope','heart')" onmouseout="ch2('nope','heart')"><3</td>"#
	525	)
70436f23 SW	526	);
	527	assert_eq!(
	528	render_cell(
	529	"foo",
d9bfcf4d	530	&mut Row {
88a08162 SW	531	label: "bob's".to_owned(),
88a08162 SW	532	entries: HashMap::from([("foo".to_owned(), vec![None])]),
70436f23 SW	533	}
70436f23 SW	534	),
5ffe8e3a SW	535	HTML::from(
	536	r#"<td class="yes" onmouseover="h2('bob's','foo')" onmouseout="ch2('bob's','foo')"></td>"#
	537	)
de408c29	538	);
d9bfcf4d SW	539	let mut r = Row {
	540	label: "nope".to_owned(),
	541	entries: HashMap::from([
	542	("foo".to_owned(), vec![None]),
	543	("baz".to_owned(), vec![None]),
	544	]),
	545	};
	546	assert_eq!(r.entries.len(), 2);
	547	render_cell("foo", &mut r);
	548	assert_eq!(r.entries.len(), 1);
	549	render_cell("bar", &mut r);
	550	assert_eq!(r.entries.len(), 1);
	551	render_cell("baz", &mut r);
	552	assert_eq!(r.entries.len(), 0);
de408c29	553	}
75bb888a	554	}