Deploy template from CLI (#516)

.gitignore

@@ -3,3 +3,4 @@ node_modules
 bin/railway
 .envrc
 .direnv/*
+.env

src/commands/deploy.rs

@@ -0,0 +1,268 @@
+use anyhow::bail;
+use is_terminal::IsTerminal;
+use serde::Deserialize;
+use std::{collections::BTreeMap, collections::HashMap, time::Duration};
+
+use crate::{consts::TICK_STRING, mutations::TemplateVolume, util::prompt::prompt_text};
+
+use super::*;
+
+/// Provisions a template into your project
+#[derive(Parser)]
+pub struct Args {
+    /// The code of the template to deploy
+    #[arg(short, long)]
+    template: Vec<String>,
+    /// The "{key}={value}" environment variable pair to set the template variables
+    ///
+    /// To specify the variable for a single service prefix it with "{service}."
+    /// Example:
+    ///
+    /// ```bash
+    /// railway deploy -t postgres -v "MY_SPECIAL_ENV_VAR=1" -v "Backend.Port=3000"
+    /// ```
+    #[arg(short, long)]
+    variable: Vec<String>,
+}
+
+pub async fn command(args: Args, _json: bool) -> Result<()> {
+    let configs = Configs::new()?;
+
+    let client = GQLClient::new_authorized(&configs)?;
+    let linked_project = configs.get_linked_project().await?;
+
+    let templates = if args.template.is_empty() {
+        if !std::io::stdout().is_terminal() {
+            bail!("No template specified");
+        }
+        vec![prompt_text("Select template to deploy")?]
+    } else {
+        args.template
+    };
+
+    if templates.is_empty() {
+        bail!("No template selected");
+    }
+
+    let variables: HashMap<String, String> = args
+        .variable
+        .iter()
+        .map(|v| {
+            let mut split = v.split('=');
+            let key = split
+                .next()
+                .as_deref()
+                .unwrap_or_default()
+                .trim()
+                .to_owned();
+            let value = split.collect::<Vec<&str>>().join("=").trim().to_owned();
+            (key, value)
+        })
+        .filter(|(_, value)| !value.is_empty())
+        .collect();
+
+    for template in templates {
+        if std::io::stdout().is_terminal() {
+            fetch_and_create(
+                &client,
+                &configs,
+                template.clone(),
+                &linked_project,
+                &variables,
+            )
+            .await?;
+        } else {
+            println!("Creating {}...", template);
+            fetch_and_create(&client, &configs, template, &linked_project, &variables).await?;
+        }
+    }
+
+    Ok(())
+}
+/// fetch database details via `TemplateDetail`
+/// create database via `TemplateDeploy`
+async fn fetch_and_create(
+    client: &reqwest::Client,
+    configs: &Configs,
+    template: String,
+    linked_project: &LinkedProject,
+    vars: &HashMap<String, String>,
+) -> Result<(), anyhow::Error> {
+    let details = post_graphql::<queries::TemplateDetail, _>(
+        client,
+        configs.get_backboard(),
+        queries::template_detail::Variables {
+            code: template.clone(),
+        },
+    )
+    .await?;
+
+    let config = DeserializedEnvironment::deserialize(
+        &details.template.serialized_config.unwrap_or_default(),
+    )?;
+
+    let mut services: Vec<mutations::template_deploy::TemplateDeployService> =
+        Vec::with_capacity(config.services.len());
+
+    for (id, s) in &config.services {
+        let mut variables = BTreeMap::new();
+        for (key, variable) in &s.variables {
+            let value = if let Some(value) = vars.get(&format!("{}.{key}", s.name)) {
+                value.clone()
+            } else if let Some(value) = vars.get(key) {
+                value.clone()
+            } else if let Some(value) = variable.default_value.as_ref().filter(|v| !v.is_empty()) {
+                value.clone()
+            } else if !variable.is_optional.unwrap_or_default() {
+                prompt_text(&format!(
+                    "Environment Variable {key} for service {} is required, please set a value:\n{}",
+                    s.name,
+                    variable.description.as_deref().map(|d| format!("   *{d}*\n")).unwrap_or_default(),
+                ))?
+            } else {
+                continue;
+            };
+            variables.insert(key.clone(), value);
+        }
+
+        let volumes: Vec<_> = s
+            .volume_mounts
+            .values()
+            .map(|volume| TemplateVolume {
+                mount_path: volume.mount_path.clone(),
+                name: None,
+            })
+            .collect();
+
+        services.push(mutations::template_deploy::TemplateDeployService {
+            commit: None,
+            has_domain: s.networking.as_ref().map(|n| !n.service_domains.is_empty()),
+            healthcheck_path: s.deploy.as_ref().and_then(|d| d.healthcheck_path.clone()),
+            id: Some(id.clone()),
+            is_private: None,
+            name: Some(s.name.clone()),
+            owner: None,
+            root_directory: match &s.source {
+                Some(DeserializedServiceSource::Image { .. }) => None,
+                Some(DeserializedServiceSource::Repo { root_directory, .. }) => {
+                    root_directory.clone()
+                }
+                None => None,
+            },
+            service_icon: s.icon.clone(),
+            service_name: s.name.clone(),
+            start_command: s.deploy.as_ref().and_then(|d| d.start_command.clone()),
+            tcp_proxy_application_port: s
+                .networking
+                .as_ref()
+                .and_then(|n| n.tcp_proxies.keys().next().map(|k| k.parse::<i64>()))
+                .transpose()?,
+            template: match &s.source {
+                Some(DeserializedServiceSource::Image { image }) => image.clone(),
+                Some(DeserializedServiceSource::Repo { repo, .. }) => repo.clone(),
+                None => s.name.clone(),
+            },
+            variables: (!variables.is_empty()).then_some(variables),
+            volumes: (!volumes.is_empty()).then_some(volumes),
+        });
+    }
+
+    let vars = mutations::template_deploy::Variables {
+        project_id: linked_project.project.clone(),
+        environment_id: linked_project.environment.clone(),
+        services,
+        template_code: template.clone(),
+    };
+
+    let spinner = indicatif::ProgressBar::new_spinner()
+        .with_style(
+            indicatif::ProgressStyle::default_spinner()
+                .tick_chars(TICK_STRING)
+                .template("{spinner:.green} {msg}")?,
+        )
+        .with_message(format!("Creating {template}..."));
+    spinner.enable_steady_tick(Duration::from_millis(100));
+    post_graphql::<mutations::TemplateDeploy, _>(client, configs.get_backboard(), vars).await?;
+    spinner.finish_with_message(format!("Created {template}"));
+    Ok(())
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedServiceNetworking {
+    #[serde(default)]
+    service_domains: HashMap<String, serde_json::Value>,
+    #[serde(default)]
+    tcp_proxies: HashMap<String, serde_json::Value>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedServiceVolumeMount {
+    mount_path: String,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedServiceVariable {
+    #[serde(default)]
+    default_value: Option<String>,
+    #[serde(default)]
+    description: Option<String>,
+    #[serde(default)]
+    is_optional: Option<bool>,
+    // #[serde(default)]
+    // generator: Option<String>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedServiceDeploy {
+    healthcheck_path: Option<String>,
+    start_command: Option<String>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(untagged)]
+enum DeserializedServiceSource {
+    Image {
+        image: String,
+    },
+    #[serde(rename_all = "camelCase")]
+    Repo {
+        root_directory: Option<String>,
+        repo: String,
+        // branch: Option<String>,
+    },
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedService {
+    // #[serde(default)]
+    // build: serde_json::Value,
+    #[serde(default)]
+    deploy: Option<DeserializedServiceDeploy>,
+
+    #[serde(default)]
+    icon: Option<String>,
+    name: String,
+
+    #[serde(default)]
+    networking: Option<DeserializedServiceNetworking>,
+
+    #[serde(default)]
+    source: Option<DeserializedServiceSource>,
+
+    #[serde(default)]
+    variables: HashMap<String, DeserializedServiceVariable>,
+    #[serde(default)]
+    volume_mounts: HashMap<String, DeserializedServiceVolumeMount>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DeserializedEnvironment {
+    #[serde(default)]
+    services: HashMap<String, DeserializedService>,
+}

src/commands/mod.rs

@@ -6,6 +6,7 @@ pub(super) use colored::Colorize;
 pub mod add;
 pub mod completion;
 pub mod connect;
+pub mod deploy;
 pub mod docs;
 pub mod domain;
 pub mod down;

src/gql/queries/mod.rs

@@ -139,6 +139,8 @@ pub struct Domains;
 )]
 pub struct ProjectToken;
 
+pub type SerializedTemplateConfig = serde_json::Value;
+
 #[derive(GraphQLQuery)]
 #[graphql(
     schema_path = "src/gql/schema.graphql",

src/gql/queries/strings/TemplateDetail.graphql

@@ -1,5 +1,7 @@
 query TemplateDetail($code: String!) {
   template(code: $code) {
+    id
+    serializedConfig
     services {
       edges {
         node {
@@ -9,4 +11,4 @@ query TemplateDetail($code: String!) {
       }
     }
   }
-}
+}

src/main.rs

@@ -36,6 +36,7 @@ commands_enum!(
     add,
     completion,
     connect,
+    deploy,
     domain,
     docs,
     down,