only show post-season teams by default in /plangame

Author: Chew

src/main/java/pw/chew/mlb/util/MLBAPIUtil.java

@@ -10,6 +10,7 @@
 
 import java.time.Duration;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
@@ -143,10 +144,25 @@ public List<Command.Choice> asChoices() {
             return choices.subList(0, Math.min(choices.size(), 25));
         }
 
+        public List<Command.Choice> asPostSeasonChoices() {
+            // we only want people in the post-season. These teams are: 147, 110, 114, 118, 116, 117, 143, 121, 158, 119, 135, 109, 144
+            List<Integer> postSeason = Arrays.asList(147, 110, 114, 118, 116, 117, 143, 121, 158, 119, 135, 109, 144);
+
+            List<Command.Choice> choices = new ArrayList<>();
+            for (int i = 0; i < raw.length(); i++) {
+                JSONObject team = raw.getJSONObject(i);
+                if (postSeason.contains(team.getInt("id"))) {
+                    choices.add(new Command.Choice(team.getString("name"), team.getInt("id")));
+                }
+            }
+
+            return choices;
+        }
+
         public List<Command.Choice> potentialChoices(String query) {
             // If we don't have an input, return all choices
             if (query.isBlank()) {
-                return asChoices();
+                return asPostSeasonChoices();
             }
 
             List<Command.Choice> potential = new ArrayList<>();