komodo/bin/core/src/api/read/schedule.rs

use futures_util::future::join_all;
use komodo_client::{
  api::read::*,
  entities::{
    ResourceTarget,
    action::Action,
    permission::PermissionLevel,
    procedure::Procedure,
    resource::{ResourceQuery, TemplatesQueryBehavior},
    schedule::Schedule,
  },
};
use mogh_resolver::Resolve;

use crate::{
  helpers::query::{get_all_tags, get_last_run_at},
  resource::list_full_for_user,
  schedule::get_schedule_item_info,
};

use super::ReadArgs;

impl Resolve<ReadArgs> for ListSchedules {
  async fn resolve(
    self,
    args: &ReadArgs,
  ) -> mogh_error::Result<Vec<Schedule>> {
    let all_tags = get_all_tags(None).await?;
    let (actions, procedures) = tokio::try_join!(
      list_full_for_user::<Action>(
        ResourceQuery {
          names: Default::default(),
          templates: TemplatesQueryBehavior::Include,
          tag_behavior: self.tag_behavior,
          tags: self.tags.clone(),
          specific: Default::default(),
        },
        &args.user,
        PermissionLevel::Read.into(),
        &all_tags,
      ),
      list_full_for_user::<Procedure>(
        ResourceQuery {
          names: Default::default(),
          templates: TemplatesQueryBehavior::Include,
          tag_behavior: self.tag_behavior,
          tags: self.tags.clone(),
          specific: Default::default(),
        },
        &args.user,
        PermissionLevel::Read.into(),
        &all_tags,
      )
    )?;
    let actions = actions.into_iter().map(async |action| {
      let (next_scheduled_run, schedule_error) =
        get_schedule_item_info(&ResourceTarget::Action(
          action.id.clone(),
        ));
      let last_run_at =
        get_last_run_at::<Action>(&action.id).await.unwrap_or(None);
      Schedule {
        target: ResourceTarget::Action(action.id),
        name: action.name,
        enabled: action.config.schedule_enabled,
        schedule_format: action.config.schedule_format,
        schedule: action.config.schedule,
        schedule_timezone: action.config.schedule_timezone,
        tags: action.tags,
        last_run_at,
        next_scheduled_run,
        schedule_error,
      }
    });
    let procedures = procedures.into_iter().map(async |procedure| {
      let (next_scheduled_run, schedule_error) =
        get_schedule_item_info(&ResourceTarget::Procedure(
          procedure.id.clone(),
        ));
      let last_run_at = get_last_run_at::<Procedure>(&procedure.id)
        .await
        .unwrap_or(None);
      Schedule {
        target: ResourceTarget::Procedure(procedure.id),
        name: procedure.name,
        enabled: procedure.config.schedule_enabled,
        schedule_format: procedure.config.schedule_format,
        schedule: procedure.config.schedule,
        schedule_timezone: procedure.config.schedule_timezone,
        tags: procedure.tags,
        last_run_at,
        next_scheduled_run,
        schedule_error,
      }
    });
    let (actions, procedures) =
      tokio::join!(join_all(actions), join_all(procedures));

    Ok(
      actions
        .into_iter()
        .chain(procedures)
        .filter(|s| !s.schedule.is_empty())
        .collect(),
    )
  }
}