1 files changed, 39 insertions, 6 deletions
diff --git a/src/main/java/design/model/ScrambleLeague.java b/src/main/java/design/model/ScrambleLeague.java
index ad115d4..1fb8f1d 100644
--- a/src/main/java/design/model/ScrambleLeague.java
+++ b/src/main/java/design/model/ScrambleLeague.java
@@ -4,31 +4,41 @@ import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonTypeName;
 
 import java.util.ArrayList;
-import java.util.Arrays;
+import java.util.Collections;
+import java.util.Comparator;
 import java.util.Date;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 @JsonTypeName("scramble")
 public class ScrambleLeague extends League {
-    private final List<Team> participants;
+    private List<Team> participants;
+    private Map<Team, Integer> totalTeamScores;
 
     @JsonCreator
     private ScrambleLeague(int id, String name, Date registrationDate, Date startDate, Date endDate, Golfer owner, List<Team> participants, List<Match> schedule) {
         super(id, name, registrationDate, startDate, endDate, owner, schedule);
         this.participants = participants;
+        this.totalTeamScores = new HashMap<>();
+        participants.forEach(t -> totalTeamScores.putIfAbsent(t, 0));
     }
 
     public ScrambleLeague(String name, Date registrationDate, Date startDate, Date endDate, Golfer owner) {
         super(name, registrationDate, startDate, endDate, owner);
         this.participants = new ArrayList<>();
+        this.totalTeamScores = new HashMap<>();
     }
 
-    public boolean addParticipants(Team e) {
-        return participants.add(e);
+    public boolean addParticipants(Team t) {
+        boolean added = participants.add(t);
+        if(added) totalTeamScores.putIfAbsent(t, 0);
+        return added;
     }
 
-    public boolean removeParticipants(Team o) {
-        return participants.remove(o);
+    public boolean removeParticipants(Team t) {
+        totalTeamScores.remove(t);
+        return participants.remove(t);
     }
 
     public Team[] getParticipants() {
@@ -48,4 +58,27 @@ public class ScrambleLeague extends League {
     public String getType() {
         return "scramble";
     }
+    @Override
+    public void recordPlay(Golfer player, Match match, Round round){
+        if(!isPlayable()) return;
+        Team team = player.getTeam();
+        if (team == null) return;
+        int strokes = round.getTotalSwings();
+        team.addMemberRound(player, strokes);
+        match.addRound(round);
+    }
+
+    @Override
+    public void finalizeLeague(){
+        markCompleted();
+        for(Team team : participants){
+            int score = team.computeTotalBestScore();
+            totalTeamScores.put(team, score);
+        }
+        participants.sort(Comparator.comparingInt(totalTeamScores::get));
+    }
+
+    public Map<Team, Integer> getTotalTeamScores(){
+        return Collections.unmodifiableMap(totalTeamScores);
+    }
 }