diff --git a/flake.nix b/flake.nix index 159aa6b..6ad64be 100644 --- a/flake.nix +++ b/flake.nix @@ -23,6 +23,8 @@ devShells = forAllSystems (system: pkgs: { default = pkgs.mkShell { buildInputs = with pkgs; [ + sqlite + sqlitebrowser go golangci-lint docker-compose diff --git a/go.mod b/go.mod index 3c7b474..1ceca27 100644 --- a/go.mod +++ b/go.mod @@ -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 ) diff --git a/go.sum b/go.sum index e5a04a3..6c801c5 100644 --- a/go.sum +++ b/go.sum @@ -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= diff --git a/main.go b/main.go index d464080..a4f0a9b 100644 --- a/main.go +++ b/main.go @@ -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") }