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

/**** 
* Classes
****/ 
// ENEMY STATES
// 0: normal
// 1: shielded (red)
// 2: fading (about to die)
var Enemy = Container.expand(function () {
	var self = Container.call(this);
	// Default: normal enemy
	self.state = 0; // 0: normal, 1: shielded, 2: fading
	self.isWizard = false;
	self.isProtected = false; // If protected by wizard
	self.protector = null; // Wizard protecting this enemy
	// Asset
	self.enemyAsset = self.attachAsset('enemy', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.shieldAsset = null; // For shield effect
	// For movement
	self.speed = 2 + Math.random() * 1.5; // Slightly random speed
	self.angle = 0; // Will be set on spawn
	self.radius = 0; // Distance from center
	// For fade-out
	self.fadeTimer = 0;
	// For collision
	self.lastLaserHit = false;
	// For wizard
	self.isWizard = false;
	self.shieldedEnemy = null; // For wizard: the enemy it protects
	// Set state (change color/asset)
	self.setState = function (state) {
		self.state = state;
		if (self.isWizard) {
			// Always wizard asset
			if (self.enemyAsset) self.enemyAsset.destroy();
			self.enemyAsset = self.attachAsset('wizard', {
				anchorX: 0.5,
				anchorY: 0.5
			});
		} else if (state === 0) {
			if (self.enemyAsset) self.enemyAsset.destroy();
			self.enemyAsset = self.attachAsset('enemy', {
				anchorX: 0.5,
				anchorY: 0.5
			});
		} else if (state === 1) {
			if (self.enemyAsset) self.enemyAsset.destroy();
			self.enemyAsset = self.attachAsset('enemy_shielded', {
				anchorX: 0.5,
				anchorY: 0.5
			});
		} else if (state === 2) {
			if (self.enemyAsset) self.enemyAsset.destroy();
			self.enemyAsset = self.attachAsset('enemy_fading', {
				anchorX: 0.5,
				anchorY: 0.5
			});
		}
	};
	// Show/hide shield
	self.setShield = function (on) {
		if (on && !self.shieldAsset) {
			self.shieldAsset = self.attachAsset('shield', {
				anchorX: 0.5,
				anchorY: 0.5
			});
			self.shieldAsset.alpha = 0.35;
		} else if (!on && self.shieldAsset) {
			self.shieldAsset.destroy();
			self.shieldAsset = null;
		}
	};
	// Called every tick
	self.update = function () {
		// Move towards center
		var dx = 1024 - self.x;
		var dy = 1366 - self.y;
		var dist = Math.sqrt(dx * dx + dy * dy);
		if (dist > 0) {
			var moveDist = Math.min(self.speed, dist);
			self.x += dx / dist * moveDist;
			self.y += dy / dist * moveDist;
		}
		// Fading out
		if (self.state === 2) {
			self.fadeTimer += 1;
			if (self.fadeTimer > 30) {
				// Remove after fade
				self.destroy();
			} else {
				self.alpha = 1 - self.fadeTimer / 30;
			}
		}
	};
	// For wizard: assign protected enemy
	self.setShieldedEnemy = function (enemy) {
		self.shieldedEnemy = enemy;
		if (enemy) {
			enemy.isProtected = true;
			enemy.protector = self;
			enemy.setShield(true);
		}
	};
	// For wizard: remove shield from protected enemy
	self.removeShieldedEnemy = function () {
		if (self.shieldedEnemy) {
			self.shieldedEnemy.isProtected = false;
			self.shieldedEnemy.protector = null;
			self.shieldedEnemy.setShield(false);
			self.shieldedEnemy = null;
		}
	};
	// For protected enemy: called if wizard dies
	self.removeProtection = function () {
		self.isProtected = false;
		self.protector = null;
		self.setShield(false);
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// Sound for enemy destroyed
// Shield effect (for protected enemies)
// Enemy: wizard
// Enemy: fading (second hit)
// Enemy: shielded (first hit)
// Enemy: normal
// Lazer beam (thin, long rectangle)
// Main character (center)
// Center of screen
var CENTER_X = 1024;
var CENTER_Y = 1366;
// Main character
var hero = new Container();
var heroAsset = hero.attachAsset('hero', {
	anchorX: 0.5,
	anchorY: 0.5
});
hero.x = CENTER_X;
hero.y = CENTER_Y;
game.addChild(hero);
// Laser
var laser = new Container();
var laserAsset = laser.attachAsset('laser', {
	anchorX: 0.5,
	anchorY: 0
});
laser.x = CENTER_X;
laser.y = CENTER_Y;
laser.rotation = 0;
game.addChild(laser);
// Laser direction (in radians)
var laserAngle = 0; // 0 = up
var laserTargetAngle = 0; // For smooth tweening
// Touch drag control
var dragging = false;
var dragStart = null;
var dragStartAngle = 0;
// Enemies
var enemies = [];
var spawnWaveTimer = 0;
var waveNumber = 1;
// Score
var score = 0;
var scoreTxt = new Text2('0', {
	size: 120,
	fill: "#fff"
});
scoreTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(scoreTxt);
// Helper: spawn a single enemy at random angle/distance
function spawnEnemy(type) {
	var enemy = new Enemy();
	enemy.isWizard = type === 'wizard';
	enemy.setState(enemy.isWizard ? 0 : 0);
	// Spawn at random angle, at edge of screen
	var angle = Math.random() * Math.PI * 2;
	var dist = 1200 + Math.random() * 400;
	enemy.x = CENTER_X + Math.cos(angle) * dist;
	enemy.y = CENTER_Y + Math.sin(angle) * dist;
	enemy.angle = angle;
	enemy.radius = dist;
	enemy.speed = 2 + Math.random() * 1.5 + (enemy.isWizard ? 0.5 : 0);
	game.addChild(enemy);
	enemies.push(enemy);
	return enemy;
}
// Helper: assign wizards to protect nearest enemy
function assignWizards() {
	// Get all wizards
	var wizards = [];
	var normalEnemies = [];
	for (var i = 0; i < enemies.length; ++i) {
		var e = enemies[i];
		if (e.isWizard) wizards.push(e);else if (!e.isProtected) normalEnemies.push(e);
	}
	// For each wizard, find nearest unprotected enemy
	for (var j = 0; j < wizards.length; ++j) {
		var wiz = wizards[j];
		// Remove old shield
		wiz.removeShieldedEnemy();
		// Find nearest
		var minDist = 99999;
		var nearest = null;
		for (var k = 0; k < enemies.length; ++k) {
			var e2 = enemies[k];
			if (e2 === wiz) continue;
			if (e2.isProtected) continue;
			if (e2.state === 2) continue; // Fading out
			var dx = wiz.x - e2.x;
			var dy = wiz.y - e2.y;
			var d = dx * dx + dy * dy;
			if (d < minDist) {
				minDist = d;
				nearest = e2;
			}
		}
		if (nearest) {
			wiz.setShieldedEnemy(nearest);
		}
	}
}
// Helper: spawn a wave of enemies
function spawnWave() {
	var numEnemies = 3 + Math.floor(waveNumber * 0.7);
	var numWizards = Math.min(1 + Math.floor(waveNumber / 3), Math.floor(numEnemies / 3));
	// Spawn wizards
	for (var i = 0; i < numWizards; ++i) {
		spawnEnemy('wizard');
	}
	// Spawn normal enemies
	for (var j = 0; j < numEnemies; ++j) {
		spawnEnemy('normal');
	}
	assignWizards();
	waveNumber += 1;
}
// Helper: check if a point is inside the laser beam
function pointInLaser(px, py) {
	// Laser is a rectangle from (CENTER_X, CENTER_Y) in direction laserAngle, length = laserAsset.height
	var dx = px - CENTER_X;
	var dy = py - CENTER_Y;
	var len = Math.sqrt(dx * dx + dy * dy);
	if (len < 60) return false; // Don't hit at center
	var angleToPoint = Math.atan2(dy, dx);
	var diff = Math.abs(angleToPoint - laserAngle);
	// Normalize diff to [0, PI]
	while (diff > Math.PI) diff = Math.abs(diff - 2 * Math.PI);
	// If within 0.13 rad (~7.5 deg) of laser direction, and within laser length
	if (diff < 0.13 && len < laserAsset.height) {
		return true;
	}
	return false;
}
// Helper: remove enemy from array and game
function removeEnemy(enemy) {
	// Remove shield if wizard
	if (enemy.isWizard) {
		enemy.removeShieldedEnemy();
	}
	// Remove protection if protected
	if (enemy.isProtected && enemy.protector) {
		enemy.protector.removeShieldedEnemy();
	}
	for (var i = enemies.length - 1; i >= 0; --i) {
		if (enemies[i] === enemy) {
			enemies.splice(i, 1);
			break;
		}
	}
	enemy.destroy();
}
// Touch/drag controls
game.down = function (x, y, obj) {
	// Only start drag if not in top left 100x100
	if (x < 100 && y < 100) return;
	dragging = true;
	dragStart = {
		x: x,
		y: y
	};
	dragStartAngle = laserTargetAngle;
};
game.move = function (x, y, obj) {
	if (dragging && dragStart) {
		// Calculate angle from center to current touch
		var dx = x - CENTER_X;
		var dy = y - CENTER_Y;
		var angle = Math.atan2(dy, dx);
		// Clamp to [-PI, PI]
		if (angle > Math.PI) angle -= 2 * Math.PI;
		if (angle < -Math.PI) angle += 2 * Math.PI;
		laserTargetAngle = angle;
	}
};
game.up = function (x, y, obj) {
	dragging = false;
	dragStart = null;
};
// Main update loop
game.update = function () {
	// Smoothly rotate laser towards target angle
	var diff = laserTargetAngle - laserAngle;
	while (diff > Math.PI) diff -= 2 * Math.PI;
	while (diff < -Math.PI) diff += 2 * Math.PI;
	laserAngle += diff * 0.18; // Smooth
	// Clamp
	if (laserAngle > Math.PI) laserAngle -= 2 * Math.PI;
	if (laserAngle < -Math.PI) laserAngle += 2 * Math.PI;
	laser.rotation = laserAngle - Math.PI / 2; // Because asset is vertical
	// Update all enemies
	for (var i = enemies.length - 1; i >= 0; --i) {
		var e = enemies[i];
		e.update();
		// Check if reached center (game over)
		var dx = e.x - CENTER_X;
		var dy = e.y - CENTER_Y;
		var dist = Math.sqrt(dx * dx + dy * dy);
		if (dist < 100 && e.state !== 2) {
			// Game over
			LK.effects.flashScreen(0xff0000, 1000);
			LK.showGameOver();
			return;
		}
	}
	// Laser collision
	var anyLaserHit = false;
	for (var j = enemies.length - 1; j >= 0; --j) {
		var e2 = enemies[j];
		if (e2.state === 2) continue; // Already dying
		if (pointInLaser(e2.x, e2.y)) {
			// If protected, can't be hit
			if (e2.isProtected) {
				// Show shield effect
				e2.setShield(true);
				continue;
			}
			// Wizards: die in one hit
			if (e2.isWizard) {
				e2.setState(2);
				e2.fadeTimer = 0;
				// Remove shield from protected enemy
				e2.removeShieldedEnemy();
				LK.getSound('wizard_down').play();
				score += 5;
				scoreTxt.setText(score);
				anyLaserHit = true;
				continue;
			}
			// Normal enemy: 2 hits to die
			if (e2.state === 0) {
				e2.setState(1);
				LK.getSound('laser_hit').play();
				anyLaserHit = true;
			} else if (e2.state === 1) {
				e2.setState(2);
				e2.fadeTimer = 0;
				LK.getSound('enemy_down').play();
				score += 1;
				scoreTxt.setText(score);
				anyLaserHit = true;
			}
		} else {
			// Not hit, remove shield effect if any
			if (!e2.isWizard && !e2.isProtected) {
				e2.setShield(false);
			}
		}
	}
	// Remove dead enemies
	for (var k = enemies.length - 1; k >= 0; --k) {
		var e3 = enemies[k];
		if (e3.state === 2 && e3.fadeTimer > 30) {
			removeEnemy(e3);
		}
	}
	// If all enemies gone, spawn next wave
	var living = 0;
	for (var m = 0; m < enemies.length; ++m) {
		if (enemies[m].state !== 2) living += 1;
	}
	if (living === 0) {
		spawnWaveTimer += 1;
		if (spawnWaveTimer > 40) {
			spawnWave();
			spawnWaveTimer = 0;
		}
	} else {
		spawnWaveTimer = 0;
	}
	// Win condition: e.g. after 10 waves
	if (waveNumber > 10 && living === 0) {
		LK.showYouWin();
	}
};
// Start first wave
spawnWave();
scoreTxt.setText(score);

===================================================================
--- original.js
+++ change.js