bb/command/time.go

package command

import (
	"fmt"
	"strings"
	"time"

	"github.com/bwmarrin/discordgo"
	log "github.com/sirupsen/logrus"
)

type (
	TimeHandler struct {
		config Config
	}
)

func NewTimeHandler() *TimeHandler {
	return new(TimeHandler)
}

func (h *TimeHandler) SetConfig(config 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))
}