railwayapp 3.13.0

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::*;

const GITHUB_BASE_URL: &str = "https://github.com/";

/// 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().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`
pub 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: 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(),
                // The `repo` is in the `${owner}/${repo}` format so we need to add the prefix
                Some(DeserializedServiceSource::Repo { repo, .. }) => {
                    format!("{}{}", GITHUB_BASE_URL, repo)
                }
                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}"));

    let hostname = configs.get_host();
    let url = format!("https://{hostname}/project/{}", linked_project.project);
    println!("url: {}", url);

    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>,
}