X-Git-Url: http://git.scottworley.com/voter/blobdiff_plain/dd0a124698a0ded987f7b5b426607308b0e79ffc..381eeda53fb8a9a574b96ce5cc91b28c5a99b105:/src/main.rs diff --git a/src/main.rs b/src/main.rs index 273a719..994ffac 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,7 +4,7 @@ use std::path::{Path, PathBuf}; const DATA_PATH: &str = "/var/lib/voter"; const COOKIE_NAME: &[u8] = b"__Secure-id"; -const COOKIE_LENGTH: usize = 32; +const COOKIE_LENGTH: usize = 12; fn validate_path(path: &str) -> Result { let invalid_path = || cgi::text_response(404, "Invalid path"); @@ -75,7 +75,20 @@ fn set_cookie(mut response: cgi::Response, path: &str) -> Result Result { let voter = get_voter(&request); - let mut response = cgi::html_response(200, "You should vote"); + 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("".to_owned())) + .chain( + std::io::BufReader::new(cfile) + .lines() + .map(|rc| rc.map(|c| format!(""))), + ) + .chain(std::iter::once(Ok("

{c}

".to_owned()))) + .collect::>() + .map_err(|_| cgi::text_response(503, "Missing candidates"))?, + ); if voter.is_err() { response = set_cookie(response, request.uri().path())? }