about summary refs log tree commit diff
path: root/cmd/vm
AgeCommit message (Expand)Author
2022-05-27Add debug info to compiled VM codeMel
2022-05-20Add VM cli toolMel
='n19' href='#n19'>19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
package bot

import (
	"context"
	"errors"
	"jinx/pkg/discord"
	"jinx/pkg/discord/events"

	"github.com/rs/zerolog"
)

type Bot struct {
	client        *discord.Discord
	logger        *zerolog.Logger
	cancelContext context.CancelFunc
}

func NewBot(token string, logger *zerolog.Logger) *Bot {
	return &Bot{
		client:        discord.NewClient(token, logger),
		logger:        logger,
		cancelContext: nil,
	}
}

func (b *Bot) Start() error {
	var err error
	ctx, cancel := context.WithCancel(context.Background())
	b.cancelContext = cancel

	defer func() {
		if err != nil {
			cancel()
		}
	}()

	b.client.On(events.READY, func(_ any) {
		b.logger.Info().Msg("ready!")
	})

	b.client.On(events.MESSAGE, func(m any) {
		msg := m.(events.Message)
		b.logger.Debug().Msgf("message: %s", msg.Content)

		if msg.Content == "ping" {
			if err := b.client.SendMessage(msg.ChannelID, "pong"); err != nil {
				b.logger.Error().Err(err).Msg("error sending message")
			}
		}
	})

	if err := b.client.Connect(ctx); err != nil {
		return err
	}

	return nil
}

func (b *Bot) Stop() error {
	if b.cancelContext == nil {
		return errors.New("bot has no context")
	}

	b.cancelContext()
	return b.client.Disconnect()
}