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

/**** 
* Classes
****/ 
// No plugins needed for MVP
// Soldier class
var Soldier = Container.expand(function () {
	var self = Container.call(this);
	// Properties: team ('player' or 'enemy'), health, attack, speed
	self.team = 'player';
	self.health = 1;
	self.attack = 1;
	self.speed = 4; // px per frame
	// Attach correct asset
	var assetId = self.team === 'player' ? 'soldier_player' : 'soldier_enemy';
	var soldierGfx = self.attachAsset(assetId, {
		anchorX: 0.5,
		anchorY: 0.5
	});
	// Set stats and team
	self.init = function (team, health, attack) {
		self.team = team;
		self.health = health;
		self.attack = attack;
		assetId = team === 'player' ? 'soldier_player' : 'soldier_enemy';
		// Remove old asset if any
		if (soldierGfx) soldierGfx.destroy();
		soldierGfx = self.attachAsset(assetId, {
			anchorX: 0.5,
			anchorY: 0.5
		});
	};
	// For displaying health above soldier
	self.healthTxt = new Text2('', {
		size: 40,
		fill: '#fff'
	});
	self.healthTxt.anchor.set(0.5, 1.2);
	self.addChild(self.healthTxt);
	// Update per frame
	self.update = function () {
		// Update health display
		self.healthTxt.setText(self.health);
		// Move forward if not in combat
		if (!self.inCombat) {
			if (self.team === 'player') {
				self.x += self.speed;
			} else {
				self.x -= self.speed;
			}
		}
	};
	// Mark if in combat
	self.inCombat = false;
	return self;
});

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

/**** 
* Game Code
****/ 
// --- Add background image ---
var background = LK.getAsset('background', {
	anchorX: 0,
	anchorY: 0,
	x: 0,
	y: 0,
	width: GAME_W,
	height: GAME_H
});
game.addChild(background);
// --- Game constants ---
// Player castle (blue)
// Enemy castle (red)
// Player soldier (green)
// Enemy soldier (orange)
var GAME_W = 2048,
	GAME_H = 2732;
var CASTLE_OFFSET_X = 120,
	CASTLE_OFFSET_Y = GAME_H - 520; // moved up by 200px
var CASTLE_HEALTH_INIT = 1000;
var GOLD_INIT = 200;
var GOLD_PER_TICK = 5;
var GOLD_TICK_MS = 200;
var SOLDIER_COST = 100;
var SOLDIER_HEALTH_MIN = 80,
	SOLDIER_HEALTH_MAX = 160;
var SOLDIER_ATTACK_MIN = 20,
	SOLDIER_ATTACK_MAX = 50;
var SOLDIER_SPEED = 14 / 5; // px per frame (slowed down by 5x)
// --- Difficulty selection ---
var difficulty = null;
var ENEMY_SOLDIER_COST = 100; // default, will be set by difficulty
// Show difficulty selection overlay
var diffOverlay = new Container();
diffOverlay.zIndex = 10000; // ensure on top
diffOverlay.width = GAME_W;
diffOverlay.height = GAME_H;
var overlayBg = LK.getAsset('centerCircle', {
	anchorX: 0.5,
	anchorY: 0.5,
	scaleX: 20,
	scaleY: 20,
	x: GAME_W / 2,
	y: GAME_H / 2,
	tint: 0x000000,
	alpha: 0.7
});
diffOverlay.addChild(overlayBg);
var diffTitle = new Text2("Select Difficulty", {
	size: 120,
	fill: "#fff"
});
diffTitle.anchor.set(0.5, 0.5);
diffTitle.x = GAME_W / 2;
diffTitle.y = GAME_H / 2 - 300;
diffOverlay.addChild(diffTitle);
var btnEasy = new Text2("Easy", {
	size: 100,
	fill: 0x00FF00
});
btnEasy.anchor.set(0.5, 0.5);
btnEasy.x = GAME_W / 2;
btnEasy.y = GAME_H / 2 - 80;
diffOverlay.addChild(btnEasy);
var btnNormal = new Text2("Normal", {
	size: 100,
	fill: 0xFFFF00
});
btnNormal.anchor.set(0.5, 0.5);
btnNormal.x = GAME_W / 2;
btnNormal.y = GAME_H / 2 + 80;
diffOverlay.addChild(btnNormal);
var btnHard = new Text2("Hard", {
	size: 100,
	fill: 0xFF0000
});
btnHard.anchor.set(0.5, 0.5);
btnHard.x = GAME_W / 2;
btnHard.y = GAME_H / 2 + 240;
diffOverlay.addChild(btnHard);
game.addChild(diffOverlay);
// Disable game input until difficulty is chosen
var gameInputEnabled = false;
// Helper to start game with selected difficulty
function selectDifficulty(level) {
	difficulty = level;
	if (difficulty === "easy") {
		ENEMY_SOLDIER_COST = 110;
	} else if (difficulty === "normal") {
		ENEMY_SOLDIER_COST = 105;
	} else if (difficulty === "hard") {
		ENEMY_SOLDIER_COST = 99;
	}
	gameInputEnabled = true;
	diffOverlay.destroy();
}
// Add touch/click handlers for buttons
btnEasy.down = function (x, y, obj) {
	selectDifficulty("easy");
};
btnNormal.down = function (x, y, obj) {
	selectDifficulty("normal");
};
btnHard.down = function (x, y, obj) {
	selectDifficulty("hard");
};
// --- State variables ---
var playerHealth = CASTLE_HEALTH_INIT;
var enemyHealth = CASTLE_HEALTH_INIT;
var playerGold = GOLD_INIT;
var enemyGold = GOLD_INIT;
var playerSoldiers = [];
var enemySoldiers = [];
// --- Castles ---
var castlePlayer = game.addChild(LK.getAsset('castle_player', {
	anchorX: 0.5,
	anchorY: 0.5
}));
castlePlayer.x = CASTLE_OFFSET_X;
castlePlayer.y = CASTLE_OFFSET_Y;
var castleEnemy = game.addChild(LK.getAsset('castle_enemy', {
	anchorX: 0.5,
	anchorY: 0.5
}));
castleEnemy.x = GAME_W - CASTLE_OFFSET_X;
castleEnemy.y = CASTLE_OFFSET_Y;
// --- Enemy castle health display above castle ---
var castleEnemyHealthTxt = new Text2(enemyHealth + '', {
	size: 60,
	fill: '#fff'
});
castleEnemyHealthTxt.anchor.set(0.5, 1.2);
castleEnemyHealthTxt.x = castleEnemy.x;
castleEnemyHealthTxt.y = castleEnemy.y - 260; // above the castle
game.addChild(castleEnemyHealthTxt);
// --- GUI: Health and Gold displays ---
var playerHealthTxt = new Text2('1000', {
	size: 70,
	fill: '#fff'
});
playerHealthTxt.anchor.set(0, 0);
LK.gui.top.addChild(playerHealthTxt);
var playerGoldTxt = new Text2('200', {
	size: 60,
	fill: '#ffe600'
});
playerGoldTxt.anchor.set(0, 0);
LK.gui.top.addChild(playerGoldTxt);
var enemyHealthTxt = new Text2('1000', {
	size: 70,
	fill: '#fff'
});
enemyHealthTxt.anchor.set(1, 0);
LK.gui.top.addChild(enemyHealthTxt);
var enemyGoldTxt = new Text2('200', {
	size: 60,
	fill: '#ffe600'
});
enemyGoldTxt.anchor.set(1, 0);
LK.gui.top.addChild(enemyGoldTxt);
// --- Win Counter (Rounds Won) ---
// Use storage plugin for persistence
var roundsWon = storage.roundsWon || 0;
var roundsWonTxt = new Text2('Rounds Won: ' + roundsWon, {
	size: 60,
	fill: '#00ff00'
});
roundsWonTxt.anchor.set(1, 0);
LK.gui.top.addChild(roundsWonTxt);
roundsWonTxt.x = LK.gui.width - 40;
roundsWonTxt.y = 220;
// Position GUI elements
playerHealthTxt.x = 120;
playerHealthTxt.y = 40;
playerGoldTxt.x = 120;
playerGoldTxt.y = 120;
enemyHealthTxt.x = LK.gui.width - 120;
enemyHealthTxt.y = 40;
enemyGoldTxt.x = LK.gui.width - 120;
enemyGoldTxt.y = 120;
// --- Gold income timer ---
var goldTimer = LK.setInterval(function () {
	playerGold += GOLD_PER_TICK;
	enemyGold += GOLD_PER_TICK;
	updateGui();
}, GOLD_TICK_MS);
// --- GUI update function ---
function updateGui() {
	playerHealthTxt.setText(playerHealth);
	playerGoldTxt.setText(playerGold);
	enemyHealthTxt.setText(enemyHealth);
	enemyGoldTxt.setText(enemyGold);
	// Update enemy castle health text and position
	castleEnemyHealthTxt.setText(enemyHealth);
	castleEnemyHealthTxt.x = castleEnemy.x;
	castleEnemyHealthTxt.y = castleEnemy.y - 260;
}
// --- Deploy soldier function ---
function deploySoldier(team) {
	// Block deployment if difficulty not chosen
	if (!gameInputEnabled) return false;
	var gold = team === 'player' ? playerGold : enemyGold;
	var cost = team === 'player' ? SOLDIER_COST : ENEMY_SOLDIER_COST;
	if (gold < cost) return false;
	// Deduct gold
	if (team === 'player') playerGold -= SOLDIER_COST;else enemyGold -= ENEMY_SOLDIER_COST;
	// Fixed health and random attack
	var health = 25;
	var attack = 5 + Math.floor(Math.random() * 6); // 5-10 inclusive
	// Create soldier
	var s = new Soldier();
	s.init(team, health, attack);
	s.speed = SOLDIER_SPEED;
	s.inCombat = false;
	// Position
	s.y = CASTLE_OFFSET_Y + 120; // Move soldiers' exit position down by 120px
	if (team === 'player') {
		s.x = castlePlayer.x + 120;
		playerSoldiers.push(s);
	} else {
		s.x = castleEnemy.x - 120;
		enemySoldiers.push(s);
	}
	game.addChild(s);
	return true;
}
// --- Player tap to deploy soldier ---
game.down = function (x, y, obj) {
	// Block input until difficulty is chosen
	if (!gameInputEnabled) return;
	// Only allow tap in lower half of screen (avoid top left menu)
	if (y > GAME_H / 2 && playerGold >= SOLDIER_COST) {
		deploySoldier('player');
		updateGui();
	}
};
// --- Enemy AI: deploy soldier automatically if enough gold ---
var enemyDeployTimer = LK.setInterval(function () {
	// Block AI until difficulty is chosen
	if (!gameInputEnabled) return;
	if (enemyGold >= ENEMY_SOLDIER_COST) {
		deploySoldier('enemy');
		updateGui();
	}
}, 600);
// --- Main update loop ---
game.update = function () {
	// --- Update all soldiers ---
	// 1. Move soldiers forward if not in combat
	// 2. Check for combat with opposing soldiers
	// 3. If at enemy castle, attack castle
	// --- Player soldiers ---
	for (var i = playerSoldiers.length - 1; i >= 0; i--) {
		var s = playerSoldiers[i];
		s.inCombat = false;
		// Check for enemy soldier in range
		var engaged = false;
		for (var j = 0; j < enemySoldiers.length; j++) {
			var e = enemySoldiers[j];
			// If close enough (overlap)
			if (Math.abs(s.x - e.x) < 80) {
				// Engage in combat
				s.inCombat = true;
				e.inCombat = true;
				// Both attack each other
				if (LK.ticks % 12 === 0) {
					// Attack every 12 frames (~5 times/sec)
					e.health -= s.attack;
					s.health -= e.attack;
				}
				// Remove dead soldiers
				if (e.health <= 0) {
					e.destroy();
					enemySoldiers.splice(j, 1);
					j--;
					// Award 10 gold to player for killing enemy soldier
					playerGold += 10;
					updateGui();
				}
				if (s.health <= 0) {
					s.destroy();
					playerSoldiers.splice(i, 1);
					i--;
					// Award 10 gold to enemy for killing player soldier
					enemyGold += 10;
					updateGui();
					engaged = true;
					break;
				}
				engaged = true;
				break;
			}
		}
		if (engaged) continue;
		// If not in combat, check if at enemy castle
		if (Math.abs(s.x - castleEnemy.x) < 120) {
			s.inCombat = true;
			if (LK.ticks % 12 === 0) {
				enemyHealth -= s.attack;
				if (enemyHealth < 0) enemyHealth = 0;
				updateGui();
			}
			// Remove soldier if castle destroyed
			if (enemyHealth <= 0) {
				s.destroy();
				playerSoldiers.splice(i, 1);
				i--;
			}
		}
	}
	// --- Enemy soldiers ---
	for (var i = enemySoldiers.length - 1; i >= 0; i--) {
		var s = enemySoldiers[i];
		s.inCombat = false;
		// Check for player soldier in range
		var engaged = false;
		for (var j = 0; j < playerSoldiers.length; j++) {
			var e = playerSoldiers[j];
			if (Math.abs(s.x - e.x) < 80) {
				s.inCombat = true;
				e.inCombat = true;
				if (LK.ticks % 12 === 0) {
					e.health -= s.attack;
					s.health -= e.attack;
				}
				if (e.health <= 0) {
					e.destroy();
					playerSoldiers.splice(j, 1);
					j--;
					// Award 10 gold to enemy for killing player soldier
					enemyGold += 10;
					updateGui();
				}
				if (s.health <= 0) {
					s.destroy();
					enemySoldiers.splice(i, 1);
					i--;
					// Award 10 gold to player for killing enemy soldier
					playerGold += 10;
					updateGui();
					engaged = true;
					break;
				}
				engaged = true;
				break;
			}
		}
		if (engaged) continue;
		// If not in combat, check if at player castle
		if (Math.abs(s.x - castlePlayer.x) < 120) {
			s.inCombat = true;
			if (LK.ticks % 12 === 0) {
				playerHealth -= s.attack;
				if (playerHealth < 0) playerHealth = 0;
				updateGui();
			}
			if (playerHealth <= 0) {
				s.destroy();
				enemySoldiers.splice(i, 1);
				i--;
			}
		}
	}
	// --- Move soldiers ---
	for (var i = 0; i < playerSoldiers.length; i++) {
		playerSoldiers[i].update();
	}
	for (var i = 0; i < enemySoldiers.length; i++) {
		enemySoldiers[i].update();
	}
	// --- Check win/lose ---
	if (playerHealth <= 0) {
		LK.effects.flashScreen(0xff0000, 1000);
		LK.showGameOver();
		return;
	}
	if (enemyHealth <= 0) {
		// Increase and persist rounds won
		roundsWon = (storage.roundsWon || 0) + 1;
		storage.roundsWon = roundsWon;
		roundsWonTxt.setText('Rounds Won: ' + roundsWon);
		LK.effects.flashScreen(0x00ff00, 1000);
		LK.showYouWin();
		return;
	}
};
// --- Clean up timers on game over ---
game.destroy = function () {
	LK.clearInterval(goldTimer);
	LK.clearInterval(enemyDeployTimer);
};

===================================================================
--- original.js
+++ change.js
@@ -155,13 +155,13 @@
 // Helper to start game with selected difficulty
 function selectDifficulty(level) {
 	difficulty = level;
 	if (difficulty === "easy") {
-		ENEMY_SOLDIER_COST = 115;
-	} else if (difficulty === "normal") {
 		ENEMY_SOLDIER_COST = 110;
+	} else if (difficulty === "normal") {
+		ENEMY_SOLDIER_COST = 105;
 	} else if (difficulty === "hard") {
-		ENEMY_SOLDIER_COST = 100;
+		ENEMY_SOLDIER_COST = 99;
 	}
 	gameInputEnabled = true;
 	diffOverlay.destroy();
 }