klein-projects/wazuh-notify
2
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases 2 Wiki Activity

message refactor

Browse Source
This commit is contained in:
darius 2024-05-27 13:48:08 +02:00
parent 50b4baec2d
commit 9a1f982ea6
3 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wazuh-notify-go/notification/discord.go
View File

@ -42,12 +42,12 @@ func SendDiscord(params types.Params) {
"**Threat level:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Level) + "\n" +
"**Times fired:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Firedtimes) +
"\n\n" +
"Priority: " + strconv.Itoa(params.Priority) + "\n" +
"Tags: " + params.Tags + "\n\n" +
"**Priority:** " + strconv.Itoa(params.Priority) + "\n" +
"**Tags:** " + params.Tags + "\n\n" +
params.General.Click
}
message := types.Message{
message := types.DiscordMessage{
Username: params.General.Sender,
Content: params.Mention,
Embeds: []types.Embed{

23
wazuh-notify-go/notification/slack.go
View File

@ -3,7 +3,6 @@ package notification
import (
"bytes"
"encoding/json"
"fmt"
"log"
"net/http"
"os"
@ -35,20 +34,22 @@ func SendSlack(params types.Params) {
params.General.Click
} else {
embedDescription = "\n\n" +
"**Timestamp: **" + time.Now().Format(time.DateTime) + "\n" +
"**Agent:** " + params.WazuhMessage.Parameters.Alert.Agent.Name + "\n" +
"**Event id:** " + params.WazuhMessage.Parameters.Alert.Rule.ID + "\n" +
"**Rule:** " + params.WazuhMessage.Parameters.Alert.Rule.Description + "\n" +
"**Description: **" + params.WazuhMessage.Parameters.Alert.FullLog + "\n" +
"**Threat level:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Level) + "\n" +
"**Times fired:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Firedtimes) +
"*Timestamp:* " + time.Now().Format(time.DateTime) + "\n" +
"*Agent:* " + params.WazuhMessage.Parameters.Alert.Agent.Name + "\n" +
"*Event id:* " + params.WazuhMessage.Parameters.Alert.Rule.ID + "\n" +
"*Rule:* " + params.WazuhMessage.Parameters.Alert.Rule.Description + "\n" +
"*Description:* " + params.WazuhMessage.Parameters.Alert.FullLog + "\n" +
"*Threat level:* " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Level) + "\n" +
"*Times fired:* " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Firedtimes) +
"\n\n" +
"Priority: " + strconv.Itoa(params.Priority) + "\n" +
"Tags: " + params.Tags + "\n\n" +
"*Priority:* " + strconv.Itoa(params.Priority) + "\n" +
"*Tags:* " + params.Tags + "\n\n" +
params.General.Click
}
message := fmt.Sprintf("{\"text\": %s}", embedDescription)
message := types.SlackMessage{
Text: embedDescription,
}
payload := new(bytes.Buffer)

8
wazuh-notify-go/types/types.go
View File

@ -32,7 +32,8 @@ type MarkdownEmphasis struct {
Discord string `toml:"discord"`
}
type Message struct {
// Discord
type DiscordMessage struct {
Username string `json:"username,omitempty"`
AvatarUrl string `json:"avatar_url,omitempty"`
Content string `json:"content,omitempty"`
@ -44,3 +45,8 @@ type Embed struct {
Description string `json:"description,omitempty"`
Color int `json:"color,omitempty"`
}
// slack
type SlackMessage struct {
Text string `json:"text,omitempty"`
}