feat: 支持自定义 webhook 模板,支持发送测试信息 (#551)

This commit is contained in:
ᴀᴍᴛᴏᴀᴇʀ
2025-12-05 00:21:36 +08:00
committed by GitHub
parent e76673d076
commit 7ef38a38ed
7 changed files with 114 additions and 13 deletions

View File

@@ -1,17 +1,35 @@
use anyhow::Result;
use futures::future;
use reqwest::header;
use serde::{Deserialize, Serialize};
use crate::config::TEMPLATE;
#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "camelCase", tag = "type")]
pub enum Notifier {
Telegram { bot_token: String, chat_id: String },
Webhook { url: String },
Telegram {
bot_token: String,
chat_id: String,
},
Webhook {
url: String,
template: Option<String>,
#[serde(skip)]
// 一个内部辅助字段,用于决定是否强制渲染当前模板,在测试时使用
ignore_cache: Option<()>,
},
}
#[derive(Serialize)]
struct WebhookPayload<'a> {
text: &'a str,
pub fn webhook_template_key(url: &str) -> String {
format!("payload_{}", url)
}
pub fn webhook_template_content(template: &Option<String>) -> &str {
template
.as_deref()
.filter(|t| !t.trim().is_empty())
.unwrap_or(r#"{"text": "{{{message}}}"}"#)
}
pub trait NotifierAllExt {
@@ -33,9 +51,28 @@ impl Notifier {
let params = [("chat_id", chat_id.as_str()), ("text", message)];
client.post(&url).form(&params).send().await?;
}
Notifier::Webhook { url } => {
let payload = WebhookPayload { text: message };
client.post(url).json(&payload).send().await?;
Notifier::Webhook {
url,
template,
ignore_cache,
} => {
let key = webhook_template_key(url);
let data = serde_json::json!(
{
"message": message,
}
);
let handlebar = TEMPLATE.read();
let payload = match ignore_cache {
Some(_) => handlebar.render_template(webhook_template_content(template), &data)?,
None => handlebar.render(&key, &data)?,
};
client
.post(url)
.header(header::CONTENT_TYPE, "application/json")
.body(payload)
.send()
.await?;
}
}
Ok(())