capytal/dislate
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(events,messages): send translated message to linked channel

Browse Source
This commit is contained in:
Gustavo "Guz" L. de Mello
2024-08-21 21:49:00 -03:00
parent eca9152d47
commit c43fffd14a
2 changed files with 186 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internals/discord/bot/events.go
View File

@@ -5,7 +5,7 @@ import "dislate/internals/discord/bot/events"
func (b *Bot) registerEventHandlers() {
ehs := []any{
events.NewGuildCreate(b.logger, b.db).Serve,
events.NewMessageCreate(b.logger, b.db).Serve,
events.NewMessageCreate(b.logger, b.db, b.translator).Serve,
events.NewReady(b.logger, b.db).Serve,
}
for _, h := range ehs {

190
internals/discord/bot/events/messages.go
View File

@@ -1,9 +1,13 @@
package events
import (
"errors"
"fmt"
"log/slog"
"dislate/internals/guilddb"
"dislate/internals/translator"
"dislate/internals/translator/lang"
dgo "github.com/bwmarrin/discordgo"
)
@@ -13,13 +17,189 @@ type EventHandler[E any] interface {
}
type MessageCreate struct {
log *slog.Logger
db guilddb.GuildDB
log *slog.Logger
db guilddb.GuildDB
translator translator.Translator
}
func NewMessageCreate(log *slog.Logger, db guilddb.GuildDB) MessageCreate {
return MessageCreate{log, db}
func NewMessageCreate(log *slog.Logger, db guilddb.GuildDB, t translator.Translator) MessageCreate {
return MessageCreate{log, db, t}
}
func (h MessageCreate) Serve(s *dgo.Session, e *dgo.MessageCreate) {
if e.Message.Author.Bot {
return
}
ch, err := h.db.Channel(e.GuildID, e.ChannelID)
if err != nil && errors.Is(err, guilddb.ErrNotFound) {
h.log.Debug("Channel is not in database, ignoring.", slog.String("guild", e.GuildID), slog.String("channel", e.ChannelID))
return
} else if err != nil {
h.log.Error("Error while trying to get channel from database",
slog.String("guild", e.GuildID),
slog.String("channel", e.ChannelID),
slog.String("err", err.Error()),
)
return
}
gc, err := h.db.ChannelGroup(ch.GuildID, ch.ID)
if err != nil && errors.Is(err, guilddb.ErrNotFound) {
h.log.Debug("Channel is not in a group, ignoring.", slog.String("guild", e.GuildID), slog.String("channel", e.ChannelID))
return
} else if err != nil {
h.log.Error("Error while trying to get channel group from database",
slog.String("guild", e.GuildID),
slog.String("channel", e.ChannelID),
slog.String("err", err.Error()),
)
return
}
_, err = h.getMessage(e.Message, ch.Language)
if err != nil {
h.log.Error("Error while trying to get/set message to database",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
_, err := s.ChannelMessageSendReply(
e.Message.ChannelID,
fmt.Sprintf("Error while trying to send message to database. %s", err.Error()),
e.Message.Reference(),
)
if err != nil {
h.log.Error("Error while trying to send error message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
}
return
}
for _, c := range gc {
if c.ID == ch.ID && c.GuildID == ch.GuildID {
continue
}
go func(c guilddb.Channel) {
t, err := h.translator.Translate(ch.Language, c.Language, e.Message.Content)
if err != nil {
h.log.Error("Error while trying to translate message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("content", e.Message.Content),
slog.String("err", err.Error()),
)
_, err := s.ChannelMessageSendReply(
e.Message.ChannelID,
fmt.Sprintf("Error while trying to translate message. %s", err.Error()),
e.Message.Reference(),
)
if err != nil {
h.log.Error("Error while trying to send error message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
}
}
tdm, err := s.ChannelMessageSend(c.ID, t)
if err != nil {
h.log.Error("Error while trying to send translated message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("content", e.Message.Content),
slog.String("err", err.Error()),
)
_, err := s.ChannelMessageSendReply(
e.Message.ChannelID,
fmt.Sprintf("Error while trying to send translated message. %s", err.Error()),
e.Message.Reference(),
)
if err != nil {
h.log.Error("Error while trying to send error message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
}
}
_, err = h.getTranslatedMessage(tdm, e.Message, c.Language)
if err != nil {
h.log.Error("Error while trying to get/set translated message to database",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
_, err := s.ChannelMessageSendReply(
e.Message.ChannelID,
fmt.Sprintf("Error while trying to send translated message to database. %s", err.Error()),
e.Message.Reference(),
)
if err != nil {
h.log.Error("Error while trying to send error message",
slog.String("guild", e.Message.GuildID),
slog.String("channel", e.Message.ChannelID),
slog.String("message", e.Message.ID),
slog.String("err", err.Error()),
)
}
}
}(c)
}
}
func (h MessageCreate) getMessage(m *dgo.Message, lang lang.Language) (guilddb.Message, error) {
msg, err := h.db.Message(m.GuildID, m.ChannelID, m.ID)
if err != nil && errors.Is(err, guilddb.ErrNotFound) {
if err := h.db.MessageInsert(guilddb.NewMessage(m.GuildID, m.ChannelID, m.ID, lang)); err != nil {
return guilddb.Message{}, err
}
msg, err = h.db.Message(m.GuildID, m.ChannelID, m.ID)
if err != nil {
return guilddb.Message{}, err
}
}
return msg, nil
}
func (h MessageCreate) getTranslatedMessage(m, original *dgo.Message, lang lang.Language) (guilddb.Message, error) {
msg, err := h.db.Message(m.GuildID, m.ChannelID, m.ID)
if err != nil && errors.Is(err, guilddb.ErrNotFound) {
if err := h.db.MessageInsert(guilddb.NewTranslatedMessage(
m.GuildID,
m.ChannelID,
m.ID,
lang,
original.ChannelID,
original.ID,
)); err != nil {
return guilddb.Message{}, err
}
msg, err = h.db.Message(m.GuildID, m.ChannelID, m.ID)
if err != nil {
return guilddb.Message{}, err
}
} else if err != nil {
return guilddb.Message{}, err
}
return msg, nil
}