From f2cac349c6dd98629e28fbdea35cbc20e81f2555 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thorben=20G=C3=BCnther?= <admin@xenrox.net>
Date: Wed, 11 Sep 2024 22:45:56 +0200
Subject: [PATCH] Support logging to file

---
 config.scfg           |  3 +++
 config/config.go      |  8 ++++++++
 config/config_test.go |  2 ++
 main.go               | 17 ++++++++++++++++-
 4 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/config.scfg b/config.scfg
index 5a42f0f..8076443 100644
--- a/config.scfg
+++ b/config.scfg
@@ -19,6 +19,9 @@ log-level info
 # Options: text, json
 # Default: text
 log-format text
+# Write logs to this file. If unset, the logs will be written to stderr.
+# Default: ""
+log-file /var/log/ntfy-alertmanager.log
 # When multiple alerts are grouped together by Alertmanager, they can either be sent
 # each on their own (single mode) or be kept together (multi mode)
 # Options: single, multi
diff --git a/config/config.go b/config/config.go
index e9d6d33..1273bd7 100644
--- a/config/config.go
+++ b/config/config.go
@@ -25,6 +25,7 @@ type Config struct {
 	HTTPAddress string
 	LogLevel    string
 	LogFormat   string
+	LogFile     string
 	AlertMode   AlertMode
 	User        string
 	Password    string
@@ -97,6 +98,13 @@ func parseBlock(block scfg.Block, config *Config) error {
 		}
 	}
 
+	d = block.Get("log-file")
+	if d != nil {
+		if err := d.ParseParams(&config.LogFile); err != nil {
+			return err
+		}
+	}
+
 	d = block.Get("http-address")
 	if d != nil {
 		if err := d.ParseParams(&config.HTTPAddress); err != nil {
diff --git a/config/config_test.go b/config/config_test.go
index 6fbf6b3..010dc09 100644
--- a/config/config_test.go
+++ b/config/config_test.go
@@ -14,6 +14,7 @@ base-url https://ntfy-alertmanager.example.com
 http-address :8080
 log-level info
 log-format json
+log-file /var/log/ntfy-alertmanager.log
 alert-mode multi
 user webhookUser
 password webhookPass
@@ -70,6 +71,7 @@ cache {
 		HTTPAddress: ":8080",
 		LogLevel:    "info",
 		LogFormat:   "json",
+		LogFile:     "/var/log/ntfy-alertmanager.log",
 		AlertMode:   Multi,
 		User:        "webhookUser",
 		Password:    "webhookPass",
diff --git a/main.go b/main.go
index 5ae251e..33d0bcf 100644
--- a/main.go
+++ b/main.go
@@ -14,6 +14,7 @@ import (
 	"errors"
 	"flag"
 	"fmt"
+	"io"
 	"log/slog"
 	"net/http"
 	"os"
@@ -505,7 +506,21 @@ func main() {
 		os.Exit(1)
 	}
 
-	logger, err := logging.New(cfg.LogLevel, cfg.LogFormat, os.Stderr)
+	var logOutput io.Writer
+	if cfg.LogFile != "" {
+		f, err := os.OpenFile(cfg.LogFile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0600)
+		if err != nil {
+			slog.Error("Failed to open logfile",
+				slog.String("error", err.Error()))
+			os.Exit(1)
+		}
+		defer f.Close()
+		logOutput = f
+	} else {
+		logOutput = os.Stderr
+	}
+
+	logger, err := logging.New(cfg.LogLevel, cfg.LogFormat, logOutput)
 	if err != nil {
 		slog.Error("Failed to create logger",
 			slog.String("error", err.Error()))