|
1 | | -use super::ConfigUpdate; |
2 | | -use crate::config; |
3 | | -use crate::git; |
| 1 | +use super::{ConfigUpdate, GetRunnerConfig}; |
| 2 | +use crate::config::{self, RunnerConfig}; |
| 3 | +use crate::{env, git, webhook}; |
| 4 | +use std::collections::HashMap; |
4 | 5 | use std::path::PathBuf; |
5 | 6 | use std::time::Duration; |
6 | 7 | use tokio::time::interval; |
7 | | -use tracing::{error, info}; |
| 8 | +use tracing::{error, info, warn}; |
8 | 9 | use xtra::prelude::*; |
9 | 10 | use xtra::refcount::Weak; |
10 | 11 |
|
11 | 12 | const CONFIG_FILE: &str = "rollcron.yaml"; |
12 | 13 |
|
13 | 14 | pub async fn run<A>(source: String, pull_interval: Duration, addr: Address<A, Weak>) |
14 | 15 | where |
15 | | - A: Handler<ConfigUpdate>, |
| 16 | + A: Handler<ConfigUpdate> + Handler<GetRunnerConfig, Return = RunnerConfig>, |
16 | 17 | { |
17 | 18 | let mut ticker = interval(pull_interval); |
18 | 19 |
|
@@ -48,11 +49,64 @@ where |
48 | 49 | } |
49 | 50 | Err(e) => { |
50 | 51 | error!(target: "rollcron::runner", error = %e, "Failed to reload config"); |
| 52 | + notify_config_error(&addr, &sot_path, &e.to_string()).await; |
51 | 53 | } |
52 | 54 | } |
53 | 55 | } |
54 | 56 | } |
55 | 57 |
|
| 58 | +async fn notify_config_error<A>(addr: &Address<A, Weak>, sot_path: &PathBuf, error: &str) |
| 59 | +where |
| 60 | + A: Handler<GetRunnerConfig, Return = RunnerConfig>, |
| 61 | +{ |
| 62 | + let runner = match addr.send(GetRunnerConfig).await { |
| 63 | + Ok(r) => r, |
| 64 | + Err(_) => return, // Runner stopped |
| 65 | + }; |
| 66 | + |
| 67 | + if runner.webhook.is_empty() { |
| 68 | + return; |
| 69 | + } |
| 70 | + |
| 71 | + let runner_env = load_runner_env(sot_path, &runner); |
| 72 | + for wh in &runner.webhook { |
| 73 | + let url = wh.to_url(runner_env.as_ref()); |
| 74 | + if url.contains('$') { |
| 75 | + warn!(target: "rollcron::webhook", url = %url, "Webhook URL contains unexpanded variable, skipping"); |
| 76 | + continue; |
| 77 | + } |
| 78 | + if !url.starts_with("http://") && !url.starts_with("https://") { |
| 79 | + warn!(target: "rollcron::webhook", url = %url, "Webhook URL must start with http:// or https://, skipping"); |
| 80 | + continue; |
| 81 | + } |
| 82 | + webhook::send_config_error(&url, error).await; |
| 83 | + } |
| 84 | +} |
| 85 | + |
| 86 | +fn load_runner_env(sot_path: &PathBuf, runner: &RunnerConfig) -> Option<HashMap<String, String>> { |
| 87 | + let mut env_vars = HashMap::new(); |
| 88 | + |
| 89 | + if let Some(env_file_path) = &runner.env_file { |
| 90 | + let expanded = env::expand_string(env_file_path); |
| 91 | + let full_path = sot_path.join(&expanded); |
| 92 | + match env::load_env_from_path(&full_path) { |
| 93 | + Ok(vars) => env_vars.extend(vars), |
| 94 | + Err(e) => { |
| 95 | + warn!(target: "rollcron::webhook", error = %e, "Failed to load runner env_file"); |
| 96 | + return None; |
| 97 | + } |
| 98 | + } |
| 99 | + } |
| 100 | + |
| 101 | + if let Some(runner_env) = &runner.env { |
| 102 | + for (k, v) in runner_env { |
| 103 | + env_vars.insert(k.clone(), env::expand_string(v)); |
| 104 | + } |
| 105 | + } |
| 106 | + |
| 107 | + Some(env_vars) |
| 108 | +} |
| 109 | + |
56 | 110 | fn load_config(sot_path: &PathBuf) -> anyhow::Result<(config::RunnerConfig, Vec<config::Job>)> { |
57 | 111 | let config_path = sot_path.join(CONFIG_FILE); |
58 | 112 | let content = std::fs::read_to_string(&config_path) |
|
0 commit comments