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" + "**Threat level:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Level) + "\n" +
"**Times fired:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Firedtimes) + "**Times fired:** " + strconv.Itoa(params.WazuhMessage.Parameters.Alert.Rule.Firedtimes) +
"\n\n" + "\n\n" +
"Priority: " + strconv.Itoa(params.Priority) + "\n" + "**Priority:** " + strconv.Itoa(params.Priority) + "\n" +
"Tags: " + params.Tags + "\n\n" + "**Tags:** " + params.Tags + "\n\n" +
params.General.Click params.General.Click
} }
message := types.Message{ message := types.DiscordMessage{
Username: params.General.Sender, Username: params.General.Sender,
Content: params.Mention, Content: params.Mention,
Embeds: []types.Embed{ Embeds: []types.Embed{

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

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

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

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