csm-rs 0.99.0

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

use crate::{
  bos::template::http_client::v2::types::BosSessionTemplate, error::Error,
};
use globset::Glob;

pub fn filter(
  bos_sessiontemplate_vec: &mut Vec<BosSessionTemplate>,
  configuration_name_pattern_opt: Option<&str>,
  target_hsm_group_name_vec: &[String],
  xname_vec: &[String],
  // cfs_configuration_name_opt: Option<&str>,
  limit_number_opt: Option<&u8>,
) -> Result<Vec<BosSessionTemplate>, Error> {
  log::info!("Filter BOS sessiontemplates");

  if let Some(configuration_name_pattern) = configuration_name_pattern_opt {
    let glob = Glob::new(configuration_name_pattern)?.compile_matcher();

    bos_sessiontemplate_vec.retain(|sessiontemplate| {
      sessiontemplate
        .configuration_name()
        .is_some_and(|configuration_name| glob.is_match(configuration_name))
    });
  };

  // Filter by list of HSM group or xnames as target
  if !target_hsm_group_name_vec.is_empty() || !xname_vec.is_empty() {
    bos_sessiontemplate_vec.retain(|bos_sessiontemplate| {
      let bos_sessiontemplate_target_hsm = bos_sessiontemplate.get_target_hsm();
      let bos_sessiontemplate_target_xname =
        bos_sessiontemplate.get_target_xname();

      !bos_sessiontemplate_target_hsm.is_empty()
        && bos_sessiontemplate_target_hsm
          .iter()
          .all(|bos_st_hsm_group| {
            target_hsm_group_name_vec.iter().any(|target_hsm_group| {
              bos_st_hsm_group.contains(target_hsm_group)
            })
          })
        || !bos_sessiontemplate_target_xname.is_empty()
          && bos_sessiontemplate_target_xname
            .iter()
            .all(|target_xname| xname_vec.contains(target_xname))
    });
  }

  if let Some(limit_number) = limit_number_opt {
    // Limiting the number of results to return to client
    *bos_sessiontemplate_vec = bos_sessiontemplate_vec
      [bos_sessiontemplate_vec
        .len()
        .saturating_sub(*limit_number as usize)..]
      .to_vec();
  }

  Ok(bos_sessiontemplate_vec.to_vec())
}

pub async fn filter_by_configuration(
  bos_sessiontemplate_vec: &mut Vec<BosSessionTemplate>,
  cfs_configuration_name: &str,
) {
  bos_sessiontemplate_vec.retain(|bos_template| {
    bos_template.get_configuration().as_deref() == Some(cfs_configuration_name)
  });
}

pub fn get_image_id_cfs_configuration_target_tuple_vec(
  bos_sessiontemplate_value_vec: &Vec<BosSessionTemplate>,
) -> Vec<(String, String, Vec<String>)> {
  let mut image_id_cfs_configuration_from_bos_sessiontemplate: Vec<(
    String,
    String,
    Vec<String>,
  )> = Vec::new();

  for bos_sessiontemplate in bos_sessiontemplate_value_vec {
    let cfs_configuration_opt: Option<String> = bos_sessiontemplate
      .clone()
      .cfs
      .and_then(|cfs| cfs.configuration);

    // FIXME: use BosSessionTemplate.get_image_vec() to get the list of image ids
    let first_image_id_opt: Option<String> = bos_sessiontemplate
      .get_path_vec()
      .first()
      .and_then(|v| v.strip_prefix("s3://boot-images/"))
      .and_then(|v| v.strip_suffix("/manifest.json").map(str::to_string));

    if let (Some(cfs_configuration), Some(first_image_id)) =
      (cfs_configuration_opt, first_image_id_opt)
    {
      let target = [
        bos_sessiontemplate.get_target_hsm(),
        bos_sessiontemplate.get_target_xname(),
      ]
      .concat();

      image_id_cfs_configuration_from_bos_sessiontemplate.push((
        first_image_id.to_string(),
        cfs_configuration.to_string(),
        target,
      ));
    } else {
      log::warn!("BOS sessiontemplate '{:?}' not valid, check fields 'path' and 'cfs.configuration' have valid values. Path field should have 's3://boot-images/' as prefix and '/manifest.json' as sufix", bos_sessiontemplate.name);
    }
  }

  image_id_cfs_configuration_from_bos_sessiontemplate
}