1 files changed, 10 insertions, 4 deletions
diff --git a/battles/include/kurator/battles/Battle.h b/battles/include/kurator/battles/Battle.h
index 468a972..82a5da5 100644
--- a/battles/include/kurator/battles/Battle.h
+++ b/battles/include/kurator/battles/Battle.h
@@ -1,5 +1,7 @@
 #pragma once
 
+#include <memory>
+
 #include <entt/entity/registry.hpp>
 
 #include "Scenario.h"
@@ -11,13 +13,17 @@ namespace battles
 {
 
 
-struct Battle
+class Battle
 {
-	explicit Battle(Scenario scenario);
-	void update(float dt);
-	entt::registry registry;
+public:
+	virtual ~Battle() = default;
+	virtual entt::registry& registry() = 0;
+	virtual void update(float dt) = 0;
 };
 
 
+auto prepare(const Scenario& scenario) -> std::unique_ptr<Battle>;
+
+
 }  // namespace battles
 }  // namespace kurator