use failure::{format_err, Error};
use serde::{Deserialize, Serialize};

#[derive(Deserialize, Debug)]
struct CloudflareListResponse {
    success: bool,
    errors: Vec<String>,
    result: Vec<ObjectWithId>,
}

#[derive(Deserialize, Debug)]
struct CloudflareUpdateResponse {
    success: bool,
    errors: Vec<String>,
    result: ObjectWithId,
}

#[derive(Deserialize, Debug, PartialEq)]
struct ObjectWithId {
    id: String,
}

#[derive(Serialize, Debug)]
struct UpdateIpData {
    id: String,
    r#type: String,
    name: String,
    content: String,
}

pub fn get_zone_identifier(zone: &str, email: &str, key: &str) -> Result<String, Error> {
    let client = reqwest::Client::new();
    let url = format!("https://api.cloudflare.com/client/v4/zones?name={}", zone);
    let response: CloudflareListResponse = client
        .get(&url)
        .header("X-Auth-Email", email)
        .header("X-Auth-Key", key)
        .header("Content-Type", "application/json")
        .send()?
        .json()?;
    if !response.success {
        let err: String = response
            .errors
            .iter()
            .map(|s| format!("{}\n", s.to_owned()))
            .collect();
        return Err(format_err!("API Error: {}", err));
    }

    Ok(response.result[0].id.clone())
}

pub fn get_dns_record_id(
    zone_id: &str,
    domain: &str,
    email: &str,
    key: &str,
) -> Result<String, Error> {
    let client = reqwest::Client::new();
    let url = format!(
        "https://api.cloudflare.com/client/v4/zones/{}/dns_records?name={}",
        zone_id, domain
    );
    let response: CloudflareListResponse = client
        .get(&url)
        .header("X-Auth-Email", email)
        .header("X-Auth-Key", key)
        .header("Content-Type", "application/json")
        .send()?
        .json()?;
    if !response.success {
        let err: String = response
            .errors
            .iter()
            .map(|s| format!("{}\n", s.to_owned()))
            .collect();
        return Err(format_err!("API Error: {}", err));
    }

    let id = match response.result.first() {
        Some(v) => v.id.clone(),
        None => {
            return Err(format_err!(
                "Unexpected API result for DNS record. Check if you passed the right options."
            ))
        }
    };

    Ok(id)
}

pub fn get_current_ip() -> Result<String, Error> {
    Ok(reqwest::Client::new()
        .get("http://ipv4.icanhazip.com")
        .send()?
        .text()?
        .trim()
        .into())
}

pub fn update_ddns(
    ip: &str,
    domain: &str,
    zone_id: &str,
    record_id: &str,
    email: &str,
    key: &str,
) -> Result<(), Error> {
    let client = reqwest::Client::new();
    let url = format!(
        "https://api.cloudflare.com/client/v4/zones/{}/dns_records/{}",
        zone_id, record_id
    );

    let update_data = UpdateIpData {
        id: zone_id.to_owned(),
        r#type: "A".to_owned(),
        name: domain.to_owned(),
        content: ip.to_owned(),
    };

    let response: CloudflareUpdateResponse = client
        .put(&url)
        .header("X-Auth-Email", email)
        .header("X-Auth-Key", key)
        .header("Content-Type", "application/json")
        .json(&update_data)
        .send()?
        .json()?;

    if !response.success {
        let err: String = response
            .errors
            .iter()
            .map(|s| format!("{}\n", s.to_owned()))
            .collect();
        return Err(format_err!("Unsuccessful update of DNS record: {}", err));
    }

    Ok(())
}