Reimplement initial kv handling

Signed-off-by: Ryan Levick <ryan.levick@fermyon.com>

Author: rylev

Diff for: crates/factor-key-value/src/lib.rs

@@ -133,6 +133,11 @@ impl AppState {
             .values()
             .any(|stores| stores.contains(label))
     }
+
+    /// Get a store by label.
+    pub async fn get_store(&self, label: &str) -> Option<Arc<dyn Store>> {
+        self.store_manager.get(label).await.ok()
+    }
 }
 
 pub struct InstanceBuilder {

Diff for: crates/trigger/src/cli.rs

@@ -1,3 +1,4 @@
+mod initial_kv_setter;
 mod launch_metadata;
 mod sqlite_statements;
 mod stdio;
@@ -17,6 +18,7 @@ use spin_factors::RuntimeFactors;
 use spin_factors_executor::{ComponentLoader, FactorsExecutor};
 
 use crate::{Trigger, TriggerApp};
+pub use initial_kv_setter::InitialKvSetterHook;
 pub use launch_metadata::LaunchMetadata;
 pub use sqlite_statements::SqlStatementExecutorHook;
 use stdio::FollowComponents;

Diff for: crates/trigger/src/cli/initial_kv_setter.rs

@@ -0,0 +1,41 @@
+use anyhow::Context as _;
+use spin_core::async_trait;
+use spin_factor_key_value::KeyValueFactor;
+use spin_factors::RuntimeFactors;
+use spin_factors_executor::ExecutorHooks;
+
+pub struct InitialKvSetterHook {
+    kv_pairs: Vec<(String, String)>,
+}
+
+impl InitialKvSetterHook {
+    pub fn new(kv_pairs: Vec<(String, String)>) -> Self {
+        Self { kv_pairs }
+    }
+}
+
+const DEFAULT_KEY_VALUE_STORE_LABEL: &str = "default";
+
+#[async_trait]
+impl<F: RuntimeFactors, U> ExecutorHooks<F, U> for InitialKvSetterHook {
+    async fn configure_app(
+        &mut self,
+        configured_app: &spin_factors::ConfiguredApp<F>,
+    ) -> anyhow::Result<()> {
+        let Some(kv) = configured_app.app_state::<KeyValueFactor>().ok() else {
+            return Ok(());
+        };
+        let store = kv
+            .get_store(DEFAULT_KEY_VALUE_STORE_LABEL)
+            .await
+            .expect("trigger was misconfigured and lacks a default store");
+        for (key, value) in &self.kv_pairs {
+            store
+                .set(key, value.as_bytes())
+                .await
+                .context("failed to set key-value pair")?;
+        }
+
+        Ok(())
+    }
+}

Diff for: examples/spin-timer/src/main.rs

@@ -1,6 +1,7 @@
 use anyhow::Error;
-use spin_trigger::cli::FactorsTriggerCommand;
+use clap::Parser;
 use spin_cli::runtime_factors::FactorsBuilder;
+use spin_trigger::cli::FactorsTriggerCommand;
 
 use trigger_timer::TimerTrigger;
 

Diff for: src/runtime_factors.rs

@@ -5,8 +5,8 @@ use spin_factors_executor::FactorsExecutor;
 use spin_runtime_config::ResolvedRuntimeConfig;
 use spin_runtime_factors::{TriggerAppOptions, TriggerFactors, TriggerFactorsRuntimeConfig};
 use spin_trigger::cli::{
-    CommonTriggerOptions, KeyValueDefaultStoreSummaryHook, RuntimeFactorsBuilder,
-    SqlStatementExecutorHook, StdioLoggingExecutorHooks,
+    CommonTriggerOptions, InitialKvSetterHook, KeyValueDefaultStoreSummaryHook,
+    RuntimeFactorsBuilder, SqlStatementExecutorHook, StdioLoggingExecutorHooks,
 };
 
 pub struct FactorsBuilder;
@@ -56,10 +56,7 @@ impl RuntimeFactorsBuilder for FactorsBuilder {
             runtime_config.log_dir(),
         ));
         executor.add_hooks(KeyValueDefaultStoreSummaryHook);
-        // TODO: implement initial key values as a hook
-        // runtime_config
-        //     .set_initial_key_values(&options.initial_key_values)
-        //     .await?;
+        executor.add_hooks(InitialKvSetterHook::new(options.key_values.clone()));
         // builder.hooks(SummariseRuntimeConfigHook::new(&self.runtime_config_file));
         // builder.hooks(SqlitePersistenceMessageHook);
         Ok(())