logmania/output/websocket/main.go

package xmpp

import (
	"net/http"
	"regexp"

	"dev.sum7.eu/genofire/golang-lib/websocket"
	"github.com/mitchellh/mapstructure"
	log "github.com/sirupsen/logrus"

	"dev.sum7.eu/genofire/logmania/bot"
	"dev.sum7.eu/genofire/logmania/database"
	"dev.sum7.eu/genofire/logmania/output"
)

const (
	proto = "ws"
)

var logger = log.WithField("output", proto)

type Output struct {
	output.Output
	defaults  []*database.Notify
	ws        *websocket.Server
	formatter log.Formatter
}

type OutputConfig struct {
	Default string `mapstructure:"default"`
}

func Init(configInterface interface{}, db *database.DB, bot *bot.Bot) output.Output {
	var config OutputConfig
	if err := mapstructure.Decode(configInterface, &config); err != nil {
		logger.Warnf("not able to decode data: %s", err)
		return nil
	}
	inputMSG := make(chan *websocket.Message)
	ws := websocket.NewServer(inputMSG, nil)

	http.HandleFunc("/output/ws", ws.Handler)

	go func() {
		for msg := range inputMSG {
			if msg.Subject != "bot" {
				logger.Warnf("receive unknown websocket message: %s", msg.Subject)
				continue
			}
			bot.Handle(func(answer string) {
				msg.Answer("bot", answer)
			}, "", msg.Body.(string))
		}
	}()

	logger.Info("startup")

	var defaults []*database.Notify
	if config.Default != "" {
		defaults = append(defaults, &database.Notify{
			Protocol:  proto,
			To:        config.Default,
			RegexIn:   make(map[string]*regexp.Regexp),
			MaxPrioIn: log.DebugLevel,
		})
	}
	return &Output{
		defaults: defaults,
		ws:       ws,
		formatter: &log.TextFormatter{
			DisableTimestamp: true,
		},
	}
}

func (out *Output) Default() []*database.Notify {
	return out.defaults
}

func (out *Output) Send(e *log.Entry, to *database.Notify) bool {
	if to.Protocol != proto {
		return false
	}

	out.ws.SendAll(&websocket.Message{
		Subject: to.Address(),
		Body: &log.Entry{
			Buffer:  e.Buffer,
			Data:    e.Data,
			Level:   e.Level,
			Logger:  e.Logger,
			Message: to.RunReplace(e.Message),
			Time:    e.Time,
		},
	})
	return true
}

func (out *Output) Close() {
}

func init() {
	output.Add("websocket", Init)
}