7 files changed, 310 insertions, 2 deletions
diff --git a/src/main/java/design/model/Club.java b/src/main/java/design/model/Club.java
index b12fd99..c19811c 100644
--- a/src/main/java/design/model/Club.java
+++ b/src/main/java/design/model/Club.java
@@ -1,5 +1,6 @@
 package design.model;
 
+import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonIdentityInfo;
 import com.fasterxml.jackson.annotation.ObjectIdGenerators;
 
@@ -21,18 +22,26 @@ public class Club {
         PUTTER
     }
 
-    private final int id;
+    private int id;
     private final String manufacture;
     private final String nickname;
     private final ClubType clubType;
 
-    public Club(int id, String manufacture, String nickname, ClubType clubType) {
+    @JsonCreator
+    private Club(int id, String manufacture, String nickname, ClubType clubType) {
         this.id = id;
         this.manufacture = manufacture;
         this.nickname = nickname;
         this.clubType = clubType;
     }
 
+    public Club(String manufacture, String nickname, ClubType clubType) {
+        this.id = -1;
+        this.manufacture = manufacture;
+        this.nickname = nickname;
+        this.clubType = clubType;
+    }
+
     public int getId() {
         return id;
     }
@@ -49,6 +58,11 @@ public class Club {
         return clubType;
     }
 
+    public void setId(int id) {
+        assert this.id == -1;
+        this.id = id;
+    }
+
     @Override
     public String toString() {
         return String.format("#%d %s - %s (%s)", id, nickname, manufacture, clubType);
diff --git a/src/main/java/design/model/Invite.java b/src/main/java/design/model/Invite.java
new file mode 100644
index 0000000..479f520
--- /dev/null
+++ b/src/main/java/design/model/Invite.java
@@ -0,0 +1,25 @@
+package design.model;
+
+import java.util.Date;
+
+public class Invite {
+    private final Team team;
+    private final Date sendDate;
+
+    public Invite(Team team, Date sendDate) {
+        this.team = team;
+        this.sendDate = sendDate;
+    }
+
+    public Team getTeam() {
+        return team;
+    }
+
+    public Date getSendDate() {
+        return sendDate;
+    }
+
+    public void accept(Golfer g) {
+        team.addMember(g);
+    }
+}
diff --git a/src/main/java/design/model/League.java b/src/main/java/design/model/League.java
new file mode 100644
index 0000000..881e25f
--- /dev/null
+++ b/src/main/java/design/model/League.java
@@ -0,0 +1,90 @@
+package design.model;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonSubTypes;
+import com.fasterxml.jackson.annotation.JsonTypeInfo;
+
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+@JsonTypeInfo(
+        use = JsonTypeInfo.Id.NAME,
+        include = JsonTypeInfo.As.PROPERTY,
+        property = "type"
+)
+@JsonSubTypes({
+        @JsonSubTypes.Type(value = ScrambleLeague.class, name = "scramble"),
+        @JsonSubTypes.Type(value = StrokeLeague.class, name = "stroke")
+})
+public abstract class League {
+    private int id;
+    private final String name;
+    private final Date registrationDate;
+    private final Date startDate;
+    private final Date endDate;
+    private final Golfer owner;
+    private final List<Match> schedule;
+
+    @JsonCreator
+    protected League(int id, String name, Date registrationDate, Date startDate, Date endDate, Golfer owner, List<Match> schedule) {
+        this.id = id;
+        this.name = name;
+        this.registrationDate = registrationDate;
+        this.startDate = startDate;
+        this.endDate = endDate;
+        this.owner = owner;
+        this.schedule = schedule != null ? schedule : new ArrayList<>();
+    }
+
+    public League(String name, Date registrationDate, Date startDate, Date endDate, Golfer owner) {
+        this.id = -1;
+        this.name = name;
+        this.registrationDate = registrationDate;
+        this.startDate = startDate;
+        this.endDate = endDate;
+        this.owner = owner;
+        this.schedule = new ArrayList<>();
+    }
+
+    public int getId() {
+        return id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public Date getRegistrationDate() {
+        return registrationDate;
+    }
+
+    public Date getStartDate() {
+        return startDate;
+    }
+
+    public Date getEndDate() {
+        return endDate;
+    }
+
+    public Golfer getOwner() {
+        return owner;
+    }
+
+    public Match[] getSchedule() {
+        return schedule.toArray(Match[]::new);
+    }
+
+    public void addMatchToSchedule(Match match) {
+        Date date = match.getDateScheduled();
+        if(date.after(endDate)){
+            throw new IllegalArgumentException("Cannot create match after league has ended");
+        }
+        schedule.add(match);
+    }
+
+    public void setId(int id) {
+        assert this.id == -1;
+        this.id = id;
+    }
+}
diff --git a/src/main/java/design/model/Match.java b/src/main/java/design/model/Match.java
new file mode 100644
index 0000000..4f01c98
--- /dev/null
+++ b/src/main/java/design/model/Match.java
@@ -0,0 +1,57 @@
+package design.model;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+public class Match {
+    private final Course course;
+    private final Date dateScheduled;
+    private final LocalDateTime start;
+    private final int holeCount;
+    private final List<Round> rounds;
+
+    @JsonCreator
+    private Match(Course course, Date dateScheduled, LocalDateTime start, int holeCount, List<Round> rounds) {
+        this.course = course;
+        this.dateScheduled = dateScheduled;
+        this.start = start;
+        this.holeCount = holeCount;
+        this.rounds = rounds;
+    }
+
+    public Match(Course course, Date dateScheduled, LocalDateTime start, int holeCount) {
+        this.course = course;
+        this.dateScheduled = dateScheduled;
+        this.start = start;
+        this.holeCount = holeCount;
+        this.rounds = new ArrayList<>();
+    }
+
+    public Course getCourse() {
+        return course;
+    }
+
+    public Date getDateScheduled() {
+        return dateScheduled;
+    }
+
+    public LocalDateTime getStart() {
+        return start;
+    }
+
+    public int getHoleCount() {
+        return holeCount;
+    }
+
+    public boolean addRound(Round round) {
+        return rounds.add(round);
+    }
+
+    public Round[] getRounds() {
+        return rounds.toArray(Round[]::new);
+    }
+}
diff --git a/src/main/java/design/model/ScrambleLeague.java b/src/main/java/design/model/ScrambleLeague.java
new file mode 100644
index 0000000..d372264
--- /dev/null
+++ b/src/main/java/design/model/ScrambleLeague.java
@@ -0,0 +1,46 @@
+package design.model;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonTypeName;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Date;
+import java.util.List;
+
+@JsonTypeName("scramble")
+public class ScrambleLeague extends League {
+    private final List<Team> participants;
+
+    @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;
+    }
+
+    public ScrambleLeague(String name, Date registrationDate, Date startDate, Date endDate, Golfer owner) {
+        super(name, registrationDate, startDate, endDate, owner);
+        this.participants = new ArrayList<>();
+    }
+
+    public boolean addParticipants(Team e) {
+        return participants.add(e);
+    }
+
+    public boolean removeParticipants(Team o) {
+        return participants.remove(o);
+    }
+
+    public Team[] getParticipants() {
+        return participants.toArray(Team[]::new);
+    }
+
+    public Team locateTeam(Golfer golfer) {
+        for (Team participant : participants) {
+            if (List.of(participant.getMembers()).contains(golfer)) {
+                return participant;
+            }
+        }
+        return null;
+    }
+}
diff --git a/src/main/java/design/model/StrokeLeague.java b/src/main/java/design/model/StrokeLeague.java
new file mode 100644
index 0000000..2b787b5
--- /dev/null
+++ b/src/main/java/design/model/StrokeLeague.java
@@ -0,0 +1,36 @@
+package design.model;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonTypeName;
+
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+@JsonTypeName("stroke")
+public class StrokeLeague extends League {
+    private final List<Golfer> participants;
+
+    @JsonCreator
+    private StrokeLeague(int id, String name, Date registrationDate, Date startDate, Date endDate, Golfer owner, List<Golfer> participants, List<Match> schedule) {
+        super(id, name, registrationDate, startDate, endDate, owner, schedule);
+        this.participants = participants;
+    }
+
+    public StrokeLeague(String name, Date registrationDate, Date startDate, Date endDate, Golfer owner) {
+        super(name, registrationDate, startDate, endDate, owner);
+        this.participants = new ArrayList<>();
+    }
+
+    public boolean addParticipants(Golfer e) {
+        return participants.add(e);
+    }
+
+    public boolean removeParticipants(Golfer o) {
+        return participants.remove(o);
+    }
+
+    public Golfer[] getParticipants() {
+        return participants.toArray(Golfer[]::new);
+    }
+}
diff --git a/src/main/java/design/model/Team.java b/src/main/java/design/model/Team.java
new file mode 100644
index 0000000..53b276e
--- /dev/null
+++ b/src/main/java/design/model/Team.java
@@ -0,0 +1,40 @@
+package design.model;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class Team {
+    private String name;
+    private final List<Golfer> members;
+    private final Golfer owner;
+
+    public Team(String name, Golfer owner) {
+        this.name = name;
+        this.owner = owner;
+        this.members = new ArrayList<>();
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public Golfer[] getMembers() {
+        return members.toArray(Golfer[]::new);
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public boolean addMember(Golfer golfer) {
+        return members.add(golfer);
+    }
+
+    public boolean removeMember(Golfer g) {
+        return members.remove(g);
+    }
+
+    public Golfer getOwner() {
+        return owner;
+    }
+}