Consider DISCs in InfShift

1 files changed, 6 insertions, 0 deletions

diff --git a/rules.js b/rules.js
index 9ba6717..51abd46 100644
--- a/rules.js
+++ b/rules.js
@@ -1296,6 +1296,10 @@ function goto_attack_space() {
 		if (piece_space(p) === game.cmd.where)
 			set_add(game.cmd.selected, p)
 	})
+	for_each_piece(game.cmd.attacker, DISC, p => {
+		if (piece_space(p) === game.cmd.where)
+			set_add(game.cmd.selected, p)
+	})
 	for_each_movable(game.cmd.attacker, p => {
 		if (piece_space(p) === game.cmd.where) {
 			set_add(game.cmd.selected, p)
@@ -1535,6 +1539,8 @@ states.attack_casualties = {
 			log_summary_remove(p)
 		else
 			log_summary_remove_from(p)
+		if (piece_type(p) === DISC)
+			game.cmd.n_units[game.cmd.step === 2 ? 1 : 0] += 1
 		remove_piece(p)
 		set_delete(game.cmd.selected, p)
 		game.cmd.count -= 1