csm-rs 0.7.1

A library for Shasta
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

pub mod http_client;
pub mod utils;

use crate::cfs;
use http_client::v2::types::{CfsSessionGetResponse, CfsSessionPostRequest};

use crate::{
  common::{
    kubernetes::{self, print_cfs_session_logs},
    vault::http_client::fetch_shasta_k8s_secrets_from_vault,
  },
  error::Error,
};

/// Fetch CFS sessions ref --> https://apidocs.svc.cscs.ch/paas/cfs/operation/get_sessions/
/// Returns list of CFS sessions ordered by start time.
/// This methods filter by either HSM group name or HSM group members or both
pub async fn get_and_sort(
  shasta_token: &str,
  shasta_base_url: &str,
  shasta_root_cert: &[u8],
  min_age_opt: Option<&String>,
  max_age_opt: Option<&String>,
  status_opt: Option<&String>,
  session_name_opt: Option<&String>,
  is_succeded_opt: Option<bool>,
) -> Result<Vec<CfsSessionGetResponse>, Error> {
  let mut cfs_session_vec = cfs::session::http_client::v2::get(
    shasta_token,
    shasta_base_url,
    shasta_root_cert,
    min_age_opt,
    max_age_opt,
    status_opt,
    session_name_opt,
    is_succeded_opt,
  )
  .await?;

  // Sort CFS sessions by start time order ASC
  cfs_session_vec.sort_by(|a, b| {
    a.status
      .as_ref()
      .unwrap()
      .session
      .as_ref()
      .unwrap()
      .start_time
      .as_ref()
      .unwrap()
      .cmp(
        b.status
          .as_ref()
          .unwrap()
          .session
          .as_ref()
          .unwrap()
          .start_time
          .as_ref()
          .unwrap(),
      )
  });

  Ok(cfs_session_vec)
}

pub async fn post(
  shasta_token: &str,
  shasta_base_url: &str,
  shasta_root_cert: &[u8],
  session: &CfsSessionPostRequest,
) -> Result<CfsSessionGetResponse, Error> {
  log::info!("Create CFS session '{}'", session.name);
  log::debug!("Create CFS session request payload:\n{:#?}", session);

  cfs::session::http_client::v2::post(
    shasta_token,
    shasta_base_url,
    shasta_root_cert,
    session,
  )
  .await
}

/// Creates a CFS session and waits for it to finish.
/// Optionally, it can also print the CFS session logs if `watch_logs` is set to true.
pub async fn post_sync(
  shasta_token: &str,
  shasta_base_url: &str,
  shasta_root_cert: &[u8],
  vault_base_url: &str,
  site_name: &str,
  k8s_api_url: &str,
  session: &CfsSessionPostRequest,
  watch_logs: bool,
) -> Result<CfsSessionGetResponse, Error> {
  // Create CFS session
  log::info!("Create CFS session '{}'", session.name);
  let cfs_session: CfsSessionGetResponse = cfs::session::post(
    shasta_token,
    shasta_base_url,
    shasta_root_cert,
    session,
  )
  .await?;

  let cfs_session_name: String = cfs_session.name.unwrap();

  // FIXME: refactor becase this code is duplicated in command `manta apply sat-file` and also in
  // `manta logs`
  if watch_logs {
    log::info!("Fetching logs form CFS session {} ...", session.name);
    let shasta_k8s_secrets = fetch_shasta_k8s_secrets_from_vault(
      vault_base_url,
      shasta_token,
      site_name,
    )
    .await?;

    let client = kubernetes::get_k8s_client_programmatically(
      k8s_api_url,
      shasta_k8s_secrets,
    )
    .await
    .unwrap();

    let _ = print_cfs_session_logs(client, &cfs_session_name).await;
  }

  // User does not want the CFS logs but we still need to wayt the CFS session to
  // finis. Wait till the CFS session finishes
  utils::wait_cfs_session_to_finish(
    shasta_token,
    shasta_base_url,
    shasta_root_cert,
    &cfs_session_name,
  )
  .await?;

  // Get most recent CFS session status
  let cfs_session: CfsSessionGetResponse = get_and_sort(
    shasta_token,
    shasta_base_url,
    shasta_root_cert,
    None,
    None,
    None,
    Some(&cfs_session_name),
    None,
  )
  .await?
  .first()
  .unwrap()
  .clone();

  Ok(cfs_session)
}