/**** 
* Classes
****/ 
// Ball class to represent falling balls
var Ball = Container.expand(function (color) {
	var self = Container.call(this);
	self.attachAsset(color + 'Ball', {
		anchorX: 0.5,
		anchorY: 0.5,
		shape: 'ellipse'
	});
	self.color = color;
	self.speed = 2; // Speed of falling balls
	self.dragging = false;
	self.update = function () {
		if (!self.dragging) {
			self.y += self.speed;
		}
		if (self.x < 40) {
			self.x = 40;
		}
		if (self.x > 2008) {
			self.x = 2008;
		}
	};
	self.startDrag = function (x, y) {
		self.dragging = true;
		self.dragStartX = x - self.x;
		self.dragStartY = y - self.y;
	};
	self.drag = function (x, y) {
		if (self.dragging) {
			self.x = x - self.dragStartX;
			self.y = y - self.dragStartY;
		}
	};
	self.stopDrag = function () {
		self.dragging = false;
	};
	self.intersects = function (obj) {
		var dx = this.x - obj.x;
		var dy = this.y - obj.y;
		var distance = Math.sqrt(dx * dx + dy * dy);
		return distance < this.width / 2 + obj.width / 2;
	};
	self.destroy = function () {
		this.parent.removeChild(this);
	};
	return self;
});
// BlueBall class to represent blue falling balls
var BlueBall = Ball.expand(function () {
	var self = Ball.call(this, 'Blue');
	return self;
});
// Bucket class to represent buckets at the bottom
var Bucket = Container.expand(function (color, xPosition) {
	var self = Container.call(this);
	self.attachAsset(color + 'Bucket', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.color = color;
	self.x = xPosition;
	self.y = 2600; // Position buckets at the bottom
	return self;
});

/**** 
* Initialize Game
****/ 
var game = new LK.Game({
	backgroundColor: 0x40E0D0 // Init game with turquoise background
});

/**** 
* Game Code
****/ 
// Create yellow margins
var leftMargin = game.attachAsset('leftMargin', {
	anchorX: 0.0,
	anchorY: 0.0,
	x: 0,
	y: 0
});
var rightMargin = game.attachAsset('rightMargin', {
	anchorX: 0.0,
	anchorY: 0.0,
	x: 2008,
	y: 0
});
// Initialize arrays and variables
var balls = [];
var buckets = [];
var score = 0;
var scoreTxt = new Text2('0', {
	size: 150,
	fill: 0xFFFFFF
});
scoreTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(scoreTxt);
// Create buckets
buckets.push(game.addChild(new Bucket('Blue', 400)));
buckets.push(game.addChild(new Bucket('Green', 1024)));
buckets.push(game.addChild(new Bucket('Red', 1648)));
// Function to spawn a new ball
function spawnBall() {
	var colors = ['Red', 'Blue', 'Green'];
	var color = colors[Math.floor(Math.random() * colors.length)];
	var newBall = new Ball(color);
	newBall.x = Math.random() * 2048;
	newBall.y = 0;
	balls.push(newBall);
	game.addChild(newBall);
}
// Add balls on the game canvas
var ball1 = game.addChild(new Ball('Blue'));
ball1.x = 1024;
ball1.y = 0;
balls.push(ball1);
var ball2 = game.addChild(new Ball('Red'));
ball2.x = 512;
ball2.y = 0;
balls.push(ball2);
var ball3 = game.addChild(new Ball('Green'));
ball3.x = 1536;
ball3.y = 0;
balls.push(ball3);
// Handle game update
game.update = function () {
	for (var i = balls.length - 1; i >= 0; i--) {
		balls[i].update();
		// Check if ball is in a bucket
		for (var j = 0; j < buckets.length; j++) {
			if (balls[i].intersects(buckets[j])) {
				if (balls[i].color === buckets[j].color) {
					score++;
					scoreTxt.setText(score);
					LK.getSound('Poof').play();
					balls[i].destroy();
					balls.splice(i, 1);
					break;
				} else {
					LK.effects.flashScreen(0xff0000, 1000);
					LK.getSound('GameEnd').play();
					LK.showGameOver();
					return;
				}
			}
		}
		// Remove balls that are not collected
		if (balls[i] && balls[i].y > 2732) {
			balls[i].destroy();
			balls.splice(i, 1);
		}
	}
	// Increase the speed of the balls dropping after collecting ten balls
	if (score >= 10) {
		Ball.prototype.speed = 10;
	}
	// Increase the speed of the balls to 4 once the points reach 20
	if (score >= 20) {
		Ball.prototype.speed = 4;
	}
	// Spawn a new ball every 60 ticks
	if (LK.ticks % 60 === 0) {
		spawnBall();
	}
};
// Handle touch/mouse events to direct balls
game.down = function (x, y, obj) {
	var game_position = game.toLocal(obj.global);
	for (var i = 0; i < balls.length; i++) {
		if (balls[i].intersects({
			x: game_position.x,
			y: game_position.y,
			width: 1,
			height: 1
		})) {
			balls[i].startDrag(game_position.x, game_position.y);
		}
	}
};
game.move = function (x, y, obj) {
	var game_position = game.toLocal(obj.global);
	for (var i = 0; i < balls.length; i++) {
		if (balls[i].dragging) {
			balls[i].drag(game_position.x, game_position.y);
		}
	}
};
game.up = function (x, y, obj) {
	for (var i = 0; i < balls.length; i++) {
		balls[i].stopDrag();
	}
};