SURVIVE CHIWAWA IN ALLEYWAY AT 3AM

function setup() {
  createCanvas(windowWidth, windowHeight);
  gameState = "START";
}

function initGame(mode) {
  gameMode = mode;
  wave = 1;
  coins = 0;
  hasMachineGun = (mode === "HARDCORE"); 
  walls = [];
  friends = [];
  gods = [];
  
  player = {
    pos: createVector(width / 2, height / 2 + 100),
    health: (mode === "HARDCORE") ? 1 : 100,
    maxHealth: (mode === "HARDCORE") ? 1 : 100,
    speed: 4.0,
    radius: 20,
    punchRadius: 65,
    punchFrames: 0,
    punchAngle: 0
  };
  
  if (cheatActivated) {
    coins = 10000;
    hasMachineGun = true;
    
    for (let i = 0; i < 100; i++) {
      walls.push({ 
        pos: createVector(player.pos.x + random(-350, 350), player.pos.y + random(-350, 350)), 
        radius: 25, hp: 150 
      });
    }
    
    for (let i = 0; i < 50; i++) {
      friends.push({ 
        pos: createVector(player.pos.x + random(-150, 150), player.pos.y + random(-150, 150)), 
        speed: 3.0, radius: 12 
      });
    }
    
    gods.push({ pos: createVector(player.pos.x, player.pos.y), speed: 6.0, radius: 30 });
    
    cheatActivated = false;
    secretCode = "";
  }
  
  startNextWave();
}

function startNextWave() {
  dogs = [];
  
  if (gameMode === "CLASSIC") {
    dogs.push({
      pos: createVector(width / 2, height / 2 - 300),
      speed: min(7.5, 3.8 + (wave * 0.2)),
      radius: 15,
      hitsTaken: 0,
      maxHits: floor(10 * pow(1.25, wave - 1)),
      stunFrames: 0, flashRed: 0
    });
  } 
  else if (gameMode === "SWARM") {
    let numDogs = 2 + (wave * 3); 
    for (let i = 0; i < numDogs; i++) {
      dogs.push({
        pos: createVector(width / 2 + random(-200, 200), height / 2 - 400 + random(-150, 150)),
        speed: random(2.5, 4.5) + (wave * 0.1),
        radius: 12,
        hitsTaken: 0,
        maxHits: ceil(wave * 0.5), 
        stunFrames: 0, flashRed: 0
      });
    }
  } 
  else if (gameMode === "HARDCORE") {
    dogs.push({
      pos: createVector(width / 2, height / 2 - 300),
      speed: min(9.0, 4.5 + (wave * 0.3)), 
      radius: 15,
      hitsTaken: 0,
      maxHits: floor(15 * pow(1.3, wave - 1)),
      stunFrames: 0, flashRed: 0
    });
  }

  projectiles = [];
  particles = [];
  shakeAmount = 0;
  gameState = "PLAYING";
}

function keyPressed() {
  secretCode += key;
  if (secretCode.length > 2) secretCode = secretCode.substring(secretCode.length - 2);
  
  if (secretCode === "41" && (gameState === "START" || gameState === "MODE_SELECT")) {
    cheatActivated = true;
    shakeAmount = 25;
    secretCode = ""; 
  }
}

function draw() {
  background(10); 
  
  push();
  if (shakeAmount > 0) {
    translate(random(-shakeAmount, shakeAmount), random(-shakeAmount, shakeAmount));
    shakeAmount *= 0.9;
    if (shakeAmount < 0.5) shakeAmount = 0;
  }

  if (gameState === "PLAYING") {
    updateLogic();
    drawEnvironment();
    drawWalls();
    drawFriends();
    drawGods();
    drawDogs();
    drawPlayer();
    drawProjectiles();
    drawParticles();
    drawFlashlight();
    drawHUD();
    
    if (mobileMode) drawMobileControls();
    
  } else if (gameState === "START") {
    drawStartScreen();
  } else if (gameState === "MODE_SELECT") {
    drawModeSelectScreen();
  } else if (gameState === "SHOP") {
    drawEnvironment(); 
    drawShopScreen();
  } else if (gameState === "LOSE") {
    drawLoseScreen();
  }
  pop(); 
}

function updateLogic() {
  let isMoving = false;
  
  if (keyIsDown(65) || keyIsDown(LEFT_ARROW)) { player.pos.x -= player.speed; isMoving = true; }
  if (keyIsDown(68) || keyIsDown(RIGHT_ARROW)) { player.pos.x += player.speed; isMoving = true; }
  if (keyIsDown(87) || keyIsDown(UP_ARROW)) { player.pos.y -= player.speed; isMoving = true; }
  if (keyIsDown(83) || keyIsDown(DOWN_ARROW)) { player.pos.y += player.speed; isMoving = true; }

  let joyBaseX = 120; let joyBaseY = height - 120;
  let shootBtnX = width - 120; let shootBtnY = height - 120;
  joyKnob = { x: joyBaseX, y: joyBaseY };
  let shootActive = false;

  if (mobileMode) {
    let inputs = [];
    if (touches.length > 0) {
      for (let i = 0; i < touches.length; i++) inputs.push(touches[i]);
    } else if (mouseIsPressed) {
      inputs.push({ x: mouseX, y: mouseY });
    }

    for (let t of inputs) {
      if (t.x < width / 2 && t.y > height / 2) {
        let d = dist(t.x, t.y, joyBaseX, joyBaseY);
        if (d < 150) { 
          let angle = atan2(t.y - joyBaseY, t.x - joyBaseX);
          let mag = min(d, 50);
          joyKnob.x = joyBaseX + cos(angle) * mag;
          joyKnob.y = joyBaseY + sin(angle) * mag;
          let moveVec = createVector(cos(angle), sin(angle)).mult(player.speed * (mag/50));
          player.pos.add(moveVec);
          isMoving = true;
        }
      }
      if (t.x > width / 2 && t.y > height / 2) {
        if (dist(t.x, t.y, shootBtnX, shootBtnY) < 100) shootActive = true;
      }
    }

    if (shootActive) {
      let closestDog = null; let record = Infinity;
      for (let d of dogs) {
        let dDist = dist(player.pos.x, player.pos.y, d.pos.x, d.pos.y);
        if (dDist < record) { record = dDist; closestDog = d; }
      }
      let targetAngle = player.punchAngle;
      if (closestDog) targetAngle = atan2(closestDog.pos.y - player.pos.y, closestDog.pos.x - player.pos.x);

      if (hasMachineGun && player.punchFrames === 0) {
        if (frameCount % 6 === 0) {
          let dir = p5.Vector.fromAngle(targetAngle).mult(15);
          projectiles.push({ pos: player.pos.copy(), vel: dir, life: 60, isGod: false, damage: 1 });
          player.punchAngle = targetAngle;
          shakeAmount = 1.5; 
        }
      } else if (!hasMachineGun && player.punchFrames === 0) {
        player.punchFrames = 15;
        player.punchAngle = targetAngle;
        executePunch(targetAngle);
      }
    }
  } 
  else {
    if (touches.length > 0 && !isMoving) {
      let t = touches[0];
      let dir = createVector(t.x - player.pos.x, t.y - player.pos.y);
      if (dir.mag() > 20) {
        dir.normalize().mult(player.speed);
        player.pos.add(dir);
      }
    }

    if (mouseIsPressed && hasMachineGun && player.punchFrames === 0) {
      if (frameCount % 6 === 0) {
        let tx = touches.length > 0 ? touches[0].x : mouseX;
        let ty = touches.length > 0 ? touches[0].y : mouseY;
        let dir = createVector(tx - player.pos.x, ty - player.pos.y).normalize().mult(15);
        projectiles.push({ pos: player.pos.copy(), vel: dir, life: 60, isGod: false, damage: 1 });
        player.punchAngle = dir.heading();
        shakeAmount = 1.5; 
      }
    }
  }

  let leftWall = width / 2 - alleyWidth / 2;
  let rightWall = width / 2 + alleyWidth / 2;
  player.pos.x = constrain(player.pos.x, leftWall + player.radius, rightWall - player.radius);
  player.pos.y = constrain(player.pos.y, player.radius, height - player.radius);

  if (player.punchFrames > 0) player.punchFrames--;

  for (let f of friends) {
    let closest = null; let record = Infinity;
    for (let d of dogs) {
      let dToDog = dist(f.pos.x, f.pos.y, d.pos.x, d.pos.y);
      if (dToDog < record) { record = dToDog; closest = d; }
    }
    if (closest) {
      if (record > 180) {
        let dir = p5.Vector.sub(closest.pos, f.pos).normalize().mult(f.speed);
        f.pos.add(dir);
      } else if (frameCount % 40 === 0) {
        let dir = p5.Vector.sub(closest.pos, f.pos).normalize().mult(12);
        projectiles.push({ pos: f.pos.copy(), vel: dir, life: 60, isGod: false, damage: 1 });
      }
    }
  }

  let godFireRate = max(2, 8 - floor(wave / 3));
  let godDamage = floor(25 * pow(1.2, wave - 1));

  for (let g of gods) {
    let closest = null; let record = Infinity;
    for (let d of dogs) {
      let dToDog = dist(g.pos.x, g.pos.y, d.pos.x, d.pos.y);
      if (dToDog < record) { record = dToDog; closest = d; }
    }
    if (closest) {
      if (record > 200) {
        let dir = p5.Vector.sub(closest.pos, g.pos).normalize().mult(g.speed);
        g.pos.add(dir);
      } else if (frameCount % godFireRate === 0) {
        let dir = p5.Vector.sub(closest.pos, g.pos).normalize().mult(25);
        projectiles.push({ pos: g.pos.copy(), vel: dir, life: 100, isGod: true, damage: godDamage });
      }
    }
  }

  for (let i = projectiles.length - 1; i >= 0; i--) {
    let p = projectiles[i];
    p.pos.add(p.vel);
    p.life--;
    
    let hit = false;
    for (let j = dogs.length - 1; j >= 0; j--) {
      let d = dogs[j];
      let hitRadius = p.isGod ? 25 : 15;
      
      if (dist(p.pos.x, p.pos.y, d.pos.x, d.pos.y) < d.radius + hitRadius) {
        let dmg = p.damage || 1;
        d.hitsTaken += dmg;
        d.flashRed = 8;
        d.pos.add(p.vel.copy().setMag(p.isGod ? 8 : 2)); 
        
        for(let k=0; k < (p.isGod ? 8 : 3); k++) {
          particles.push({pos: d.pos.copy(), vel: p5.Vector.random2D().mult(random(2,8)), life: 255});
        }
        
        if (d.hitsTaken >= d.maxHits) dogs.splice(j, 1);
        hit = true; break;
      }
    }
    if (hit || p.life <= 0) projectiles.splice(i, 1);
  }

  for (let j = dogs.length - 1; j >= 0; j--) {
    let d = dogs[j];
    if (d.stunFrames > 0) {
      d.stunFrames--;
    } else {
      let dirToPlayer = p5.Vector.sub(player.pos, d.pos);
      let distToPlayer = dirToPlayer.mag();
      
      let hitWall = false;
      for (let i = walls.length - 1; i >= 0; i--) {
        let w = walls[i];
        if (dist(d.pos.x, d.pos.y, w.pos.x, w.pos.y) < d.radius + w.radius) {
          hitWall = true;
          w.hp -= 1.5; 
          d.pos.sub(p5.Vector.sub(d.pos, w.pos).setMag(3)); 
          if (w.hp <= 0) walls.splice(i, 1);
        }
      }

      if (!hitWall && distToPlayer > d.radius + player.radius) {
        dirToPlayer.normalize().mult(d.speed);
        d.pos.add(dirToPlayer);
      }

      if (distToPlayer < d.radius + player.radius + 5) {
        player.health -= (gameMode === "HARDCORE") ? 1 : 0.8;
        shakeAmount = 2;
        if (player.health <= 0) {
          gameState = "LOSE";
          deathCooldown = 60;
        }
      }
    }

    d.pos.x = constrain(d.pos.x, leftWall + d.radius, rightWall - d.radius);
    d.pos.y = constrain(d.pos.y, d.radius, height - d.radius);
    if (d.flashRed > 0) d.flashRed--;
  }

  if (dogs.length === 0 && gameState === "PLAYING") {
    let earned = (gameMode === "SWARM") ? wave * 15 + 20 : wave * 10 + 10;
    coins += earned;
    player.health = player.maxHealth; 
    gameState = "SHOP";
  }

  for (let i = particles.length - 1; i >= 0; i--) {
    particles[i].pos.add(particles[i].vel);
    particles[i].life -= 5;
    if (particles[i].life <= 0) particles.splice(i, 1);
  }
}

function executePunch(angle) {
  for (let j = dogs.length - 1; j >= 0; j--) {
    let d = dogs[j];
    let distToDog = dist(player.pos.x, player.pos.y, d.pos.x, d.pos.y);
    let angleToDog = atan2(d.pos.y - player.pos.y, d.pos.x - player.pos.x);
    let angleDiff = abs(angle - angleToDog);
    if (angleDiff > PI) angleDiff = TWO_PI - angleDiff;

    if (distToDog < player.punchRadius && angleDiff < PI / 2) {
      shakeAmount = 5;
      let knockback = p5.Vector.fromAngle(angleToDog).mult(50);
      d.pos.add(knockback);
      d.stunFrames = 15;
      d.hitsTaken++;
      d.flashRed = 8;
      for (let i = 0; i < 5; i++) particles.push({pos: d.pos.copy(), vel: p5.Vector.random2D().mult(random(2, 6)), life: 255});
      if (d.hitsTaken >= d.maxHits) dogs.splice(j, 1);
    }
  }
}

function handleInput(tx, ty) {
  if (gameState === "START") {
    let btnY = height / 2 + 160;
    if (tx > width/2 - 130 && tx < width/2 + 130 && ty > btnY - 25 && ty < btnY + 25) {
      mobileMode = !mobileMode;
      return; 
    }
    gameState = "MODE_SELECT";
    return;
  }
  
  if (gameState === "MODE_SELECT") {
    handleModeSelectClick(tx, ty);
    return;
  }
  
  if (gameState === "LOSE") {
    if (deathCooldown <= 0) gameState = "MODE_SELECT";
    return;
  }
  
  if (gameState === "SHOP") {
    handleShopClick(tx, ty);
    return;
  }

  if (gameState === "PLAYING" && player.punchFrames === 0 && !mobileMode) {
    player.punchFrames = 15;
    player.punchAngle = atan2(ty - player.pos.y, tx - player.pos.x);
    executePunch(player.punchAngle);
  }
}

function drawPlayer() {
  push(); translate(player.pos.x, player.pos.y);
  fill(100, 150, 255); stroke(0); strokeWeight(2); circle(0, 0, player.radius * 2);
  if (player.punchFrames > 0 || (mouseIsPressed && hasMachineGun && !mobileMode)) {
    rotate(player.punchAngle); fill(255, 200, 150);
    let ext = hasMachineGun ? 25 : map(player.punchFrames, 15, 0, player.punchRadius - 20, 0);
    circle(ext, 0, 15);
  }
  pop();
}

function drawDogs() {
  for (let d of dogs) {
    push();
    let jitterX = d.stunFrames > 0 ? 0 : random(-2, 2);
    let jitterY = d.stunFrames > 0 ? 0 : random(-2, 2);
    translate(d.pos.x + jitterX, d.pos.y + jitterY);
    
    fill(d.flashRed > 0 ? color(255, 0, 0) : color(139, 69, 19));
    stroke(0); strokeWeight(2); circle(0, 0, d.radius * 2);
    
    let dir = p5.Vector.sub(player.pos, d.pos); rotate(dir.heading());
    triangle(0, -d.radius, 15, -d.radius - 15, 10, -d.radius); 
    triangle(0, d.radius, 15, d.radius + 15, 10, d.radius); 
    fill(255, 0, 0); noStroke(); circle(5, -5, 4); circle(5, 5, 4);
    pop();
  }
}

function drawFlashlight() {
  let ctx = drawingContext;
  let gradient = ctx.createRadialGradient(player.pos.x, player.pos.y, player.radius * 2, player.pos.x, player.pos.y, width * 0.4);
  gradient.addColorStop(0, "rgba(0, 0, 0, 0)"); gradient.addColorStop(0.3, "rgba(0, 0, 0, 0.4)"); gradient.addColorStop(1, "rgba(0, 0, 0, 0.98)");
  ctx.fillStyle = gradient; ctx.fillRect(0, 0, width, height);
}

function drawHUD() {
  fill(50); rect(20, 20, 200, 20);
  fill(player.health > 30 ? color(0, 255, 0) : color(255, 0, 0));
  rect(20, 20, map(player.health, 0, player.maxHealth, 0, 200), 20);
  
  fill(255); noStroke(); textSize(16); textAlign(LEFT, TOP);
  text(`HP: ${ceil(player.health)} / ${player.maxHealth}`, 20, 45);
  fill(255, 215, 0); textSize(20); text(`Coins: ${coins}`, 20, 70);
  fill(255); text(`Wave: ${wave}`, 20, 95);

  textAlign(RIGHT, TOP); textSize(24);
  if (gameMode === "SWARM") {
    text(`Dogs Remaining: ${dogs.length}`, width - 20, 20);
  } else {
    let bossHp = dogs.length > 0 ? (dogs[0].maxHits - dogs[0].hitsTaken) : 0;
    text(`Boss HP: ${bossHp}`, width - 20, 20);
  }
  
  textAlign(CENTER, BOTTOM); textSize(14); fill(200);
  if (mobileMode) {
    text("Use Joystick to move • Tap Button to Auto-Attack", width / 2, height - 20);
  } else {
    if (hasMachineGun) text("WASD to move • HOLD Click to Fire Machine Gun", width / 2, height - 10);
    else text("WASD to move • Click towards dog to PUNCH", width / 2, height - 10);
  }
}

function drawStartScreen() {
  textAlign(CENTER, CENTER);
  fill(255, 0, 0); textSize(60);
  text("3AM ALLEYWAY", width / 2, height / 2 - 50);
  fill(255); textSize(20);
  text("Survive endless waves of the rabid beast.", width / 2, height / 2 + 20);
  
  let alpha = map(sin(frameCount * 0.05), -1, 1, 100, 255);
  fill(100, 255, 100, alpha); textSize(18);
  text("Click or Tap to begin", width / 2, height / 2 + 90);
  
  let btnY = height / 2 + 160; let btnW = 260; let btnH = 50;
  fill(mobileMode ? color(50, 200, 100) : color(80));
  stroke(255); strokeWeight(2); rectMode(CENTER);
  rect(width / 2, btnY, btnW, btnH, 10); rectMode(CORNER);
  
  fill(255); noStroke(); textSize(20);
  text(`Mobile Controls: ${mobileMode ? "ON" : "OFF"}`, width / 2, btnY);
  
  fill(150); textSize(14); textAlign(CENTER, BOTTOM);
  text("made by corbun powerd by p5js.ai", width / 2, height - 20);
}

function drawShopScreen() {
  background(10, 10, 15, 220); 
  textAlign(CENTER, CENTER);
  fill(0, 255, 0); textSize(50);
  text(`WAVE ${wave - 1} CLEARED!`, width / 2, 100);
  fill(255, 215, 0); textSize(30);
  text(`Coins: ${coins}`, width / 2, 150);

  let btnW = 300; let btnH = 55; let startY = 220; let gap = 65;
  let items = [
    { name: "Buy Wall (Blocks Dogs)", cost: 15, action: "WALL" },
    { name: hasMachineGun ? "Machine Gun (OWNED)" : "Punch Machine Gun", cost: 50, action: "GUN" },
    { name: "Hire Friend (Shoots Dogs)", cost: 40, action: "FRIEND" },
    { name: "Hire a GOD", cost: 10000, action: "GOD" },
    { name: "START NEXT WAVE", cost: 0, action: "NEXT" }
  ];

  for (let i = 0; i < items.length; i++) {
    let item = items[i];
    let y = startY + i * gap;
    let x = width / 2 - btnW / 2;
    
    let canAfford = coins >= item.cost;
    let isOwnedGun = (item.action === "GUN" && hasMachineGun);
    
    if (item.action === "NEXT") fill(100, 255, 100);
    else if (item.action === "GOD" && canAfford) fill(255, 215, 0);
    else if (item.action === "GOD" && !canAfford) fill(80, 80, 0);
    else if (isOwnedGun) fill(50, 150, 50);
    else if (!canAfford) fill(80);
    else fill(50, 150, 255);
    
    stroke(0); strokeWeight(2);
    rect(x, y, btnW, btnH, 10);
    
    fill(255); noStroke(); textSize(18);
    let label = item.name;
    if (item.cost > 0 && !isOwnedGun) label += ` [${item.cost}c]`;
    text(label, width / 2, y + btnH / 2);
  }
}

function drawLoseScreen() {
  if (deathCooldown > 0) deathCooldown--;
  
  background(50, 0, 0, 200);
  textAlign(CENTER, CENTER);
  fill(255, 0, 0); textSize(60);
  text("YOU DIED", width / 2, height / 2 - 40);
  fill(255); textSize(24);
  text(`You survived until Wave ${wave} in ${gameMode} mode`, width / 2, height / 2 + 30);
  
  textSize(16);
  if (deathCooldown <= 0) {
    let alpha = map(sin(frameCount * 0.05), -1, 1, 100, 255);
    fill(255, alpha);
    text("Click or Tap to return to menu", width / 2, height / 2 + 90);
  } else {
    fill(150); text("...", width / 2, height / 2 + 90);
  }
}