mirror of
https://github.com/Dvorinka/beszel.git
synced 2026-06-05 05:42:56 +00:00
Initial commit: Beszel fork with Domain Locker integration
This commit is contained in:
@@ -0,0 +1,83 @@
|
||||
package providers
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/henrygd/beszel/internal/entities/notification"
|
||||
)
|
||||
|
||||
type WebhookProvider struct {
|
||||
settings notification.WebhookSettings
|
||||
}
|
||||
|
||||
func NewWebhookProvider(settings notification.WebhookSettings) *WebhookProvider {
|
||||
return &WebhookProvider{settings: settings}
|
||||
}
|
||||
|
||||
func (p *WebhookProvider) Validate() error {
|
||||
if p.settings.URL == "" {
|
||||
return fmt.Errorf("webhook URL is required")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *WebhookProvider) Send(msg *notification.NotificationMessage) error {
|
||||
if err := p.Validate(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
method := p.settings.Method
|
||||
if method == "" {
|
||||
method = "POST"
|
||||
}
|
||||
|
||||
body := p.formatBody(msg)
|
||||
req, err := http.NewRequest(method, p.settings.URL, bytes.NewBufferString(body))
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to create request: %w", err)
|
||||
}
|
||||
|
||||
for k, v := range p.settings.Headers {
|
||||
req.Header.Set(k, v)
|
||||
}
|
||||
if req.Header.Get("Content-Type") == "" {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
|
||||
client := &http.Client{Timeout: 30 * time.Second}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("webhook request failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
return fmt.Errorf("webhook returned status %d", resp.StatusCode)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *WebhookProvider) formatBody(msg *notification.NotificationMessage) string {
|
||||
if p.settings.BodyTemplate != "" {
|
||||
return p.settings.BodyTemplate
|
||||
}
|
||||
|
||||
data := map[string]interface{}{
|
||||
"title": msg.Title,
|
||||
"body": msg.Body,
|
||||
"monitor": msg.MonitorName,
|
||||
"url": msg.MonitorURL,
|
||||
"status": msg.Status,
|
||||
"timestamp": msg.Timestamp,
|
||||
"ping": msg.Ping,
|
||||
"message": msg.Message,
|
||||
}
|
||||
|
||||
b, _ := json.Marshal(data)
|
||||
return string(b)
|
||||
}
|
||||
Reference in New Issue
Block a user