/**** 
* Classes
****/ 
//<Assets used in the game will automatically appear here>
//<Write imports for supported plugins here>
// Ball class to represent falling balls
var Ball = Container.expand(function (color) {
	var self = Container.call(this);
	var ballGraphics = self.attachAsset(color + 'Ball', {
		anchorX: 0.5,
		anchorY: 0.5,
		shape: 'ellipse'
	});
	self.color = color;
	self.speed = 5; // Speed of falling balls
	self.update = function () {
		self.y += self.speed;
	};
});
// BonusBall class to represent bonus blue balls
var BonusBall = Ball.expand(function () {
	var self = Ball.call(this, 'Blue');
	self.speed = self.speed || 3; // Speed of falling bonus balls
});
// Bucket class to represent buckets at the bottom
var Bucket = Container.expand(function (color, xPosition) {
	var self = Container.call(this);
	var bucketGraphics = self.attachAsset(color + 'Bucket', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.color = color;
	self.x = xPosition;
	self.y = 2600; // Position buckets at the bottom
});

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

/**** 
* Game Code
****/ 
// 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', 512)));
buckets.push(game.addChild(new Bucket('Green', 1024)));
buckets.push(game.addChild(new Bucket('Red', 1536)));
// Create buckets
buckets.push(game.addChild(new Bucket('Blue', 512)));
buckets.push(game.addChild(new Bucket('Green', 1024)));
buckets.push(game.addChild(new Bucket('Red', 1536)));
// 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 - 200) + 100; // Ensure all balls drop within the margins
	newBall.y = 0;
	balls.push(newBall);
	game.addChild(newBall);
}
// 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] instanceof BonusBall && buckets[j].color === 'Blue') {
					score += 20;
					scoreTxt.setText(score);
					balls[i].destroy();
					balls.splice(i, 1);
					break;
				} else if (balls[i].color === 'Red' && buckets[j].color === 'Red' || balls[i].color === 'Blue' && buckets[j].color === 'Blue' || balls[i].color === 'Green' && buckets[j].color === 'Green') {
					score++;
					scoreTxt.setText(score);
					balls[i].destroy();
					balls.splice(i, 1);
					break;
				} else {
					LK.effects.flashScreen(0xff0000, 1000);
					LK.showGameOver();
				}
			}
		}
		// End game when a ball is not collected
		if (balls[i] && balls[i].y > 2732) {
			LK.effects.flashScreen(0xff0000, 1000);
			LK.showGameOver();
		}
	}
	// Increase the speed of the balls dropping after collecting ten balls
	if (score >= 10) {
		Ball.prototype.speed = 10;
	}
	// Spawn a new ball every 60 ticks
	if (LK.ticks % 60 === 0) {
		spawnBall();
		// Spawn a bonus blue ball randomly
		if (Math.random() < 0.1) {
			var newBonusBall = new BonusBall();
			newBonusBall.x = Math.random() * (2048 - 200) + 100; // Add margins to the canvas on the left and right sides
			newBonusBall.y = 0;
			balls.push(newBonusBall);
			game.addChild(newBonusBall);
		}
	}
};
// Handle touch/mouse events to direct balls
game.down = function (x, y, obj) {
	for (var i = 0; i < balls.length; i++) {
		if (balls[i].intersects({
			x: x,
			y: y,
			width: 1,
			height: 1
		})) {
			balls[i].x = x;
			balls[i].y = y;
			if (balls[i].color === 'Red') {
				console.log("RedBall was clicked");
				balls[i].x += 100; // Move RedBall to the right
			}
			if (balls[i].color === 'Blue') {
				console.log("BlueBall was clicked");
				balls[i].x -= 100; // Move BlueBall to the left
			}
			if (balls[i].color === 'Green') {
				console.log("GreenBall was clicked");
				balls[i].x += 100; // Move GreenBall to the right
			}
		}
	}
	for (var j = 0; j < buckets.length; j++) {
		if (buckets[j].color === 'Green' && buckets[j].intersects({
			x: x,
			y: y,
			width: 1,
			height: 1
		})) {
			buckets[j].x = x - buckets[j].width / 2;
			buckets[j].y = 2600;
		} else if (buckets[j].color === 'Red' && buckets[j].intersects({
			x: x,
			y: y,
			width: 1,
			height: 1
		})) {
			buckets[j].x = x - buckets[j].width / 2;
		}
	}
};
game.move = function (x, y, obj) {
	for (var i = 0; i < balls.length; i++) {
		if (balls[i].intersects({
			x: x,
			y: y,
			width: 1,
			height: 1
		})) {
			balls[i].x = x;
			balls[i].y = y;
		}
	}
	for (var j = 0; j < buckets.length; j++) {
		if (buckets[j].intersects({
			x: x,
			y: y,
			width: 1,
			height: 1
		})) {
			buckets[j].x = x - buckets[j].width / 2;
		}
	}
};
game.up = function (x, y, obj) {
	// No specific action needed on release
};