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-micro/util/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
|
|
}
|