use decon_spf::mechanism::{Kind, Mechanism, ParsedMechanism, Qualifier};
use decon_spf::Spf;

fn main() {
    let mut spf1 = Spf::new();
    spf1.set_v1();
    let ip_m_1 = ParsedMechanism::new("+ip4:203.32.160.0/24");
    let ip_m_2 = ParsedMechanism::new("+ip4:203.32.166.0/24");
    if let Ok(ip1) = ip_m_1 {
        spf1.append_ip_mechanism(ip1.network());
    }
    if let Ok(ip2) = ip_m_2 {
        spf1.append_ip_mechanism(ip2.network());
    }

    if let Ok(mx) = ParsedMechanism::new("mx") {
        spf1.append_mechanism(mx.txt());
    }

    // example.xx is not a valid domain. There is no TLD of xx
    if let Ok(m) = "a:test.xx".parse() {
        // Append does not occur
        spf1.append_mechanism(m);
    }

    println!("New spf 1: >{}<", spf1);
    assert_eq!(
        spf1.to_string(),
        "v=spf1 mx ip4:203.32.160.0/24 ip4:203.32.166.0/24"
    );

    let mut spf2 = Spf::new();
    spf2.set_v1();
    let ip = "203.32.166.0/24".parse().unwrap();
    spf2.append_ip_mechanism(Mechanism::ip(Qualifier::Pass, ip));

    println!("\nNew spf 2: >{}<", spf2);
    println!("Attempt to create invalid mx to spf2");
    match Mechanism::mx(Qualifier::Pass).with_rrdata("example.xx") {
        Ok(m) => {
            spf2.append_mechanism(m);
        }
        Err(e) => {
            println!("Error creating Mechanism: \"{}\"", e.to_string());
        }
    };
    println!("Add mx to spf2");
    spf2.append_mechanism(Mechanism::mx(Qualifier::Pass));
    println!("Altered spf 2: >{}<", spf2);
    println!("Clear mx from spf2");
    spf2.clear_mechanism(Kind::MX);
    println!("Altered spf 2: >{}<", spf2);

    let mut spf3 = Spf::new();
    spf3.set_v2_pra();
    spf3.append_mechanism(Mechanism::a(Qualifier::Pass));
    spf3.append_mechanism(Mechanism::all(Qualifier::Neutral));

    println!("\nNew spf 3: >{}<", spf3);
    println!("Change spf3 all to Fail");
    spf3.append_mechanism(Mechanism::all(Qualifier::Fail));
    println!("Altered spf 3: >{}<", spf3);
}