}
fn make_random_id() -> [u8; COOKIE_LENGTH] {
- std::iter::from_fn(random)
- .filter(|c| {
- (b'A'..=b'Z').contains(c) || (b'a'..=b'z').contains(c) || (b'0'..=b'9').contains(c)
- })
- .take(COOKIE_LENGTH)
- .collect::<Vec<_>>()
- .try_into()
- .unwrap()
+ let mut id = [0; COOKIE_LENGTH];
+ for i in 0..COOKIE_LENGTH {
+ while !(b'A'..=b'Z').contains(&id[i])
+ && !(b'a'..=b'z').contains(&id[i])
+ && !(b'0'..=b'9').contains(&id[i])
+ {
+ id[i] = random()
+ }
+ }
+ id
}
fn set_cookie(mut response: cgi::Response, path: &str) -> Result<cgi::Response, cgi::Response> {
fn prompt_for_vote(dir: PathBuf, request: cgi::Request) -> Result<cgi::Response, cgi::Response> {
let voter = get_voter(&request);
- let mut response = cgi::html_response(200, "<html><body>You should vote</body></html>");
+ let cfile = std::fs::File::open(dir.join("candidates"))
+ .map_err(|_| cgi::text_response(503, "No candidates"))?;
+ let mut response = cgi::html_response(
+ 200,
+ std::iter::once(Ok("<html><body><table>".to_owned()))
+ .chain(
+ std::io::BufReader::new(cfile)
+ .lines()
+ .map(|rc| rc.map(|c| format!("<tr><td>{c}</td></tr>"))),
+ )
+ .chain(std::iter::once(Ok("</table></body></html>".to_owned())))
+ .collect::<std::io::Result<String>>()
+ .map_err(|_| cgi::text_response(503, "Missing candidates"))?,
+ );
if voter.is_err() {
response = set_cookie(response, request.uri().path())?
}