Merge pull request #21 from RIT-SWEN-262/league-model

League model

1 files changed, 31 insertions, 0 deletions

diff --git a/src/main/java/design/model/Team.java b/src/main/java/design/model/Team.java
index 53b276e..8f30468 100644
--- a/src/main/java/design/model/Team.java
+++ b/src/main/java/design/model/Team.java
@@ -1,17 +1,22 @@
 package design.model;
 
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Objects;
 
 public class Team {
     private String name;
     private final List<Golfer> members;
+    private final Map<Golfer, Integer> memberBestRounds;
     private final Golfer owner;
 
     public Team(String name, Golfer owner) {
         this.name = name;
         this.owner = owner;
         this.members = new ArrayList<>();
+        this.memberBestRounds = new HashMap<>();
     }
 
     public String getName() {
@@ -27,14 +32,40 @@ public class Team {
     }
 
     public boolean addMember(Golfer golfer) {
+        if(members.size() >= 4){
+            throw new IllegalArgumentException("Team size limit reached!");
+        }
         return members.add(golfer);
     }
 
     public boolean removeMember(Golfer g) {
+        memberBestRounds.remove(g);
         return members.remove(g);
     }
 
     public Golfer getOwner() {
         return owner;
     }
+
+    public void addMemberRound(Golfer g, int strokes){
+        memberBestRounds.merge(g, strokes, Math::min);
+    }
+
+    public int computeTotalBestScore(){
+        if(memberBestRounds.isEmpty()) return 100; // Bad Score Penalty for not playing
+        return memberBestRounds.values().stream().min(Integer::compareTo).orElse(100);
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj) return true;
+        if (obj == null || getClass() != obj.getClass()) return false;
+        Team other = (Team) obj;
+        return Objects.equals(name, other.name) && Objects.equals(owner, other.owner);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(name, owner);
+    }
 }