/**** 
* Plugins
****/ 
var tween = LK.import("@upit/tween.v1");

/**** 
* Classes
****/ 
var AIPlayer = Container.expand(function () {
	var self = Container.call(this);
	self.makeMove = function () {
		// Only make a move if it's AI's turn
		if (game.gameState !== 'aiTurn') {
			return false;
		}
		var units = game.getTeamUnits('red');
		var moveFound = false;
		// First priority: attack if possible
		for (var i = 0; i < units.length; i++) {
			var unit = units[i];
			var attacks = unit.getPossibleAttacks(game.grid);
			if (attacks.length > 0) {
				// Prioritize attacking king
				var kingAttack = attacks.find(function (attack) {
					return game.grid[attack.row][attack.col].occupiedBy && game.grid[attack.row][attack.col].occupiedBy.isKing;
				});
				if (kingAttack) {
					game.selectUnit(unit);
					game.attackCell(game.grid[kingAttack.row][kingAttack.col]);
					return true;
				}
				// Otherwise attack first available target
				game.selectUnit(unit);
				game.attackCell(game.grid[attacks[0].row][attacks[0].col]);
				return true;
			}
		}
		// Second priority: move toward player king
		var playerKing = game.getKing('blue');
		if (playerKing) {
			// Find unit closest to player king that can move
			var closestUnit = null;
			var closestDistance = Infinity;
			for (var i = 0; i < units.length; i++) {
				var unit = units[i];
				var moves = unit.getPossibleMoves(game.grid);
				if (moves.length > 0) {
					var dist = Math.abs(unit.row - playerKing.row) + Math.abs(unit.col - playerKing.col);
					if (dist < closestDistance) {
						closestUnit = unit;
						closestDistance = dist;
					}
				}
			}
			if (closestUnit) {
				var bestMove = null;
				var bestMoveScore = -Infinity;
				var moves = closestUnit.getPossibleMoves(game.grid);
				// Score each possible move
				for (var i = 0; i < moves.length; i++) {
					var move = moves[i];
					var moveScore = closestDistance - (Math.abs(move.row - playerKing.row) + Math.abs(move.col - playerKing.col));
					if (moveScore > bestMoveScore) {
						bestMove = move;
						bestMoveScore = moveScore;
					}
				}
				if (bestMove) {
					game.selectUnit(closestUnit);
					game.moveSelectedUnit(bestMove.row, bestMove.col);
					return true;
				}
			}
		}
		// Last resort: random move with random unit
		var movableUnits = units.filter(function (unit) {
			return unit.getPossibleMoves(game.grid).length > 0;
		});
		if (movableUnits.length > 0) {
			var randomUnit = movableUnits[Math.floor(Math.random() * movableUnits.length)];
			var possibleMoves = randomUnit.getPossibleMoves(game.grid);
			var randomMove = possibleMoves[Math.floor(Math.random() * possibleMoves.length)];
			game.selectUnit(randomUnit);
			game.moveSelectedUnit(randomMove.row, randomMove.col);
			return true;
		}
		return false; // No move possible
	};
	return self;
});
var CriticalDamageEffect = Container.expand(function () {
	var self = Container.call(this);
	self.init = function (damage, x, y) {
		// Create the critical hit text with animation
		var critText = new Text2('CRITICAL HIT!', {
			size: 80,
			fill: '#FF0000'
		});
		critText.anchor.set(0.5, 0.5);
		self.addChild(critText);
		// Create the damage text
		var damageText = new Text2('-' + damage, {
			size: 100,
			fill: '#FFFF00'
		});
		damageText.anchor.set(0.5, 0.5);
		damageText.y = 80;
		self.addChild(damageText);
		// Position the effect
		self.x = x;
		self.y = y;
		// Initial scale and alpha
		self.scale.set(0.1);
		self.alpha = 0;
		// Animate in
		tween(self, {
			scaleX: 1.2,
			scaleY: 1.2,
			alpha: 1
		}, {
			duration: 300,
			easing: tween.elasticOut
		});
		// Animate out after showing
		LK.setTimeout(function () {
			tween(self, {
				scaleX: 1.5,
				scaleY: 1.5,
				alpha: 0,
				y: self.y - 100
			}, {
				duration: 700,
				easing: tween.easeOutQuad,
				onFinish: function onFinish() {
					self.parent.removeChild(self);
				}
			});
		}, 1200);
		return self;
	};
	return self;
});
var GridCell = Container.expand(function () {
	var self = Container.call(this);
	// Cell properties
	self.row = 0;
	self.col = 0;
	self.occupied = false;
	self.occupiedBy = null;
	self.highlighted = false;
	self.highlightType = null;
	// Create cell graphic
	var cellGraphic = self.attachAsset('grid_cell', {
		anchorX: 0.5,
		anchorY: 0.5,
		alpha: 0.5
	});
	// Highlighting methods
	self.highlight = function (type) {
		if (self.highlightAsset) {
			self.removeChild(self.highlightAsset);
		}
		self.highlighted = true;
		self.highlightType = type;
		if (type === 'move') {
			self.highlightAsset = self.attachAsset('highlight_move', {
				anchorX: 0.5,
				anchorY: 0.5,
				alpha: 0.5
			});
		} else if (type === 'attack') {
			self.highlightAsset = self.attachAsset('highlight_attack', {
				anchorX: 0.5,
				anchorY: 0.5,
				alpha: 0.5
			});
		} else if (type === 'selected') {
			self.highlightAsset = self.attachAsset('highlight_selected', {
				anchorX: 0.5,
				anchorY: 0.5,
				alpha: 0.5
			});
		}
	};
	self.clearHighlight = function () {
		if (self.highlightAsset) {
			self.removeChild(self.highlightAsset);
			self.highlightAsset = null;
		}
		self.highlighted = false;
		self.highlightType = null;
	};
	// Event handlers
	self.down = function (x, y, obj) {
		if (game.gameState === 'playerTurn') {
			game.handleCellClick(self);
		}
	};
	return self;
});
var Unit = Container.expand(function () {
	var self = Container.call(this);
	// Unit properties
	self.type = "";
	self.team = "";
	self.row = 0;
	self.col = 0;
	self.isKing = false;
	self.alive = true;
	self.selected = false;
	self.health = 100; // Base health for all units
	self.attackStrength = 20; // Base attack strength for all units
	// Initialize with type and team
	self.init = function (type, team, row, col) {
		self.type = type;
		self.team = team;
		self.row = row;
		self.col = col;
		self.isKing = type === 'king';
		// Set different health and attack values based on unit type
		switch (type) {
			case 'king':
				self.health = 200;
				self.attackStrength = 30;
				break;
			case 'warrior':
				self.health = 250;
				self.attackStrength = 40;
				break;
			case 'knight':
				self.health = 130;
				self.attackStrength = 35;
				break;
			case 'wizard':
				self.health = 80;
				self.attackStrength = 50;
				break;
			case 'archer':
				self.health = 90;
				self.attackStrength = 45;
				break;
		}
		var assetId = type + '_' + team;
		self.unitGraphic = self.attachAsset(assetId, {
			anchorX: 0.5,
			anchorY: 0.5
		});
		return self;
	};
	// Movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		// Base movement logic to be overridden by specific unit types
		return false;
	};
	// Attack validation
	self.canAttack = function (targetRow, targetCol) {
		// Base attack logic to be overridden by specific unit types
		return false;
	};
	// Get possible move cells
	self.getPossibleMoves = function (grid) {
		var moves = [];
		for (var r = 0; r < grid.length; r++) {
			for (var c = 0; c < grid[r].length; c++) {
				if (this.canMoveTo(r, c) && !grid[r][c].occupied) {
					moves.push({
						row: r,
						col: c
					});
				}
			}
		}
		return moves;
	};
	// Get possible attack cells
	self.getPossibleAttacks = function (grid) {
		var attacks = [];
		for (var r = 0; r < grid.length; r++) {
			for (var c = 0; c < grid[r].length; c++) {
				if (this.canAttack(r, c) && grid[r][c].occupied && grid[r][c].occupiedBy.team !== this.team) {
					attacks.push({
						row: r,
						col: c
					});
				}
			}
		}
		return attacks;
	};
	// Move unit to new position
	self.moveTo = function (targetRow, targetCol) {
		self.row = targetRow;
		self.col = targetCol;
	};
	// Select unit
	self.select = function () {
		self.selected = true;
		self.unitGraphic.alpha = 1.0;
	};
	// Deselect unit
	self.deselect = function () {
		self.selected = false;
		self.unitGraphic.alpha = 0.8;
	};
	return self;
});
var Wizard = Unit.expand(function () {
	var self = Unit.call(this);
	// Override movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		var rowDiff = Math.abs(targetRow - self.row);
		var colDiff = Math.abs(targetCol - self.col);
		// Wizard can move diagonally or in straight lines
		return rowDiff <= 2 && colDiff <= 2;
	};
	// Override attack validation
	self.canAttack = function (targetRow, targetCol) {
		// Calculate the attack zone (6 tiles ahead based on team direction)
		var attackTileRow = self.team === 'blue' ? self.row - 6 : self.row + 6;
		var attackTileCol = self.col;
		// Calculate distance from attack tile
		var rowDiff = Math.abs(targetRow - attackTileRow);
		var colDiff = Math.abs(targetCol - attackTileCol);
		// Can only attack the attack tile or one tile around it
		return rowDiff <= 1 && colDiff <= 1;
	};
	// Wizards do area damage
	self.isAreaAttacker = true;
	return self;
});
var Warrior = Unit.expand(function () {
	var self = Unit.call(this);
	// Override movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		var rowDiff = Math.abs(targetRow - self.row);
		var colDiff = Math.abs(targetCol - self.col);
		// Warrior can only move 1 tile in any direction
		return rowDiff <= 1 && colDiff <= 1 && !(rowDiff === 0 && colDiff === 0);
	};
	// Override attack validation
	self.canAttack = function (targetRow, targetCol) {
		// Instant-kill attack (1 tile forward)
		if (self.team === 'blue') {
			return targetRow === self.row - 1 && targetCol === self.col;
		} else {
			return targetRow === self.row + 1 && targetCol === self.col;
		}
	};
	// Warriors have instant kill ability
	self.hasInstantKill = true;
	return self;
});
var Knight = Unit.expand(function () {
	var self = Unit.call(this);
	// Override movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		// Knight moves straight or sideways (1-3 tiles)
		var rowDiff = Math.abs(targetRow - self.row);
		var colDiff = Math.abs(targetCol - self.col);
		// Can move horizontally or vertically up to 3 tiles
		return rowDiff === 0 && colDiff > 0 && colDiff <= 3 || colDiff === 0 && rowDiff > 0 && rowDiff <= 3;
	};
	// Override attack validation
	self.canAttack = function (targetRow, targetCol) {
		// Same as movement - can attack in movement range
		return self.canMoveTo(targetRow, targetCol);
	};
	return self;
});
var King = Unit.expand(function () {
	var self = Unit.call(this);
	// Override movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		// King moves like a chess queen but limited distance
		var rowDiff = Math.abs(targetRow - self.row);
		var colDiff = Math.abs(targetCol - self.col);
		// Can move in any direction but maximum 1 space
		return rowDiff <= 1 && colDiff <= 1 && !(rowDiff === 0 && colDiff === 0);
	};
	// Override attack validation
	self.canAttack = function (targetRow, targetCol) {
		// King only attacks 1 tile forward based on team direction
		var forwardRow = self.team === 'blue' ? self.row - 1 : self.row + 1;
		return targetRow === forwardRow && targetCol === self.col;
	};
	return self;
});
var Archer = Unit.expand(function () {
	var self = Unit.call(this);
	// Override movement validation
	self.canMoveTo = function (targetRow, targetCol) {
		var rowDiff = Math.abs(targetRow - self.row);
		var colDiff = Math.abs(targetCol - self.col);
		// Can move sideways freely, but only one tile forward
		if (self.team === 'blue') {
			return colDiff > 0 && rowDiff === 0 ||
			// sideways
			targetRow === self.row - 1 && colDiff === 0; // one forward
		} else {
			return colDiff > 0 && rowDiff === 0 ||
			// sideways
			targetRow === self.row + 1 && colDiff === 0; // one forward
		}
	};
	// Override attack validation
	self.canAttack = function (targetRow, targetCol) {
		// Can attack up to 8 tiles ahead
		if (self.team === 'blue') {
			return targetCol === self.col && targetRow < self.row && self.row - targetRow <= 8;
		} else {
			return targetCol === self.col && targetRow > self.row && targetRow - self.row <= 8;
		}
	};
	return self;
});

/**** 
* Initialize Game
****/ 
var game = new LK.Game({
	backgroundColor: 0x000000
});

/**** 
* Game Code
****/ 
// Game constants
// Import tween plugin
var GRID_ROWS = 10;
var GRID_COLS = 5;
var CELL_SIZE = 250; // Significantly increased cell size
var GRID_PADDING_X = (2048 - GRID_COLS * CELL_SIZE) / 2;
var GRID_PADDING_Y = (2732 - GRID_ROWS * CELL_SIZE) / 2;
// Game variables
var grid = [];
var units = [];
var selectedUnit = null;
var gameState = 'playerTurn'; // playerTurn, aiTurn, gameOver
var ai = new AIPlayer();
var statusText = new Text2('Player Turn', {
	size: 120,
	// Significantly increased text size
	fill: 0xFFFFFF
});
// Initialize the game board
function initializeGame() {
	// Create grid
	for (var row = 0; row < GRID_ROWS; row++) {
		grid[row] = [];
		for (var col = 0; col < GRID_COLS; col++) {
			var cell = new GridCell();
			cell.row = row;
			cell.col = col;
			cell.x = GRID_PADDING_X + col * CELL_SIZE + CELL_SIZE / 2;
			cell.y = GRID_PADDING_Y + row * CELL_SIZE + CELL_SIZE / 2;
			// Alternate cell colors
			if ((row + col) % 2 === 0) {
				cell.removeChild(cell.children[0]); // Remove default graphic
				cell.attachAsset('grid_cell_dark', {
					anchorX: 0.5,
					anchorY: 0.5,
					alpha: 0.5
				});
			}
			grid[row][col] = cell;
			game.addChild(cell);
		}
	}
	// Create units for both teams
	createUnits();
	// Add status text
	statusText.anchor.set(0.5, 0);
	statusText.x = 2048 / 2;
	statusText.y = 50;
	LK.gui.top.addChild(statusText);
}
// Create all units for both teams
function createUnits() {
	// Blue team (player) - bottom of the board
	// Formation: Knight, Warrior, Warrior, Warrior, Knight (row 8)
	// Formation: Archer, Mage, King, Mage, Archer (row 9)
	createUnit('knight', 'blue', 8, 0);
	createUnit('warrior', 'blue', 8, 1);
	createUnit('warrior', 'blue', 8, 2);
	createUnit('warrior', 'blue', 8, 3);
	createUnit('knight', 'blue', 8, 4);
	createUnit('archer', 'blue', 9, 0);
	createUnit('wizard', 'blue', 9, 1);
	createUnit('king', 'blue', 9, 2);
	createUnit('wizard', 'blue', 9, 3);
	createUnit('archer', 'blue', 9, 4);
	// Red team (AI) - top of the board
	// Formation: Knight, Warrior, Warrior, Warrior, Knight (row 1)
	// Formation: Archer, Mage, King, Mage, Archer (row 0)
	createUnit('knight', 'red', 1, 0);
	createUnit('warrior', 'red', 1, 1);
	createUnit('warrior', 'red', 1, 2);
	createUnit('warrior', 'red', 1, 3);
	createUnit('knight', 'red', 1, 4);
	createUnit('archer', 'red', 0, 0);
	createUnit('wizard', 'red', 0, 1);
	createUnit('king', 'red', 0, 2);
	createUnit('wizard', 'red', 0, 3);
	createUnit('archer', 'red', 0, 4);
}
// Create a single unit
function createUnit(type, team, row, col) {
	var unit;
	// Create the appropriate unit type
	switch (type) {
		case 'king':
			unit = new King().init(type, team, row, col);
			break;
		case 'knight':
			unit = new Knight().init(type, team, row, col);
			break;
		case 'archer':
			unit = new Archer().init(type, team, row, col);
			break;
		case 'wizard':
			unit = new Wizard().init(type, team, row, col);
			break;
		case 'warrior':
			unit = new Warrior().init(type, team, row, col);
			break;
	}
	// Position unit
	unit.x = grid[row][col].x;
	unit.y = grid[row][col].y;
	unit.unitGraphic.alpha = 0.8;
	// Add health display
	unit.healthText = new Text2(unit.health.toString(), {
		size: 50,
		fill: team === 'blue' ? "#0088ff" : "#ff0088"
	});
	unit.healthText.anchor.set(0.5, 0.5);
	unit.healthText.y = unit.unitGraphic.height / 2 + 10;
	unit.addChild(unit.healthText);
	// Add subtle entry animation
	unit.unitGraphic.scale.set(0.1);
	unit.unitGraphic.alpha = 0.3;
	// Add unit to game
	game.addChild(unit);
	units.push(unit);
	// Mark cell as occupied
	grid[row][col].occupied = true;
	grid[row][col].occupiedBy = unit;
	// Add click handler for player units
	if (team === 'blue') {
		unit.down = function (x, y, obj) {
			if (game.gameState === 'playerTurn') {
				selectUnit(unit);
			}
		};
	}
	// Animate unit entry with enhanced effect
	tween(unit.unitGraphic, {
		scaleX: 1.0,
		scaleY: 1.0,
		alpha: 0.8
	}, {
		duration: 600,
		easing: tween.elasticOut,
		delay: 100 * (row + col) % 5 // Staggered appearance for visual interest
	});
	return unit;
}
// Handle cell click
game.handleCellClick = function (cell) {
	if (!selectedUnit || game.gameState !== 'playerTurn') {
		return;
	}
	// If cell is highlighted for movement
	if (cell.highlighted && cell.highlightType === 'move') {
		moveSelectedUnit(cell.row, cell.col);
	}
	// If cell is highlighted for attack
	else if (cell.highlighted && cell.highlightType === 'attack') {
		attackCell(cell);
	}
};
// Select a unit
function selectUnit(unit) {
	// Can only select your own units
	if (unit.team !== 'blue' && gameState === 'playerTurn') {
		return;
	}
	if (unit.team !== 'red' && gameState === 'aiTurn') {
		return;
	}
	// Deselect previous unit if there was one
	if (selectedUnit) {
		selectedUnit.deselect();
		clearAllHighlights();
	}
	// Select new unit
	selectedUnit = unit;
	selectedUnit.select();
	// Enhanced selection animation with glow effect
	tween(selectedUnit.unitGraphic, {
		scaleX: 1.2,
		scaleY: 1.2,
		alpha: 1.0
	}, {
		duration: 200,
		easing: tween.elasticOut,
		onFinish: function onFinish() {
			tween(selectedUnit.unitGraphic, {
				scaleX: 1.0,
				scaleY: 1.0,
				alpha: 0.9
			}, {
				duration: 200,
				easing: tween.easeOutQuad
			});
		}
	});
	// Highlight cell under selected unit
	grid[unit.row][unit.col].highlight('selected');
	// Highlight possible moves with delay for better visual feedback
	var possibleMoves = unit.getPossibleMoves(grid);
	for (var i = 0; i < possibleMoves.length; i++) {
		(function (index) {
			LK.setTimeout(function () {
				if (selectedUnit === unit) {
					// Only highlight if still selected
					var move = possibleMoves[index];
					grid[move.row][move.col].highlight('move');
				}
			}, index * 30);
		})(i);
	}
	// Highlight possible attacks with delay
	var possibleAttacks = unit.getPossibleAttacks(grid);
	for (var i = 0; i < possibleAttacks.length; i++) {
		(function (index) {
			LK.setTimeout(function () {
				if (selectedUnit === unit) {
					// Only highlight if still selected
					var attack = possibleAttacks[index];
					grid[attack.row][attack.col].highlight('attack');
				}
			}, (possibleMoves.length + index) * 30);
		})(i);
	}
}
// Move selected unit to new position
function moveSelectedUnit(targetRow, targetCol) {
	if (!selectedUnit) {
		return;
	}
	// Update grid occupation
	grid[selectedUnit.row][selectedUnit.col].occupied = false;
	grid[selectedUnit.row][selectedUnit.col].occupiedBy = null;
	// Move unit with animation
	selectedUnit.moveTo(targetRow, targetCol);
	// Animate movement with tween
	var targetX = grid[targetRow][targetCol].x;
	var targetY = grid[targetRow][targetCol].y;
	tween(selectedUnit, {
		x: targetX,
		y: targetY
	}, {
		duration: 300,
		easing: tween.easeOutQuad,
		onFinish: function onFinish() {
			// Update grid occupation at new position
			grid[targetRow][targetCol].occupied = true;
			grid[targetRow][targetCol].occupiedBy = selectedUnit;
			// Deselect unit and clear highlights
			selectedUnit.deselect();
			clearAllHighlights();
			selectedUnit = null;
			// Check if any attacks are possible after movement
			checkForAutoAttack();
			// Switch turns
			endTurn();
		}
	});
}
// Attack a cell
function attackCell(cell) {
	if (!selectedUnit || !cell.occupied || cell.occupiedBy.team === selectedUnit.team) {
		return;
	}
	var targetUnit = cell.occupiedBy;
	// Animate the attacker - "lunge" toward target
	var originalX = selectedUnit.x;
	var originalY = selectedUnit.y;
	var directionX = targetUnit.x - selectedUnit.x;
	var directionY = targetUnit.y - selectedUnit.y;
	var distance = Math.sqrt(directionX * directionX + directionY * directionY);
	var normalizedX = directionX / distance * 30; // small lunge
	var normalizedY = directionY / distance * 30;
	// Enhanced attack animation sequence
	tween(selectedUnit, {
		x: selectedUnit.x + normalizedX * 1.5,
		// More pronounced lunge
		y: selectedUnit.y + normalizedY * 1.5
	}, {
		duration: 200,
		easing: tween.easeOutQuad,
		onFinish: function onFinish() {
			// More dramatic visual flash on target
			LK.effects.flashObject(targetUnit, 0xff0000, 400);
			// Return attacker to original position
			tween(selectedUnit, {
				x: originalX,
				y: originalY
			}, {
				duration: 200,
				easing: tween.easeOutQuad,
				onFinish: function onFinish() {
					// Check for critical damage
					var isCritical = isCriticalDamage(selectedUnit.type, targetUnit.type);
					var damageMultiplier = isCritical ? getCriticalDamageMultiplier() : 1.0;
					var actualDamage = Math.floor(selectedUnit.attackStrength * damageMultiplier);
					// Perform attack after animation
					if (selectedUnit.hasInstantKill && !targetUnit.isKing) {
						// Warrior's instant kill works on all enemies except kings
						removeUnit(targetUnit);
					} else if (targetUnit.isKing) {
						// King only dies in one hit if attacked by a warrior
						if (selectedUnit.type === 'warrior') {
							removeUnit(targetUnit);
						} else {
							// Apply normal damage to king
							var isCritical = isCriticalDamage(selectedUnit.type, targetUnit.type);
							var damageMultiplier = isCritical ? getCriticalDamageMultiplier() : 1.0;
							var actualDamage = Math.floor(selectedUnit.attackStrength * damageMultiplier);
							targetUnit.health -= actualDamage;
							targetUnit.healthText.setText(targetUnit.health.toString());
							if (targetUnit.health <= 0) {
								removeUnit(targetUnit);
							}
						}
					} else if (selectedUnit.isAreaAttacker) {
						// Area attack affects target and adjacent units of opposite team
						var adjacentCells = getAdjacentCells(cell.row, cell.col);
						// Apply damage instead of instant kill
						targetUnit.health -= actualDamage;
						targetUnit.healthText.setText(targetUnit.health.toString());
						// Show critical hit effect if applicable
						if (isCritical) {
							var critEffect = new CriticalDamageEffect().init(actualDamage, targetUnit.x, targetUnit.y - 50);
							game.addChild(critEffect);
							// Add stronger visual feedback
							tween(targetUnit, {
								scaleX: 0.8,
								scaleY: 0.8,
								rotation: 0.2
							}, {
								duration: 100,
								easing: tween.easeOutQuad,
								onFinish: function onFinish() {
									tween(targetUnit, {
										scaleX: 1.0,
										scaleY: 1.0,
										rotation: 0
									}, {
										duration: 200,
										easing: tween.elasticOut
									});
								}
							});
						}
						if (targetUnit.health <= 0) {
							removeUnit(targetUnit);
						}
						for (var i = 0; i < adjacentCells.length; i++) {
							var adjCell = adjacentCells[i];
							if (adjCell.occupied && adjCell.occupiedBy.team !== selectedUnit.team) {
								var adjTargetUnit = adjCell.occupiedBy;
								var adjDamage = Math.floor(actualDamage / 2); // Half damage for adjacent units
								adjTargetUnit.health -= adjDamage;
								adjTargetUnit.healthText.setText(adjTargetUnit.health.toString());
								if (adjTargetUnit.health <= 0) {
									removeUnit(adjTargetUnit);
								}
							}
						}
					} else {
						// Normal attack
						targetUnit.health -= actualDamage;
						targetUnit.healthText.setText(targetUnit.health.toString());
						// Show critical hit effect if applicable
						if (isCritical) {
							var critEffect = new CriticalDamageEffect().init(actualDamage, targetUnit.x, targetUnit.y - 50);
							game.addChild(critEffect);
							// Add stronger visual feedback for critical hit
							tween(targetUnit, {
								scaleX: 0.7,
								scaleY: 0.7,
								rotation: 0.3
							}, {
								duration: 100,
								easing: tween.easeOutQuad,
								onFinish: function onFinish() {
									tween(targetUnit, {
										scaleX: 1.0,
										scaleY: 1.0,
										rotation: 0
									}, {
										duration: 300,
										easing: tween.elasticOut
									});
								}
							});
						}
						if (targetUnit.health <= 0) {
							removeUnit(targetUnit);
						}
					}
					// Deselect unit and clear highlights
					selectedUnit.deselect();
					clearAllHighlights();
					selectedUnit = null;
					// Switch turns
					endTurn();
				}
			});
		}
	});
}
// Check if auto-attack is possible (for units that moved next to enemies)
function checkForAutoAttack() {
	// Not implemented in this version - all attacks are chosen by player
}
// Get adjacent cells
function getAdjacentCells(row, col) {
	var adjacent = [];
	var directions = [{
		r: -1,
		c: 0
	}, {
		r: 1,
		c: 0
	}, {
		r: 0,
		c: -1
	}, {
		r: 0,
		c: 1
	}, {
		r: -1,
		c: -1
	}, {
		r: -1,
		c: 1
	}, {
		r: 1,
		c: -1
	}, {
		r: 1,
		c: 1
	}];
	for (var i = 0; i < directions.length; i++) {
		var newRow = row + directions[i].r;
		var newCol = col + directions[i].c;
		if (newRow >= 0 && newRow < GRID_ROWS && newCol >= 0 && newCol < GRID_COLS) {
			adjacent.push(grid[newRow][newCol]);
		}
	}
	return adjacent;
}
// Remove a unit from the game
function removeUnit(unit) {
	// Clear grid cell
	grid[unit.row][unit.col].occupied = false;
	grid[unit.row][unit.col].occupiedBy = null;
	// Remove from units array
	var index = units.indexOf(unit);
	if (index > -1) {
		units.splice(index, 1);
	}
	// Remove from display
	game.removeChild(unit);
	// Check for game over if a king was killed
	if (unit.isKing) {
		if (unit.team === 'blue') {
			endGame('red');
		} else {
			endGame('blue');
		}
	}
}
// Clear all highlights from the grid
function clearAllHighlights() {
	for (var row = 0; row < GRID_ROWS; row++) {
		for (var col = 0; col < GRID_COLS; col++) {
			grid[row][col].clearHighlight();
		}
	}
}
// End the current turn
function endTurn() {
	if (gameState === 'playerTurn') {
		gameState = 'aiTurn';
		statusText.setText('AI Turn');
		// Execute AI turn with delay for better visibility
		LK.setTimeout(function () {
			if (ai.makeMove()) {
				// AI successfully made a move, turn will be ended in the attack/move handlers
			} else {
				// AI couldn't make a move, switch back to player
				gameState = 'playerTurn';
				statusText.setText('Player Turn');
			}
		}, 500); // Small delay for better user experience
	} else if (gameState === 'aiTurn') {
		gameState = 'playerTurn';
		statusText.setText('Player Turn');
	}
}
// End the game with a winner
function endGame(winner) {
	gameState = 'gameOver';
	if (winner === 'blue') {
		statusText.setText('You Win!');
		LK.showYouWin();
	} else {
		statusText.setText('Game Over');
		LK.showGameOver();
	}
}
// Helper functions to determine critical damage relationships
function isCriticalDamage(attackerType, targetType) {
	// Knights are strong against archers
	if (attackerType === 'knight' && targetType === 'archer') {
		return true;
	}
	// Archers are strong against mages/wizards
	if (attackerType === 'archer' && targetType === 'wizard') {
		return true;
	}
	// Mages/wizards are strong against knights
	if (attackerType === 'wizard' && targetType === 'knight') {
		return true;
	}
	return false;
}
function getCriticalDamageMultiplier() {
	// Critical damage is 2x normal damage
	return 2.0;
}
// Helper functions
game.getTeamUnits = function (team) {
	return units.filter(function (unit) {
		return unit.team === team;
	});
};
game.getKing = function (team) {
	for (var i = 0; i < units.length; i++) {
		if (units[i].isKing && units[i].team === team) {
			return units[i];
		}
	}
	return null;
};
// Expose functions to game object for access from other classes
game.selectUnit = selectUnit;
game.moveSelectedUnit = moveSelectedUnit;
game.attackCell = attackCell;
game.grid = grid;
game.gameState = gameState;
// Create a getter/setter for game state to ensure consistency
Object.defineProperty(game, 'gameState', {
	get: function get() {
		return gameState;
	},
	set: function set(value) {
		gameState = value;
	}
});
// Initialize game
initializeGame();

===================================================================
--- original.js
+++ change.js
@@ -9,9 +9,11 @@
 var AIPlayer = Container.expand(function () {
 	var self = Container.call(this);
 	self.makeMove = function () {
 		// Only make a move if it's AI's turn
-		if (game.gameState !== 'aiTurn') return false;
+		if (game.gameState !== 'aiTurn') {
+			return false;
+		}
 		var units = game.getTeamUnits('red');
 		var moveFound = false;
 		// First priority: attack if possible
 		for (var i = 0; i < units.length; i++) {
@@ -217,13 +219,13 @@
 		self.isKing = type === 'king';
 		// Set different health and attack values based on unit type
 		switch (type) {
 			case 'king':
-				self.health = 120;
+				self.health = 200;
 				self.attackStrength = 30;
 				break;
 			case 'warrior':
-				self.health = 150;
+				self.health = 250;
 				self.attackStrength = 40;
 				break;
 			case 'knight':
 				self.health = 130;
@@ -564,9 +566,11 @@
 	return unit;
 }
 // Handle cell click
 game.handleCellClick = function (cell) {
-	if (!selectedUnit || game.gameState !== 'playerTurn') return;
+	if (!selectedUnit || game.gameState !== 'playerTurn') {
+		return;
+	}
 	// If cell is highlighted for movement
 	if (cell.highlighted && cell.highlightType === 'move') {
 		moveSelectedUnit(cell.row, cell.col);
 	}
@@ -577,10 +581,14 @@
 };
 // Select a unit
 function selectUnit(unit) {
 	// Can only select your own units
-	if (unit.team !== 'blue' && gameState === 'playerTurn') return;
-	if (unit.team !== 'red' && gameState === 'aiTurn') return;
+	if (unit.team !== 'blue' && gameState === 'playerTurn') {
+		return;
+	}
+	if (unit.team !== 'red' && gameState === 'aiTurn') {
+		return;
+	}
 	// Deselect previous unit if there was one
 	if (selectedUnit) {
 		selectedUnit.deselect();
 		clearAllHighlights();
@@ -637,9 +645,11 @@
 	}
 }
 // Move selected unit to new position
 function moveSelectedUnit(targetRow, targetCol) {
-	if (!selectedUnit) return;
+	if (!selectedUnit) {
+		return;
+	}
 	// Update grid occupation
 	grid[selectedUnit.row][selectedUnit.col].occupied = false;
 	grid[selectedUnit.row][selectedUnit.col].occupiedBy = null;
 	// Move unit with animation
@@ -669,9 +679,11 @@
 	});
 }
 // Attack a cell
 function attackCell(cell) {
-	if (!selectedUnit || !cell.occupied || cell.occupiedBy.team === selectedUnit.team) return;
+	if (!selectedUnit || !cell.occupied || cell.occupiedBy.team === selectedUnit.team) {
+		return;
+	}
 	var targetUnit = cell.occupiedBy;
 	// Animate the attacker - "lunge" toward target
 	var originalX = selectedUnit.x;
 	var originalY = selectedUnit.y;