Skip to main content

stackless_integrations/providers/upstash/
search.rs

1//! `upstash/search` 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};
12
13pub const RESOURCE_KIND: &str = "integration-upstash-search";
14
15#[derive(Debug, Serialize)]
16pub struct UpstashSearchConfig {
17    #[serde(skip_serializing_if = "Option::is_none")]
18    pub name: Option<String>,
19    #[serde(skip_serializing_if = "Option::is_none")]
20    pub price: Option<String>,
21    #[serde(skip_serializing_if = "Option::is_none")]
22    pub region: Option<String>,
23}
24
25impl CatalogService for UpstashSearchConfig {
26    const REFERENCE: &'static str = "upstash/search";
27}
28
29#[derive(Debug)]
30pub struct UpstashSearch;
31
32impl Hostable for UpstashSearch {
33    const PROVIDER: &'static str = "upstash-search";
34    const HOSTING: IntegrationHosting = IntegrationHosting::Managed;
35    const CONFIG_SCOPE: ConfigScope = ConfigScope::GlobalOnly;
36    const RESOURCE_KIND: &'static str = RESOURCE_KIND;
37    const OUTPUTS: &'static [&'static str] = &["token", "url"];
38}
39
40impl FamilyResource for UpstashSearch {
41    type Config = UpstashSearchConfig;
42    const PROVIDER_PREFIX: &'static str = "UPSTASH";
43    const OUTPUT_FIELDS: &'static [(&'static str, &'static str, bool)] =
44        &[("TOKEN", "token", true), ("URL", "url", true)];
45
46    fn build_config(ctx: &ProvisionContext<'_>) -> Result<UpstashSearchConfig, IntegrationError> {
47        let config = super::integration_config(ctx)?;
48        Ok(UpstashSearchConfig {
49            name: super::interp_optional(ctx, &config, "name")?,
50            price: super::interp_optional(ctx, &config, "price")?,
51            region: super::interp_optional(ctx, &config, "region")?,
52        })
53    }
54}
55
56pub fn validate_config(
57    name: &str,
58    config: &BTreeMap<String, toml::Value>,
59) -> Result<(), IntegrationError> {
60    let _ = (name, config);
61    Ok(())
62}
63
64#[cfg(test)]
65mod tests {
66    use super::*;
67    use crate::ProviderOps;
68    use crate::resource::ResourcePayload;
69    use stackless_core::def::StackDef;
70    use stackless_stripe_projects::stripe::StripeProjects;
71    use stackless_stripe_projects::test_support;
72
73    #[test]
74    fn config_matches_catalog() {
75        const FIXTURE: &str = include_str!(concat!(
76            env!("CARGO_MANIFEST_DIR"),
77            "/../stackless-stripe-projects/tests/fixtures/catalog.json"
78        ));
79        let catalog = stackless_stripe_projects::Catalog::from_json_envelope(FIXTURE).unwrap();
80        let failures = stackless_stripe_projects::verify_service(
81            &catalog,
82            &UpstashSearchConfig {
83                name: None,
84                price: None,
85                region: None,
86            },
87        );
88        assert!(
89            failures.is_empty(),
90            "upstash/search catalog gaps:\n{}",
91            failures.join("\n")
92        );
93    }
94
95    const CATALOG_ENVELOPE: &str = r##"{"ok":true,"command":"projects catalog","data":{"last_updated":"2026-07-11T00:00:00Z","services":[{"id":"prvsvc_search","object":"v2.provisioning.provider_service_detail","provider_id":"prvdr_upstash","provider_name":"Upstash","service_id":"search","categories":["database"],"kind":"deployable","scope":"project","availability":"available","development":false,"livemode":true,"pricing":{"type":"paid"},"configuration_schema":{"properties":{"name":{"description":"Search index name","minLength":1,"type":"string"},"price":{"default":"payg","description":"Pricing tier. 'free' provisions on the free plan (subject to free-tier limits and one-per-account). 'payg' provisions pay-as-you-go. Defaults to 'payg'.","enum":["free","payg"],"type":"string"},"region":{"description":"Region for the search index","enum":["eu-west-1","us-central1"],"type":"string"}},"type":"object"}}]}}"##;
96
97    fn test_def() -> StackDef {
98        StackDef::parse(
99            r#"
100[stack]
101name = "atto"
102[stack.projects.stripe]
103project = "project_1"
104[integrations.res]
105provider = "upstash-search"
106[services.api]
107source = { repo = "r", ref = "main" }
108env = { OUT = "${integrations.res.url}" }
109health = { path = "/health" }
110[services.api.local]
111run = "true"
112"#,
113        )
114        .unwrap()
115    }
116
117    #[tokio::test]
118    async fn provision_records_outputs() {
119        let runner = test_support::provision_script(
120            CATALOG_ENVELOPE,
121            serde_json::json!({"UPSTASH_TOKEN": "val_token", "UPSTASH_URL": "val_url"}),
122            0,
123        );
124        let dir = tempfile::tempdir().unwrap();
125        std::fs::write(
126            dir.path().join("stackless.toml"),
127            "[stack]\nname=\"atto\"\n",
128        )
129        .unwrap();
130        let stripe = StripeProjects::new(&runner, dir.path());
131
132        let resource = UpstashSearch
133            .provision(
134                &stripe.as_dyn(),
135                &test_def(),
136                dir.path(),
137                "demo",
138                "res",
139                "local",
140                false,
141            )
142            .await
143            .unwrap();
144        assert_eq!(resource.resource_kind, "integration-upstash-search");
145        let payload: ResourcePayload = serde_json::from_str(&resource.payload).unwrap();
146        assert_eq!(payload.outputs["url"], "val_url");
147    }
148}