Implement wasi-config

Signed-off-by: itowlson <[email protected]>

Author: itowlson

Cargo.lock

@@ -7,6 +7,7 @@ edition = { workspace = true }
 [dependencies]
 anyhow = { workspace = true }
 async-trait = { workspace = true }
+futures = { workspace = true }
 serde = { workspace = true }
 spin-locked-app = { path = "../locked-app" }
 thiserror = { workspace = true }

crates/expressions/Cargo.toml

@@ -52,6 +52,21 @@ impl ProviderResolver {
         self.resolve_template(template).await
     }
 
+    pub async fn resolve_all(&self, component_id: &str) -> Result<Vec<(String, String)>> {
+        use futures::FutureExt;
+
+        let Some(keys2templates) = self.internal.component_configs.get(component_id) else {
+            return Ok(vec![]);
+        };
+
+        let resolve_futs = keys2templates.iter().map(|(key, template)| {
+            self.resolve_template(template)
+                .map(|r| r.map(|value| (key.to_string(), value)))
+        });
+
+        futures::future::try_join_all(resolve_futs).await
+    }
+
     /// Resolves the given template.
     pub async fn resolve_template(&self, template: &Template) -> Result<String> {
         let mut resolved_parts: Vec<Cow<str>> = Vec::with_capacity(template.parts().len());

crates/expressions/src/lib.rs

@@ -37,6 +37,61 @@ impl v1::config::Host for InstanceState {
     }
 }
 
+#[async_trait]
+impl spin_world::wasi::config::store::Host for InstanceState {
+    async fn get(
+        &mut self,
+        key: String,
+    ) -> Result<Option<String>, spin_world::wasi::config::store::Error> {
+        match <Self as variables::Host>::get(self, key).await {
+            Ok(value) => Ok(Some(value)),
+            // Err(e) if matches!(e, variables::Error::Undefined(_)) => Ok(Ok(None)),
+            Err(e) => {
+                match e {
+                    variables::Error::Undefined(_) => Ok(None),
+                    variables::Error::InvalidName(msg) => {
+                        Err(spin_world::wasi::config::store::Error::Io(msg))
+                    } // TODO: this doesn't feel ideal, but wasi-config only allows two error cases
+                    variables::Error::Provider(msg) => {
+                        Err(spin_world::wasi::config::store::Error::Upstream(msg))
+                    }
+                    variables::Error::Other(msg) => {
+                        Err(spin_world::wasi::config::store::Error::Io(msg))
+                    } // TODO: again, hard to know how to map this (and the original expressions::Error is not more helpful)
+                }
+            }
+        }
+    }
+
+    async fn get_all(
+        &mut self,
+    ) -> Result<Vec<(String, String)>, spin_world::wasi::config::store::Error> {
+        let all = self
+            .expression_resolver
+            .resolve_all(&self.component_id)
+            .await;
+        all.map_err(|e| {
+            match expressions_to_variables_err(e) {
+                variables::Error::Undefined(msg) => spin_world::wasi::config::store::Error::Io(msg), // TODO: this should presumably not occur here
+                variables::Error::InvalidName(msg) => {
+                    spin_world::wasi::config::store::Error::Io(msg)
+                } // TODO: this doesn't feel ideal, but wasi-config only allows two error cases
+                variables::Error::Provider(msg) => {
+                    spin_world::wasi::config::store::Error::Upstream(msg)
+                }
+                variables::Error::Other(msg) => spin_world::wasi::config::store::Error::Io(msg), // TODO: again, hard to know how to map this (and the original expressions::Error is not more helpful)
+            }
+        })
+    }
+
+    fn convert_error(
+        &mut self,
+        err: spin_world::wasi::config::store::Error,
+    ) -> anyhow::Result<spin_world::wasi::config::store::Error> {
+        Ok(err)
+    }
+}
+
 fn expressions_to_variables_err(err: spin_expressions::Error) -> variables::Error {
     use spin_expressions::Error;
     match err {

crates/factor-variables/src/host.rs

@@ -31,6 +31,7 @@ impl Factor for VariablesFactor {
     fn init<T: Send + 'static>(&mut self, mut ctx: InitContext<T, Self>) -> anyhow::Result<()> {
         ctx.link_bindings(spin_world::v1::config::add_to_linker)?;
         ctx.link_bindings(spin_world::v2::variables::add_to_linker)?;
+        ctx.link_bindings(spin_world::wasi::config::store::add_to_linker)?;
         Ok(())
     }
 

crates/factor-variables/src/lib.rs

@@ -27,6 +27,7 @@ wasmtime::component::bindgen!({
         "fermyon:spin/[email protected]/error" => v2::sqlite::Error,
         "fermyon:spin/sqlite/error" => v1::sqlite::Error,
         "fermyon:spin/[email protected]/error" => v2::variables::Error,
+        "wasi:config/store/error" => wasi::config::store::Error,
     },
     trappable_imports: true,
 });

crates/world/src/lib.rs

@@ -0,0 +1,17 @@
+spin_manifest_version = "1"
+authors = [""]
+description = ""
+name = "variables"
+trigger = { type = "http" }
+version = "0.1.0"
+
+[variables]
+variable = { default = "value" }
+
+[[component]]
+id = "variables"
+source = "%{source=variables}"
+[component.trigger]
+route = "/..."
+[component.config]
+variable = "{{ variable }}"

tests/runtime-tests/tests/variables/spin.toml

@@ -0,0 +1,17 @@
+spin_manifest_version = "1"
+authors = [""]
+description = ""
+name = "wasi-config"
+trigger = { type = "http" }
+version = "0.1.0"
+
+[variables]
+variable = { default = "value" }
+
+[[component]]
+id = "wasi-config"
+source = "%{source=wasi-config}"
+[component.trigger]
+route = "/..."
+[component.config]
+variable = "{{ variable }}"

tests/runtime-tests/tests/wasi-config/spin.toml

@@ -0,0 +1,11 @@
+[package]
+name = "wasi-config"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+helper = { path = "../../helper" }
+wit-bindgen = "0.16.0"