1 files changed, 112 insertions, 0 deletions
diff --git a/src/main/java/design/model/holeplay/HolePlayContext.java b/src/main/java/design/model/holeplay/HolePlayContext.java
new file mode 100644
index 0000000..7a5d8ef
--- /dev/null
+++ b/src/main/java/design/model/holeplay/HolePlayContext.java
@@ -0,0 +1,112 @@
+package design.model.holeplay;
+
+import java.io.IOException;
+import java.util.ArrayList;
+
+import design.model.*;
+import design.persistence.PersonalDatabase;
+
+public class HolePlayContext {
+    private final Golfer golfer;
+    private final Round round;
+    private final PersonalDatabase pdb;
+
+    private HoleState state;
+    private Play currentPlay;
+    private int strokes;
+
+    public HolePlayContext(Golfer golfer, Round round, PersonalDatabase pdb) {
+        this.golfer = golfer;
+        this.round = round;
+        this.pdb = pdb;
+        setState(new SetupState());
+    }
+
+    public void setState(HoleState next) {
+        this.state = next;
+        next.enter(this);
+    }
+
+    // API the PTUI/Commands will call:
+    public void startHole() {
+        state.handleStart(this);
+    }
+
+    public void recordShot(Club club, Integer distanceYds) {
+        state.handleShot(this, club, distanceYds);
+    }
+
+    public void holeOut() {
+        state.handleHoleOut(this);
+    }
+
+    void beginNewPlay(int holeNumber) {
+        this.currentPlay = new Play(holeNumber);
+        this.strokes = 0;
+    }
+
+    void addSwing(Club club, Integer distanceYds) {
+        // Checks golfer owns club maybe not necessary
+        // if (golfer.hasClub(club)) {
+        // throw new IllegalArgumentException("Club not in golfer's bag");
+        // }
+        int d = (distanceYds == null ? 0 : distanceYds);
+        this.currentPlay.addSwing(new Swing(d, club));
+        this.strokes++;
+
+    }
+
+    void finalizePlayAndPersist() {
+        round.addPlay(currentPlay);
+        round.nextHole();
+        currentPlay = null;
+        strokes = 0;
+        // Not sure if try catch is necessary
+        try {
+            pdb.updateGolfer(golfer);
+        } catch (IOException e) {
+            // TODO Auto-generated catch block
+            e.printStackTrace();
+        }
+    }
+
+    public void clearInProgressPlay() {
+        this.setCurrentPlay(null);
+        this.setStrokes(0);
+    }
+
+    public void endRoundNow() {
+        try {
+            pdb.updateGolfer(golfer); // persist whatever’s done so far
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        setState(new SetupState());
+    }
+
+    // getters you’ll want:
+    public Golfer getGolfer() {
+        return golfer;
+    }
+
+    public Round getRound() {
+        return round;
+    }
+
+    public Play getCurrentPlay() {
+        return currentPlay;
+    }
+
+    public int getStrokes() {
+        return strokes;
+    }
+
+    // setters used internally
+    void setCurrentPlay(Play p) {
+        this.currentPlay = p;
+    }
+
+    void setStrokes(int s) {
+        this.strokes = s;
+    }
+}
+\ No newline at end of file