mirror of
https://github.com/go-micro/go-micro.git
synced 2024-12-12 08:23:58 +02:00
95 lines
1.7 KiB
Go
95 lines
1.7 KiB
Go
|
package discord
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"strings"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/bwmarrin/discordgo"
|
||
|
"github.com/micro/go-micro/agent/input"
|
||
|
"github.com/micro/go-log"
|
||
|
)
|
||
|
|
||
|
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 {
|
||
|
log.Log("[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
|
||
|
}
|