Skip to main content

stackless_integrations/providers/flyio/
mpg.rs

1//! `flyio/mpg` integration.
2
3use std::collections::BTreeMap;
4
5use serde::Serialize;
6use stackless_stripe_projects::catalog::verify::CatalogService;
7use stackless_stripe_projects::provision::ProvisionContext;
8
9use super::FamilyResource;
10use crate::error::IntegrationError;
11use crate::hostable::{ConfigScope, Hostable, IntegrationHosting};
12use crate::registry;
13
14pub const RESOURCE_KIND: &str = "integration-flyio-mpg";
15
16#[derive(Debug, Serialize)]
17pub struct FlyioMpgConfig {
18    pub name: String,
19    pub plan: String,
20    pub region: String,
21}
22
23impl CatalogService for FlyioMpgConfig {
24    const REFERENCE: &'static str = "flyio/mpg";
25}
26
27#[derive(Debug)]
28pub struct FlyioMpg;
29
30impl Hostable for FlyioMpg {
31    const PROVIDER: &'static str = "flyio-mpg";
32    const HOSTING: IntegrationHosting = IntegrationHosting::Managed;
33    const CONFIG_SCOPE: ConfigScope = ConfigScope::GlobalOnly;
34    const RESOURCE_KIND: &'static str = RESOURCE_KIND;
35    const OUTPUTS: &'static [&'static str] = &["database_url"];
36}
37
38impl FamilyResource for FlyioMpg {
39    type Config = FlyioMpgConfig;
40    const PROVIDER_PREFIX: &'static str = "FLYIO";
41    // Provisional until pinned by `mise run discover flyio/mpg`.
42    const OUTPUT_FIELDS: &'static [(&'static str, &'static str, bool)] =
43        &[("DATABASE_URL", "database_url", true)];
44
45    fn build_config(ctx: &ProvisionContext<'_>) -> Result<FlyioMpgConfig, IntegrationError> {
46        let config = super::integration_config(ctx)?;
47        Ok(FlyioMpgConfig {
48            name: super::interp_required(ctx, &config, "name")?,
49            plan: super::interp_required(ctx, &config, "plan")?,
50            region: super::interp_required(ctx, &config, "region")?,
51        })
52    }
53}
54
55pub fn validate_config(
56    name: &str,
57    config: &BTreeMap<String, toml::Value>,
58) -> Result<(), IntegrationError> {
59    registry::config_string(config, "name").map_err(|err| IntegrationError::ConfigInvalid {
60        location: format!("integrations.{name}.name"),
61        detail: err.to_string(),
62    })?;
63    registry::config_string(config, "plan").map_err(|err| IntegrationError::ConfigInvalid {
64        location: format!("integrations.{name}.plan"),
65        detail: err.to_string(),
66    })?;
67    registry::config_string(config, "region").map_err(|err| IntegrationError::ConfigInvalid {
68        location: format!("integrations.{name}.region"),
69        detail: err.to_string(),
70    })?;
71    let _ = (name, config);
72    Ok(())
73}
74
75#[cfg(test)]
76mod tests {
77    use super::*;
78    use crate::ProviderOps;
79    use crate::resource::ResourcePayload;
80    use stackless_core::def::StackDef;
81    use stackless_stripe_projects::stripe::StripeProjects;
82    use stackless_stripe_projects::test_support;
83
84    #[test]
85    fn config_matches_catalog() {
86        const FIXTURE: &str = include_str!(concat!(
87            env!("CARGO_MANIFEST_DIR"),
88            "/../stackless-stripe-projects/tests/fixtures/catalog.json"
89        ));
90        let catalog = stackless_stripe_projects::Catalog::from_json_envelope(FIXTURE).unwrap();
91        let failures = stackless_stripe_projects::verify_service(
92            &catalog,
93            &FlyioMpgConfig {
94                name: "test-name".into(),
95                plan: "basic".into(),
96                region: "ams".into(),
97            },
98        );
99        assert!(
100            failures.is_empty(),
101            "flyio/mpg catalog gaps:\n{}",
102            failures.join("\n")
103        );
104    }
105
106    const CATALOG_ENVELOPE: &str = r##"{"ok":true,"command":"projects catalog","data":{"last_updated":"2026-07-11T00:00:00Z","services":[{"id":"prvsvc_mpg","object":"v2.provisioning.provider_service_detail","provider_id":"prvdr_flyio","provider_name":"Flyio","service_id":"mpg","categories":["database"],"kind":"deployable","scope":"project","availability":"available","development":false,"livemode":true,"pricing":{"type":"paid"},"configuration_schema":{"properties":{"name":{"description":"Cluster name","maxLength":63,"minLength":5,"pattern":"^[a-z0-9][a-z0-9-]*[a-z0-9]$","type":"string"},"plan":{"enum":["basic","starter","launch","scale","Performance"],"type":"string"},"region":{"enum":["ams","fra","gru","iad","lax","lhr","nrt","ord","sin","sjc","syd","yyz"],"type":"string"}},"required":["name","region","plan"],"type":"object"}}]}}"##;
107
108    fn test_def() -> StackDef {
109        StackDef::parse(
110            r#"
111[stack]
112name = "atto"
113[stack.projects.stripe]
114project = "project_1"
115[integrations.res]
116provider = "flyio-mpg"
117name = "test-name"
118plan = "basic"
119region = "ams"
120[services.api]
121source = { repo = "r", ref = "main" }
122env = { OUT = "${integrations.res.database_url}" }
123health = { path = "/health" }
124[services.api.local]
125run = "true"
126"#,
127        )
128        .unwrap()
129    }
130
131    #[tokio::test]
132    async fn provision_records_outputs() {
133        let runner = test_support::provision_script(
134            CATALOG_ENVELOPE,
135            serde_json::json!({"FLYIO_DATABASE_URL": "val_database_url"}),
136            0,
137        );
138        let dir = tempfile::tempdir().unwrap();
139        std::fs::write(
140            dir.path().join("stackless.toml"),
141            "[stack]\nname=\"atto\"\n",
142        )
143        .unwrap();
144        let stripe = StripeProjects::new(&runner, dir.path());
145
146        let resource = FlyioMpg
147            .provision(
148                &stripe.as_dyn(),
149                &test_def(),
150                dir.path(),
151                "demo",
152                "res",
153                "local",
154                false,
155            )
156            .await
157            .unwrap();
158        assert_eq!(resource.resource_kind, "integration-flyio-mpg");
159        let payload: ResourcePayload = serde_json::from_str(&resource.payload).unwrap();
160        assert_eq!(payload.outputs["database_url"], "val_database_url");
161    }
162}