ochami-rs 0.1.33

use std::time;

use serde_json::Value;

use crate::{
    error::Error,
    pcs::transitions::types::{Location, Operation},
};

use super::types::Transition;

pub async fn get(
    shasta_base_url: &str,
    shasta_token: &str,
    shasta_root_cert: &[u8],
) -> Result<Vec<Value>, Error> {
    let client;

    let client_builder = reqwest::Client::builder()
        .add_root_certificate(reqwest::Certificate::from_pem(shasta_root_cert)?);

    // Build client
    if std::env::var("SOCKS5").is_ok() {
        // socks5 proxy
        log::debug!("SOCKS5 enabled");
        let socks5proxy = reqwest::Proxy::all(std::env::var("SOCKS5").unwrap())?;

        // rest client to authenticate
        client = client_builder.proxy(socks5proxy).build()?;
    } else {
        client = client_builder.build()?;
    }

    let api_url = format!("{}/power-control/v1/transitions", shasta_base_url);

    let response = client
        .get(api_url)
        .bearer_auth(shasta_token)
        .send()
        .await
        .map_err(|error| Error::NetError(error))?;

    if response.status().is_success() {
        let resp_payload = response
            .json::<Value>()
            .await
            .map_err(|error| Error::NetError(error))?;

        serde_json::from_value::<Vec<Value>>(resp_payload["transitions"].clone())
            .map_err(|error| Error::SerdeError(error))
    } else {
        let payload = response
            .json::<Value>()
            .await
            .map_err(|error| Error::NetError(error))?;

        Err(Error::CsmError(payload))
    }
}

pub async fn get_by_id(
    shasta_token: &str,
    shasta_base_url: &str,
    shasta_root_cert: &[u8],
    id: &str,
) -> Result<Value, Error> {
    let client;

    let client_builder = reqwest::Client::builder()
        .add_root_certificate(reqwest::Certificate::from_pem(shasta_root_cert)?);

    // Build client
    if std::env::var("SOCKS5").is_ok() {
        // socks5 proxy
        log::debug!("SOCKS5 enabled");
        let socks5proxy = reqwest::Proxy::all(std::env::var("SOCKS5").unwrap())?;

        // rest client to authenticate
        client = client_builder.proxy(socks5proxy).build()?;
    } else {
        client = client_builder.build()?;
    }

    let api_url = format!("{}/power-control/v1/transitions/{}", shasta_base_url, id);

    let response = client
        .get(api_url)
        .bearer_auth(shasta_token)
        .send()
        .await
        .map_err(|error| Error::NetError(error))?;

    if response.status().is_success() {
        let payload = response
            .json()
            .await
            .map_err(|error| Error::NetError(error));

        log::debug!("PCS transition details\n{:#?}", payload);

        payload
    } else {
        let payload = response
            .json::<Value>()
            .await
            .map_err(|error| Error::NetError(error))?;

        Err(Error::CsmError(payload))
    }
}

pub async fn post(
    shasta_base_url: &str,
    shasta_token: &str,
    shasta_root_cert: &[u8],
    operation: &str,
    xname_vec: &Vec<String>,
) -> Result<Value, Error> {
    log::info!("Create PCS transition '{}' on {:?}", operation, xname_vec);

    //Create request payload
    //
    // Create 'location' list with all the xnames to operate
    let mut location_vec: Vec<Location> = Vec::new();

    for xname in xname_vec {
        let location: Location = Location {
            xname: xname.to_string(),
            deputy_key: None,
        };

        location_vec.push(location);
    }

    // Create 'transition'
    let request_payload = Transition {
        operation: Operation::from_str(operation)?,
        task_deadline_minutes: None,
        location: location_vec,
    };

    // Build http client
    let client_builder = reqwest::Client::builder()
        .add_root_certificate(reqwest::Certificate::from_pem(shasta_root_cert)?);

    let client = if let Ok(socks5_env) = std::env::var("SOCKS5") {
        // socks5 proxy
        log::debug!("SOCKS5 enabled");
        let socks5proxy = reqwest::Proxy::all(socks5_env)?;

        // rest client to authenticate
        client_builder.proxy(socks5proxy).build()?
    } else {
        client_builder.build()?
    };

    let api_url = shasta_base_url.to_owned() + "/power-control/v1/transitions";

    // Submit call to http api
    let response = client
        .post(api_url)
        .json(&request_payload)
        .bearer_auth(shasta_token)
        .send()
        .await
        .map_err(|error| Error::NetError(error))?;

    if response.status().is_success() {
        Ok(response.json::<Value>().await.unwrap())
    } else {
        let payload = response.json().await.map_err(|e| Error::NetError(e))?;

        Err(Error::CsmError(payload))
    }
}

// Creates a task on CSM for power management nodes.
// Returns a serde_json::Value with the power task management
pub async fn post_block(
    shasta_base_url: &str,
    shasta_token: &str,
    shasta_root_cert: &[u8],
    operation: &str,
    xname_vec: &Vec<String>,
) -> Result<Value, Error> {
    let node_reset = post(
        shasta_base_url,
        shasta_token,
        shasta_root_cert,
        operation,
        xname_vec,
    )
    .await?;

    let transition_id = node_reset["transitionID"].as_str().unwrap();

    log::info!("PCS transition ID: {}", transition_id);

    let power_management_status: Value = wait_to_complete(
        shasta_base_url,
        shasta_token,
        shasta_root_cert,
        transition_id,
    )
    .await?;

    Ok(power_management_status)
}

pub async fn wait_to_complete(
    shasta_base_url: &str,
    shasta_token: &str,
    shasta_root_cert: &[u8],
    transition_id: &str,
) -> Result<Value, Error> {
    let mut transition_status = "";

    let mut transition: serde_json::Value = get_by_id(
        shasta_token,
        shasta_base_url,
        shasta_root_cert,
        transition_id,
    )
    .await?;

    let mut i = 1;
    let max_attempt = 300;

    while i <= max_attempt && transition_status != "completed" {
        // Check PCS transition status
        transition = get_by_id(
            shasta_token,
            shasta_base_url,
            shasta_root_cert,
            transition_id,
        )
        .await?;

        transition_status = transition["transitionStatus"].as_str().unwrap();

        let operation = transition["operation"].as_str().unwrap();

        let failed = transition
            .pointer("/taskCounts/failed")
            .unwrap()
            .as_number()
            .unwrap();

        let in_progress = transition
            .pointer("/taskCounts/in-progress")
            .unwrap()
            .as_number()
            .unwrap();

        /* let new = transition
        .pointer("/taskCounts/new")
        .unwrap()
        .as_number()
        .unwrap(); */

        let succeeded = transition
            .pointer("/taskCounts/succeeded")
            .unwrap()
            .as_number()
            .unwrap();

        let total = transition
            .pointer("/taskCounts/total")
            .unwrap()
            .as_number()
            .unwrap();

        /* let un_supported = transition
        .pointer("/taskCounts/un-supported")
        .unwrap()
        .as_number()
        .unwrap(); */

        eprintln!(
                    "Power '{}' summary - status: {}, failed: {}, in-progress: {}, succeeded: {}, total: {}. Attempt {} of {}",
                    operation, transition_status, failed, in_progress, succeeded, total, i, max_attempt
                );

        tokio::time::sleep(time::Duration::from_secs(3)).await;
        i += 1;
    }

    if transition_status == "completed" {
        Ok(transition)
    } else {
        Err(Error::CsmError(transition))
    }
}