Lua: Library quest: Move HOOK_STAIR code to C

1 files changed, 46 insertions, 0 deletions

diff --git a/src/q_library.c b/src/q_library.c
index d7a098cd..8804e2dc 100644
--- a/src/q_library.c
+++ b/src/q_library.c
@@ -12,6 +12,16 @@ static int LIBRARY_QUEST()
 	return get_lua_int("LIBRARY_QUEST");
 }
 
+static int library_quest_get_status()
+{
+	return exec_lua("return quest(LIBRARY_QUEST).status");
+}
+
+static void library_quest_set_status(int new_status)
+{
+	exec_lua(format("quest(LIBRARY_QUEST).status = %d", new_status));
+}
+
 static s16b library_quest_place_random(int minY, int minX, int maxY, int maxX, int r_idx)
 {
 	int y = randint(maxY - minY + 1) + minY;
@@ -262,6 +272,42 @@ bool_ quest_library_gen_hook()
 	return TRUE;
 }
 
+bool_ quest_library_stair_hook()
+{
+	bool_ ret;
+
+	/* only ask this if player about to go up stairs of quest and hasn't won yet */
+	if ((p_ptr->inside_quest != LIBRARY_QUEST()) ||
+	    (library_quest_get_status() == QUEST_STATUS_COMPLETED))
+	{
+		return FALSE;
+	}
+
+	if (cave[p_ptr->py][p_ptr->px].feat != FEAT_LESS)
+	{
+		return FALSE;
+	}
+
+	/* flush all pending input */
+	flush();
+
+	/* confirm */
+	ret = get_check("Really abandon the quest?");
+
+	/* if yes, then */
+	if (ret == TRUE)
+	{
+		/* fail the quest */
+		library_quest_set_status(QUEST_STATUS_FAILED);
+		return FALSE;
+	}
+	else
+	{
+		/* if no, they stay in the quest */
+		return TRUE;
+	}
+}
+
 static int get_status()
 {
 	return exec_lua("return quest(LIBRARY_QUEST).status");