zamm_yang 0.1.8

use path_abs::PathAbs;
use std::cell::RefCell;
use std::collections::HashSet;
use std::fs::{File, OpenOptions};
use std::io::{BufRead, BufReader, Result, Write};

thread_local! {
    static AUTOGEN_FILES: RefCell<HashSet<String>> = RefCell::new(HashSet::new());
}

/// The path to the file that tracks autogenerated files.
pub const AUTOGEN_TRACKER: &str = ".autogen.txt";

/// Track a file as being autogenerated.
pub fn track_autogen(filename: String) {
    AUTOGEN_FILES.with(|f| f.borrow_mut().insert(filename));
}

/// Output all autogenerated files to the tracker file.
pub fn save_autogen() {
    let tracker_path = PathAbs::new(AUTOGEN_TRACKER).expect("Cannot open autogen tracker");
    AUTOGEN_FILES.with(|f| {
        for filename in f.borrow().iter() {
            // todo: cut down on the number of file reads here
            add_to_file(&tracker_path, filename).expect("Cannot add to autogen tracker");
        }
    });
    // add AUTOGEN_TRACKER to project-level .gitignore
    add_to_file(&PathAbs::new(".gitignore").unwrap(), AUTOGEN_TRACKER)
        .expect("Cannot ignore autogen tracker in top-level .gitignore");
    println!(
        "Generated {} files in total.",
        AUTOGEN_FILES.with(|f| f.borrow().len())
    );
}

/// Ensures that the given line will be in the file.
pub fn add_to_file(file: &PathAbs, content: &str) -> Result<()> {
    let mut already_in_file = false;
    match File::open(file.as_path()) {
        Ok(existing_ignore) => {
            for line in BufReader::new(existing_ignore).lines() {
                if line.unwrap() == content {
                    already_in_file = true;
                    break;
                }
            }
        }
        Err(_) => {
            writeln!(File::create(file.as_path())?, "{}", content)?;
            already_in_file = true;
        }
    }
    if !already_in_file {
        let mut existing_file = OpenOptions::new().append(true).open(file.as_path())?;
        writeln!(existing_file, "{}", content)
    } else {
        Ok(())
    }
}