questPortrait callback

1 files changed, 15 insertions, 11 deletions

diff --git a/source/game/scripting/StarPlayerLuaBindings.cpp b/source/game/scripting/StarPlayerLuaBindings.cpp
index 4d3b316..5af86d0 100644
--- a/source/game/scripting/StarPlayerLuaBindings.cpp
+++ b/source/game/scripting/StarPlayerLuaBindings.cpp
@@ -478,24 +478,28 @@ LuaCallbacks LuaBindings::makePlayerCallbacks(Player* player) {
   });
 
   callbacks.registerCallback("quest", [player](String const& questId) -> Json {
-    if (auto quest = player->questManager()->getQuest(questId))
-      return quest->diskStore();
+    if (!player->questManager()->hasQuest(questId))
+      return {};
+    return player->questManager()->getQuest(questId)->diskStore();
+  });
 
-    return {};
+  callbacks.registerCallback("questPortrait", [player](String const& questId, String const& portraitName) -> Maybe<List<Drawable>> {
+    if (!player->questManager()->hasQuest(questId))
+      return {};
+    return player->questManager()->getQuest(questId)->portrait(portraitName);
   });
 
-  callbacks.registerCallback("questState", [player](String const& questId) -> Maybe<String> {
-    if (auto quest = player->questManager()->getQuest(questId))
-      return QuestStateNames.getRight(quest->state());
 
-    return {};
+  callbacks.registerCallback("questState", [player](String const& questId) -> Maybe<String> {
+    if (!player->questManager()->hasQuest(questId))
+      return {};
+    return QuestStateNames.getRight(player->questManager()->getQuest(questId)->state());
   });
 
   callbacks.registerCallback("callQuest", [player](String const& questId, String const& func, LuaVariadic<LuaValue> const& args) -> Maybe<LuaValue> {
-    if (auto quest = player->questManager()->getQuest(questId))
-      return quest->callScript(func, args);
-    
-    return {};
+    if (!player->questManager()->hasQuest(questId))
+      return {};
+    return player->questManager()->getQuest(questId)->callScript(func, args);
   });
 
   callbacks.registerCallback("hasQuest", [player](String const& questId) {