komodo/bin/cli/src/sync/resources/procedure.rs

use std::collections::HashMap;

use monitor_client::{
  api::{
    execute::Execution,
    read::GetProcedure,
    write::{CreateProcedure, UpdateProcedure},
  },
  entities::{
    procedure::{
      PartialProcedureConfig, Procedure, ProcedureConfig, ProcedureConfigDiff, ProcedureListItemInfo
    },
    resource::{Resource, ResourceListItem},
    toml::ResourceToml,
    update::ResourceTarget,
  },
};
use partial_derive2::{MaybeNone, PartialDiff};

use crate::{
  maps::{
    id_to_build, id_to_deployment, id_to_procedure, id_to_repo,
    id_to_server, name_to_procedure,
  },
  monitor_client,
  sync::resources::ToUpdateItem,
};

use super::{ResourceSync, ToCreate, ToUpdate};

impl ResourceSync for Procedure {
  type Config = ProcedureConfig;
  type Info = ();
  type PartialConfig = PartialProcedureConfig;
  type ConfigDiff = ProcedureConfigDiff;
  type ListItemInfo = ProcedureListItemInfo;

  fn display() -> &'static str {
    "procedure"
  }

  fn resource_target(id: String) -> ResourceTarget {
    ResourceTarget::Procedure(id)
  }

  fn name_to_resource(
  ) -> &'static HashMap<String, ResourceListItem<Self::ListItemInfo>>
  {
    name_to_procedure()
  }

  async fn create(
    resource: ResourceToml<Self::PartialConfig>,
  ) -> anyhow::Result<String> {
    monitor_client()
      .write(CreateProcedure {
        name: resource.name,
        config: resource.config,
      })
      .await
      .map(|p| p.id)
  }

  async fn update(
    id: String,
    resource: ResourceToml<Self::PartialConfig>,
  ) -> anyhow::Result<()> {
    monitor_client()
      .write(UpdateProcedure {
        id,
        config: resource.config,
      })
      .await?;
    Ok(())
  }

  async fn run_updates(
    mut to_create: ToCreate<Self::PartialConfig>,
    mut to_update: ToUpdate<Self::PartialConfig>,
  ) {
    if to_update.is_empty() && to_create.is_empty() {
      return;
    }

    for i in 0..10 {
      let mut to_pull = Vec::new();
      for ToUpdateItem {
        id,
        resource,
        update_description,
        update_tags,
      } in &to_update
      {
        // Update resource
        let name = resource.name.clone();
        let tags = resource.tags.clone();
        let description = resource.description.clone();
        if *update_description {
          Self::update_description(id.clone(), &name, description)
            .await;
        }
        if *update_tags {
          Self::update_tags(id.clone(), &name, tags).await;
        }
        if !resource.config.is_none() {
          if let Err(e) =
            Self::update(id.clone(), resource.clone()).await
          {
            if i == 9 {
              warn!(
                "failed to update {} {name} | {e:#}",
                Self::display()
              );
            }
          }
        }

        info!("{} {name} updated", Self::display());
        // have to clone out so to_update is mutable
        to_pull.push(id.clone());
      }
      //
      to_update.retain(|resource| !to_pull.contains(&resource.id));

      let mut to_pull = Vec::new();
      for resource in &to_create {
        let name = resource.name.clone();
        let tags = resource.tags.clone();
        let description = resource.description.clone();
        let id = match Self::create(resource.clone()).await {
          Ok(id) => id,
          Err(e) => {
            if i == 9 {
              warn!(
                "failed to create {} {name} | {e:#}",
                Self::display(),
              );
            }
            continue;
          }
        };
        Self::update_tags(id.clone(), &name, tags).await;
        Self::update_description(id, &name, description).await;
        info!("{} {name} created", Self::display());
        to_pull.push(name);
      }
      to_create.retain(|resource| !to_pull.contains(&resource.name));

      if to_update.is_empty() && to_create.is_empty() {
        info!(
          "============ {}s synced ✅ ============",
          Self::display()
        );
        return;
      }
    }
    warn!("procedure sync loop exited after max iterations");
  }

  async fn get(
    id: String,
  ) -> anyhow::Result<Resource<Self::Config, Self::Info>> {
    monitor_client().read(GetProcedure { procedure: id }).await
  }

  async fn get_diff(
    mut original: Self::Config,
    update: Self::PartialConfig,
  ) -> anyhow::Result<Self::ConfigDiff> {
    for execution in &mut original.executions {
      match &mut execution.execution {
        Execution::None(_) => {}
        Execution::RunProcedure(config) => {
          config.procedure = id_to_procedure()
            .get(&config.procedure)
            .map(|p| p.name.clone())
            .unwrap_or_default();
        }
        Execution::RunBuild(config) => {
          config.build = id_to_build()
            .get(&config.build)
            .map(|b| b.name.clone())
            .unwrap_or_default();
        }
        Execution::Deploy(config) => {
          config.deployment = id_to_deployment()
            .get(&config.deployment)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::StartContainer(config) => {
          config.deployment = id_to_deployment()
            .get(&config.deployment)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::StopContainer(config) => {
          config.deployment = id_to_deployment()
            .get(&config.deployment)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::RemoveContainer(config) => {
          config.deployment = id_to_deployment()
            .get(&config.deployment)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::CloneRepo(config) => {
          config.repo = id_to_repo()
            .get(&config.repo)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::PullRepo(config) => {
          config.repo = id_to_repo()
            .get(&config.repo)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::StopAllContainers(config) => {
          config.server = id_to_server()
            .get(&config.server)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::PruneDockerNetworks(config) => {
          config.server = id_to_server()
            .get(&config.server)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::PruneDockerImages(config) => {
          config.server = id_to_server()
            .get(&config.server)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
        Execution::PruneDockerContainers(config) => {
          config.server = id_to_server()
            .get(&config.server)
            .map(|d| d.name.clone())
            .unwrap_or_default();
        }
      }
    }

    Ok(original.partial_diff(update))
  }
}