/**********************************************************************
 * PAN & ZOOM
 **********************************************************************/
function setupPanZoom() {
  canvas.addEventListener('mousedown', (e) => {
    isDragging = true;
    lastMouseX = e.clientX;
    lastMouseY = e.clientY;
  });
  
  canvas.addEventListener('mouseup', () => { isDragging = false; });
  canvas.addEventListener('mouseleave', () => { isDragging = false; });
  
  canvas.addEventListener('mousemove', (e) => {
    if(isDragging) {
      let dx = e.clientX - lastMouseX;
      let dy = e.clientY - lastMouseY;
      offsetX += dx;
      offsetY += dy;
      lastMouseX = e.clientX;
      lastMouseY = e.clientY;
    }
  });
  
  canvas.addEventListener('wheel', (e) => {
    e.preventDefault();
    let zoomSpeed = 0.001;
    let delta = e.deltaY * zoomSpeed;
    let oldScale = scale;
    scale -= delta;
    if(scale < 0.1) scale = 0.1;
    if(scale > 5) scale = 5;

    let mouseX = e.clientX - (canvas.width/2 + offsetX);
    let mouseY = e.clientY - (canvas.height/2 + offsetY);
    offsetX -= mouseX * (scale - oldScale);
    offsetY -= mouseY * (scale - oldScale);
  }, {passive: false});
}

/**********************************************************************
 * BUY MENU LOGIC
 **********************************************************************/
function setupBuyButtons() {
  document.getElementById('buyHouseBtn').addEventListener('click', () => {
    purchaseMode = "House";
    logAction("Click on map to place a House site.");
  });
  
  document.getElementById('buyRoadBtn').addEventListener('click', () => {
    purchaseMode = "Road";
    logAction("Click on map to place a Road site.");
  });
  
  document.getElementById('buyBuilderBtn').addEventListener('click', () => {
    purchaseMode = "Builder";
    logAction("Click on map to place a new Builder citizen.");
  });
  
  document.getElementById('buyFarmerBtn').addEventListener('click', () => {
    purchaseMode = "Farmer";
    logAction("Click on map to place a new Farmer citizen.");
  });
  
  document.getElementById('buyMerchantBtn').addEventListener('click', () => {
    purchaseMode = "Merchant";
    logAction("Click on map to place a new Merchant citizen.");
  });
  
  document.getElementById('buyDoctorBtn').addEventListener('click', () => {
    purchaseMode = "Doctor";
    logAction("Click on map to place a new Doctor citizen.");
  });
  
  document.getElementById('buyTeacherBtn').addEventListener('click', () => {
    purchaseMode = "Teacher";
    logAction("Click on map to place a new Teacher citizen.");
  });
  
  document.getElementById('buyMarketBtn').addEventListener('click', () => {
    purchaseMode = "Market";
    logAction("Click on map to place a Market site.");
  });
  
  document.getElementById('buyHospitalBtn').addEventListener('click', () => {
    purchaseMode = "Hospital";
    logAction("Click on map to place a Hospital site.");
  });
  
  document.getElementById('buySchoolBtn').addEventListener('click', () => {
    purchaseMode = "School";
    logAction("Click on map to place a School site.");
  });
  
  document.getElementById('buySpawnerBtn').addEventListener('click', () => {
    purchaseMode = "Spawner";
    logAction("Click on map to place a Spawner building.");
  });
  
  document.getElementById('buyTreeBtn').addEventListener('click', () => {
    purchaseMode = "Tree";
    logAction("Click on map to place a new Tree.");
  });
  
  document.getElementById('toggleLogsBtn').addEventListener('click', (e) => {
    if(logContainer.style.display === "none") {
      logContainer.style.display = "block";
      e.target.textContent = "Hide Logs";
    } else {
      logContainer.style.display = "none";
      e.target.textContent = "Show Logs";
    }
  });
}

function setupCanvasClick() {
  canvas.addEventListener('click', (e) => {
    if(!purchaseMode) return;
    const rect = canvas.getBoundingClientRect();
    let cx = e.clientX - rect.left;
    let cy = e.clientY - rect.top;
    let worldX = (cx - (canvas.width/2) - offsetX) / scale;
    let worldY = (cy - (canvas.height/2) - offsetY) / scale;

    switch(purchaseMode) {
      case "House":
        if(money >= COST_HOUSE) {
          addMoney(-COST_HOUSE, "Buy House");
          let site = createHouseSite(worldX, worldY);
          buildings.push(site);
          logAction(`Purchased House site @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy House!");
        }
        break;
        
      case "Road":
        if(money >= COST_ROAD) {
          addMoney(-COST_ROAD, "Buy Road");
          // For simplicity, we create a short horizontal road
          let site = createRoadSite(worldX-50, worldY, worldX+50, worldY);
          buildings.push(site);
          logAction(`Purchased Road site @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Road!");
        }
        break;
        
      case "Market":
        if(money >= COST_MARKET) {
          addMoney(-COST_MARKET, "Buy Market");
          let site = createMarketSite(worldX, worldY);
          buildings.push(site);
          logAction(`Purchased Market site @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Market!");
        }
        break;
        
      case "Hospital":
        if(money >= COST_HOSPITAL) {
          addMoney(-COST_HOSPITAL, "Buy Hospital");
          let site = createHospitalSite(worldX, worldY);
          buildings.push(site);
          logAction(`Purchased Hospital site @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Hospital!");
        }
        break;
        
      case "School":
        if(money >= COST_SCHOOL) {
          addMoney(-COST_SCHOOL, "Buy School");
          let site = createSchoolSite(worldX, worldY);
          buildings.push(site);
          logAction(`Purchased School site @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy School!");
        }
        break;
        
      case "Builder":
        if(money >= COST_BUILDER) {
          addMoney(-COST_BUILDER, "Buy Builder");
          let c = createCitizen(randomName(), worldX, worldY, "Builder");
          citizens.push(c);
          logAction(`Purchased new Builder @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Builder!");
        }
        break;
        
      case "Farmer":
        if(money >= COST_FARMER) {
          addMoney(-COST_FARMER, "Buy Farmer");
          let c = createCitizen(randomName(), worldX, worldY, "Farmer");
          citizens.push(c);
          logAction(`Purchased new Farmer @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Farmer!");
        }
        break;
        
      case "Merchant":
        if(money >= COST_MERCHANT) {
          addMoney(-COST_MERCHANT, "Buy Merchant");
          let c = createCitizen(randomName(), worldX, worldY, "Merchant");
          citizens.push(c);
          logAction(`Purchased new Merchant @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Merchant!");
        }
        break;
        
      case "Doctor":
        if(money >= COST_DOCTOR) {
          addMoney(-COST_DOCTOR, "Buy Doctor");
          let c = createCitizen(randomName(), worldX, worldY, "Doctor");
          citizens.push(c);
          logAction(`Purchased new Doctor @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Doctor!");
        }
        break;
        
      case "Teacher":
        if(money >= COST_TEACHER) {
          addMoney(-COST_TEACHER, "Buy Teacher");
          let c = createCitizen(randomName(), worldX, worldY, "Teacher");
          citizens.push(c);
          logAction(`Purchased new Teacher @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Teacher!");
        }
        break;
        
      case "Spawner":
        if(money >= COST_SPAWNER) {
          addMoney(-COST_SPAWNER, "Buy Spawner");
          let spawnBuild = {
            buildingType: "Spawner",
            x: worldX, y: worldY,
            completed: true,
            lastSpawnTime: frameCount
          };
          buildings.push(spawnBuild);
          logAction(`Purchased Spawner @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Spawner!");
        }
        break;
        
      case "Tree":
        if(money >= COST_TREE) {
          addMoney(-COST_TREE, "Buy Tree");
          let t = createResource("Tree", worldX, worldY, 100);
          resources.push(t);
          logAction(`Purchased a new Tree @(${Math.floor(worldX)},${Math.floor(worldY)})`);
        } else {
          logAction("Not enough money to buy Tree!");
        }
        break;
    }

    purchaseMode = null;
  });
}