axtra 0.3.1

Axtra is a Rust library for building web applications with Axum and Astro providing utilities for error handling, notifications, and more.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

//! Discord webhook notification provider.

use reqwest::Client;

use crate::notifier::{ErrorEvent, ErrorNotifier, NotifyFuture};

/// Configuration for the Discord provider.
#[derive(Debug, Clone)]
pub struct DiscordConfig {
    /// Webhook URL for the Discord channel
    pub webhook_url: String,
    /// Optional mention to include (e.g., "@oncall", "<@123456789>", "<@&role_id>")
    pub mention: Option<String>,
}

impl DiscordConfig {
    /// Create a new config with just the webhook URL.
    pub fn new(webhook_url: impl Into<String>) -> Self {
        Self {
            webhook_url: webhook_url.into(),
            mention: None,
        }
    }

    /// Set a mention to include in notifications.
    ///
    /// Examples:
    /// - `"<@123456789>"` - mention a specific user by ID
    /// - `"<@&987654321>"` - mention a role by ID
    /// - `"@everyone"` - mention everyone (if webhook has permission)
    pub fn with_mention(mut self, mention: impl Into<String>) -> Self {
        self.mention = Some(mention.into());
        self
    }
}

/// Discord notification provider using webhooks.
///
/// Sends formatted error messages to a Discord channel via webhook.
///
/// # Example
///
/// ```rust,ignore
/// use axtra::notifier::{NotificationManager, DiscordConfig};
///
/// // Simple usage
/// let manager = NotificationManager::builder()
///     .with_discord(DiscordConfig::new("https://discord.com/api/webhooks/..."))
///     .build();
///
/// // With mention
/// let manager = NotificationManager::builder()
///     .with_discord(DiscordConfig::new("https://discord.com/api/webhooks/...")
///         .with_mention("<@&123456789>"))
///     .build();
/// ```
pub struct DiscordProvider {
    config: DiscordConfig,
}

impl DiscordProvider {
    /// Create a new Discord provider with the given configuration.
    pub fn new(config: DiscordConfig) -> Self {
        Self { config }
    }
}

impl ErrorNotifier for DiscordProvider {
    fn notify<'a>(&'a self, client: &'a Client, event: &'a ErrorEvent) -> NotifyFuture<'a> {
        Box::pin(async move {
            let formatted_message = format!("[{}] {}", event.location, event.message);

            let mut fields = vec![serde_json::json!({
                "name": "Details",
                "value": format!("```{}```", formatted_message),
                "inline": false
            })];

            if let Some(ref mention) = self.config.mention {
                fields.push(serde_json::json!({
                    "name": "\u{200B}",
                    "value": mention,
                    "inline": false
                }));
            }

            let embeds = serde_json::json!([
                {
                    "title": format!(":red_circle: {:?} — {}", event.error_code, event.app_name),
                    "color": 16711680, // Red
                    "fields": fields
                }
            ]);

            let payload = serde_json::json!({ "embeds": embeds });

            client
                .post(&self.config.webhook_url)
                .json(&payload)
                .send()
                .await?
                .error_for_status()?;

            Ok(())
        })
    }

    fn name(&self) -> &'static str {
        "discord"
    }
}