An internal tool used to track how much of my game's code is on Rojo. Also my first experience writing Rust.

rojocoverage.rs

extern crate regex;
extern crate serde_json;
extern crate walkdir;

use regex::Regex;
use serde_json::{ Value };
use std::env;
use std::fs;
use std::path::Path;
use walkdir::WalkDir;

fn main() -> Result<(), std::io::Error> {
	println!("Rojo Coverage Calculator");

	let args:Vec<String> = env::args().collect();
	let (rojo_path, output_path, readme_path) = (&args[1], &args[2], &args[3]);

	let file = fs::File::open(rojo_path)?;
	let data:Value = serde_json::from_reader(file)?;
	let partition_paths:Vec<&Path> = data["partitions"].as_object().unwrap().into_iter().map(|(_, partition)| Path::new(partition["path"].as_str().unwrap())).collect();

	println!("Figured out partitions from rojo.json");

	let (mut total_files, mut covered_files) = (0, 0);

	for entry in WalkDir::new(output_path).into_iter().filter_entry(|e| e.path().is_dir()).filter_map(|e| e.ok()) {
		total_files += 1;
		
		if entry.path().strip_prefix(Path::new(output_path).parent().unwrap()).unwrap().ancestors().find(|path| partition_paths.contains(path)) != None {
			covered_files += 1;
		}
	}

	let percentage = (covered_files / total_files) * 100;

	println!("{} total directories, {} covered, {}% covered", total_files, covered_files, percentage);

	let color = match percentage {
		 0...20 => "red",
		20...40 => "yellow",
		40...75 => "yellowgreen",
		75...90 => "green",
		_ => "brightgreen",
	};

	println!("color: {}", color);

	let re = Regex::new(r"!\[rojo-coverage]\((.+)\)").unwrap();
	let badge = format!("https://img.shields.io/badge/rojo%20coverage-{}%25-{}.svg", percentage, color);
	let readme = fs::read_to_string(readme_path).expect("Couldn't read README");
	
	println!("badge: {}", badge);

	let updated = re.replace(readme.as_str(), format!("![rojo-coverage]({})", badge).as_str());
	println!("{}", updated);

	fs::write(readme_path, updated.into_owned())?;

	Ok(())
}