package discord import ( "errors" "strings" "sync" "github.com/bwmarrin/discordgo" "github.com/micro/go-micro/v2/agent/input" "github.com/micro/go-micro/v2/logger" ) type discordConn struct { master *discordInput exit chan struct{} recv chan *discordgo.Message sync.Mutex } func newConn(master *discordInput) *discordConn { conn := &discordConn{ master: master, exit: make(chan struct{}), recv: make(chan *discordgo.Message), } conn.master.session.AddHandler(func(s *discordgo.Session, m *discordgo.MessageCreate) { if m.Author.ID == master.botID { return } whitelisted := false for _, ID := range conn.master.whitelist { if m.Author.ID == ID { whitelisted = true break } } if len(master.whitelist) > 0 && !whitelisted { return } var valid bool m.Message.Content, valid = conn.master.prefixfn(m.Message.Content) if !valid { return } conn.recv <- m.Message }) return conn } func (dc *discordConn) Recv(event *input.Event) error { for { select { case <-dc.exit: return errors.New("connection closed") case msg := <-dc.recv: event.From = msg.ChannelID + ":" + msg.Author.ID event.To = dc.master.botID event.Type = input.TextEvent event.Data = []byte(msg.Content) return nil } } } func (dc *discordConn) Send(e *input.Event) error { fields := strings.Split(e.To, ":") _, err := dc.master.session.ChannelMessageSend(fields[0], string(e.Data)) if err != nil { if logger.V(logger.ErrorLevel, logger.DefaultLogger) { logger.Error("[bot][loop][send]", err) } } return nil } func (dc *discordConn) Close() error { if err := dc.master.session.Close(); err != nil { return err } select { case <-dc.exit: return nil default: close(dc.exit) } return nil }