1 files changed, 136 insertions, 0 deletions
diff --git a/source/game/StarPlayerCompanions.cpp b/source/game/StarPlayerCompanions.cpp
new file mode 100644
index 0000000..ef59ef0
--- /dev/null
+++ b/source/game/StarPlayerCompanions.cpp
@@ -0,0 +1,136 @@
+#include "StarPlayerCompanions.hpp"
+#include "StarConfigLuaBindings.hpp"
+#include "StarEntityLuaBindings.hpp"
+#include "StarJsonExtra.hpp"
+#include "StarPlayer.hpp"
+#include "StarPlayerLuaBindings.hpp"
+#include "StarStatusControllerLuaBindings.hpp"
+
+namespace Star {
+
+Companion::Companion(Json const& json) : m_json(json) {
+  m_portrait = json.getArray("portrait", JsonArray{}).transformed(construct<Drawable>());
+}
+
+Json Companion::toJson() const {
+  return m_json;
+}
+
+Uuid Companion::podUuid() const {
+  return Uuid(m_json.getString("podUuid"));
+}
+
+Maybe<String> Companion::name() const {
+  return m_json.optString("name");
+}
+
+Maybe<String> Companion::description() const {
+  return m_json.optString("description");
+}
+
+List<Drawable> Companion::portrait() const {
+  return m_portrait;
+}
+
+Maybe<float> Companion::resource(String const& resourceName) const {
+  if (auto status = m_json.opt("status"))
+    if (auto resources = status->opt("resources"))
+      return resources->optFloat(resourceName);
+  return {};
+}
+
+Maybe<float> Companion::resourceMax(String const& resourceName) const {
+  if (auto status = m_json.opt("status"))
+    if (auto resources = status->opt("resourceMax"))
+      return resources->optFloat(resourceName);
+  return {};
+}
+
+Maybe<float> Companion::stat(String const& statName) const {
+  if (auto status = m_json.opt("status"))
+    if (auto stats = status->opt("stats"))
+      return stats->optFloat(statName);
+  return {};
+}
+
+PlayerCompanions::PlayerCompanions(Json const& config) : m_config(config) {}
+
+void PlayerCompanions::diskLoad(Json const& diskStore) {
+  m_scriptComponent.setScriptStorage(diskStore.getObject("scriptStorage", JsonObject{}));
+  m_companions = jsonToMapV<StringMap<List<CompanionPtr>>>(diskStore.getObject("companions", JsonObject{}),
+      [](Json const& companions) {
+        return companions.toArray().transformed([](Json const& json) { return make_shared<Companion>(json); });
+      });
+}
+
+Json PlayerCompanions::diskStore() const {
+  JsonObject result;
+  result["scriptStorage"] = m_scriptComponent.getScriptStorage();
+  result["companions"] = jsonFromMapV(m_companions,
+      [](List<CompanionPtr> const& companions) { return companions.transformed(mem_fn(&Companion::toJson)); });
+  return result;
+}
+
+List<CompanionPtr> PlayerCompanions::getCompanions(String const& category) const {
+  if (m_companions.contains(category))
+    return m_companions.get(category);
+  return {};
+}
+
+void PlayerCompanions::init(Entity* player, World* world) {
+  m_world = world;
+
+  m_scriptComponent.setScripts(jsonToStringList(m_config.getArray("scripts", JsonArray())));
+  m_scriptComponent.setUpdateDelta(m_config.getInt("scriptDelta", 10));
+
+  m_scriptComponent.addCallbacks("entity", LuaBindings::makeEntityCallbacks(player));
+  m_scriptComponent.addCallbacks("player", LuaBindings::makePlayerCallbacks(as<Player>(player)));
+  m_scriptComponent.addCallbacks(
+      "status", LuaBindings::makeStatusControllerCallbacks(as<Player>(player)->statusController()));
+  m_scriptComponent.addCallbacks("playerCompanions", makeCompanionsCallbacks());
+
+  m_scriptComponent.addCallbacks("config",
+      LuaBindings::makeConfigCallbacks([this](
+          String const& name, Json const& def) { return m_config.query(name, def); }));
+
+  m_scriptComponent.init(world);
+}
+
+void PlayerCompanions::uninit() {
+  m_scriptComponent.uninit();
+  m_scriptComponent.removeCallbacks("entity");
+  m_scriptComponent.removeCallbacks("player");
+  m_scriptComponent.removeCallbacks("status");
+  m_scriptComponent.removeCallbacks("playerCompanions");
+  m_scriptComponent.removeCallbacks("config");
+  m_world = nullptr;
+}
+
+void PlayerCompanions::dismissCompanion(String const& category, Uuid const& podUuid) {
+  m_scriptComponent.invoke("dismissCompanion", category, podUuid.hex());
+}
+
+Maybe<Json> PlayerCompanions::receiveMessage(String const& message, bool localMessage, JsonArray const& args) {
+  return m_scriptComponent.handleMessage(message, localMessage, args);
+}
+
+void PlayerCompanions::update() {
+  m_scriptComponent.update(m_scriptComponent.updateDt());
+}
+
+LuaCallbacks PlayerCompanions::makeCompanionsCallbacks() {
+  LuaCallbacks callbacks;
+
+  callbacks.registerCallback("getCompanions",
+      [this](String const& category) {
+        return m_companions[category].transformed([](CompanionPtr const& companion) { return companion->toJson(); });
+      });
+  callbacks.registerCallback("setCompanions",
+      [this](String const& category, JsonArray const& companions) {
+        m_companions[category] = companions.transformed([](Json const& json) { return make_shared<Companion>(json); });
+      });
+
+  return callbacks;
+}
+
+}