game.js

const canvas = document.getElementById('mainCanvas');
const ctx = canvas.getContext('2d');
const overlay = document.getElementById('game-overlay');
const instructions = document.getElementById('game-instructions');

let currentGame = 'defuse'; // 'defuse' or 'aim'
let gameState = 'waiting'; // 'waiting', 'playing', 'won', 'lost'
let score = 0;
let timer = 0;

// Game Logic Variables
let explosionTimer = 0;
let targetCode = "";
let currentCode = "";
let wires = [];
let bots = [];
let lastSpawn = 0;

// Aesthetic Config
const csGreen = "#00FF00";
const csDarkGreen = "#004400";
const csGold = "#FFD700";

// --- CORE UTILS ---

function switchGame(type) {
    currentGame = type;
    gameState = 'waiting';
    overlay.style.display = 'flex';
    overlay.innerText = `START ${type.toUpperCase()}`;

    // Update instructions and button UI
    document.querySelectorAll('.game-selector button').forEach(b => b.classList.remove('active'));
    if (type === 'defuse') {
        instructions.innerText = "DEFUSE: Input the correct 4-digit code before the timer hits zero!";
        document.querySelector('button[onclick*="defuse"]').classList.add('active');
    } else {
        instructions.innerText = "AIM_BOTZ: Click the 'bots' as they appear. Don't miss!";
        document.querySelector('button[onclick*="aim"]').classList.add('active');
    }
    resetGame();
}

function resetGame() {
    score = 0;
    if (currentGame === 'defuse') {
        explosionTimer = 400; // frames
        targetCode = Math.floor(1000 + Math.random() * 9000).toString();
        currentCode = "";
    } else {
        bots = [];
        timer = 600; // 10 seconds approx
    }
}

overlay.addEventListener('click', () => {
    gameState = 'playing';
    overlay.style.display = 'none';
    resetGame();
    if (gameLoopId === null) loop();
});

// --- DEFUSE LOGIC ---

function updateDefuse() {
    explosionTimer--;
    if (explosionTimer <= 0) {
        gameState = 'lost';
        overlay.innerText = "TERRORISTS WIN (BOOM)";
        overlay.style.display = 'flex';
    }

    // Win check
    if (currentCode === targetCode) {
        gameState = 'won';
        overlay.innerText = "COUNTER-TERRORISTS WIN (DEFUSED)";
        overlay.style.display = 'flex';
    }
}

function drawDefuse() {
    // C4 Appearance
    ctx.fillStyle = "#333";
    ctx.fillRect(150, 50, 300, 200);
    ctx.strokeStyle = "#444";
    ctx.strokeRect(150, 50, 300, 200);

    // Timer Display
    ctx.fillStyle = "red";
    ctx.font = "30px 'Press Start 2P'";
    ctx.textAlign = "center";
    let displayTime = (explosionTimer / 60).toFixed(2);
    ctx.fillText(displayTime, 300, 110);

    // Code Display
    ctx.fillStyle = csGreen;
    ctx.font = "20px Courier Prime";
    ctx.fillText("ENTER CODE:", 300, 160);
    ctx.fillStyle = "white";
    ctx.font = "40px Courier Prime";
    ctx.fillText(currentCode.padEnd(4, "*"), 300, 210);

    ctx.font = "12px Press Start 2P";
    ctx.fillStyle = csGold;
    ctx.fillText("[PRESS 0-9 KEYS TO DEFUSE]", 300, 240);
}

// --- AIM_BOTZ LOGIC ---

function updateAim() {
    timer--;
    if (Date.now() - lastSpawn > 800) {
        bots.push({
            x: Math.random() * (canvas.width - 40) + 20,
            y: Math.random() * (canvas.height - 40) + 20,
            size: 20,
            life: 60
        });
        lastSpawn = Date.now();
    }

    for (let i = bots.length - 1; i >= 0; i--) {
        bots[i].life--;
        if (bots[i].life <= 0) bots.splice(i, 1);
    }

    if (timer <= 0) {
        gameState = 'won';
        overlay.innerText = `ROUND OVER. KILLS: ${score}`;
        overlay.style.display = 'flex';
    }
}

function drawAim() {
    ctx.strokeStyle = csGreen;
    ctx.lineWidth = 1;
    for (let b of bots) {
        ctx.beginPath();
        ctx.arc(b.x, b.y, b.size, 0, Math.PI * 2);
        ctx.stroke();
        // Crosshair inside bot
        ctx.moveTo(b.x - 5, b.y); ctx.lineTo(b.x + 5, b.y);
        ctx.moveTo(b.x, b.y - 5); ctx.lineTo(b.x, b.y + 5);
        ctx.stroke();
    }

    ctx.fillStyle = csWhite;
    ctx.font = "16px Courier Prime";
    ctx.textAlign = "left";
    ctx.fillText(`KILLS: ${score}`, 10, 20);
    ctx.fillText(`TIME: ${(timer / 60).toFixed(1)}s`, 10, 40);
}

// --- MAIN LOOP ---

window.addEventListener('keydown', (e) => {
    if (gameState !== 'playing') return;
    if (currentGame === 'defuse') {
        if (e.key >= '0' && e.key <= '9' && currentCode.length < 4) {
            currentCode += e.key;
        }
    }
});

canvas.addEventListener('mousedown', (e) => {
    if (gameState !== 'playing' || currentGame !== 'aim') return;
    let rect = canvas.getBoundingClientRect();
    let mx = e.clientX - rect.left;
    let my = e.clientY - rect.top;

    for (let i = bots.length - 1; i >= 0; i--) {
        let b = bots[i];
        let dist = Math.sqrt((mx - b.x) ** 2 + (my - b.y) ** 2);
        if (dist < b.size) {
            bots.splice(i, 1);
            score++;
            break;
        }
    }
});

let gameLoopId = null;
function loop() {
    ctx.clearRect(0, 0, canvas.width, canvas.height);

    if (gameState === 'playing') {
        if (currentGame === 'defuse') {
            updateDefuse();
            drawDefuse();
        } else {
            updateAim();
            drawAim();
        }
    }

    gameLoopId = requestAnimationFrame(loop);
}

// Global aesthetic fallback
const csWhite = "#FFF";

switchGame('defuse');