[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 = "<!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: &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(" "),
        )
    };
    HTML(format!("<td class=\"{class}\" onmouseover=\"h2('{row_label}','{col_label}')\" onmouseout=\"ch2('{row_label}','{col_label}')\">{contents}</td>"))
}

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