extern crate webscan;
use webscan::{UriScanner, ScanStatus, RequestMethod};
use tokio;
use tokio::runtime::Runtime;
use std::fs::{read_to_string, read};
use std::time::Duration;
use std::thread;

#[tokio::main]
async fn main(){
    let mut uri_scanner = match UriScanner::new(){
        Ok(scanner) => (scanner),
        Err(e) => panic!("Error creating scanner: {}", e),
    };
    let base_uri = String::from("http://localhost:8000/");
    uri_scanner.set_base_uri(base_uri);
    let data = read_to_string("uris.txt");
    let text = match data {
        Ok(content) => content,
        Err(e) => {panic!("Could not open or find uris.txt file: {}", e);}
    };
    let word_list: Vec<&str> = text.trim().split("\n").collect();
    for word in word_list {
        uri_scanner.add_word(word.to_string());
    }
    match read("content.txt") {
        Ok(ct) => {
            let sep = b'\n';
            ct.split(|b| b == &sep )
                .for_each(|c| uri_scanner.add_content(c.to_vec()));
        },
        Err(e) => {panic!("Could not open or find content.txt file: {}", e);}
    }
    uri_scanner.set_request_method(RequestMethod::Get);
    uri_scanner.set_timeout(Duration::from_millis(20000));
    uri_scanner.set_accept_invalid_certs(false);
    let rx = uri_scanner.get_progress_receiver();
    let rt = Runtime::new().unwrap();
    // Run scan 
    let handle = thread::spawn(move|| {
        rt.block_on(async {
            uri_scanner.scan().await
        })
    });
    // Print progress
    while let Ok(_uri) = rx.lock().unwrap().recv() {
        //println!("Check: {}", uri);
    }
    let result = handle.join().unwrap();
    print!("Status: ");
    match result.scan_status {
        ScanStatus::Done => {println!("Done")},
        ScanStatus::Timeout => {println!("Timed out")},
        _ => {println!("Error")},
    }
    println!("URI Scan Result:");
    for (uri, status) in result.responses {
        println!("{} {} {}", uri, status.code(), status.reason());
    }
    println!("Scan Time: {:?}", result.scan_time);
}