bb/bot/handler/time.go

package handler

import (
	"fmt"
	"strings"
	"time"

	"git.kill0.net/chill9/beepboop/bot"
	"github.com/bwmarrin/discordgo"
	log "github.com/sirupsen/logrus"
)

type (
	TimeHandler struct {
		config bot.Config
		Name   string
	}
)

func NewTimeHandler(s string) *TimeHandler {
	h := new(TimeHandler)
	h.Name = s
	return h
}

func (h *TimeHandler) SetConfig(config bot.Config) {
	h.config = config
}

func (h *TimeHandler) Handle(s *discordgo.Session, m *discordgo.MessageCreate) {
	var (
		t  time.Time
		tz string
	)

	if m.Author.ID == s.State.User.ID {
		return
	}

	if !strings.HasPrefix(m.Content, "!time") {
		return
	}

	x := strings.SplitN(m.Content, " ", 2)

	if len(x) > 2 {
		s.ChannelMessageSend(m.ChannelID, "help: `!time TIMEZONE`")
		return
	}

	now := time.Now()

	if len(x) == 2 {
		tz = x[1]
		loc, err := time.LoadLocation(tz)
		if err != nil {
			log.Warnf("failed to load location: %s", err)
			s.ChannelMessageSend(m.ChannelID, err.Error())
			return
		}
		t = now.In(loc)
	} else {
		t = now
	}

	s.ChannelMessageSend(m.ChannelID, fmt.Sprint(t))
}