cloud-seed 0.1.0

Fetch user-data from a cloud provider and write files described within to the filesystem
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

mod execute;
mod schema;
mod source;

use anyhow::{bail, Result};
use source::Source;
use tracing::{debug, info};

use crate::execute::execute_user_data;

const SOURCES: &[&dyn Source] = &[
  #[cfg(feature = "source-alibaba")]
  &source::alibaba::AlibabaSource,
  #[cfg(feature = "source-amazon")]
  &source::amazon::AmazonSource,
  #[cfg(feature = "source-exoscale")]
  &source::exoscale::ExoscaleSource,
  #[cfg(feature = "source-google")]
  &source::google::GoogleSource,
  #[cfg(feature = "source-openstack")]
  &source::openstack::OpenstackSource,
  #[cfg(feature = "source-oracle")]
  &source::oracle::OracleSource,
  #[cfg(feature = "source-vultr")]
  &source::vultr::VultrSource,
];

const ALLOWED_SHEBANGS: [&str; 2] = [
  "#cloud-seed",
  // Compatibility with cloud-init's write_files
  "#cloud-config",
];

#[tokio::main]
async fn main() -> Result<()> {
  tracing_subscriber::fmt::init();

  for source in SOURCES {
    debug!("Trying {:?}", source);
    if let Some(user_data) = source.try_fetch().await? {
      info!("Got user data from {:?}", source);
      if let Some((shebang, content)) = user_data.split_once('\n') {
        if ALLOWED_SHEBANGS.contains(&shebang.trim()) {
          let user_data = serde_yaml::from_str(content)?;
          info!("Executing user data from {:?}", source);
          execute_user_data(user_data).await;
          return Ok(());
        }

        bail!("Unhandled shebang in user data");
      }

      bail!("Malformed user data");
    }
  }

  bail!("No suitable source found");
}