support betting on games

Author: Chew

src/main/java/pw/chew/mlb/commands/BettingCommand.java

@@ -4,11 +4,17 @@
 import com.jagrosh.jdautilities.command.SlashCommandEvent;
 import net.dv8tion.jda.api.EmbedBuilder;
 import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
+import net.dv8tion.jda.api.interactions.commands.OptionType;
+import net.dv8tion.jda.api.interactions.commands.build.OptionData;
 import net.dv8tion.jda.api.utils.TimeFormat;
 import org.hibernate.Transaction;
 import pw.chew.chewbotcca.util.DatabaseHelper;
 import pw.chew.mlb.models.Bet;
+import pw.chew.mlb.models.BetKind;
 import pw.chew.mlb.models.Profile;
+import pw.chew.mlb.objects.GameBlurb;
+import pw.chew.mlb.objects.GameState;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -18,7 +24,8 @@ public BettingCommand() {
         this.name = "betting";
         this.help = "Bet on a team";
         this.children = new SlashCommand[]{
-            new BettingProfileSubCommand()
+            new BettingProfileSubCommand(),
+            new BettingBetSubCommand()
         };
     }
 
@@ -40,7 +47,7 @@ protected void execute(SlashCommandEvent event) {
             List<Bet> bets = retrieveBets(event.getUser().getIdLong());
 
             // build embed
-            event.replyEmbeds(buildProfileEmbed(event, profile, bets)).queue();
+            event.replyEmbeds(buildProfileEmbed(event, profile, bets)).setEphemeral(true).queue();
         }
 
         private MessageEmbed buildProfileEmbed(SlashCommandEvent event, Profile profile, List<Bet> bets) {
@@ -58,8 +65,8 @@ private MessageEmbed buildProfileEmbed(SlashCommandEvent event, Profile profile,
                 Bet bet = bets.get(i);
                 betString.add("%s | %s%s - %s".formatted(
                     TimeFormat.DATE_SHORT.format(bet.getCreatedAt()),
-                    bet.getSuccessful() ? "+" : "-",
-                    bet.getAmount(),
+                    bet.amount() > 0 ? "+" : "",
+                    bet.amount(),
                     bet.getReason()
                 ));
             }
@@ -70,6 +77,122 @@ private MessageEmbed buildProfileEmbed(SlashCommandEvent event, Profile profile,
         }
     }
 
+    public static class BettingBetSubCommand extends SlashCommand {
+        public BettingBetSubCommand() {
+            this.name = "bet";
+            this.help = "Bet on a team";
+            this.options = List.of(
+                new OptionData(OptionType.INTEGER, "team", "Which team to bet on", true, true),
+                new OptionData(OptionType.INTEGER, "game", "Which game to bet on", true, true),
+                new OptionData(OptionType.INTEGER, "amount", "How much to bet. 0 to remove bet.", true, false)
+                    .setMinValue(0)
+            );
+        }
+
+        @Override
+        protected void execute(SlashCommandEvent event) {
+            // options
+            int teamId = (int) event.optLong("team", 0);
+            int gamePk = (int) event.optLong("game", 0);
+            int amount = (int) event.optLong("amount", 0);
+
+            // Check the game status
+            GameState state = GameState.fromPk(String.valueOf(gamePk));
+            if (state.isFinal()) {
+                event.reply("This game is already over!").setEphemeral(true).queue();
+                return;
+            }
+            if (state.inning() > 4) {
+                event.reply("Bets cannot be placed or changed past the 4th inning! To see your bet, run /betting placed").setEphemeral(true).queue();
+                return;
+            }
+
+            // start session
+            var session = DatabaseHelper.getSessionFactory().openSession();
+
+            // check for a bet, check by user_id, game_pk, and team_id
+            Bet bet = session.createQuery("from Bet where userId = :userId and gamePk = :gamePk and teamId = :teamId", Bet.class)
+                .setParameter("userId", event.getUser().getIdLong())
+                .setParameter("gamePk", gamePk)
+                .setParameter("teamId", teamId)
+                .uniqueResult();
+
+            // Ensure we have enough credit to bet
+            Profile user = retrieveProfile(event.getUser().getIdLong());
+
+            // Get game blurb
+            GameBlurb blurb = new GameBlurb(gamePk + "");
+
+            // if bet exists, update it
+            if (bet == null) {
+                if (amount == 0) {
+                    event.reply("You must bet at least 1 credit!").setEphemeral(true).queue();
+                    return;
+                }
+
+                if (user.getCredits() < amount) {
+                    event.reply("You don't have enough credits to bet that much! You only have " + user.getCredits() + " credits!").setEphemeral(true).queue();
+                    return;
+                }
+
+                String team = blurb.away().id() == teamId ? blurb.away().name() : blurb.home().name();
+
+                Bet newBet = new Bet();
+                newBet.setUserId(event.getUser().getIdLong());
+                newBet.setGamePk(gamePk);
+                newBet.setTeamId(teamId);
+                newBet.setBet(amount);
+                newBet.setKind(BetKind.PENDING);
+                newBet.setReason("Bet on " + team + " for " + blurb.name());
+
+                user.setCredits(user.getCredits() - amount);
+
+                Transaction trans = session.beginTransaction();
+                session.save(newBet);
+                session.update(user);
+                trans.commit();
+
+                event.reply("Bet placed!").setEphemeral(true).queue();
+            } else if (amount == 0) {
+                int currentBet = bet.getBet();
+
+                Transaction trans = session.beginTransaction();
+                session.delete(bet);
+
+                user.setCredits(user.getCredits() + currentBet);
+                session.update(user);
+
+                trans.commit();
+
+                event.reply("Bet removed!").setEphemeral(true).queue();
+            } else {
+                int currentBet = bet.getBet();
+                int creditsToDeduct = amount - currentBet;
+                if (amount > currentBet && creditsToDeduct > user.getCredits()) {
+                    event.reply("You don't have enough credits to bet that much! You only have " + user.getCredits() + " credits!").queue();
+                    return;
+                }
+
+                user.setCredits(user.getCredits() - creditsToDeduct);
+                bet.setBet(amount);
+
+                Transaction trans = session.beginTransaction();
+                session.update(bet);
+                session.update(user);
+                trans.commit();
+
+                event.reply("Bet updated!").setEphemeral(true).queue();
+            }
+
+            session.close();
+        }
+
+        @Override
+        public void onAutoComplete(CommandAutoCompleteInteractionEvent event) {
+            event.replyChoices(PlanGameCommand.handleAutoComplete(event)).queue();
+        }
+    }
+
     public static Profile retrieveProfile(long userId) {
         var session = DatabaseHelper.getSessionFactory().openSession();
         Profile profile = session.find(Profile.class, userId);
@@ -81,9 +204,9 @@ public static Profile retrieveProfile(long userId) {
 
             // add initial betting credits
             Bet bet = new Bet();
-            bet.setAmount(100);
-            bet.setAutomated(true);
-            bet.setSuccessful(true);
+            bet.setKind(BetKind.AUTOMATED);
+            bet.setBet(0);
+            bet.setPayout(100);
             bet.setReason("Initial betting credits");
             bet.setUserId(userId);
             session.save(bet);

src/main/java/pw/chew/mlb/commands/PlanGameCommand.java

@@ -27,6 +27,7 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Calendar;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -171,6 +172,16 @@ public void handle(InteractionHook event, String gamePk, GuildChannelUnion chann
 
     @Override
     public void onAutoComplete(CommandAutoCompleteInteractionEvent event) {
+        event.replyChoices(handleAutoComplete(event)).queue();
+    }
+
+    /**
+     * Handles auto complete for: teams (MLB), sports, and dates (for the team)
+     *
+     * @param event the event
+     * @return a list of choices, maybe empty, maybe 1 choice, maybe 25 choices, who knows.
+     */
+    public static List<Command.Choice> handleAutoComplete(CommandAutoCompleteInteractionEvent event) {
         switch (event.getFocusedOption().getName()) {
             case "team" -> {
                 // get current value of sport
@@ -187,20 +198,17 @@ public void onAutoComplete(CommandAutoCompleteInteractionEvent event) {
                 // Ensure no duplicates and no more than 25 choices
                 choices = choices.stream().distinct().limit(25).toList();
 
-                event.replyChoices(choices).queue();
-                return;
+                return choices;
             }
             case "sport" -> {
-                event.replyChoices(MLBAPIUtil.getSports().asChoices()).queue();
-                return;
+                return MLBAPIUtil.getSports().asChoices();
             }
             case "date" -> {
                 int teamId = event.getOption("team", -1, OptionMapping::getAsInt);
                 String sport = event.getOption("sport", "1", OptionMapping::getAsString);
 
                 if (teamId == -1) {
-                    event.replyChoices(new Command.Choice("Please select a team first!", -1)).queue();
-                    return;
+                    return Collections.singletonList(new Command.Choice("Please select a team first!", -1));
                 }
 
                 JSONArray games = new JSONObject(RestClient.get("https://statsapi.mlb.com/api/v1/schedule?lang=en&sportId=%S&season=2023&teamId=%S&fields=dates,date,games,gamePk,teams,away,team,teamName,id&hydrate=team".formatted(sport, teamId)))
@@ -245,13 +253,11 @@ public void onAutoComplete(CommandAutoCompleteInteractionEvent event) {
                 // Ensure no more than 25 choices, and no duplicates
                 choices = choices.stream().distinct().limit(25).toList();
 
-                event.replyChoices(choices).queue();
-
-                return;
+                return choices;
             }
         }
 
-        event.replyChoices().queue();
+        return Collections.emptyList();
     }
 
     public static List<Button> buildButtons(String gamePk, GameBlurb blurb) {

src/main/java/pw/chew/mlb/models/Bet.kt

@@ -14,18 +14,36 @@ open class Bet {
     @Column(name = "user_id", nullable = false)
     open var userId: Long? = null
 
-    @Column(name = "amount", nullable = false)
-    open var amount: Int = 0
+    @Column(name = "bet", nullable = false)
+    open var bet: Int = 0
 
-    @Column(name = "successful", nullable = false)
-    open var successful: Boolean = false
+    @Column(name = "payout", nullable = false)
+    open var payout: Int = 0
 
-    @Column(name = "automated", nullable = false)
-    open var automated: Boolean = false
+    @Column(name = "game_pk")
+    open var gamePk: Int? = null
+
+    @Column(name = "team_id")
+    open var teamId: Int? = null
+
+    @Enumerated(EnumType.STRING)
+    @Column(name = "kind", nullable = false)
+    open var kind: BetKind = BetKind.PENDING
 
     @Column(name = "reason", nullable = false, length = 128)
     open var reason: String = ""
 
     @Column(name = "created_at", nullable = false)
     open var createdAt: Instant = Instant.now()
+
+    fun amount(): Int {
+        return payout - bet
+    }
 }
+
+enum class BetKind {
+    AUTOMATED,
+    PENDING,
+    WIN,
+    LOSS
+}

src/main/java/pw/chew/mlb/objects/GameState.java

@@ -69,7 +69,7 @@ public OffsetDateTime officialDate() {
     }
 
     public int inning() {
-        return lineScore().getInt("currentInning");
+        return lineScore().optInt("currentInning", 0);
     }
 
     public String inningOrdinal() {