feat(mongodb): added guildID to joke database structure

Wittano · Oct 29, 2023 · 640a963 · 640a963
1 parent 677a724
commit 640a963
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 4 deletions.
diff --git a/internal/command/joke.go b/internal/command/joke.go
@@ -54,7 +54,9 @@ var (
 )
 
 func executeAddJokeCommand(ctx context.Context, s *discordgo.Session, i *discordgo.InteractionCreate) {
-	var j mongo.JokeDB
+	j := mongo.JokeDB{
+		GuildID: i.GuildID,
+	}
 
 	for _, o := range i.Data.(discordgo.ApplicationCommandInteractionData).Options {
 		switch o.Name {

diff --git a/internal/interaction/joke.go b/internal/interaction/joke.go
@@ -29,6 +29,8 @@ var (
 func SendJoke(ctx context.Context, s *discordgo.Session, i *discordgo.InteractionCreate, t types.JokeType, c types.JokeCategory) {
 	var msg *discordgo.InteractionResponseData
 
+	ctx = context.WithValue(ctx, "GUILD_ID", i.GuildID)
+
 	switch t {
 	case types.Single:
 		j, err := getSingleTypeJokeGenerator()(ctx, c)

diff --git a/internal/mongo/joke.go b/internal/mongo/joke.go
@@ -17,6 +17,7 @@ type JokeDB struct {
 	ContentRes string             `bson:"content"`
 	Type       types.JokeType     `bson:"type"`
 	Category   types.JokeCategory `bson:"category"`
+	GuildID    string             `bson:"guild_id"`
 	ExternalID int64              `bson:"externalID"`
 }
 
@@ -83,9 +84,18 @@ func findRandomJoke(ctx context.Context, j JokeSearch) (JokeDB, error) {
 		}},
 	}},
 		{{
-			"$match", bson.D{{
-				"type", jokeType,
-			}},
+			"$match", bson.D{
+				{
+					"type", jokeType,
+				}, {
+					"$or", []bson.D{
+						{{
+							"guild_id", "",
+						}},
+						{{
+							"guild_id", ctx.Value("GUILD_ID"),
+						}}},
+				}},
 		}},
 	}