stackless_integrations/providers/upstash/
search.rs1use 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] = &["rest_url", "rest_token"];
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)] = &[
45 ("REST_URL", "rest_url", true),
46 ("REST_TOKEN", "rest_token", true),
47 ];
48
49 fn build_config(ctx: &ProvisionContext<'_>) -> Result<UpstashSearchConfig, IntegrationError> {
50 let config = super::integration_config(ctx)?;
51 Ok(UpstashSearchConfig {
52 name: super::interp_optional(ctx, &config, "name")?,
53 price: super::interp_optional(ctx, &config, "price")?,
54 region: super::interp_optional(ctx, &config, "region")?,
55 })
56 }
57}
58
59pub fn validate_config(
60 name: &str,
61 config: &BTreeMap<String, toml::Value>,
62) -> Result<(), IntegrationError> {
63 let _ = (name, config);
64 Ok(())
65}
66
67#[cfg(test)]
68mod tests {
69 use super::*;
70 use crate::ProviderOps;
71 use crate::resource::ResourcePayload;
72 use stackless_core::def::StackDef;
73 use stackless_stripe_projects::stripe::StripeProjects;
74 use stackless_stripe_projects::test_support;
75
76 #[test]
77 fn config_matches_catalog() {
78 const FIXTURE: &str = include_str!(concat!(
79 env!("CARGO_MANIFEST_DIR"),
80 "/../stackless-stripe-projects/tests/fixtures/catalog.json"
81 ));
82 let catalog = stackless_stripe_projects::Catalog::from_json_envelope(FIXTURE).unwrap();
83 let failures = stackless_stripe_projects::verify_service(
84 &catalog,
85 &UpstashSearchConfig {
86 name: None,
87 price: None,
88 region: None,
89 },
90 );
91 assert!(
92 failures.is_empty(),
93 "upstash/search catalog gaps:\n{}",
94 failures.join("\n")
95 );
96 }
97
98 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"}}]}}"##;
99
100 fn test_def() -> StackDef {
101 StackDef::parse(
102 r#"
103[stack]
104name = "atto"
105[stack.projects.stripe]
106project = "project_1"
107[integrations.res]
108provider = "upstash-search"
109[services.api]
110source = { repo = "r", ref = "main" }
111env = { OUT = "${integrations.res.rest_url}" }
112health = { path = "/health" }
113[services.api.local]
114run = "true"
115"#,
116 )
117 .unwrap()
118 }
119
120 #[tokio::test]
121 async fn provision_records_outputs() {
122 let runner = test_support::provision_script(
123 CATALOG_ENVELOPE,
124 serde_json::json!({"UPSTASH_REST_URL": "val_rest_url", "UPSTASH_REST_TOKEN": "val_rest_token"}),
125 0,
126 );
127 let dir = tempfile::tempdir().unwrap();
128 std::fs::write(
129 dir.path().join("stackless.toml"),
130 "[stack]\nname=\"atto\"\n",
131 )
132 .unwrap();
133 let stripe = StripeProjects::new(&runner, dir.path());
134
135 let resource = UpstashSearch
136 .provision(
137 &stripe.as_dyn(),
138 &test_def(),
139 dir.path(),
140 "demo",
141 "res",
142 "local",
143 false,
144 )
145 .await
146 .unwrap();
147 assert_eq!(resource.resource_kind, "integration-upstash-search");
148 let payload: ResourcePayload = serde_json::from_str(&resource.payload).unwrap();
149 assert_eq!(payload.outputs["rest_url"], "val_rest_url");
150 }
151}