logmania/notify/file/main.go

80 lines
1.6 KiB
Go

package file
import (
"os"
"path"
"regexp"
log "github.com/sirupsen/logrus"
"dev.sum7.eu/genofire/logmania/bot"
"dev.sum7.eu/genofire/logmania/database"
"dev.sum7.eu/genofire/logmania/lib"
"dev.sum7.eu/genofire/logmania/notify"
)
const (
proto = "file"
)
var logger = log.WithField("notify", proto)
type Notifier struct {
notify.Notifier
files map[string]*os.File
formatter log.Formatter
path string
}
func Init(config *lib.NotifyConfig, db *database.DB, bot *bot.Bot) notify.Notifier {
logger.Info("startup")
if config.FileDirectory == "" {
return nil
}
return &Notifier{
files: make(map[string]*os.File),
formatter: &log.JSONFormatter{},
path: config.FileDirectory,
}
}
func (n *Notifier) getFile(name string) *os.File {
if file, ok := n.files[name]; ok {
return file
}
if m, err := regexp.MatchString(`^[0-9A-Za-z_-]*$`, name); err != nil || !m {
logger.Errorf("not allowed to use '%s:%s'", proto, name)
return nil
}
filename := path.Join(n.path, name+".json")
file, err := os.OpenFile(filename, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
logger.Errorf("could not open file: %s", err.Error())
return nil
}
n.files[name] = file
return file
}
func (n *Notifier) Send(e *log.Entry, to *database.Notify) bool {
if to.Protocol != proto {
return false
}
byteText, err := n.formatter.Format(e)
if err != nil {
return false
}
text := to.RunReplace(string(byteText))
file := n.getFile(to.To)
if file == nil {
return false
}
_, err = file.WriteString(text)
return err == nil
}
func init() {
notify.AddNotifier(Init)
}