fixed some other comments, now also the code side seems to be at a level to be reviewed by experienced programmers - suckless/dwm - Unnamed repository; edit this file 'description' to name the repository.

diff options

author	Anselm R. Garbe <arg@10kloc.org>	2006-09-11 11:28:28 +0200
committer	Anselm R. Garbe <arg@10kloc.org>	2006-09-11 11:28:28 +0200
commit	b597fa46370af6594c0980f6c2ddefbe943ee933 (patch)
tree	5000414fe9f80a09e99d199f848e3ff40107f045 /main.c
parent	d2d394eccf2d23eb264a321fe187bb2084beb4b9 (diff)
download	dwm-b597fa46370af6594c0980f6c2ddefbe943ee933.tar.gz

fixed some other comments, now also the code side seems to be at a level to be reviewed by experienced programmers

Diffstat (limited to 'main.c')

-rw-r--r--

main.c

1 files changed, 2 insertions, 2 deletions

use chrono::prelude::*; use std::env; use std::fs; use std::path::Path; use std::process; use serde::{Deserialize, Serialize}; use serde_yaml; use walkdir::WalkDir; const VERS: &str = "v0.1"; const CONF_PATH: &str = "instistats.yml"; #[derive(Debug, Deserialize, Serialize)] struct Server { name: String, url: String, signup_url: String, user_count: usize, want_users: bool, admin_email: String, description: String, last_generated: String, users: Vec<User>, } #[derive(Debug, Deserialize, Serialize, Clone)] struct User { name: String, title: String, mtime: String, } fn main() { println!("instistats {}", VERS); println!("(c) 2019 Ben Morrison - ISC License"); println!(); let args = env::args().collect::<Vec<String>>(); let out_path = match args[1].trim() { "-h" | "--help" => { println!("The only argument should be the path to save the tilde.json file.\nEx: /var/www/htdocs/tilde.json"); process::exit(0); } out_path => { println!("Output Location: {}", out_path); out_path } }; println!(); let home_dir = WalkDir::new("/home").follow_links(true).max_depth(1); let user_count = fs::read_dir("/home") .unwrap() .map(|d| d.unwrap()) .collect::<Vec<fs::DirEntry>>() .len(); let mut users_list = Vec::new(); home_dir.into_iter().for_each(|d| { if let Ok(p) = d { let p = p.path().strip_prefix("/home").unwrap(); let p = p.to_str().unwrap(); if p.len() > 1 { let user = p .chars() .map(|c| { if c != '"' { c.to_string() } else { "".to_string() } }) .collect::<String>(); users_list.push(user); } } }); let mut users_struct = Vec::new(); users_list.iter().for_each(|user| { let path = format!("/home/{}/public_html/index.html", user); let path = Path::new(&path); let index_file = if let Ok(file) = fs::read_to_string(path) { file } else { return; }; let mut title = String::new(); index_file.split("\n").for_each(|line| { if line.contains("<title>") { let title_line = line .split("<title>") .map(|s| s.to_string()) .collect::<Vec<String>>(); let title_line = title_line .iter() .map(|s| { s.split("</title>") .map(|s| s.to_string()) .collect::<Vec<String>>() }) .flatten() .collect::<Vec<String>>(); title_line.iter().for_each(|e| { if !e.contains("<title>") && !e.contains("</title>") { title.push_str(e); } }) } }); let meta = fs::metadata(path); let mtime = format!( "{}", meta.unwrap() .modified() .unwrap() .duration_since(std::time::SystemTime::UNIX_EPOCH) .unwrap() .as_secs() ); users_struct.push(User { name: user.to_string(), title, mtime, }); }); let conf = fs::read_to_string(CONF_PATH).expect("Could not read config file"); let conf_yaml: serde_yaml::Value = serde_yaml::from_str(&conf).expect("Could not parse config data as YAML"); let last_generated = Utc::now().to_rfc2822(); let mut conf_yaml = Server { name: conf_yaml["name"].as_str().unwrap().to_string(), url: conf_yaml["url"].as_str().unwrap().to_string(), signup_url: conf_yaml["signup_url"].as_str().unwrap().to_string(), user_count: 0, want_users: conf_yaml["want_users"].as_bool().unwrap(), admin_email: conf_yaml["admin_email"].as_str().unwrap().to_string(), description: conf_yaml["description"].as_str().unwrap().to_string(), last_generated, users: Vec::new(), }; users_struct.iter().for_each(|user| { conf_yaml.users.push(user.clone()); }); conf_yaml.user_count = user_count; let json = serde_json::to_string(&conf_yaml).unwrap(); fs::write(out_path, &json).unwrap(); println!("File written successfully! Please inspect it: {}", out_path); }


context:
space:
mode: