Command trigger with arguments

Signed-off-by: Radu Matei <[email protected]>

containerd-shim-spin/Cargo.toml

@@ -23,6 +23,7 @@ spin-trigger-http = { git = "https://github.com/fermyon/spin", tag = "v2.4.0" }
 spin-trigger-redis = { git = "https://github.com/fermyon/spin", tag = "v2.4.0" }
 # TODO: change to the original repo once https://github.com/fermyon/spin-trigger-sqs/pull/19 is merged.
 trigger-sqs = { git = "https://github.com/radu-matei/spin-trigger-sqs", rev = "8e173e5c7023415cd6b29d1f0a37296d2861d5a2" }
+trigger-command = { git = "https://github.com/fermyon/spin-trigger-command" , rev = "47ab3d6da7c577cac722b300b8db29c85f80a80f" }
 spin-manifest = { git = "https://github.com/fermyon/spin", tag = "v2.4.0" }
 spin-loader = { git = "https://github.com/fermyon/spin", tag = "v2.4.0" }
 spin-oci = { git = "https://github.com/fermyon/spin", tag = "v2.4.0" }

containerd-shim-spin/src/engine.rs

@@ -21,6 +21,7 @@ use std::net::SocketAddr;
 use std::net::ToSocketAddrs;
 use std::path::{Path, PathBuf};
 use tokio::runtime::Runtime;
+use trigger_command::CommandTrigger;
 use trigger_sqs::SqsTrigger;
 use url::Url;
 
@@ -190,11 +191,13 @@ impl SpinEngine {
         let trigger_cmd = trigger_command_for_resolved_app_source(&resolved_app_source)
             .with_context(|| format!("Couldn't find trigger executor for {app_source:?}"))?;
         let locked_app = self.load_resolved_app_source(resolved_app_source).await?;
-        self.run_trigger(&trigger_cmd, locked_app, app_source).await
+        self.run_trigger(ctx, &trigger_cmd, locked_app, app_source)
+            .await
     }
 
     async fn run_trigger(
         &self,
+        ctx: &impl RuntimeContext,
         trigger_type: &str,
         app: LockedApp,
         app_source: AppSource,
@@ -232,6 +235,17 @@ impl SpinEngine {
                 info!(" >>> running spin trigger");
                 sqs_trigger.run(spin_trigger::cli::NoArgs)
             }
+            CommandTrigger::TRIGGER_TYPE => {
+                let command_trigger: CommandTrigger = self
+                    .build_spin_trigger(working_dir, app, app_source)
+                    .await
+                    .context("failed to build spin trigger")?;
+
+                info!(" >>> running spin trigger");
+                command_trigger.run(trigger_command::CliArgs {
+                    guest_args: ctx.args().to_vec(),
+                })
+            }
             _ => {
                 todo!("Only Http, Redis and SQS triggers are currently supported.")
             }
@@ -442,9 +456,10 @@ fn trigger_command_for_resolved_app_source(resolved: &ResolvedAppSource) -> Resu
     let trigger_type = resolved.trigger_type()?;
 
     match trigger_type {
-        RedisTrigger::TRIGGER_TYPE | HttpTrigger::TRIGGER_TYPE | SqsTrigger::TRIGGER_TYPE => {
-            Ok(trigger_type.to_owned())
-        }
+        RedisTrigger::TRIGGER_TYPE
+        | HttpTrigger::TRIGGER_TYPE
+        | SqsTrigger::TRIGGER_TYPE
+        | CommandTrigger::TRIGGER_TYPE => Ok(trigger_type.to_owned()),
         _ => {
             todo!("Only Http, Redis and SQS triggers are currently supported.")
         }