Made output random

2 files changed, 17 insertions, 0 deletions

diff --git a/include/move.h b/include/move.h
new file mode 100644
index 0000000..e0e9b6b
--- /dev/null
+++ b/include/move.h
@@ -0,0 +1,14 @@
+#ifndef MOVE_H
+#define MOVE_H
+
+enum class Move: int {
+    NOTHING = 0,
+    MOVE_LEFT = 1,
+    MOVE_RIGHT = 2,
+    SHOOT = 3,
+    BUILD_ALIEN_FACTORY = 4,
+    BUILD_MISSILE_CONTROLLER = 5,
+    BUILD_SHIELD = 6
+};
+
+#endif
diff --git a/include/spacebot.h b/include/spacebot.h
index 920eb73..a5c23c2 100644
--- a/include/spacebot.h
+++ b/include/spacebot.h
@@ -2,6 +2,7 @@
 
 #include <iostream>
 #include <fstream>
+#include "move.h"
 
 class Spacebot {
 public:
@@ -10,4 +11,6 @@ public:
 private:
     std::ifstream mapStream;
     std::ofstream resultStream;
+    void writeMove(const Move& move);
+    Move chooseMove();
 };