Units count in attack

2 files changed, 3 insertions, 2 deletions

diff --git a/play.js b/play.js
index 0a0b96c..e72d022 100644
--- a/play.js
+++ b/play.js
@@ -906,8 +906,8 @@ function update_player_info(name, x) {
 
 function update_dice_box() {
 	ui.attack_header.textContent = "Attack in " + data.space_name[view.attack.where]
-	ui.attack_attacker.textContent = "Attacker - " + faction_name[view.attack.attacker]
-	ui.attack_defender.textContent = "Defender - " + faction_name[view.attack.target]
+	ui.attack_attacker.textContent = faction_name[view.attack.attacker] + " - " + view.attack.n_units[0] + " units"
+	ui.attack_defender.textContent = faction_name[view.attack.target] + " - " + view.attack.n_units[1] + " units"
 
 	ui.attacker_flag.forEach(function(element){
         element.src = faction_flags[view.attack.attacker];
diff --git a/rules.js b/rules.js
index 6d7cc59..45ceb40 100644
--- a/rules.js
+++ b/rules.js
@@ -135,6 +135,7 @@ exports.view = function (state, role) {
 			attacker: game.cmd.attacker,
 			target: game.cmd.target
 		}
+		if ("n_units" in game.cmd) view.attack.n_units = game.cmd.n_units
 	}
 
 	if (game.result) {