use log::{log, Level};
use rand::{distributions::Alphanumeric, thread_rng, Rng};
use scopeguard::defer;
use std::{process::Command, str};
use winlog2::{deregister, init, register};

#[test]
fn end_to_end() {
    let rand_string = String::from_iter(
        thread_rng()
            .sample_iter(&Alphanumeric)
            .take(16)
            .map(|b| b as char),
    );
    let log_source = format!("winlog-test-{}", rand_string);

    // Add log source to Windows registry
    register(&log_source).unwrap();
    // Remove log source from Windows registry
    defer!(deregister(&log_source).unwrap());

    // Do some logging and verification
    init(&log_source).unwrap();
    log_and_verify_one(Level::Error, &log_source, "Error", "1", "Error!!");
    log_and_verify_one(Level::Warn, &log_source, "Warning", "2", "Warning!!");
    log_and_verify_one(Level::Info, &log_source, "Information", "3", "Info!!");
    log_and_verify_one(Level::Debug, &log_source, "Information", "4", "Debug!!");
    log_and_verify_one(Level::Trace, &log_source, "Information", "5", "Trace!!");
}

fn log_and_verify_one(level: Level, log_source: &str, entry_type: &str, entry_id: &str, msg: &str) {
    log!(level, "{}", msg);

    // Use PowerShell to extract formatted entries from the event log.
    let mut command = Command::new("powershell");
    command.arg("-Command").arg(format!(
        "Get-EventLog -Newest 1 -LogName Application -Source {} \
         | Select-Object Source, EntryType, EventID, Message \
         | foreach {{ \"$_\" }}",
        log_source
    ));
    let out = command.output().unwrap();

    assert_eq!(
        format!(
            "@{{Source={}; EntryType={}; EventID={}; Message={}}}\r\n",
            &log_source, &entry_type, entry_id, msg
        ),
        String::from_utf8_lossy(&out.stdout)
    );
}