Add lua downloadablePack:IsQueued()

etternagame · Jul 18, 2018 · 6e13714 · 6e13714
1 parent 6c5cb16
commit 6e13714
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/DownloadManager.cpp b/src/DownloadManager.cpp
@@ -1649,6 +1649,7 @@ class LunaDownloadablePack : public Luna<DownloadablePack>
 		}
 		else
 			lua_pushnil(L);
+		IsQueued(p, L);
 		return 1;
 	}
 	static int GetName(T* p, lua_State* L)
@@ -1666,6 +1667,12 @@ class LunaDownloadablePack : public Luna<DownloadablePack>
 		lua_pushnumber(L, p->avgDifficulty);
 		return 1;
 	}
+	static int IsQueued(T* p, lua_State* L)
+	{
+		auto it = std::find(DLMAN->DownloadQueue.begin(), DLMAN->DownloadQueue.end(), p);
+		lua_pushboolean(L, it != DLMAN->DownloadQueue.end());
+		return 1;
+	}
 	static int IsDownloading(T* p, lua_State* L)
 	{
 		lua_pushboolean(L, p->downloading ==0);
@@ -1683,6 +1690,7 @@ class LunaDownloadablePack : public Luna<DownloadablePack>
 	{
 		ADD_METHOD(DownloadAndInstall);
 		ADD_METHOD(IsDownloading);
+		ADD_METHOD(IsQueued);
 		ADD_METHOD(GetAvgDifficulty);
 		ADD_METHOD(GetName);
 		ADD_METHOD(GetSize);