1 files changed, 33 insertions, 0 deletions
diff --git a/src/main/java/design/model/holeplay/SetupState.java b/src/main/java/design/model/holeplay/SetupState.java
new file mode 100644
index 0000000..d49ea0e
--- /dev/null
+++ b/src/main/java/design/model/holeplay/SetupState.java
@@ -0,0 +1,33 @@
+package design.model.holeplay;
+
+import design.model.Hole;
+
+public class SetupState implements HoleState {
+
+    @Override
+    public void enter(HolePlayContext ctx) {
+        ctx.clearInProgressPlay(); // Maybe we don't need this?
+    }
+
+    @Override
+    public void handleStart(HolePlayContext ctx) {
+        Hole hole = ctx.getRound().getCurrentHole();
+        ctx.beginNewPlay(hole.getNumber());
+        ctx.setState(new PlayState());
+    }
+
+    @Override
+    public void handleShot(HolePlayContext ctx, design.model.Club club, Integer distanceYds) {
+        throw new IllegalStateException("Cannot record a shot during setup.");
+    }
+
+    @Override
+    public void handleHoleOut(HolePlayContext ctx) {
+        throw new IllegalStateException("Cannot hole out during setup.");
+    }
+
+    @Override
+    public String name() {
+        return "SetupState";
+    }
+}