Hotfixed an untrain bug

2019-05-20 15:53:01 +10:00
parent 5ad864103f
commit 2298cb7e16
1 changed files with 14 additions and 2 deletions
@@ -295,8 +295,20 @@ const untrain = (connection) => (req, res) => {
 			connection.query(query, [fields.id], (err, results) => {
 				if (err) throw err;

-				if (results[0][fields.role] <= 0) {
-					res.status(400).write('Not enough ' + fields.role);
+				if (fields.role === 'soldier' && results[0].soldiers <= 0) {
+					res.status(400).write('Not enough soldiers');
+					res.end();
+					return;
+				}
+
+				if (fields.role === 'spy' && results[0].spies <= 0) {
+					res.status(400).write('Not enough spies');
+					res.end();
+					return;
+				}
+
+				if (fields.role === 'scientist' && results[0].scientists <= 0) {
+					res.status(400).write('Not enough scientists');
 					res.end();
 					return;
 				}