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

fix path

Browse Source
This commit is contained in:
Darius 2024-05-09 23:19:59 +02:00
parent a18ae36dfb
commit a6504842f2
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
wazuh-notify-go/services/init.go
View File

@ -4,6 +4,7 @@ import (
"bufio"
"encoding/json"
"flag"
"fmt"
"github.com/joho/godotenv"
"gopkg.in/yaml.v2"
"os"
@ -19,22 +20,23 @@ var wazuhData types.WazuhMessage
var BasePath string
func InitNotify() types.Params {
BasePath, _ := os.Executable()
BaseFilePath, _ := os.Executable()
BaseDirPath := path.Dir(BaseFilePath)
log.OpenLogFile(BasePath)
log.OpenLogFile(BaseDirPath)
err := godotenv.Load(path.Join(BasePath, "../../etc/.env"))
err := godotenv.Load(path.Join(BaseDirPath, "../../etc/.env"))
if err != nil {
log.Log("env failed to load")
godotenv.Load(path.Join(BasePath, ".env"))
godotenv.Load(path.Join(BaseDirPath, ".env"))
} else {
log.Log("env loaded")
}
yamlFile, err := os.ReadFile(path.Join(BasePath, "../../etc/wazuh-notify-config.yaml"))
yamlFile, err := os.ReadFile(path.Join(BaseDirPath, "../../etc/wazuh-notify-config.yaml"))
if err != nil {
log.Log("yaml failed to load")
yamlFile, err = os.ReadFile(path.Join(BasePath, "wazuh-notify-config.yaml"))
yamlFile, err = os.ReadFile(path.Join(BaseDirPath, "wazuh-notify-config.yaml"))
}
yaml.Unmarshal(yamlFile, &configParams)