feat: add webhook server with payload parsing and chat/stream event handling

Made-with: Cursor
This commit is contained in:
cottongin
2026-03-10 21:56:12 -04:00
parent 09fbf76873
commit c41eba1040
2 changed files with 198 additions and 0 deletions

View File

@@ -3,6 +3,7 @@ mod events;
mod health;
mod html;
mod owncast_api;
mod webhook;
fn main() {
println!("owncast-irc-bridge");

197
src/webhook.rs Normal file
View File

@@ -0,0 +1,197 @@
use axum::{extract::State, http::StatusCode, routing::post, Json, Router};
use serde::Deserialize;
use tokio::sync::mpsc;
use tracing::{info, warn};
use crate::events::{BridgeEvent, Source};
use crate::html::strip_html;
#[derive(Debug, Deserialize)]
pub struct WebhookPayload {
#[serde(rename = "type")]
pub event_type: String,
#[serde(rename = "eventData")]
pub event_data: serde_json::Value,
}
#[derive(Debug, Deserialize)]
struct ChatEventData {
user: ChatUser,
body: String,
id: String,
#[serde(default = "default_visible")]
visible: bool,
}
fn default_visible() -> bool { true }
#[derive(Debug, Deserialize)]
struct ChatUser {
#[serde(rename = "displayName")]
display_name: String,
#[serde(default, rename = "isBot")]
is_bot: bool,
}
#[derive(Debug, Deserialize)]
struct StreamEventData {
#[serde(default, rename = "streamTitle")]
stream_title: Option<String>,
}
impl WebhookPayload {
pub fn into_bridge_event(self) -> Option<BridgeEvent> {
match self.event_type.as_str() {
"CHAT" => {
let data: ChatEventData = serde_json::from_value(self.event_data).ok()?;
if data.user.is_bot || !data.visible {
return None;
}
Some(BridgeEvent::ChatMessage {
source: Source::Owncast,
username: data.user.display_name,
body: strip_html(&data.body),
id: Some(data.id),
})
}
"STREAM_STARTED" => {
let data: StreamEventData = serde_json::from_value(self.event_data).ok()?;
Some(BridgeEvent::StreamStarted {
title: data.stream_title.unwrap_or_default(),
})
}
"STREAM_STOPPED" => Some(BridgeEvent::StreamStopped),
_ => None,
}
}
}
#[derive(Clone)]
struct WebhookState {
event_tx: mpsc::Sender<BridgeEvent>,
}
async fn handle_webhook(
State(state): State<WebhookState>,
Json(payload): Json<WebhookPayload>,
) -> StatusCode {
info!(event_type = %payload.event_type, "Received webhook");
match payload.into_bridge_event() {
Some(event) => {
if state.event_tx.send(event).await.is_err() {
warn!("Router channel closed");
return StatusCode::INTERNAL_SERVER_ERROR;
}
StatusCode::OK
}
None => StatusCode::OK,
}
}
pub async fn run_webhook_server(
port: u16,
event_tx: mpsc::Sender<BridgeEvent>,
) -> anyhow::Result<()> {
let state = WebhookState { event_tx };
let app = Router::new()
.route("/webhook", post(handle_webhook))
.with_state(state);
let addr = std::net::SocketAddr::from(([0, 0, 0, 0], port));
info!(%addr, "Starting webhook server");
let listener = tokio::net::TcpListener::bind(addr).await?;
axum::serve(listener, app).await?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_parse_chat_event() {
let json = serde_json::json!({
"type": "CHAT",
"eventData": {
"user": { "displayName": "viewer42", "isBot": false },
"body": "hello <b>world</b>",
"id": "abc123",
"visible": true
}
});
let payload: WebhookPayload = serde_json::from_value(json).unwrap();
let event = payload.into_bridge_event();
assert!(event.is_some());
if let Some(BridgeEvent::ChatMessage { source, username, body, id }) = event {
assert_eq!(source, Source::Owncast);
assert_eq!(username, "viewer42");
assert_eq!(body, "hello world");
assert_eq!(id, Some("abc123".to_string()));
} else {
panic!("Expected ChatMessage");
}
}
#[test]
fn test_parse_stream_started() {
let json = serde_json::json!({
"type": "STREAM_STARTED",
"eventData": {
"id": "x",
"name": "Test",
"streamTitle": "Friday bowls",
"summary": "",
"timestamp": "2026-01-01T00:00:00Z"
}
});
let payload: WebhookPayload = serde_json::from_value(json).unwrap();
let event = payload.into_bridge_event();
assert!(matches!(event, Some(BridgeEvent::StreamStarted { title }) if title == "Friday bowls"));
}
#[test]
fn test_parse_stream_stopped() {
let json = serde_json::json!({
"type": "STREAM_STOPPED",
"eventData": {
"id": "x",
"name": "Test",
"streamTitle": "",
"summary": "",
"timestamp": "2026-01-01T00:00:00Z"
}
});
let payload: WebhookPayload = serde_json::from_value(json).unwrap();
let event = payload.into_bridge_event();
assert!(matches!(event, Some(BridgeEvent::StreamStopped)));
}
#[test]
fn test_ignores_bot_messages() {
let json = serde_json::json!({
"type": "CHAT",
"eventData": {
"user": { "displayName": "bot", "isBot": true },
"body": "automated message",
"id": "x",
"visible": true
}
});
let payload: WebhookPayload = serde_json::from_value(json).unwrap();
let event = payload.into_bridge_event();
assert!(event.is_none());
}
#[test]
fn test_ignores_unknown_event_type() {
let json = serde_json::json!({
"type": "USER_JOINED",
"eventData": { "id": "x" }
});
let payload: WebhookPayload = serde_json::from_value(json).unwrap();
let event = payload.into_bridge_event();
assert!(event.is_none());
}
}