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

/**** 
* Classes
****/ 
/* Lightning, Aura effect, Molotov flame can be lightweight containers */
var AuraField = Container.expand(function () {
	var self = Container.call(this);
	var gfx = self.attachAsset('aura', {
		anchorX: 0.5,
		anchorY: 0.5,
		alpha: 0.35
	});
	self.radius = 150;
	self.dmg = 5;
	self.slow = 0;
	self.update = function () {/* position follows player by external binding */};
	return self;
});
/****-----------------------------------------------------------------
* 4.  Enemy hierarchy (unchanged aside from helper intersectsCircle)
*------------------------------------------------------------------*/ 
var Enemy = Container.expand(function () {
	var self = Container.call(this);
	self.hp = 20;
	self.maxHp = 20;
	self.damage = 10;
	self.speed = 2;
	self.expValue = 5;
	self.type = 'normal';
	self.attachAsset('zombie', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.takeDamage = function (a) {
		self.hp -= a;
		LK.effects.flashObject(self, 0xffffff, 100);
		return self.hp <= 0;
	};
	self.update = function () {
		if (!player || skillSelectionActive) {
			return;
		}
		var dx = player.x - self.x,
			dy = player.y - self.y,
			d = Math.sqrt(dx * dx + dy * dy);
		if (d > 0) {
			self.x += dx / d * self.speed;
			self.y += dy / d * self.speed;
		}
	};
	self.intersectsCircle = function (cx, cy, r) {
		var dx = self.x - cx,
			dy = self.y - cy;
		return dx * dx + dy * dy <= r * r;
	};
	return self;
});
var TankZombie = Enemy.expand(function () {
	var self = Enemy.call(this);
	var s = Enemy.call(this);
	s.hp = 60;
	s.damage = 20;
	s.speed = 1;
	s.expValue = 15;
	s.attachAsset('tankZombie', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	return s;
});
var FastZombie = Enemy.expand(function () {
	var self = Enemy.call(this);
	var s = Enemy.call(this);
	s.hp = 15;
	s.speed = 4;
	s.expValue = 8;
	s.attachAsset('fastZombie', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	return s;
});
var Boss = Enemy.expand(function () {
	var self = Enemy.call(this);
	var s = Enemy.call(this);
	s.hp = 500;
	s.damage = 30;
	s.speed = 1.5;
	s.expValue = 100;
	s.type = 'boss';
	s.attachAsset('boss', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	return s;
});
/****-----------------------------------------------------------------
* 5.  Exp orb / pickups  (unchanged except helper magnetSpeed)
*------------------------------------------------------------------*/ 
var ExpOrb = Container.expand(function () {
	var self = Container.call(this);
	self.value = 5;
	self.magnetSpeed = 0;
	self.attachAsset('expOrb', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.update = function () {
		if (!player || !self.magnetSpeed) {
			return;
		}
		var dx = player.x - self.x,
			dy = player.y - self.y,
			d = Math.sqrt(dx * dx + dy * dy);
		if (d > 0) {
			self.x += dx / d * self.magnetSpeed;
			self.y += dy / d * self.magnetSpeed;
		}
	};
	return self;
});
var FlameField = Container.expand(function () {
	var self = Container.call(this);
	var gfx = self.attachAsset('molotov', {
		anchorX: 0.5,
		anchorY: 0.5,
		alpha: 0.4
	});
	self.life = 60;
	self.dmg = 12;
	self.radius = 70;
	self.update = function () {
		if (--self.life <= 0) {
			self.destroy();
		}
	};
	return self;
});
/****-----------------------------------------------------------------
* 6.  Joystick (same as before)
*------------------------------------------------------------------*/ 
var Joystick = Container.expand(function () {
	var self = Container.call(this);
	self.active = false;
	self.dirX = 0;
	self.dirY = 0;
	var base = self.attachAsset('joystickBase', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	base.alpha = 0.5;
	var knob = self.attachAsset('joystickKnob', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	knob.alpha = 0.7;
	self.setKnobPosition = function (x, y) {
		var dx = x - base.x,
			dy = y - base.y,
			d = Math.sqrt(dx * dx + dy * dy),
			m = 60;
		if (d > m) {
			dx = dx / d * m;
			dy = dy / d * m;
		}
		knob.x = dx;
		knob.y = dy;
		self.dirX = dx / m;
		self.dirY = dy / m;
	};
	self.reset = function () {
		knob.x = 0;
		knob.y = 0;
		self.dirX = self.dirY = 0;
		self.active = false;
	};
	return self;
});
var LightningStrike = Container.expand(function () {
	var self = Container.call(this);
	self.attachAsset('lightning', {
		anchorX: 0.5,
		anchorY: 1
	});
	self.life = 30;
	self.dmg = 30;
	self.update = function () {
		if (--self.life <= 0) {
			self.destroy();
		}
	};
	return self;
});
/****-----------------------------------------------------------------
* 7.  Player  (now supports weapons / ench / upgrades)
*------------------------------------------------------------------*/ 
var Player = Container.expand(function () {
	var self = Container.call(this);
	self.level = 1;
	self.exp = 0;
	self.expToNext = 10;
	self.hp = 100;
	self.maxHp = 100;
	self.moveSpeed = 5;
	self.attachAsset('player', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	/* ---- WEAPON INVENTORY ---- */
	self.weapons = {}; // key→data per weapon instance
	addWeapon('sword'); // starts with sword
	function addWeapon(k) {
		if (self.weapons[k]) {
			return false;
		}
		var d = WEAPONS[k];
		self.weapons[k] = {
			lvl: 1,
			cd: d.baseCooldown,
			cdCounter: 0,
			dmgMul: 1,
			rngMul: 1,
			sizeMul: 1,
			amount: 1,
			upgradesGot: 0,
			enchGot: []
		};
		return true;
	}
	/* ---- Level / EXP ---- */
	self.gainExp = function (v) {
		self.exp += v;
		if (self.exp >= self.expToNext) {
			levelUp();
		}
	};
	function levelUp() {
		self.level++;
		self.exp = 0;
		self.expToNext = self.level * 15;
		self.maxHp += 10;
		self.hp = Math.min(self.maxHp, self.hp + self.maxHp * 0.3); // auto-heal 30% after check below
		LK.getSound('levelup').play();
		openChoiceUI();
	}
	/* ---- Combat Tick ---- */
	self.update = function () {
		// traverse owned weapons, fire if CD ready
		for (var w in self.weapons) {
			var inst = self.weapons[w],
				meta = WEAPONS[w];
			inst.cdCounter--;
			if (inst.cdCounter <= 0) {
				fireWeapon(w, inst, meta);
				inst.cdCounter = inst.cd | 0;
			}
		}
	};
	/* ---- Weapon Firing Logic (very lean) ---- */
	function fireWeapon(k, inst, meta) {
		if (Object.keys(enemies).length === 0) {
			return;
		}
		switch (meta.type) {
			case 'melee':
				swingSword(inst);
				break;
			case 'boomerang':
				throwBoomerang(inst);
				break;
			case 'ball':
				throwBall(inst);
				break;
			case 'rocket':
				fireRocket(inst);
				break;
			case 'brick':
				throwBrick(inst, 75);
				if (inst.upgradesGot > 0) throwBrick(inst, -75);
				if (inst.upgradesGot > 1) throwBrick(inst, 45);
				if (inst.upgradesGot > 2) throwBrick(inst, -45);
				break;
			case 'lightning':
				for (var i = 0; i < inst.amount; i++) {
					strikeLightning(inst);
				}
				break;
			case 'aura':
				/* aura handled by passive field following player */break;
			case 'molotov':
				for (var i = 0; i < inst.amount; i++) {
					throwMolotov(inst);
				}
				break;
			case 'shuriken':
				/* spinning handled elsewhere */break;
		}
	}
	/* ----- individual weapon helpers (compact) ----- */
	function nearestEnemy() {
		var n = null,
			d = 1e9;
		for (var i = 0; i < enemies.length; i++) {
			var e = enemies[i],
				dx = e.x - self.x,
				dy = e.y - self.y,
				di = dx * dx + dy * dy;
			if (di < d) {
				d = di;
				n = e;
			}
		}
		return n;
	}
	function directionTo(e) {
		var dx = e.x - self.x,
			dy = e.y - self.y,
			d = Math.sqrt(dx * dx + dy * dy);
		return {
			dx: dx / d,
			dy: dy / d
		};
	}
	function swingSword(inst) {
		var arc = new Container();
		arc.x = self.x;
		arc.y = self.y;
		arc.attachAsset('swordHit', {
			anchorX: 0,
			anchorY: 0.5,
			rotation: self.rotation,
			scaleX: inst.sizeMul,
			scaleY: inst.sizeMul
		});
		arc.life = 10;
		arc.dmg = 10 * inst.dmgMul;
		arc.update = function () {
			if (--arc.life <= 0) {
				arc.destroy();
				return;
			}
			for (var i = enemies.length - 1; i >= 0; i--) {
				var e = enemies[i];
				if (e.intersectsCircle(arc.x, arc.y, meta.range * inst.rngMul) && e.takeDamage(arc.dmg)) {
					killEnemy(i);
				}
			}
		};
		gameContainer.addChild(arc);
	}
	function throwBoomerang(inst) {
		var e = nearestEnemy();
		if (!e) return;
		var dir = directionTo(e);
		for (var i = 0; i < inst.amount; i++) {
			var p = new BoomerangProj();
			p.x = self.x;
			p.y = self.y;
			p.speed = 12;
			p.dirX = dir.dx;
			p.dirY = dir.dy;
			p.dmg = 10 * inst.dmgMul;
			p.returnPhase = false;
			p.life = 240;
			p.update = function () {
				if (!p.returnPhase && --p.life < 180) {
					p.returnPhase = true;
				}
				if (p.returnPhase) {
					var dx = self.x - p.x,
						dy = self.y - p.y,
						d = Math.sqrt(dx * dx + dy * dy);
					p.dirX = dx / d;
					p.dirY = dy / d;
				}
				BoomerangProj.parent.update.call(p);
				for (var i = enemies.length - 1; i >= 0; i--) {
					var en = enemies[i];
					if (p.intersects(en) && en.takeDamage(p.dmg)) {
						killEnemy(i);
					}
				}
			};
			bullets.push(p);
			gameContainer.addChild(p);
		}
	}
	function throwBall(inst) {
		var e = nearestEnemy();
		if (!e) return;
		var dir = directionTo(e);
		for (var i = 0; i < inst.amount; i++) {
			var b = new BallProj();
			b.x = self.x;
			b.y = self.y;
			b.dirX = dir.dx;
			b.dirY = dir.dy;
			b.dmg = 12 * inst.dmgMul;
			b.bounced = false;
			b.update = function () {
				BallProj.parent.update.call(b);
				if (!b.bounced && (Math.abs(b.x - self.x) > 1150 || Math.abs(b.y - self.y) > 1550)) {
					b.dirX *= -1;
					b.dirY *= -1;
					b.bounced = true;
				}
				for (var i = enemies.length - 1; i >= 0; i--) {
					var en = enemies[i];
					if (b.intersects(en) && en.takeDamage(b.dmg)) {
						killEnemy(i);
					}
				}
			};
			bullets.push(b);
			gameContainer.addChild(b);
		}
	}
	function fireRocket(inst) {
		var e = nearestEnemy();
		if (!e) return;
		var dir = directionTo(e);
		for (var i = 0; i < inst.amount; i++) {
			var r = new RocketProj();
			r.x = self.x;
			r.y = self.y;
			r.dirX = dir.dx;
			r.dirY = dir.dy;
			r.dmg = 20 * inst.dmgMul;
			r.update = function () {
				RocketProj.parent.update.call(r);
				for (var i = enemies.length - 1; i >= 0; i--) {
					var en = enemies[i];
					if (r.intersects(en)) {
						rocketAoE(r.x, r.y, 120 * inst.rngMul, r.dmg);
						r.destroy();
						bullets.splice(bullets.indexOf(r), 1);
						break;
					}
				}
			};
			bullets.push(r);
			gameContainer.addChild(r);
		}
	}
	function rocketAoE(cx, cy, rad, dmg) {
		for (var i = enemies.length - 1; i >= 0; i--) {
			var en = enemies[i];
			if (en.intersectsCircle(cx, cy, rad) && en.takeDamage(dmg)) {
				killEnemy(i);
			}
		}
		LK.effects.flashScreen(0xff6622, 100);
	}
	function throwBrick(inst, angle) {
		var rad = self.rotation + angle * Math.PI / 180;
		var dirX = Math.cos(rad),
			dirY = Math.sin(rad);
		var br = new BrickProj();
		br.x = self.x;
		br.y = self.y;
		br.dirX = dirX;
		br.dirY = dirY;
		br.gravity = 0.4;
		br.speed = 14;
		br.dmg = 15 * inst.dmgMul;
		br.update = function () {
			br.y += br.gravity;
			BrickProj.parent.update.call(br);
			if (Math.abs(br.x - self.x) > 1200 || br.y > self.y + 1400) {
				br.destroy();
				bullets.splice(bullets.indexOf(br), 1);
				return;
			}
			for (var i = enemies.length - 1; i >= 0; i--) {
				var en = enemies[i];
				if (br.intersects(en) && en.takeDamage(br.dmg)) {
					killEnemy(i);
				}
			}
		};
		bullets.push(br);
		gameContainer.addChild(br);
	}
	function strikeLightning(inst) {
		var e = enemies[Math.floor(Math.random() * enemies.length)];
		if (!e) return;
		var l = new LightningStrike();
		l.x = e.x;
		l.y = e.y;
		bullets.push(l);
		gameContainer.addChild(l);
		for (var i = enemies.length - 1; i >= 0; i--) {
			var en = enemies[i];
			if (en.x === e.x && en.y === e.y && en.takeDamage(25 * inst.dmgMul)) {
				killEnemy(i);
			}
		}
	}
	function throwMolotov(inst) {
		var angle = Math.random() * Math.PI * 2,
			dist = 200 + Math.random() * 500,
			tx = self.x + Math.cos(angle) * dist,
			ty = self.y + Math.sin(angle) * dist;
		var m = new MolotovProjectile(tx, ty, inst);
		bullets.push(m);
		gameContainer.addChild(m);
	}
	function MolotovProjectile(tx, ty, inst) {
		var mp = new BrickProj();
		mp.x = self.x;
		mp.y = self.y;
		var dx = tx - mp.x,
			dy = ty - mp.y,
			d = Math.sqrt(dx * dx + dy * dy);
		mp.dirX = dx / d;
		mp.dirY = dy / d;
		mp.speed = 15;
		mp.update = function () {
			BrickProj.parent.update.call(mp);
			if (Math.sqrt(Math.pow(mp.x - tx, 2) + Math.pow(mp.y - ty, 2)) < 20) {
				spawnFlame(mp.x, mp.y, inst);
				mp.destroy();
				bullets.splice(bullets.indexOf(mp), 1);
			}
		};
		return mp;
	}
	function spawnFlame(x, y, inst) {
		var f = new FlameField();
		f.x = x;
		f.y = y;
		bullets.push(f);
		gameContainer.addChild(f);
		f.update = function () {
			FlameField.parent.update.call(f);
			for (var i = enemies.length - 1; i >= 0; i--) {
				var en = enemies[i];
				if (en.intersectsCircle(f.x, f.y, f.radius) && en.takeDamage(f.dmg * inst.dmgMul)) {
					killEnemy(i);
				}
			}
		};
	}
	function killEnemy(idx) {
		var en = enemies[idx];
		dropExpOrb(en.x, en.y, en.expValue);
		en.destroy();
		enemies.splice(idx, 1);
	}
	/* --------------------------------------------------------------*/
	/* ---- Choice UI (enchant vs weapon selection) ---- */
	function openChoiceUI() {
		skillSelectionActive = true;
		var overlay = game.addChild(new Container());
		overlay.x = 1024;
		overlay.y = 1366;
		var cards = [];
		var choices = [];
		if (self.level % 5 === 0) {
			// weapon choices
			var pool = Object.keys(WEAPONS);
			while (choices.length < 3 && pool.length) {
				var pick = pool.splice(Math.random() * pool.length | 0, 1)[0];
				choices.push(pick);
			}
			choices.forEach(function (wKey, i) {
				var data = WEAPONS[wKey];
				var already = !!self.weapons[wKey];
				var bg = addCard(i, data.display + (already ? " Upgrade" : " Unlock"));
				bg.effect = function () {
					if (already) {
						var inst = self.weapons[wKey];
						var upArr = data.upgrades;
						if (inst.upgradesGot < 3) {
							applyUpgrade(inst, data);
						} else {
							self.hp = Math.min(self.maxHp, self.hp + self.maxHp * 0.3);
						} // full upgrades → heal 30 %
					} else {
						addWeapon(wKey);
					}
				};
			});
		} else {
			// enchantments for current weapon in hand (sword if only one)
			var currentKey = Object.keys(self.weapons)[0];
			var inst = self.weapons[currentKey],
				meta = WEAPONS[currentKey];
			var pool = meta.ench.filter(function (e) {
				return inst.enchGot.indexOf(e.label) === -1;
			});
			if (pool.length === 0) {
				self.hp = Math.min(self.maxHp, self.hp + self.maxHp * 0.3);
				skillSelectionActive = false;
				overlay.destroy();
				return;
			}
			while (choices.length < 3 && pool.length) {
				var pick = pool.splice(Math.random() * pool.length | 0, 1)[0];
				choices.push(pick);
			}
			choices.forEach(function (e, i) {
				var bg = addCard(i, e.label);
				bg.effect = function () {
					applyEnchant(inst, e);
				};
			});
		}
		function addCard(idx, text) {
			var card = overlay.addChild(new Container());
			card.attachAsset('skillCard', {
				anchorX: 0.5,
				anchorY: 0.5
			});
			card.x = (idx - 1) * 450;
			card.y = 0;
			var t = new Text2(text, {
				size: 38,
				fill: 0xffffff
			});
			t.anchor.set(0.5);
			card.addChild(t);
			card.down = function () {
				this.effect();
				overlay.destroy();
				skillSelectionActive = false;
				levelText.setText('Level ' + self.level);
			};
			return card;
		}
	}
	function applyEnchant(inst, e) {
		inst.enchGot.push(e.label);
		if (e.k === 'dmg') inst.dmgMul *= e.v;
		if (e.k === 'cd') inst.cd *= e.v;
		if (e.k === 'rng') inst.rngMul *= e.v;
		if (e.k === 'size') inst.sizeMul *= e.v;
	}
	function applyUpgrade(inst, meta) {
		inst.upgradesGot++;
		if (meta.type !== 'sword') inst.amount++;
	}
	return self;
});

/**** 
* Initialize Game
****/ 
/****-----------------------------------------------------------------
* 8.  Game bootstrap (everything else from your original file)
*------------------------------------------------------------------*/ 
var game = new LK.Game({
	backgroundColor: 0x111111
});

/**** 
* Game Code
****/ 
/****-----------------------------------------------------------------
* 2.  Weapon / Enchant / Upgrade data tables  (single source of truth)
*------------------------------------------------------------------*/ 
/****-----------------------------------------------------------------
* 1.  Plugins
*------------------------------------------------------------------*/ 
/*— existing assets you already had —*/
/****-----------------------------------------------------------------
* 0.  Assets
*------------------------------------------------------------------*/ 
/******************************************************************** 
*  SURVIVOR-LIKE — HUGE UPDATE  (nine weapons, enchants, upgrades)  *
********************************************************************/ 
var WEAPONS = {
	sword: {
		display: "Sword",
		baseCooldown: 120,
		range: 90,
		type: 'melee',
		ench: [{
			k: 'dmg',
			v: 1.10,
			label: "+10% Damage"
		}, {
			k: 'cd',
			v: 0.90,
			label: "+10% Attack Speed"
		}, {
			k: 'rng',
			v: 1.10,
			label: "+10% Range"
		}, {
			k: 'size',
			v: 1.10,
			label: "+10% Size"
		}],
		upgrades: [{
			id: 1,
			label: "Back Swing"
		}, {
			id: 2,
			label: "Up Swing"
		}, {
			id: 3,
			label: "Down Swing"
		}]
	},
	boomerang: {
		display: "Boomerang",
		baseCooldown: 180,
		range: 900,
		type: 'boomerang',
		ench: [{
			k: 'dmg',
			v: 1.10,
			label: "+10% Damage"
		}, {
			k: 'cd',
			v: 0.90,
			label: "+10% Attack Speed"
		}, {
			k: 'rng',
			v: 1.10,
			label: "+10% Range"
		}, {
			k: 'size',
			v: 1.10,
			label: "+10% Size"
		}],
		upgrades: [{
			id: 1,
			label: "+1 Boomerang"
		}, {
			id: 2,
			label: "+1 Boomerang"
		}, {
			id: 3,
			label: "+1 Boomerang"
		}]
	},
	ball: {
		display: "Rebound Ball",
		baseCooldown: 180,
		range: 900,
		type: 'ball',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "+1 Ball"
		}, {
			id: 2,
			label: "+1 Ball"
		}, {
			id: 3,
			label: "+1 Ball"
		}]
	},
	rocket: {
		display: "Rocket",
		baseCooldown: 180,
		range: 650,
		type: 'rocket',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "+1 Rocket"
		}, {
			id: 2,
			label: "+1 Rocket"
		}, {
			id: 3,
			label: "+1 Rocket"
		}]
	},
	brick: {
		display: "Brick",
		baseCooldown: 180,
		range: 800,
		type: 'brick',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "Back 75° Brick"
		}, {
			id: 2,
			label: "Front 45° Brick"
		}, {
			id: 3,
			label: "Back 45° Brick"
		}]
	},
	lightning: {
		display: "Lightning",
		baseCooldown: 180,
		range: 9999,
		type: 'lightning',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "+1 Lightning"
		}, {
			id: 2,
			label: "+1 Lightning"
		}, {
			id: 3,
			label: "+1 Lightning"
		}]
	},
	aura: {
		display: "Aura",
		baseCooldown: 60,
		range: 150,
		type: 'aura',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "Slow 25%"
		}, {
			id: 2,
			label: "Slow 50%"
		}, {
			id: 3,
			label: "Slow 75%"
		}]
	},
	molotov: {
		display: "Molotov",
		baseCooldown: 180,
		range: 850,
		type: 'molotov',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "+1 Molotov"
		}, {
			id: 2,
			label: "+1 Molotov"
		}, {
			id: 3,
			label: "+1 Molotov"
		}]
	},
	shuriken: {
		display: "Shuriken",
		baseCooldown: 180,
		range: 250,
		type: 'shuriken',
		ench: [/*same four*/],
		upgrades: [{
			id: 1,
			label: "+1 Shuriken"
		}, {
			id: 2,
			label: "+1 Shuriken"
		}, {
			id: 3,
			label: "+1 Shuriken"
		}]
	}
};
// copy generic ench array where needed
for (var w in WEAPONS) {
	if (!WEAPONS[w].ench.length) {
		WEAPONS[w].ench = [{
			k: 'dmg',
			v: 1.10,
			label: "+10% Damage"
		}, {
			k: 'cd',
			v: 0.90,
			label: "+10% Attack Speed"
		}, {
			k: 'rng',
			v: 1.10,
			label: "+10% Range"
		}, {
			k: 'size',
			v: 1.10,
			label: "+10% Size"
		}];
	}
}
/****-----------------------------------------------------------------
* 3.  Base projectile / area containers  
*     (Boomerang, Ball, Rocket, Brick, Lightning, Shuriken, Aura, Molotov Flame)
*------------------------------------------------------------------*/ 
function makeSimpleProj(shapeKey) {
	// generic projectile helper
	return Container.expand(function () {
		var self = Container.call(this);
		self.speed = 10;
		self.dirX = 0;
		self.dirY = 0;
		self.dmg = 10;
		self.life = 600;
		self.attachAsset(shapeKey, {
			anchorX: 0.5,
			anchorY: 0.5
		});
		self.update = function () {
			self.x += self.dirX * self.speed;
			self.y += self.dirY * self.speed;
			if (--self.life <= 0) {
				self.destroy();
			}
		};
		return self;
	});
}
var Bullet = makeSimpleProj('bullet'); // kept for pickup-bomb etc.
var BallProj = makeSimpleProj('ball');
var RocketProj = makeSimpleProj('rocket');
var BrickProj = makeSimpleProj('brick');
var BoomerangProj = makeSimpleProj('boomerang');
var ShurikenProj = makeSimpleProj('shuriken');
var player,
	enemies = [],
	bullets = [],
	expOrbs = [],
	joystick,
	gameTime = 0,
	waveNumber = 1,
	bossSpawned = false;
var skillSelectionActive = false,
	gameContainer,
	camera = {
		x: 0,
		y: 0
	};
/* — create world, player, joystick, UI — */
gameContainer = game.addChild(new Container());
player = gameContainer.addChild(new Player());
player.x = 1024;
player.y = 1366;
joystick = new Joystick();
joystick.x = 0;
joystick.y = -200;
LK.gui.bottom.addChild(joystick);
/* timer / hp / level text (unchanged) … (keep your original UI init code) */
/* — wave spawn / exp orb / helper funcs remain mostly identical — */
/*  ...   (you can keep your original spawnEnemy, spawnWave, dropExpOrb, etc.) */
/* -----------------------------------------------------------------
*  9.  Input handlers  (same as before)
* ----------------------------------------------------------------*/ 
game.down = function (x, y) {
	if (skillSelectionActive) return;
	var loc = game.toLocal({
		x: x,
		y: y
	});
	if (loc.y > 2000) {
		joystick.active = true;
		joystick.setKnobPosition(0, 0);
	}
};
game.move = function (x, y) {
	if (!joystick.active || skillSelectionActive) return;
	var loc = LK.gui.bottom.toLocal({
		x: x,
		y: y
	});
	var dx = loc.x - joystick.x,
		dy = loc.y - joystick.y;
	joystick.setKnobPosition(dx, dy);
};
game.up = function () {
	joystick.reset();
};
/* -----------------------------------------------------------------
* 10.  Update loop (integrates new player & weapon logic)
* ----------------------------------------------------------------*/ 
game.update = function () {
	if (skillSelectionActive) {
		return;
	}
	gameTime++;
	/* --- player movement --- */
	if (joystick.active) {
		player.x += joystick.dirX * player.moveSpeed;
		player.y += joystick.dirY * player.moveSpeed;
	}
	/* --- camera follow --- */
	camera.x = player.x - 1024;
	camera.y = player.y - 1366;
	gameContainer.x = -camera.x;
	gameContainer.y = -camera.y;
	/* --- spawn waves / boss exactly like before (copy your original logic) --- */
	/* --- update containers (player already inside its own update) --- */
	for (var i = enemies.length - 1; i >= 0; i--) {
		var e = enemies[i];
		if (e.intersects(player)) {
			player.hp -= e.damage;
			e.destroy();
			enemies.splice(i, 1);
		}
	}
	bullets.forEach(function (b) {
		if (b.update) b.update();
	});
	expOrbs.forEach(function (o) {
		if (o.update) o.update();
	});
	for (var i = expOrbs.length - 1; i >= 0; i--) {
		var o = expOrbs[i];
		if (o.intersects(player)) {
			player.gainExp(o.value);
			o.destroy();
			expOrbs.splice(i, 1);
		}
	}
};
/* initial wave */
function spawnWave() {
	// Only allow up to 30 enemies at once
	if (enemies.length >= 30) return;
	// Spawn a wave of enemies off-screen, coming toward the player
	var numToSpawn = Math.min(5 + waveNumber, 30 - enemies.length);
	for (var i = 0; i < numToSpawn; i++) {
		// Pick a random angle around the player
		var angle = Math.random() * Math.PI * 2;
		// Spawn distance: just off screen (300px outside the visible area)
		var dist = Math.max(1100, 1500) + 300;
		var ex = player.x + Math.cos(angle) * dist;
		var ey = player.y + Math.sin(angle) * dist;
		// Randomly pick enemy type
		var e;
		var r = Math.random();
		if (r < 0.7) {
			e = new Enemy();
		} else if (r < 0.85) {
			e = new FastZombie();
		} else {
			e = new TankZombie();
		}
		e.x = ex;
		e.y = ey;
		enemies.push(e);
		gameContainer.addChild(e);
	}
	waveNumber++;
	// Boss spawn every 10 waves, only if not already present and under 30 enemies
	if (waveNumber % 10 === 0 && !bossSpawned && enemies.length < 30) {
		var boss = new Boss();
		var bossAngle = Math.random() * Math.PI * 2;
		var bossDist = Math.max(1100, 1500) + 400;
		boss.x = player.x + Math.cos(bossAngle) * bossDist;
		boss.y = player.y + Math.sin(bossAngle) * bossDist;
		enemies.push(boss);
		gameContainer.addChild(boss);
		bossSpawned = true;
	}
}
spawnWave();
/******************************************************************** 
*  END OF HUGE UPDATE
********************************************************************/ 

===================================================================
--- original.js
+++ change.js
@@ -1012,8 +1012,48 @@
 		}
 	}
 };
 /* initial wave */
+function spawnWave() {
+	// Only allow up to 30 enemies at once
+	if (enemies.length >= 30) return;
+	// Spawn a wave of enemies off-screen, coming toward the player
+	var numToSpawn = Math.min(5 + waveNumber, 30 - enemies.length);
+	for (var i = 0; i < numToSpawn; i++) {
+		// Pick a random angle around the player
+		var angle = Math.random() * Math.PI * 2;
+		// Spawn distance: just off screen (300px outside the visible area)
+		var dist = Math.max(1100, 1500) + 300;
+		var ex = player.x + Math.cos(angle) * dist;
+		var ey = player.y + Math.sin(angle) * dist;
+		// Randomly pick enemy type
+		var e;
+		var r = Math.random();
+		if (r < 0.7) {
+			e = new Enemy();
+		} else if (r < 0.85) {
+			e = new FastZombie();
+		} else {
+			e = new TankZombie();
+		}
+		e.x = ex;
+		e.y = ey;
+		enemies.push(e);
+		gameContainer.addChild(e);
+	}
+	waveNumber++;
+	// Boss spawn every 10 waves, only if not already present and under 30 enemies
+	if (waveNumber % 10 === 0 && !bossSpawned && enemies.length < 30) {
+		var boss = new Boss();
+		var bossAngle = Math.random() * Math.PI * 2;
+		var bossDist = Math.max(1100, 1500) + 400;
+		boss.x = player.x + Math.cos(bossAngle) * bossDist;
+		boss.y = player.y + Math.sin(bossAngle) * bossDist;
+		enemies.push(boss);
+		gameContainer.addChild(boss);
+		bossSpawned = true;
+	}
+}
 spawnWave();
 /******************************************************************** 
 *  END OF HUGE UPDATE
 ********************************************************************/ 
\ No newline at end of file