Hotfixed an untrain bug

This commit is contained in:
2019-05-20 15:53:01 +10:00
parent 5ad864103f
commit 2298cb7e16
+14 -2
View File
@@ -295,8 +295,20 @@ const untrain = (connection) => (req, res) => {
connection.query(query, [fields.id], (err, results) => { connection.query(query, [fields.id], (err, results) => {
if (err) throw err; if (err) throw err;
if (results[0][fields.role] <= 0) { if (fields.role === 'soldier' && results[0].soldiers <= 0) {
res.status(400).write('Not enough ' + fields.role); 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(); res.end();
return; return;
} }