1 files changed, 124 insertions, 0 deletions

diff --git a/minesweeper/game.py b/minesweeper/game.py
new file mode 100644
index 0000000..9ef7b16
--- /dev/null
+++ b/minesweeper/game.py
@@ -0,0 +1,124 @@
+from minesweeper.field import Field
+
+
+class Help:
+    def set_state(self, loop):
+        self.loop = loop
+
+    def print(self):
+        print("""Listed commands:
+ try    <x>    <y>    Tests for mines
+ flag   <x>    <y>    Places flag
+ restart              Starts new game
+ quit or exit         Quits game
+ help                 Prints list of commands
+""")
+        self.loop.command()
+
+
+class Setup:
+    def set_state(self, loop):
+        self.loop = loop
+
+    def setup(self):
+        print("""Select diffeculty:
+ 1. Beginner        (10 mines, 9x9)
+ 2. Intermediate    (40 mines, 16x16)
+ 3. Expert          (99 mines, 30x16)
+ 4. Custom"
+""")
+
+        choice = int(
+                input("Choice: ").split()[0])
+
+        if choice == 1:
+            width = 9
+            height = 9
+            mines = 10
+        elif choice == 2:
+            width = 16
+            height = 16
+            mines = 40
+        elif choice == 3:
+            width = 30
+            height = 16
+            mines = 99
+        elif choice == 4:
+            width = int(input("Width: "))
+            height = int(input("Height: "))
+            mines = int(input("Mines: "))
+        else:
+            print(str(choice) + " is not a option.")
+            self.setup()
+
+        minefield = Field(width, height, mines)
+        self.loop.set_minefield(minefield)
+        self.loop.command()
+
+
+class Loop:
+    def set_states(self, setup, help, end):
+        self.setup = setup
+        self.end = end
+        self.help = help
+
+    def set_minefield(self, minefield):
+        self.minefield = minefield
+
+    def command(self):
+        self.minefield.print()
+        if self.minefield.cleared:
+            self.end.end_game(False)
+
+        command = input("Command >> ").split(' ')
+
+        if len(command) == 3:
+            coordinate = (ord(command[1]) - 65, int(command[2]) - 1)
+
+        if command[0] == "try":
+            if self.minefield.guess(*coordinate):
+                self.minefield.print()
+                self.end.end_game(dead=True)
+                return
+            self.command()
+        elif command[0] == "flag":
+            self.minefield.flag(*coordinate)
+            self.command()
+        #  elif command[0] == "?":
+        #      self.minefield.question(coordinate)
+        #      self.command()
+        elif command[0] == "restart":
+            self.setup.setup()
+        elif command[0] == "quit" or command[0] == "exit":
+            exit()
+        elif command[0] == "help":
+            self.help.print()
+        else:
+            print(command[0] + " is not a recognized command")
+            self.help.print()
+
+
+class End:
+    def set_state(self, setup):
+        self.setup = setup
+
+    def restart(self):
+        choice = input("Do you want to start again? [y/n] ").split()[0]
+
+        if choice == "y":
+            self.setup.setup()
+        elif choice == "n":
+            exit()
+        else:
+            print(str(choice) + " is not a valid choice")
+            self.restart()
+
+    def end_game(self, dead):
+        print("")
+
+        if dead:
+            print("You hit a mine :o")
+        else:
+            print("You won :D")
+
+        self.restart()