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

/**** 
* Classes
****/ 
// Bird class
var Bird = Container.expand(function () {
	var self = Container.call(this);
	var birdSprite = self.attachAsset('bird', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	var birdDrinkSprite = self.attachAsset('Bird_drink', {
		anchorX: 0.5,
		anchorY: 0.5,
		visible: false
	});
	// Position drink in front of bird
	birdDrinkSprite.x = birdSprite.width * 0.7; // Offset to the right (in front)
	self.currentSprite = birdSprite;
	self.radius = birdSprite.width * 0.5 * 0.85; // for collision
	self.vy = 0; // vertical speed
	self.gravity = 0.6; // gravity per frame (slower fall)
	self.flapStrength = -22; // negative = up, less strong flap
	// Flap method
	self.flap = function () {
		self.vy = self.flapStrength;
		LK.getSound('flap').play();
	};
	// Toggle drink method
	self.toggleDrink = function () {
		// Simply toggle visibility of drink sprite
		birdDrinkSprite.visible = !birdDrinkSprite.visible;
		// Save the current drink state to storage
		storage.lastDrink = birdDrinkSprite.visible;
	};
	// Update method
	self.update = function () {
		self.vy += self.gravity;
		self.y += self.vy;
		// Rotate bird based on velocity (visual only)
		var maxAngle = Math.PI / 4;
		var minAngle = -Math.PI / 6;
		var angle = self.vy / 40 * maxAngle;
		if (angle > maxAngle) angle = maxAngle;
		if (angle < minAngle) angle = minAngle;
		birdSprite.rotation = angle;
		// Keep drink upright and in front
		birdDrinkSprite.rotation = 0;
	};
	return self;
});
// PipePair class (top and bottom pipes)
var PipePair = Container.expand(function () {
	var self = Container.call(this);
	// Pipe config
	var pipeWidth = 200;
	var pipeHeight = 1200;
	var gapHeight = 480; // vertical gap between pipes
	// Randomly decide if pipes are flipped (top/bottom swap orientation)
	var flip = Math.random() < 0.5;
	// Top pipe (can be flipped)
	var topPipe = self.attachAsset('pipe', {
		anchorX: 0,
		anchorY: flip ? 0 : 1,
		width: pipeWidth,
		height: pipeHeight,
		rotation: flip ? Math.PI : 0
	});
	// Bottom pipe (can be flipped)
	var bottomPipe = self.attachAsset('pipe', {
		anchorX: 0,
		anchorY: flip ? 1 : 0,
		width: pipeWidth,
		height: pipeHeight,
		rotation: flip ? Math.PI : 0
	});
	self.pipeWidth = pipeWidth;
	self.gapHeight = gapHeight;
	self.passed = false; // for scoring
	// Set pipes' vertical positions based on gapY (top of gap)
	self.setGap = function (gapY) {
		// gapY: y position of top of gap
		topPipe.y = gapY;
		bottomPipe.y = gapY + gapHeight;
	};
	// Move pipes left
	self.update = function () {
		self.x -= pipeSpeed;
	};
	// Get bounding rects for collision
	self.getTopRect = function () {
		return {
			x: self.x,
			y: topPipe.y - pipeHeight,
			width: pipeWidth,
			height: pipeHeight
		};
	};
	self.getBottomRect = function () {
		return {
			x: self.x,
			y: bottomPipe.y,
			width: pipeWidth,
			height: pipeHeight
		};
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// Sound: Hit
// Sound: Score
// Sound: Flap
// Ground: brown box
// Pipe: green box
// Bird: yellow ellipse
// Game constants
var BIRD_START_X = 600;
var BIRD_START_Y = 1200;
var PIPE_INTERVAL = 1300; // px between pipes horizontally (slower spawn rate)
var PIPE_MIN_Y = 350; // min y for top of gap
var PIPE_MAX_Y = 1800; // max y for top of gap
var GROUND_Y = 2732 - 120; // ground top y
var pipeSpeed = 8; // px per frame (slower pipes)
// Game state
var bird;
var pipes = [];
var ground;
var score = 0;
var scoreTxt;
var gameOver = false;
var started = false;
var lastPipeX = 0;
// GUI: Score
scoreTxt = new Text2('0', {
	size: 160,
	fill: 0xFFFFFF
});
scoreTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(scoreTxt);
// GUI: Game Over text
var gameOverTxt = new Text2('Game Over', {
	size: 180,
	fill: 0xFF4444
});
gameOverTxt.anchor.set(0.5, 0.5);
gameOverTxt.visible = false;
LK.gui.center.addChild(gameOverTxt);
// GUI: Tap to restart
var restartTxt = new Text2('Tap to restart', {
	size: 100,
	fill: 0xFFFFFF
});
restartTxt.anchor.set(0.5, 0.5);
restartTxt.visible = false;
LK.gui.center.addChild(restartTxt);
// GUI: Drink toggle button
var drinkToggleBtn = new Container();
var drinkBtnBg = drinkToggleBtn.attachAsset('centerCircle', {
	anchorX: 0.5,
	anchorY: 0.5,
	width: 150,
	height: 150,
	tint: 0x4444ff
});
var drinkBtnText = new Text2('🥤', {
	size: 80,
	fill: 0xFFFFFF
});
drinkBtnText.anchor.set(0.5, 0.5);
drinkToggleBtn.addChild(drinkBtnText);
drinkToggleBtn.x = -100;
drinkToggleBtn.y = 100;
LK.gui.topRight.addChild(drinkToggleBtn);
// Drink toggle handler
drinkToggleBtn.down = function () {
	if (bird) {
		bird.toggleDrink();
	}
};
// Add ground
ground = LK.getAsset('ground', {
	anchorX: 0,
	anchorY: 0,
	x: 0,
	y: GROUND_Y
});
game.addChild(ground);
// Helper: Reset game state
function resetGame() {
	// Remove pipes
	for (var i = 0; i < pipes.length; i++) {
		pipes[i].destroy();
	}
	pipes = [];
	// Remove bird if exists
	if (bird) bird.destroy();
	// Reset variables
	score = 0;
	scoreTxt.setText('0');
	gameOver = false;
	started = false;
	lastPipeX = 0;
	// Hide game over
	gameOverTxt.visible = false;
	restartTxt.visible = false;
	// Add bird
	bird = new Bird();
	bird.x = BIRD_START_X;
	bird.y = BIRD_START_Y;
	game.addChild(bird);
	// Restore score if available
	if (typeof storage.lastScore !== "undefined") {
		score = storage.lastScore;
		scoreTxt.setText(score + '');
	}
	// Restore drink state if available
	if (typeof storage.lastDrink !== "undefined" && storage.lastDrink) {
		bird.toggleDrink();
	}
	// Add first pipes
	for (var i = 0; i < 3; i++) {
		spawnPipe(2048 + i * PIPE_INTERVAL);
	}
}
// Helper: Spawn a pipe pair at x
function spawnPipe(x) {
	var gapY = PIPE_MIN_Y + Math.floor(Math.random() * (PIPE_MAX_Y - PIPE_MIN_Y));
	var pipePair = new PipePair();
	pipePair.x = x;
	pipePair.setGap(gapY);
	pipes.push(pipePair);
	game.addChild(pipePair);
	lastPipeX = x;
}
// Helper: Check collision between bird and a pipe rect
function birdHitsRect(rect) {
	// Bird is a circle, rect is {x, y, width, height}
	var cx = bird.x;
	var cy = bird.y;
	var r = bird.radius;
	// Find closest point in rect to bird center
	var closestX = cx;
	if (cx < rect.x) closestX = rect.x;else if (cx > rect.x + rect.width) closestX = rect.x + rect.width;
	var closestY = cy;
	if (cy < rect.y) closestY = rect.y;else if (cy > rect.y + rect.height) closestY = rect.y + rect.height;
	var dx = cx - closestX;
	var dy = cy - closestY;
	return dx * dx + dy * dy < r * r;
}
// Helper: End game
function triggerGameOver() {
	if (gameOver) return;
	gameOver = true;
	// Save score and drink state to storage
	storage.lastScore = score;
	// Access birdDrinkSprite via bird instance if available
	storage.lastDrink = bird && bird.hasOwnProperty('children') && bird.children.length > 1 && bird.children[1].visible ? true : false;
	LK.getSound('hit').play();
	gameOverTxt.visible = true;
	restartTxt.visible = true;
	// Flash screen
	LK.effects.flashScreen(0xff0000, 600);
	// Show Game Over popup (will pause game)
	// LK.showGameOver();
}
// Game tap/flap handler
function handleTap(x, y, obj) {
	if (gameOver) {
		resetGame();
		return;
	}
	if (!started) {
		started = true;
	}
	bird.flap();
}
// Attach tap handler
game.down = handleTap;
// Main update loop
game.update = function () {
	if (gameOver) return;
	// Bird update
	bird.update();
	// Pipes update
	for (var i = pipes.length - 1; i >= 0; i--) {
		var pipe = pipes[i];
		pipe.update();
		// Remove pipes off screen
		if (pipe.x + pipe.pipeWidth < 0) {
			pipe.destroy();
			pipes.splice(i, 1);
			continue;
		}
		// Scoring: passed pipe
		if (!pipe.passed && pipe.x + pipe.pipeWidth < bird.x) {
			pipe.passed = true;
			score += 1;
			scoreTxt.setText(score + '');
			LK.getSound('score').play();
		}
	}
	// Spawn new pipes
	if (pipes.length > 0) {
		var rightmost = pipes[pipes.length - 1];
		if (2048 - (rightmost.x + rightmost.pipeWidth) >= PIPE_INTERVAL - 10) {
			spawnPipe(2048);
		}
	}
	// Collision: pipes
	for (var i = 0; i < pipes.length; i++) {
		var pipe = pipes[i];
		if (birdHitsRect(pipe.getTopRect()) || birdHitsRect(pipe.getBottomRect())) {
			triggerGameOver();
			return;
		}
	}
	// Collision: ground
	if (bird.y + bird.radius > GROUND_Y) {
		bird.y = GROUND_Y - bird.radius;
		triggerGameOver();
		return;
	}
	// Collision: ceiling
	if (bird.y - bird.radius < 0) {
		bird.y = bird.radius;
		bird.vy = 0;
	}
	// Out of bounds (left/right)
	if (bird.x - bird.radius < 0 || bird.x + bird.radius > 2048) {
		triggerGameOver();
		return;
	}
};
// Initial game state
resetGame();

===================================================================
--- original.js
+++ change.js
@@ -34,8 +34,10 @@
 	// Toggle drink method
 	self.toggleDrink = function () {
 		// Simply toggle visibility of drink sprite
 		birdDrinkSprite.visible = !birdDrinkSprite.visible;
+		// Save the current drink state to storage
+		storage.lastDrink = birdDrinkSprite.visible;
 	};
 	// Update method
 	self.update = function () {
 		self.vy += self.gravity;