komodo/bin/monrun/src/execution.rs

use std::path::Path;

use anyhow::{anyhow, Context};
use futures::future::join_all;
use monitor_client::api::execute;
use serde::Deserialize;
use strum::Display;

use crate::{
  maps::{name_to_build, name_to_deployment, names_to_ids},
  monitor_client,
};

pub async fn run_execution(path: &Path) -> anyhow::Result<()> {
  let ExecutionFile { name, stages } = crate::parse_toml_file(path)?;

  info!("EXECUTION: {name}");
  info!("path: {path:?}");
  println!("{stages:#?}");

  crate::wait_for_enter("EXECUTE")?;

  run_stages(stages)
    .await
    .context("failed during a stage. terminating run.")?;

  info!("finished successfully ✅");

  Ok(())
}

/// Specifies sequence of stages (build / deploy) on resources
#[derive(Debug, Clone, Deserialize)]
pub struct ExecutionFile {
  pub name: String,
  #[serde(rename = "stage")]
  pub stages: Vec<Stage>,
}

#[derive(Debug, Clone, Deserialize)]
pub struct Stage {
  pub name: String,
  pub action: ExecutionType,
  /// resource names
  pub targets: Vec<String>,
}

#[derive(Debug, Clone, Copy, Deserialize, Display)]
#[serde(rename_all = "snake_case")]
#[strum(serialize_all = "snake_case")]
pub enum ExecutionType {
  Build,
  Deploy,
  StartContainer,
  StopContainer,
  DestroyContainer,
}

pub async fn run_stages(stages: Vec<Stage>) -> anyhow::Result<()> {
  for Stage {
    name,
    action,
    targets,
  } in stages
  {
    info!("running {action} stage: {name}... ⏳");
    let targets = match action {
      ExecutionType::Build => {
        names_to_ids(&targets, name_to_build())?
      }
      _ => names_to_ids(&targets, name_to_deployment())?,
    };
    match action {
      ExecutionType::Build => {
        trigger_builds_in_parallel(&targets).await?;
      }
      ExecutionType::Deploy => {
        redeploy_deployments_in_parallel(&targets).await?;
      }
      ExecutionType::StartContainer => {
        start_containers_in_parallel(&targets).await?
      }
      ExecutionType::StopContainer => {
        stop_containers_in_parallel(&targets).await?
      }
      ExecutionType::DestroyContainer => {
        destroy_containers_in_parallel(&targets).await?;
      }
    }
    info!("finished {action} stage: {name} ✅");
  }
  Ok(())
}

async fn redeploy_deployments_in_parallel(
  deployment_ids: &[&String],
) -> anyhow::Result<()> {
  let futes = deployment_ids.iter().map(|id| async move {
    monitor_client()
      .execute(execute::Deploy { deployment_id: id.to_string(), stop_signal: None, stop_time: None })
      .await
      .with_context(|| format!("failed to deploy {id}"))
      .and_then(|update| {
        if update.success {
          Ok(())
        } else {
          Err(anyhow!(
            "failed to deploy {id}. operation unsuccessful, see monitor update"
          ))
        }
      })
  });
  join_all(futes).await.into_iter().collect()
}

async fn start_containers_in_parallel(
  deployment_ids: &[&String],
) -> anyhow::Result<()> {
  let futes = deployment_ids.iter().map(|id| async move {
    monitor_client()
    .execute(execute::StartContainer { deployment_id: id.to_string() })
      .await
      .with_context(|| format!("failed to start container {id}"))
      .and_then(|update| {
        if update.success {
          Ok(())
        } else {
          Err(anyhow!(
            "failed to start container {id}. operation unsuccessful, see monitor update"
          ))
        }
      })
  });
  join_all(futes).await.into_iter().collect()
}

async fn stop_containers_in_parallel(
  deployment_ids: &[&String],
) -> anyhow::Result<()> {
  let futes = deployment_ids.iter().map(|id| async move {
    monitor_client()
      .execute(execute::StopContainer { deployment_id: id.to_string(), signal: None, time: None })
      .await
      .with_context(|| format!("failed to stop container {id}"))
      .and_then(|update| {
        if update.success {
          Ok(())
        } else {
          Err(anyhow!(
            "failed to stop container {id}. operation unsuccessful, see monitor update"
          ))
        }
      })
  });
  join_all(futes).await.into_iter().collect()
}

async fn destroy_containers_in_parallel(
  deployment_ids: &[&String],
) -> anyhow::Result<()> {
  let futes = deployment_ids.iter().map(|id| async move {
    monitor_client()
      .execute(execute::RemoveContainer { deployment_id: id.to_string(), signal: None, time: None })
      .await
      .with_context(|| format!("failed to destroy container {id}"))
      .and_then(|update| {
        if update.success {
          Ok(())
        } else {
          Err(anyhow!(
            "failed to destroy container {id}. operation unsuccessful, see monitor update"
          ))
        }
      })
  });
  join_all(futes).await.into_iter().collect()
}

async fn trigger_builds_in_parallel(
  build_ids: &[&String],
) -> anyhow::Result<()> {
  let futes = build_ids.iter().map(|id| async move {
    monitor_client()
      .execute(execute::RunBuild { build_id: id.to_string() })
      .await
      .with_context(|| format!("failed to build {id}"))
      .and_then(|update| {
        if update.success {
          Ok(())
        } else {
          Err(anyhow!(
            "failed to build {id}. operation unsuccessful, see monitor update"
          ))
        }
      })
  });
  join_all(futes).await.into_iter().collect()
}