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) }