railwayapp 3.5.2

use reqwest::Client;

use crate::{
    client::post_graphql,
    commands::{
        queries::{self},
        subscriptions::{self, build_logs, deployment_logs},
        Configs,
    },
    errors::RailwayError,
    subscription::subscribe_graphql,
};
use anyhow::{Context, Result};
use futures::StreamExt;

pub async fn get_deployment(
    client: &Client,
    configs: &Configs,
    deployment_id: String,
) -> Result<queries::RailwayDeployment, RailwayError> {
    let vars = queries::deployment::Variables { id: deployment_id };

    let deployment = post_graphql::<queries::Deployment, _>(client, configs.get_backboard(), vars)
        .await?
        .deployment;

    Ok(deployment)
}

pub async fn stream_build_logs(
    deployment_id: String,
    on_log: impl Fn(build_logs::LogFields),
) -> Result<()> {
    let vars = subscriptions::build_logs::Variables {
        deployment_id: deployment_id.clone(),
        filter: Some(String::new()),
        limit: Some(500),
    };

    let (_client, mut stream) = subscribe_graphql::<subscriptions::BuildLogs>(vars).await?;
    while let Some(Ok(log)) = stream.next().await {
        let log = log.data.context("Failed to retrieve build log")?;
        for line in log.build_logs {
            on_log(line);
        }
    }

    Ok(())
}

pub async fn stream_deploy_logs(
    deployment_id: String,
    on_log: impl Fn(deployment_logs::LogFields),
) -> Result<()> {
    let vars = subscriptions::deployment_logs::Variables {
        deployment_id: deployment_id.clone(),
        filter: Some(String::new()),
        limit: Some(500),
    };

    let (_client, mut stream) = subscribe_graphql::<subscriptions::DeploymentLogs>(vars).await?;
    while let Some(Ok(log)) = stream.next().await {
        let log = log.data.context("Failed to retrieve deploy log")?;
        for line in log.deployment_logs {
            on_log(line);
        }
    }

    Ok(())
}