Merge pull request #2453 from itowlson/summarise-runtime-config

Summarise runtime config

Author: itowlson

crates/trigger/src/cli.rs

@@ -9,6 +9,7 @@ use spin_common::{arg_parser::parse_kv, sloth};
 use crate::network::Network;
 use crate::runtime_config::llm::LLmOptions;
 use crate::runtime_config::sqlite::SqlitePersistenceMessageHook;
+use crate::runtime_config::SummariseRuntimeConfigHook;
 use crate::stdio::StdioLoggingTriggerHooks;
 use crate::{
     loader::TriggerLoader,
@@ -208,6 +209,7 @@ where
 
         builder.hooks(StdioLoggingTriggerHooks::new(self.follow_components()));
         builder.hooks(Network::default());
+        builder.hooks(SummariseRuntimeConfigHook::new(&self.runtime_config_file));
         builder.hooks(KeyValuePersistenceMessageHook);
         builder.hooks(SqlitePersistenceMessageHook);
 

crates/trigger/src/runtime_config.rs

@@ -15,6 +15,8 @@ use serde::Deserialize;
 use spin_common::ui::quoted_path;
 use spin_sqlite::Connection;
 
+use crate::TriggerHooks;
+
 use self::{
     key_value::{KeyValueStore, KeyValueStoreOpts},
     llm::LlmComputeOpts,
@@ -259,6 +261,77 @@ fn resolve_config_path(path: &Path, config_opts: &RuntimeConfigOpts) -> Result<P
     Ok(base_path.join(path))
 }
 
+pub(crate) struct SummariseRuntimeConfigHook {
+    runtime_config_file: Option<PathBuf>,
+}
+
+impl SummariseRuntimeConfigHook {
+    pub(crate) fn new(runtime_config_file: &Option<PathBuf>) -> Self {
+        Self {
+            runtime_config_file: runtime_config_file.clone(),
+        }
+    }
+}
+
+impl TriggerHooks for SummariseRuntimeConfigHook {
+    fn app_loaded(
+        &mut self,
+        _app: &spin_app::App,
+        runtime_config: &RuntimeConfig,
+        _resolver: &Arc<spin_expressions::PreparedResolver>,
+    ) -> anyhow::Result<()> {
+        if let Some(path) = &self.runtime_config_file {
+            let mut opts = vec![];
+            for opt in runtime_config.opts_layers() {
+                for (id, opt) in &opt.key_value_stores {
+                    opts.push(Self::summarise_kv(id, opt));
+                }
+                for (id, opt) in &opt.sqlite_databases {
+                    opts.push(Self::summarise_sqlite(id, opt));
+                }
+                for opt in &opt.llm_compute {
+                    opts.push(Self::summarise_llm(opt));
+                }
+            }
+            if !opts.is_empty() {
+                let opts_text = opts.join(", ");
+                println!(
+                    "Using {opts_text} runtime config from {}",
+                    quoted_path(path)
+                );
+            }
+        }
+        Ok(())
+    }
+}
+
+impl SummariseRuntimeConfigHook {
+    fn summarise_kv(id: &str, opt: &KeyValueStoreOpts) -> String {
+        let source = match opt {
+            KeyValueStoreOpts::Spin(_) => "spin",
+            KeyValueStoreOpts::Redis(_) => "redis",
+            KeyValueStoreOpts::AzureCosmos(_) => "cosmos",
+        };
+        format!("[key_value_store.{id}: {}]", source)
+    }
+
+    fn summarise_sqlite(id: &str, opt: &SqliteDatabaseOpts) -> String {
+        let source = match opt {
+            SqliteDatabaseOpts::Spin(_) => "spin",
+            SqliteDatabaseOpts::Libsql(_) => "libsql",
+        };
+        format!("[sqlite_database.{id}: {}]", source)
+    }
+
+    fn summarise_llm(opt: &LlmComputeOpts) -> String {
+        let source = match opt {
+            LlmComputeOpts::Spin => "spin",
+            LlmComputeOpts::RemoteHttp(_) => "remote-http",
+        };
+        format!("[llm_compute: {}]", source)
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use std::io::Write;