notifications.go

package main

import (
	"fmt"
	"strings"

	"github.com/containrrr/shoutrrr"
	"github.com/containrrr/shoutrrr/pkg/types"
)

func send_notification(message string, screenshot_url string, correlated_request string) {
	notify_urls := get_env("NOTIFY")
	if notify_urls == "" {
		return
	}

	if !get_send_alerts() {
		return
	}

	message_with_screenshot := message + " " + screenshot_url

	if correlated_request != "No correlated request found for this injection." {
		message_with_screenshot += " At " + correlated_request
	}

	urls := strings.Split(notify_urls, ",")

	sender, err := shoutrrr.CreateSender(urls...)
	params := &types.Params{}
	sender.Send(message_with_screenshot, params)
	if err != nil {
		fmt.Println("Error sending notification:", err)
	}
}