From 2298cb7e164cdc9ede743e4058492ef6641932cb Mon Sep 17 00:00:00 2001 From: Kayne Ruse Date: Mon, 20 May 2019 15:53:01 +1000 Subject: [PATCH] Hotfixed an untrain bug --- server/profiles.js | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/server/profiles.js b/server/profiles.js index 5b01814..ab6fcbb 100644 --- a/server/profiles.js +++ b/server/profiles.js @@ -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; }