Various fixes

1 files changed, 11 insertions, 3 deletions

diff --git a/inc/mine.inc b/inc/mine.inc
index 752fc69..76dfb50 100644
--- a/inc/mine.inc
+++ b/inc/mine.inc
@@ -1,18 +1,26 @@
 <?php
 
-function initCraftMine() {
-	$_SESSION["mine"] = array("gold" => 0, "miners" => 0);
+function initMineIfNeeded() {
+	if(empty($_SESSION["mine"]))
+		$_SESSION["mine"] = array("gold" => 0);
+	else if(!array_key_exists("gold", $_SESSION["mine"]))
+		$_SESSION["mine"]["gold"] = 0;
 }
 
 function withdrawMine() {
 	$amount = intval($_POST["amount"]);
+	if(!empty($_SESSION["dungeon"])) {
+		sendError("cant_withdraw_in_dungeon");
+		return;
+	}
 	if($amount == 0) return;
+	initMineIfNeeded();
 	$_SESSION["mine"]["gold"] += $amount;
 	echo json_encode($_SESSION["mine"]["gold"]);
 }
 
 function sendMine() {
-	if(empty($_SESSION["mine"])) initCraftMine();
+	initMineIfNeeded();
 	$mine = $_SESSION["mine"];
 	return $mine["gold"];
 }