use ansi_term::Colour;
use std::{env, io, io::Write, process};
use uwuizer::*;

pub fn inf(a: std::string::String) {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        println!(
            "{} {}",
            Colour::Purple.paint("❖"),
            Colour::White.paint(uwuize!(&a))
        );
    } else {
        println!("{} {}", Colour::Purple.paint("❖"), Colour::White.paint(a));
    }
}

pub fn sec(a: std::string::String) {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        println!(
            "{} {}",
            Colour::Purple.bold().paint("❖"),
            Colour::White.bold().paint(uwuize!(&a))
        );
    } else {
        println!(
            "{} {}",
            Colour::Purple.bold().paint("❖"),
            Colour::White.bold().paint(a)
        );
    }
}

pub fn succ(a: std::string::String) {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        println!(
            "{} {}",
            Colour::Green.bold().paint("✓"),
            Colour::Green.paint(uwuize!(&a))
        );
    } else {
        println!(
            "{} {}",
            Colour::Green.bold().paint("✓"),
            Colour::Green.paint(&a)
        );
    }
}

pub fn prompt(a: std::string::String) -> bool {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        print!(
            "{} {} {}",
            Colour::Purple.bold().paint("❖"),
            Colour::White.bold().paint(uwuize!(&a)),
            Colour::White.bold().paint("(Y/n): ")
        );
        io::stdout().flush().ok().expect("Couldn't flush stdout");
        let mut yn: String = String::new();
        let _ = std::io::stdin().read_line(&mut yn);
        if yn.trim() == "n" || yn.trim() == "N" || yn.trim() == "no" || yn.trim() == "No" {
            false
        } else {
            true
        }
    } else {
        print!(
            "{} {} {}",
            Colour::Purple.bold().paint("❖"),
            Colour::White.bold().paint(&a),
            Colour::White.bold().paint("(Y/n): ")
        );
        io::stdout().flush().ok().expect("Couldn't flush stdout");
        let mut yn: String = String::new();
        let _ = std::io::stdin().read_line(&mut yn);
        if yn.trim() == "n" || yn.trim() == "N" || yn.trim() == "no" || yn.trim() == "No" {
            false
        } else {
            true
        }
    }
}

pub fn err_unrec(a: std::string::String) {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        println!(
            "{} {} {}",
            Colour::Red.bold().paint(uwuize!("✖ Unrecoverable error:")),
            Colour::Red.paint(uwuize!(&a)),
            Colour::Red.bold().paint(uwuize!("Terminating."))
        );
        process::exit(1);
    } else {
        println!(
            "{} {} {}",
            Colour::Red.bold().paint("✖ Unrecoverable error:"),
            Colour::Red.paint(a),
            Colour::Red.bold().paint("Terminating.")
        );
        process::exit(1);
    }
}

pub fn err_rec(a: std::string::String) {
    if env::var("AME_UWU").unwrap_or("n/a".to_string()) == "YES" {
        println!(
            "{} {}",
            Colour::Yellow.bold().paint(uwuize!("⚠ WARNING:")),
            Colour::Yellow.paint(uwuize!(&a))
        );
    } else {
        println!(
            "{} {}",
            Colour::Yellow.bold().paint("⚠ WARNING:"),
            Colour::Yellow.paint(a)
        );
    }
}