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

feat: message editing

Browse Source
This commit is contained in:
Gustavo "Guz" L. de Mello
2024-08-08 11:08:17 -03:00
parent 8a3fe0a82e
commit 5d1528621f
4 changed files with 119 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flake.nix
View File

@@ -23,6 +23,8 @@
devShells = forAllSystems (system: pkgs: {
default = pkgs.mkShell {
buildInputs = with pkgs; [
sqlite
sqlitebrowser
go
golangci-lint
docker-compose

8
go.mod
View File

@@ -2,10 +2,16 @@ module dislate
go 1.22.5
require github.com/bwmarrin/discordgo v0.28.1
require (
github.com/bwmarrin/discordgo v0.28.1
github.com/tursodatabase/go-libsql v0.0.0-20240725130945-f44f2b84c8c8
)
require (
github.com/antlr4-go/antlr/v4 v4.13.0 // indirect
github.com/gorilla/websocket v1.4.2 // indirect
github.com/libsql/sqlite-antlr4-parser v0.0.0-20240327125255-dbf53b6cbf06 // indirect
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect
golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 // indirect
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 // indirect
)

16
go.sum
View File

@@ -1,12 +1,28 @@
github.com/antlr4-go/antlr/v4 v4.13.0 h1:lxCg3LAv+EUK6t1i0y1V6/SLeUi0eKEKdhQAlS8TVTI=
github.com/antlr4-go/antlr/v4 v4.13.0/go.mod h1:pfChB/xh/Unjila75QW7+VU4TSnWnnk9UTnmpPaOR2g=
github.com/bwmarrin/discordgo v0.28.1 h1:gXsuo2GBO7NbR6uqmrrBDplPUx2T3nzu775q/Rd1aG4=
github.com/bwmarrin/discordgo v0.28.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/libsql/sqlite-antlr4-parser v0.0.0-20240327125255-dbf53b6cbf06 h1:JLvn7D+wXjH9g4Jsjo+VqmzTUpl/LX7vfr6VOfSWTdM=
github.com/libsql/sqlite-antlr4-parser v0.0.0-20240327125255-dbf53b6cbf06/go.mod h1:FUkZ5OHjlGPjnM2UyGJz9TypXQFgYqw6AFNO1UiROTM=
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/tursodatabase/go-libsql v0.0.0-20240725130945-f44f2b84c8c8 h1:nxpR20uTcKWd+IcojEUCCieKTmBhrEnIhl0SiwUMBPk=
github.com/tursodatabase/go-libsql v0.0.0-20240725130945-f44f2b84c8c8/go.mod h1:TjsB2miB8RW2Sse8sdxzVTdeGlx74GloD5zJYUC38d8=
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg=
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 h1:aAcj0Da7eBAtrTp03QXWvm88pSyOt+UgdZw2BFZ+lEw=
golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8/go.mod h1:CQ1k9gNrJ50XIzaKCRR2hssIjF07kZFEiieALBM/ARQ=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
gotest.tools v2.2.0+incompatible h1:VsBPFP1AI068pPrMxtb/S8Zkgf9xEmTLJjfM+P5UIEo=
gotest.tools v2.2.0+incompatible/go.mod h1:DsYFclhRJ6vuDpmuTbkuFWG+y2sxOXAzmJt81HFBacw=

103
main.go
View File

@@ -1,6 +1,7 @@
package main
import (
"database/sql"
"fmt"
"log"
"os"
@@ -9,6 +10,7 @@ import (
"syscall"
"github.com/bwmarrin/discordgo"
_ "github.com/tursodatabase/go-libsql"
)
const DEST_CHANNEL = "1270407366617333920"
@@ -18,9 +20,34 @@ const USER_WEBHOOK_FORMAT = "dislate-user-%s"
func main() {
log.Printf("Hello, world")
db, err := sql.Open("libsql", "file:./guild.db")
if err != nil {
log.Printf("ERROR: failed to open database %s", err)
return
}
defer func() {
err := db.Close()
if err != nil {
log.Printf("ERROR: failed to close database %s", err)
return
}
}()
_, err = db.Exec(`
CREATE TABLE IF NOT EXISTS MessageMap (
Original text NOT NULL PRIMARY KEY UNIQUE,
Translated text NOT NULL UNIQUE
);
`)
if err != nil {
log.Printf("ERROR: failed to create MessageMap table %s", err)
return
}
discord, err := discordgo.New("Bot " + os.Getenv("DISCORD_TOKEN"))
if err != nil {
panic(err)
log.Printf("ERROR: failed to start bot %s", err)
return
}
discord.AddHandler(func(s *discordgo.Session, m *discordgo.MessageCreate) {
@@ -54,7 +81,7 @@ func main() {
w = ws[wi]
}
_, err = s.WebhookExecute(w.ID, w.Token, true, &discordgo.WebhookParams{
wm, err := s.WebhookExecute(w.ID, w.Token, true, &discordgo.WebhookParams{
Username: m.Author.GlobalName,
AvatarURL: m.Author.AvatarURL(""),
Content: m.Content,
@@ -63,21 +90,79 @@ func main() {
log.Printf("ERROR: failed to message using webhook for user %s: %s", m.Author.ID, err)
return
}
_, err = db.Exec(`
INSERT INTO MessageMap (Original, Translated)
VALUES ($1, $2) ON CONFLICT DO NOTHING
`, m.ID, wm.ID)
if err != nil {
log.Printf("ERROR: failed add message to database. Original: %s, Translated: %s: %s", m.ID, wm.ID, err)
return
}
})
discord.AddHandler(func(s *discordgo.Session, m *discordgo.MessageUpdate) {
if m.Author.Bot {
return
}
q := db.QueryRow(`
SELECT * FROM MessageMap
WHERE "Original" = $1
`, m.ID)
var original, translated string
err := q.Scan(&original, &translated)
if err != nil {
log.Printf("ERROR: failed query message to database. Original: %s: %s", m.ID, err)
return
}
ws, err := s.ChannelWebhooks(DEST_CHANNEL)
if err != nil {
log.Printf("ERROR: failed to find channel webhooks: %s", err)
return
}
var w *discordgo.Webhook
if wi := slices.IndexFunc(ws, func(w *discordgo.Webhook) bool {
return w.Name == fmt.Sprintf(USER_WEBHOOK_FORMAT, m.Author.ID)
}); wi == -1 {
w, err = s.WebhookCreate(
DEST_CHANNEL,
fmt.Sprintf(USER_WEBHOOK_FORMAT, m.Author.ID),
m.Author.AvatarURL(""),
)
if err != nil {
log.Printf("ERROR: failed to create webhook for user %s: %s", m.Author.ID, err)
return
}
} else {
w = ws[wi]
}
_, err = s.WebhookMessageEdit(w.ID, w.Token, translated, &discordgo.WebhookEdit{
Content: &m.Content,
})
if err != nil {
log.Printf("ERROR: failed to edit webhook message, Original %s, Translated %s: %s", m.ID, translated, err)
return
}
})
err = discord.Open()
if err != nil {
log.Fatalf("could not open session: %s", err)
log.Printf("could not open session: %s", err)
return
}
log.Printf("Bot session opened successfully")
defer func() {
err = discord.Close()
if err != nil {
log.Printf("could not close session: %s", err)
return
}
log.Printf("Bot session closed successfully")
}()
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt, syscall.SIGINT)
<-sig
err = discord.Close()
if err != nil {
log.Fatalf("could not close session: %s", err)
}
log.Printf("Bot session closed successfully")
}