Nobody is safe

1 files changed, 8 insertions, 1 deletions

diff --git a/rules.js b/rules.js
index 19ee2c4..e8cae32 100644
--- a/rules.js
+++ b/rules.js
@@ -1299,8 +1299,8 @@ function goto_attack_space() {
 	})
 	for_each_movable(game.cmd.attacker, p => {
 		if (piece_space(p) === game.cmd.where) {
-			set_add(game.cmd.selected, p)
 			if (game.cmd.attacker === MI || !set_has(game.cmd.pieces, p)) {
+				set_add(game.cmd.selected, p)
 				set_add(game.cmd.pieces, p)
 				game.cmd.n_units[0] += 1
 			}
@@ -1528,6 +1528,13 @@ states.attack_casualties = {
 		) {
 			view.prompt = `Attack: Remove ${game.cmd.count} piece${game.cmd.count > 1 ? "s" : ""} as casualties.`
 
+			// Adding all units in space as targets
+			if (game.cmd.step === 3) {
+				for_each_movable(game.cmd.attacker, p => {
+					if (piece_space(p) === game.cmd.where) { set_add(game.cmd.selected, p) }
+				})
+			}
+
 			for (let p of game.cmd.selected)
 				if (piece_faction(p) === game.current)
 					gen_action_piece(p)