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 } ) func NewTimeHandler() *TimeHandler { return new(TimeHandler) } 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)) }