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

/**** 
* Classes
****/ 
// StockRow: A row in the stock table, with price, holdings, buy/sell buttons
var StockRow = Container.expand(function () {
	var self = Container.call(this);
	// Properties
	self.companyIndex = 0; // 0-9
	self.companyName = '';
	self.colorId = '';
	self.price = 0;
	self.holdings = 0;
	self.lastPrice = 0;
	// UI elements
	var yPad = 0;
	// Stock icon
	var icon = self.attachAsset(self.colorId, {
		anchorX: 0.5,
		anchorY: 0.5,
		x: 80,
		y: 60 + yPad,
		scaleX: 0.7,
		scaleY: 0.7
	});
	// Company name
	var nameTxt = new Text2(self.companyName, {
		size: 48,
		fill: 0xFFFFFF
	});
	nameTxt.x = 160;
	nameTxt.y = 30 + yPad;
	nameTxt.anchor.set(0, 0);
	self.addChild(nameTxt);
	// Price text
	var priceTxt = new Text2('', {
		size: 48,
		fill: 0xF1C40F
	});
	priceTxt.x = 500;
	priceTxt.y = 30 + yPad;
	priceTxt.anchor.set(0, 0);
	self.addChild(priceTxt);
	// Holdings text
	var holdingsTxt = new Text2('', {
		size: 48,
		fill: 0x2ECC71
	});
	holdingsTxt.x = 800;
	holdingsTxt.y = 30 + yPad;
	holdingsTxt.anchor.set(0, 0);
	self.addChild(holdingsTxt);
	// Profit text
	var profitTxt = new Text2('', {
		size: 48,
		fill: 0xFFFFFF
	});
	profitTxt.x = 1100;
	profitTxt.y = 30 + yPad;
	profitTxt.anchor.set(0, 0);
	self.addChild(profitTxt);
	// Buy button
	var buyBtn = self.attachAsset('buyBtn', {
		anchorX: 0.5,
		anchorY: 0.5,
		x: 1450,
		y: 60 + yPad
	});
	var buyTxt = new Text2('Buy', {
		size: 36,
		fill: 0xFFFFFF
	});
	buyTxt.anchor.set(0.5, 0.5);
	buyTxt.x = buyBtn.x;
	buyTxt.y = buyBtn.y;
	self.addChild(buyTxt);
	// Amount textbox (between buy and sell)
	var amountBoxBg = self.attachAsset('buyBtn', {
		anchorX: 0.5,
		anchorY: 0.5,
		x: 1575,
		y: 60 + yPad,
		scaleX: 0.7,
		scaleY: 0.7
	});
	var amountTxt = new Text2(typeof selectedShareAmount !== "undefined" ? '' + selectedShareAmount : '1', {
		size: 36,
		fill: 0x222a36
	});
	amountTxt.anchor.set(0.5, 0.5);
	amountTxt.x = amountBoxBg.x;
	amountTxt.y = amountBoxBg.y;
	self.addChild(amountTxt);
	// Touch to edit amount (prompt for number)
	amountBoxBg.down = function (x, y, obj) {
		// On mobile, prompt is supported in LK sandbox for numeric input
		var current = parseInt(amountTxt.text, 10);
		if (isNaN(current) || current < 1) current = 1;
		var entered = prompt("Enter number of shares:", current);
		var val = parseInt(entered, 10);
		if (isNaN(val) || val < 1) val = 1;
		amountTxt.setText('' + val);
		// Also update global selectedShareAmount and highlight if matches a selector
		if (typeof selectedShareAmount !== "undefined") {
			selectedShareAmount = val;
			if (typeof shareAmountBtnBg !== "undefined" && typeof shareAmounts !== "undefined") {
				for (var j = 0; j < shareAmountBtnBg.length; j++) {
					shareAmountBtnBg[j].tint = shareAmounts[j] === selectedShareAmount ? 0xf1c40f : 0x2980b9;
				}
			}
			// Update all amount boxes to match
			if (typeof stockRows !== "undefined") {
				for (var k = 0; k < stockRows.length; k++) {
					if (stockRows[k].setAmountBoxValue) {
						stockRows[k].setAmountBoxValue(selectedShareAmount);
					}
				}
			}
		}
	};
	// Method to allow global selector to update this textbox
	self.setAmountBoxValue = function (val) {
		if (typeof val === "number" && val > 0) {
			amountTxt.setText('' + val);
		}
	};
	// Sell button
	var sellBtn = self.attachAsset('sellBtn', {
		anchorX: 0.5,
		anchorY: 0.5,
		x: 1700,
		y: 60 + yPad
	});
	var sellTxt = new Text2('Sell', {
		size: 36,
		fill: 0xFFFFFF
	});
	sellTxt.anchor.set(0.5, 0.5);
	sellTxt.x = sellBtn.x;
	sellTxt.y = sellBtn.y;
	self.addChild(sellTxt);
	// Update UI
	self.updateUI = function () {
		nameTxt.setText(self.companyName);
		priceTxt.setText('₲' + self.price);
		holdingsTxt.setText(self.holdings + ' shares');
		var profit = (self.price - self.lastPrice) * self.holdings;
		var profitColor = "#ffffff";
		if (profit > 0) profitColor = "#2ecc71";
		if (profit < 0) profitColor = "#e74c3c";
		profitTxt.setText((profit >= 0 ? '+' : '') + profit);
		// Use setStyle to update fill color safely
		profitTxt.setStyle({
			fill: profitColor
		});
	};
	// Buy handler
	buyBtn.down = function (x, y, obj) {
		// Use global selectedShareAmount for buy
		var amount = typeof selectedShareAmount !== "undefined" && selectedShareAmount > 0 ? selectedShareAmount : 1;
		var maxBuy = Math.floor(playerCash / self.price);
		var buyCount = Math.min(amount, maxBuy);
		if (buyCount > 0) {
			playerCash -= self.price * buyCount;
			self.holdings += buyCount;
			self.updateUI();
			updateCashUI();
		}
	};
	// Sell handler
	sellBtn.down = function (x, y, obj) {
		// Use global selectedShareAmount for sell
		var amount = typeof selectedShareAmount !== "undefined" && selectedShareAmount > 0 ? selectedShareAmount : 1;
		var sellCount = Math.min(amount, self.holdings);
		if (sellCount > 0) {
			playerCash += self.price * sellCount;
			self.holdings -= sellCount;
			self.updateUI();
			updateCashUI();
		}
	};
	// For updating price and profit
	self.setPrice = function (newPrice) {
		self.lastPrice = self.price;
		self.price = newPrice;
		self.updateUI();
	};
	// For updating holdings externally
	self.setHoldings = function (newHoldings) {
		self.holdings = newHoldings;
		self.updateUI();
	};
	// For updating profit display externally
	self.updateProfit = function () {
		self.updateUI();
	};
	// For setting up after construction
	self.init = function (companyIndex, companyName, colorId, price) {
		self.companyIndex = companyIndex;
		self.companyName = companyName;
		self.colorId = colorId;
		self.price = price;
		self.lastPrice = price;
		icon.setAsset(colorId);
		self.updateUI();
	};
	return self;
});

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

/**** 
* Game Code
****/ 
// --- Game Data ---
// 10 company colors for stocks (distinct, visually clear)
// Button shapes
var companyNames = ["AlphaTech", "BetaBank", "GammaFoods", "DeltaPharma", "EpsilonEnergy", "ZetaAuto", "EtaRetail", "ThetaMedia", "IotaLogix", "KappaSpace"];
var companyColors = ['stockA', 'stockB', 'stockC', 'stockD', 'stockE', 'stockF', 'stockG', 'stockH', 'stockI', 'stockJ'];
// Initial prices (randomized a bit for variety)
var initialPrices = [];
for (var i = 0; i < 10; i++) {
	initialPrices[i] = 80 + Math.floor(Math.random() * 40) * 5; // 80-280
}
// Player state
var playerCash = 10000;
var playerHoldings = [];
for (var i = 0; i < 10; i++) playerHoldings[i] = 0;
// Stock state
var stockPrices = [];
var stockLastPrices = [];
for (var i = 0; i < 10; i++) {
	stockPrices[i] = initialPrices[i];
	stockLastPrices[i] = initialPrices[i];
}
// Timer
var gameDurationSec = 300; // 5 minutes
var timeLeft = gameDurationSec;
var timerInterval = null;
// UI elements
var cashTxt = null;
var timerTxt = null;
var wealthTxt = null;
var stockRows = [];
var tableContainer = null;
// --- UI Setup ---
// Cash display (top right, avoid top left 100x100)
cashTxt = new Text2('Cash: ₲' + playerCash, {
	size: 72,
	fill: 0xF1C40F
});
cashTxt.anchor.set(1, 0);
cashTxt.x = 0; // x/y are ignored for gui.topRight, but set to 0 for clarity
cashTxt.y = 0;
LK.gui.topRight.addChild(cashTxt);
// Timer display (top center)
timerTxt = new Text2('05:00', {
	size: 72,
	fill: 0xFFFFFF
});
timerTxt.anchor.set(0.5, 0);
timerTxt.x = 2048 / 2;
timerTxt.y = 40;
LK.gui.top.addChild(timerTxt);
// Wealth display (below timer)
wealthTxt = new Text2('Wealth: ₲' + playerCash, {
	size: 56,
	fill: 0x2ECC71
});
wealthTxt.anchor.set(0.5, 0);
wealthTxt.x = 2048 / 2;
wealthTxt.y = 130;
LK.gui.top.addChild(wealthTxt);
// Table container (centered, scroll not needed for 10 rows)
tableContainer = new Container();
tableContainer.x = 0;
tableContainer.y = 300;
game.addChild(tableContainer);
// Table header
var headerY = 0;
var header = new Text2("Company         Price         Holdings         Profit         ", {
	size: 48,
	fill: 0xBDC3C7
});
header.anchor.set(0, 0);
header.x = 160;
header.y = headerY;
tableContainer.addChild(header);
// Total Profit/Loss label and display (at end of profit column)
var totalProfitLabel = new Text2('Total Profit:', {
	size: 48,
	fill: 0xBDC3C7
});
totalProfitLabel.anchor.set(1, 0);
totalProfitLabel.x = 1090;
totalProfitLabel.y = 80 + 10 * 120 + 10;
tableContainer.addChild(totalProfitLabel);
var totalProfitTxt = new Text2('', {
	size: 48,
	fill: 0xFFFFFF
});
totalProfitTxt.anchor.set(0, 0);
totalProfitTxt.x = 1110;
totalProfitTxt.y = 80 + 10 * 120 + 10; // below last row
tableContainer.addChild(totalProfitTxt);
// --- Global Share Amount Selector Buttons ---
var shareAmounts = [1, 5, 10, 25];
var selectedShareAmount = 1;
var shareAmountBtns = [];
var shareAmountBtnTxts = [];
var shareAmountBtnBg = [];
var shareAmountBtnY = 10; // y offset above first row
var shareAmountBtnX0 = 1450; // align with buy/sell
var shareAmountBtnSpacing = 110;
var shareAmountBtnW = 90;
var shareAmountBtnH = 60;
var shareAmountBtnColor = 0x2980b9;
var shareAmountBtnColorSelected = 0xf1c40f;
// Container for selector buttons
var selectorBtnContainer = new Container();
selectorBtnContainer.x = shareAmountBtnX0 - shareAmountBtnSpacing * 1.5;
selectorBtnContainer.y = shareAmountBtnY;
tableContainer.addChild(selectorBtnContainer);
for (var i = 0; i < shareAmounts.length; i++) {
	// Use buyBtn shape for button background
	var btnBg = LK.getAsset('buyBtn', {
		anchorX: 0.5,
		anchorY: 0.5,
		x: i * shareAmountBtnSpacing,
		y: 0,
		width: shareAmountBtnW,
		height: shareAmountBtnH
	});
	btnBg.tint = shareAmounts[i] === selectedShareAmount ? shareAmountBtnColorSelected : shareAmountBtnColor;
	selectorBtnContainer.addChild(btnBg);
	shareAmountBtnBg.push(btnBg);
	// Button label
	var btnTxt = new Text2('' + shareAmounts[i], {
		size: 36,
		fill: 0xFFFFFF
	});
	btnTxt.anchor.set(0.5, 0.5);
	btnTxt.x = btnBg.x;
	btnTxt.y = btnBg.y;
	selectorBtnContainer.addChild(btnTxt);
	shareAmountBtnTxts.push(btnTxt);
	// Button logic
	(function (idx, amount) {
		btnBg.down = function (x, y, obj) {
			selectedShareAmount = amount;
			// Update highlight
			for (var j = 0; j < shareAmountBtnBg.length; j++) {
				shareAmountBtnBg[j].tint = shareAmounts[j] === selectedShareAmount ? shareAmountBtnColorSelected : shareAmountBtnColor;
			}
			// Update all amount textboxes in all rows
			for (var k = 0; k < stockRows.length; k++) {
				if (stockRows[k].setAmountBoxValue) {
					stockRows[k].setAmountBoxValue(selectedShareAmount);
				}
			}
		};
	})(i, shareAmounts[i]);
	shareAmountBtns.push(btnBg);
}
// --- Stock Rows ---
for (var i = 0; i < 10; i++) {
	var row = new StockRow();
	row.init(i, companyNames[i], companyColors[i], stockPrices[i]);
	row.y = 80 + i * 120;
	tableContainer.addChild(row);
	stockRows.push(row);
}
// --- Company Detail Section (Chart + News) ---
// Price history for each company (last 20 prices)
var companyPriceHistory = [];
for (var i = 0; i < 10; i++) {
	companyPriceHistory[i] = [];
	for (var j = 0; j < 20; j++) companyPriceHistory[i].push(stockPrices[i]);
}
// Fictional news headlines for each company
var companyNews = [["AlphaTech launches new AI chip", "AlphaTech CEO: 'Innovation is our DNA'", "AlphaTech partners with BetaBank"], ["BetaBank expands to new markets", "BetaBank reports record profits", "BetaBank launches mobile app"], ["GammaFoods unveils plant-based burger", "GammaFoods opens 100th store", "GammaFoods: 'Healthy eating for all'"], ["DeltaPharma vaccine approved", "DeltaPharma acquires Medix", "DeltaPharma Q2 profits soar"], ["EpsilonEnergy invests in solar", "EpsilonEnergy wins green award", "EpsilonEnergy: Oil prices stable"], ["ZetaAuto reveals electric SUV", "ZetaAuto sales up 20%", "ZetaAuto opens new factory"], ["EtaRetail launches online store", "EtaRetail Black Friday success", "EtaRetail expands to Europe"], ["ThetaMedia signs streaming deal", "ThetaMedia launches new channel", "ThetaMedia ad revenue climbs"], ["IotaLogix automates warehouses", "IotaLogix wins logistics award", "IotaLogix: 'Efficiency first'"], ["KappaSpace launches satellite", "KappaSpace partners with NASA", "KappaSpace: 'Space for everyone'"]];
// Container for detail section
var companyDetailSection = new Container();
companyDetailSection.x = 0;
companyDetailSection.y = tableContainer.y + 80 + 10 * 120 + 60; // below last row + some padding
game.addChild(companyDetailSection);
companyDetailSection.visible = false; // hidden by default
// Chart and news containers
var chartContainer = new Container();
chartContainer.x = 160;
chartContainer.y = 0;
companyDetailSection.addChild(chartContainer);
var newsContainer = new Container();
newsContainer.x = 700;
newsContainer.y = 0;
companyDetailSection.addChild(newsContainer);
// Helper to clear chart/news
function clearCompanyDetailSection() {
	while (chartContainer.children.length > 0) chartContainer.removeChild(chartContainer.children[0]);
	while (newsContainer.children.length > 0) newsContainer.removeChild(newsContainer.children[0]);
}
// Draw chart for companyIndex
function drawCompanyChart(companyIndex) {
	clearCompanyDetailSection();
	// Chart area: 480x180
	var chartW = 480,
		chartH = 180;
	var margin = 30;
	var history = companyPriceHistory[companyIndex];
	// Find min/max for scaling
	var minP = history[0],
		maxP = history[0];
	for (var i = 1; i < history.length; i++) {
		if (history[i] < minP) minP = history[i];
		if (history[i] > maxP) maxP = history[i];
	}
	if (maxP === minP) maxP = minP + 1; // avoid div0
	// Draw points (simulate line by dots)
	for (var i = 0; i < history.length; i++) {
		var px = margin + (chartW - 2 * margin) * (i / (history.length - 1));
		var py = margin + (chartH - 2 * margin) * (1 - (history[i] - minP) / (maxP - minP));
		var dot = LK.getAsset(companyColors[companyIndex], {
			anchorX: 0.5,
			anchorY: 0.5,
			x: px,
			y: py,
			width: 16,
			height: 16
		});
		chartContainer.addChild(dot);
	}
	// Chart border
	var border = LK.getAsset('buyBtn', {
		anchorX: 0,
		anchorY: 0,
		x: 0,
		y: 0,
		width: chartW,
		height: chartH,
		color: 0x222a36
	});
	border.alpha = 0.3;
	chartContainer.addChild(border);
	// Chart title
	var chartTitle = new Text2(companyNames[companyIndex] + " Price History", {
		size: 36,
		fill: 0xFFFFFF
	});
	chartTitle.anchor.set(0, 0);
	chartTitle.x = 0;
	chartTitle.y = chartH + 10;
	chartContainer.addChild(chartTitle);
}
// Draw news for companyIndex
function drawCompanyNews(companyIndex) {
	var news = companyNews[companyIndex];
	for (var i = 0; i < news.length; i++) {
		var newsTxt = new Text2("- " + news[i], {
			size: 36,
			fill: 0xBDC3C7
		});
		newsTxt.anchor.set(0, 0);
		newsTxt.x = 0;
		newsTxt.y = i * 50;
		newsContainer.addChild(newsTxt);
	}
	// News title
	var newsTitle = new Text2("Recent News", {
		size: 36,
		fill: 0xFFFFFF
	});
	newsTitle.anchor.set(0, 0);
	newsTitle.x = 0;
	newsTitle.y = -50;
	newsContainer.addChild(newsTitle);
}
// Show detail section for company
var currentDetailCompany = -1;
function showCompanyDetail(companyIndex) {
	if (currentDetailCompany === companyIndex) return;
	currentDetailCompany = companyIndex;
	companyDetailSection.visible = true;
	// Move detail section to top of display list so it's not hidden by table
	if (companyDetailSection.parent && companyDetailSection.parent.children) {
		var parent = companyDetailSection.parent;
		var idx = parent.children.indexOf(companyDetailSection);
		if (idx !== -1 && idx !== parent.children.length - 1) {
			parent.removeChild(companyDetailSection);
			parent.addChild(companyDetailSection);
		}
	}
	drawCompanyChart(companyIndex);
	drawCompanyNews(companyIndex);
}
// Hide detail section
function hideCompanyDetail() {
	companyDetailSection.visible = false;
	currentDetailCompany = -1;
	clearCompanyDetailSection();
}
// Make company names clickable
for (var i = 0; i < stockRows.length; i++) {
	(function (idx) {
		var row = stockRows[idx];
		// Find the nameTxt (company name Text2) in row
		for (var j = 0; j < row.children.length; j++) {
			var child = row.children[j];
			if (child instanceof Text2 && child.text === companyNames[idx]) {
				child.interactive = true;
				child.buttonMode = true;
				// Use closure to capture idx correctly
				child.down = function (x, y, obj) {
					showCompanyDetail(idx);
				};
				break;
			}
		}
	})(i);
}
// Optionally, tap outside detail section to hide it (not required, but nice UX)
companyDetailSection.down = function (x, y, obj) {
	hideCompanyDetail();
};
// Update total profit/loss
function updateTotalProfit() {
	var totalProfit = 0;
	for (var i = 0; i < 10; i++) {
		var profit = (stockPrices[i] - stockLastPrices[i]) * stockRows[i].holdings;
		totalProfit += profit;
	}
	var profitColor = "#ffffff";
	if (totalProfit > 0) profitColor = "#2ecc71";
	if (totalProfit < 0) profitColor = "#e74c3c";
	totalProfitTxt.setText((totalProfit >= 0 ? '+' : '') + totalProfit);
	totalProfitTxt.setStyle({
		fill: profitColor
	});
}
// --- UI Update Functions ---
function updateCashUI() {
	cashTxt.setText('Cash: ₲' + playerCash);
	updateWealthUI();
	updateTotalProfit();
}
function updateWealthUI() {
	var wealth = playerCash;
	for (var i = 0; i < 10; i++) {
		wealth += playerHoldings[i] * stockPrices[i];
	}
	wealthTxt.setText('Wealth: ₲' + wealth);
}
function updateTimerUI() {
	var min = Math.floor(timeLeft / 60);
	var sec = timeLeft % 60;
	var t = (min < 10 ? '0' : '') + min + ':' + (sec < 10 ? '0' : '') + sec;
	timerTxt.setText(t);
}
// --- Stock Price Update Logic ---
// Simulate price changes every 10 seconds
function updateStockPrices() {
	for (var i = 0; i < 10; i++) {
		stockLastPrices[i] = stockPrices[i];
		// Random walk: -8% to +8%
		var pct = Math.random() * 0.16 - 0.08;
		// Occasionally, a "news event" (1 in 8 chance): -25% to +25%
		if (Math.random() < 0.125) {
			pct = Math.random() * 0.5 - 0.25;
		}
		var newPrice = Math.max(10, Math.round(stockPrices[i] * (1 + pct)));
		stockPrices[i] = newPrice;
		// Update price history for chart
		if (typeof companyPriceHistory !== "undefined" && companyPriceHistory[i]) {
			companyPriceHistory[i].push(newPrice);
			if (companyPriceHistory[i].length > 20) companyPriceHistory[i].shift();
			// If this company is currently shown in detail, redraw chart
			if (typeof currentDetailCompany !== "undefined" && currentDetailCompany === i && companyDetailSection.visible) {
				drawCompanyChart(i);
			}
		}
		// Update row
		stockRows[i].setPrice(newPrice);
		playerHoldings[i] = stockRows[i].holdings;
	}
	updateWealthUI();
	updateTotalProfit();
}
// --- Timer Logic ---
function tickTimer() {
	timeLeft -= 1;
	if (timeLeft < 0) timeLeft = 0;
	updateTimerUI();
	if (timeLeft === 0) {
		endGame();
	}
}
// --- End Game ---
function endGame() {
	// Calculate final wealth
	var finalWealth = playerCash;
	for (var i = 0; i < 10; i++) {
		finalWealth += playerHoldings[i] * stockPrices[i];
	}
	// Show game over (handled by LK)
	LK.showGameOver();
}
// --- Game Update Loop ---
game.update = function () {
	// No per-frame logic needed for this MVP
};
// --- Start Game Logic ---
function startGame() {
	// Reset state
	playerCash = 10000;
	for (var i = 0; i < 10; i++) {
		stockPrices[i] = initialPrices[i];
		stockLastPrices[i] = initialPrices[i];
		playerHoldings[i] = 0;
		stockRows[i].setPrice(stockPrices[i]);
		stockRows[i].setHoldings(0);
	}
	timeLeft = gameDurationSec;
	updateCashUI();
	updateTimerUI();
	updateWealthUI();
	hideCompanyDetail();
	// Start price update interval (every 10s)
	if (timerInterval) LK.clearInterval(timerInterval);
	timerInterval = LK.setInterval(function () {
		updateStockPrices();
	}, 10000);
	// Start timer (every 1s)
	if (game._timerTick) LK.clearInterval(game._timerTick);
	game._timerTick = LK.setInterval(function () {
		tickTimer();
	}, 1000);
}
// --- Start the game ---
startGame();

===================================================================
--- original.js
+++ change.js
@@ -517,13 +517,11 @@
 			if (child instanceof Text2 && child.text === companyNames[idx]) {
 				child.interactive = true;
 				child.buttonMode = true;
 				// Use closure to capture idx correctly
-				child.down = function (companyIdx) {
-					return function (x, y, obj) {
-						showCompanyDetail(companyIdx);
-					};
-				}(idx);
+				child.down = function (x, y, obj) {
+					showCompanyDetail(idx);
+				};
 				break;
 			}
 		}
 	})(i);