/**** 
* Classes
****/ 
// Ball class
var Ball = Container.expand(function () {
	var self = Container.call(this);
	var ballGraphics = self.attachAsset('ball', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self["throw"] = function () {
		// Throw logic for the ball
	};
	self.spawn = function () {
		self.x = Math.random() * 2048;
		self.y = Math.random() * 2732;
	};
});
// Assets will be automatically created based on usage in the code.
// Basket class
var Basket = Container.expand(function () {
	var self = Container.call(this);
	var basketGraphics = self.attachAsset('basket', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.drawArrow = function (pos) {
		// Calculate the distance and direction of the drag
		var dx = pos.x - self.x;
		var dy = pos.y - self.y;
		var distance = Math.sqrt(dx * dx + dy * dy);
		var direction = Math.atan2(dy, dx);
		// Remove the previous arrow if it exists
		if (self.arrow) {
			self.arrow.destroy();
		}
		// Create a new arrow asset
		self.arrow = LK.getAsset('arrow', {
			anchorX: 0.0,
			// Anchor at the start of the arrow
			anchorY: 0.5,
			// Center vertically
			x: self.x,
			y: self.y,
			scaleX: Math.min(Math.max(distance / 100, 1), 0.3 * Math.sqrt(2048 * 2048 + 2732 * 2732) / 100),
			// Stretch the arrow in the direction of the cursor, but ensure it's at least 1 to allow it to extend outside the screen and not more than 40% of the screen size
			scaleY: 1,
			rotation: direction
		});
		// Correctly align the basket's opening with the arrow direction without additional rotation
		basketGraphics.rotation = direction;
		// Add the arrow to the game
		game.addChild(self.arrow);
	};
	self.move = function () {
		// Move the basket
		self.x += self.vx;
		self.y += self.vy;
		// Apply friction
		self.vx *= 0.99;
		self.vy *= 0.99;
		// Check for collision with the edges of the screen and bounce off
		if (self.x < 0 || self.x > 2048) {
			self.vx *= -1;
			// Correctly adjust the basket's rotation to align with the new direction of movement, ensuring the opening follows the path
			basketGraphics.rotation = Math.atan2(self.vy, self.vx);
		}
		if (self.y < 0 || self.y > 2732) {
			self.vy *= -1;
			// Adjust the basket's rotation to follow the path of reflection, ensuring the opening is correctly aligned
			basketGraphics.rotation = Math.atan2(self.vy, self.vx) + Math.PI / 2;
		}
		// Check for collision with obstacles and boostpads, bounce off obstacles, and reflect with increased speed on boostpads
		for (var i = 0; i < obstacles.length; i++) {
			if (self.intersects(obstacles[i])) {
				// Calculate the angle of reflection
				var dx = self.x - obstacles[i].x;
				var dy = self.y - obstacles[i].y;
				var angle = Math.atan2(dy, dx);
				// Reflect the velocity
				self.vx = Math.cos(angle) * Math.sqrt(self.vx * self.vx + self.vy * self.vy);
				self.vy = Math.sin(angle) * Math.sqrt(self.vx * self.vx + self.vy * self.vy);
				// Adjust the basket's rotation to follow the path of reflection, ensuring the opening faces the correct direction
				basketGraphics.rotation = Math.atan2(self.vy, self.vx);
			}
		}
		for (var j = 0; j < boostpads.length; j++) {
			if (self.intersects(boostpads[j])) {
				// Calculate the angle of reflection
				var dxPad = self.x - boostpads[j].x;
				var dyPad = self.y - boostpads[j].y;
				var anglePad = Math.atan2(dyPad, dxPad);
				// Reflect the velocity and increase speed
				self.vx = Math.cos(anglePad) * 1.6 * Math.sqrt(self.vx * self.vx + self.vy * self.vy);
				self.vy = Math.sin(anglePad) * 1.6 * Math.sqrt(self.vx * self.vx + self.vy * self.vy);
				// Adjust the basket's rotation to follow the path of reflection with increased speed, ensuring the opening faces the correct direction
				basketGraphics.rotation = Math.atan2(self.vy, self.vx);
				// Mark the basket as having been boosted for double scoring
				self.boosted = true;
			}
		}
	};
	self.shoot = function (pos) {
		// Calculate the distance and direction of the drag
		var dx = self.x - pos.x;
		var dy = self.y - pos.y;
		var distance = Math.sqrt(dx * dx + dy * dy);
		var direction = Math.atan2(dy, dx);
		// Set the velocity of the basket to go 50% further than the cursor's end location
		self.vx = -Math.cos(direction) * distance * 1.5 / 100;
		self.vy = -Math.sin(direction) * distance * 1.5 / 100;
	};
	self.isBallHitByOpening = function (ball) {
		// Calculate the angle between the basket's direction and the ball
		var dx = ball.x - self.x;
		var dy = ball.y - self.y;
		var angle = Math.atan2(dy, dx);
		// Check if the ball is within the opening angle of the basket
		return Math.abs(angle - basketGraphics.rotation) < Math.PI / 4;
	};
});
// Boostpad class
var Boostpad = Container.expand(function () {
	var self = Container.call(this);
	var boostpadGraphics = self.attachAsset('obstacle', {
		// Reuse obstacle asset for simplicity
		anchorX: 0.5,
		anchorY: 0.5,
		tint: 0xFFD700 // Gold color to differentiate from obstacles
	});
	self.positionBoostpad = function () {
		// Positioning logic for the boostpad, similar to obstacle
	};
});
// Obstacle class
var Obstacle = Container.expand(function () {
	var self = Container.call(this);
	var obstacleGraphics = self.attachAsset('obstacle', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.positionObstacle = function () {
		// Positioning logic for the obstacle
	};
});

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

/**** 
* Game Code
****/ 
// Level menu and level-specific configurations
var currentLevel = 0;
var levelConfigurations = [{
	// Level 1
	ballCount: 5,
	obstaclePositions: [{
		x: 500,
		y: 500
	}, {
		x: 1500,
		y: 1500
	}],
	boostpadPositions: [{
		x: 1024,
		y: 1366
	}]
}, {
	// Level 2
	ballCount: 7,
	obstaclePositions: [{
		x: 300,
		y: 600
	}, {
		x: 1700,
		y: 1300
	}, {
		x: 1024,
		y: 800
	}],
	boostpadPositions: [{
		x: 512,
		y: 1366
	}, {
		x: 1536,
		y: 1366
	}]
}
// Add configurations for levels 3, 4, and 5 here
];
// Initialize game elements based on level configuration
var balls = [];
function initializeLevel(levelIndex) {
	var config = levelConfigurations[levelIndex];
	var basket = game.addChild(new Basket());
	basket.x = 1024; // Center horizontally
	basket.y = 2732 - 2732 / 4; // Position 1/4 of the screen size away from the bottom
	basket.visible = true; // Ensure the basket is always visible
	basket.arrow = null; // Initialize arrow to null
	balls = [];
	for (var i = 0; i < config.ballCount; i++) {
		var ball = game.addChild(new Ball());
		ball.spawn();
		balls.push(ball);
	}
	var obstacles = [];
	var boostpads = []; // Array to store boostpad instances
	// Create obstacles and boostpads based on level configuration
	config.obstaclePositions.forEach(function (pos) {
		var obstacle = game.addChild(new Obstacle());
		obstacle.x = pos.x;
		obstacle.y = pos.y;
		obstacles.push(obstacle);
	});
	config.boostpadPositions.forEach(function (pos) {
		var boostpad = game.addChild(new Boostpad());
		boostpad.x = pos.x;
		boostpad.y = pos.y;
		boostpads.push(boostpad);
	});
}
// Display level menu and wait for user selection
function displayLevelMenu() {
	// Create level selection assets for 5 levels
	var levelAssets = [];
	for (var i = 0; i < 5; i++) {
		(function (index) {
			var levelAsset = game.addChild(LK.getAsset('level' + (index + 1), {
				width: 200,
				height: 200,
				id: 'level' + (index + 1),
				x: 200 + index * 400,
				// Adjust x position to evenly space out level assets across the screen
				y: 1366 - 200,
				// Adjust y position to make level assets fully visible
				anchorX: 0.5,
				anchorY: 0.5
			}));
			game.addChild(levelAsset);
			levelAssets.push(levelAsset);
			// Add touch event listener for each level asset
			levelAsset.on('down', function () {
				// On selection, initialize the corresponding level
				initializeLevel(index);
				// Remove level selection assets from the game
				levelAssets.forEach(function (asset) {
					asset.destroy();
				});
				// Removed code that incorrectly sets basket visibility and shooting functionality here
			});
		})(i);
	}
}
// Start with displaying the level menu
displayLevelMenu();
// Function to spawn a new ball
function spawnBall() {
	var ball = new Ball();
	ball.x = 1024; // Start from the center horizontally
	ball.y = 2732 - 150; // Start a little above the basket
	balls.push(ball);
	game.addChild(ball);
}
// Touch event to throw a ball
var initialPos;
game.on('down', function (obj) {
	var event = obj.event;
	initialPos = event.getLocalPosition(game);
	basket.shoot(initialPos);
});
game.on('move', function (obj) {
	var event = obj.event;
	var pos = event.getLocalPosition(game);
	basket.drawArrow(pos);
});
// Game tick event
LK.on('tick', function () {
	// Update balls
	for (var i = balls.length - 1; i >= 0; i--) {
		balls[i]["throw"]();
		basket.move();
		// Check for collision with basket
		if (balls[i].intersects(basket) && basket.isBallHitByOpening(balls[i])) {
			balls[i].destroy();
			balls.splice(i, 1);
			// Increase the score and update the score counter
			score++;
			// Check if the basket was boosted and the ball was hit in the same shot
			if (basket.boosted && basket.isBallHitByOpening(balls[i])) {
				score++; // Additional score for boosted basket
				// Display bonus asset for one second in the top left corner when a bonus is hit
				var bonusAsset = LK.getAsset('bonus', {
					x: 150,
					// Move away from the corner
					y: 150,
					// Move away from the corner
					anchorX: 0,
					anchorY: 0,
					scaleX: 3,
					// Make it three times as big
					scaleY: 3 // Make it three times as big
				});
				game.addChild(bonusAsset);
				LK.setTimeout(function () {
					bonusAsset.destroy();
				}, 1000);
			}
			// Reset boosted state after checking for bonus
			basket.boosted = false;
			scoreTxt.setText('Score: ' + score);
		}
		// Remove balls that are off-screen or collide with obstacles
		if (balls[i] && (balls[i].y < 0 || obstacles.some(function (obstacle) {
			return balls[i].intersects(obstacle);
		}))) {
			balls[i].destroy();
			balls.splice(i, 1);
		}
	}
});
// Initialize the first ball
spawnBall();
// Create a score counter
var score = 0;
var scoreTxt = new Text2('Score: 0', {
	size: 50,
	fill: "#ffffff"
});
scoreTxt.anchor.set(1, 0); // Anchor to the top right corner
LK.gui.topRight.addChild(scoreTxt);