logger.go

package main

import (
	"github.com/bwmarrin/discordgo"
	"strings"
)

// Logger struct
type Logger struct {
	ch      *ChannelHandler
	session *discordgo.Session
	logchan chan string
}

// BOTLOG const for log types
const (
	BOTLOG = iota
	PERMLOG
	BANKLOG
)

// Init function
func (h *Logger) Init(ch *ChannelHandler, channel chan string, session *discordgo.Session) {
	h.ch = ch
	h.logchan = channel
	h.session = session
	go h.ReadLog()
}

// ReadLog function
func (h *Logger) ReadLog() {

	message := <-h.logchan

	if strings.HasPrefix(message, "Bot") {
		h.LogBot(message, h.session)
		go h.ReadLog()
	}
	if strings.HasPrefix(message, "Permissions") {
		h.LogPerm(message, h.session)
		go h.ReadLog()
	}
	if strings.HasPrefix(message, "Bank") {
		h.LogBank(message, h.session)
		go h.ReadLog()
	}
	h.LogBot(message, h.session)
	go h.ReadLog()
}

// LogBot function
func (h *Logger) LogBot(message string, s *discordgo.Session) {
	channelid, err := h.ch.GetBotLogChannel()
	if err != nil {
		return // Do nothing, we don't want to yell about no channel configured, just silently fail
	}

	s.ChannelMessageSend(channelid, message)
}

// LogBank function
func (h *Logger) LogBank(message string, s *discordgo.Session) {
	channelid, err := h.ch.GetBankLogChannel()
	if err != nil {
		return // Do nothing, we don't want to yell about no channel configured, just silently fail
	}

	s.ChannelMessageSend(channelid, message)
}

// LogPerm function
func (h *Logger) LogPerm(message string, s *discordgo.Session) {
	channelid, err := h.ch.GetPermissionLogChannel()
	if err != nil {
		return // Do nothing, we don't want to yell about no channel configured, just silently fail
	}
	s.ChannelMessageSend(channelid, message)
}

// Log function
func (h *Logger) Log(message string, s *discordgo.Session, level string) {

	if level == "" {
		return
	}
	if level == "bot" {
		h.LogBot(message, s)
	}
	if level == "perm" || level == "permission" {
		h.LogPerm(message, s)
	}
	if level == "bank" {
		h.LogBank(message, s)
	}
}