OneLauncher-Cracked/core/src/api/proxy/send.rs

use super::IngressId;
use crate::api::proxy::{ProxyError, Ingress, IngressType, ProcessPayloadType, ClusterPayloadType, InternetPayload};
use crate::store::{ClusterPath, IngressProcessType, IngressProcessor};
use uuid::Uuid;

#[cfg(feature = "tauri")]
use crate::api::proxy::{IngressPayload, ProcessPayload, ClusterPayload, MessagePayload};

#[cfg(feature = "tauri")]
use tauri::Manager;

#[onelauncher_debug::debugger]
pub async fn init_ingress(
    ingress_type: IngressType,
    total: f64,
    title: &str,
) -> crate::Result<IngressId> {
    let key = init_ingress_internal(ingress_type, total, title).await?;
    IngressProcessor::add_ingress(IngressProcessType::IngressFeed, key.0).await?;
    Ok(key)
}

#[cfg(feature = "cli")]
const CLI_TOTAL_INGRESS: u64 = 1000;

#[onelauncher_debug::debugger]
pub async fn init_ingress_internal(
    ingress_type: IngressType,
    total: f64,
    title: &str,
) -> crate::Result<IngressId> {
    let proxy_state = crate::ProxyState::get().await?;
    let key = IngressId(Uuid::new_v4());

    proxy_state.ingress_feeds.write().await.insert(key.0, Ingress {
        ingress_uuid: key.0,
        message: title.to_string(),
        total,
        current: 0.0,
        last_sent: 0.0,
        ingress_type,
        #[cfg(feature = "cli")]
        cli: {
            let pb = indicatif::ProgressBar::new(CLI_TOTAL_INGRESS);
            pb.set_position(0);
            pb.set_style(indicatif::ProgressStyle::default_bar().template(
                "{spinner:.green}, [{elapsed_precise}] [{bar:.lime/green}] {pos}/{len} {msg}"
            ).unwrap().progress_chars("#>-"));
            pb
        },
    });

    send_ingress(&key, 0.0, None).await?;
    Ok(key)
}

pub async fn init_or_edit_ingress(
    id: Option<IngressId>,
    ingress_type: IngressType,
    total: f64,
    title: &str,
) -> crate::Result<IngressId> {
    if let Some(id) = id {
        edit_ingress(&id, ingress_type, total, title).await?;

        Ok(id)
    } else {
        init_ingress(ingress_type, total, title).await
    }
}

pub async fn edit_ingress(
    id: &IngressId,
    ingress_type: IngressType,
    total: f64,
    title: &str,
) -> crate::Result<()> {
    let proxy_state = crate::ProxyState::get().await?;

    if let Some(ingress) = proxy_state.ingress_feeds.write().await.get_mut(&id.0) {
        ingress.ingress_type = ingress_type;
        ingress.total = total;
        ingress.message = title.to_string();
        ingress.current = 0.0;
        ingress.last_sent = 0.0;
        #[cfg(feature = "cli")]
        { ingress.cli.reset(); }
    };

    send_ingress(id, 0.0, None).await?;
    Ok(())
}

#[allow(unused_variables)]
#[tracing::instrument(level = "debug")]
#[onelauncher_debug::debugger]
pub async fn send_ingress(
    key: &IngressId,
    increment: f64,
    message: Option<&str>,
) -> crate::Result<()> {
    let proxy_state = crate::ProxyState::get().await?;
    let mut ingress = proxy_state.ingress_feeds.write().await;
    let ingress = match ingress.get_mut(&key.0) {
        Some(f) => f,
        None => {
            return Err(ProxyError::NoIngressFound(key.0).into());
        }
    };

    ingress.current += increment;
    let display = ingress.current / ingress.total;
    let display_conv = if display >= 1.0 {
        None
    } else {
        Some(display)
    };

    if f64::abs(display - ingress.last_sent) > 0.005 {
        #[cfg(feature = "cli")]
        {
            ingress.cli.set_message(message.map(|x| x.to_string()).unwrap_or(ingress.message.clone()));
            ingress.cli.set_position((display * CLI_TOTAL_INGRESS as f64).round() as u64);
        }

        #[cfg(feature = "tauri")]
        proxy_state.app.emit("ingress", IngressPayload {
            fraction: display_conv,
            message: message.unwrap_or(&ingress.message).to_string(),
            event: ingress.ingress_type.clone(),
            ingress_uuid: ingress.ingress_uuid,
        }).map_err(ProxyError::from)?;

        ingress.last_sent = display;
    }

    Ok(())
}

pub async fn send_message(message: &str) -> crate::Result<()> {
    #[cfg(feature = "tauri")]
    {
        let proxy_state = crate::ProxyState::get().await?;
        proxy_state.app.emit("message", MessagePayload { message: message.to_string() }).map_err(ProxyError::from)?;
    }

    tracing::warn!("{}", message);
    Ok(())
}

pub async fn send_offline(offline: bool) -> crate::Result<()> {
    #[cfg(feature = "tauri")]
    {
        let proxy_state = crate::ProxyState::get().await?;
        proxy_state.app.emit("offline", offline).map_err(ProxyError::from)?;
    }

    Ok(())
}

pub async fn send_internet(internet: InternetPayload) -> crate::Result<()> {
    tracing::debug!("operation {}", serde_json::to_string(&internet)?);
    #[cfg(feature = "tauri")]
    {
        let proxy_state = crate::ProxyState::get().await?;
        proxy_state.app.emit("internet", internet).map_err(ProxyError::from)?;
    }

    Ok(())
}

pub async fn send_process(
    uuid: Uuid,
    pid: u32,
    event: ProcessPayloadType,
    message: &str,
) -> crate::Result<()> {
    #[cfg(feature = "tauri")]
    {
        let proxy_state = crate::ProxyState::get().await?;
        proxy_state.app.emit("process", ProcessPayload { uuid, pid, event, message: message.to_string() }).map_err(ProxyError::from)?;
    }

    Ok(())
}

pub async fn send_cluster(
    uuid: Uuid,
    cluster_path: &ClusterPath,
    name: &str,
    event: ClusterPayloadType,
) -> crate::Result<()> {
    #[cfg(feature = "tauri")]
    {
        let path = cluster_path.full_path().await?;
        let proxy_state = crate::ProxyState::get().await?;
        proxy_state.app.emit("cluster", ClusterPayload {
            uuid,
            cluster_path: cluster_path.clone(),
            path,
            name: name.to_string(),
            event,
        }).map_err(ProxyError::from)?;
    }

    Ok(())
}