import assert from 'node:assert/strict';
import { readFileSync } from 'node:fs';
import { generateMap } from './mapGenerator.js';
import { createWorldMap } from './worldMap.js';
import { generatePatch } from './mapPatch.js';
import { MAP_W, MAP_H } from './mapUtils.js';

function worldPoint(world, p) {
  return { x: Math.round((p?.x || 0) + (world.originX || 0)), y: Math.round((p?.y || 0) + (world.originY || 0)) };
}
function idx(world,x,y){ return x>=0&&y>=0&&x<world.width&&y<world.height ? y*world.width+x : -1; }
function capitalAudit(world) {
  const pref=world.fields.prefectureRegionId, sea=world.fields.sea;
  const active=new Set();
  for(let i=0;i<pref.length;i++) if(!sea[i]&&pref[i]>=0) active.add(Math.floor(pref[i]));
  const counts=new Map([...active].map(id=>[id,0]));
  for(const city of world.sourceMap.modernCities||[]) {
    const cap=city.isPrefecturalCapital||city.isRegionalCapital||/Capital/i.test(String(city.rank||''))||/Capital/i.test(String(city.kind||''));
    if(!cap) continue;
    const p=worldPoint(world,city), i=idx(world,p.x,p.y); if(i<0||sea[i]||pref[i]<0) continue;
    const id=Math.floor(pref[i]); counts.set(id,(counts.get(id)||0)+1);
  }
  return { active:[...active], zero:[...counts].filter(([,n])=>n===0), duplicate:[...counts].filter(([,n])=>n>1) };
}
function riverSeaPointsOutsideRect(world, rect) {
  const sea=world.fields.sea; const out=[];
  for(const key of ['rivers','riverPaths','majorRivers','minorRivers']) for(const path of world.sourceMap?.[key]||[]) for(const t of path||[]) {
    if(!Array.isArray(t)||t.length<2) continue;
    const x=Math.round((t[0]||0)+world.originX), y=Math.round((t[1]||0)+world.originY), i=idx(world,x,y);
    if(i>=0 && sea[i] && !(x>=rect.x0&&x<rect.x1&&y>=rect.y0&&y<rect.y1)) out.push(`${key}:${x},${y}`);
  }
  return out.sort();
}

const seed=24681357;
const initial=generateMap(seed,{terrainType:'auto',onProgress(){}});
const world=createWorldMap(structuredClone(initial));
const rect={x0:world.originX+MAP_W-25,y0:world.originY+35,x1:world.originX+MAP_W+120,y1:world.originY+150};
const beforeRiver=riverSeaPointsOutsideRect(world,rect);
const result=generatePatch(world,rect,{patchMode:'expansion',terrainType:'auto',seed:777001,variant:0,maxQualityRetries:0,qualityTerrainAttempts:1,acceptBestAvailableQuality:true});
assert.equal(result.ok,true, result.reason||result.code);
const capitals=capitalAudit(world);
assert.equal(capitals.zero.length,0,`prefectures with no capital: ${JSON.stringify(capitals.zero)}`);
assert.equal(capitals.duplicate.length,0,`prefectures with duplicate capitals: ${JSON.stringify(capitals.duplicate)}`);
const afterRiver=riverSeaPointsOutsideRect(world,rect);
assert.deepEqual(afterRiver,beforeRiver,'river sanitation changed sea-mouth points outside patch write region');
for(const r of world.sourceMap.prefectureRegions||[]) {
  assert.equal(r.insidePrefecture,true);
  assert.ok(Number.isFinite(r.worldX)&&Number.isFinite(r.worldY));
  assert.ok(Number.isFinite(r.capitalX)&&Number.isFinite(r.capitalY));
  assert.ok(Number.isFinite(r.capitalWorldX)&&Number.isFinite(r.capitalWorldY));
}

const app=readFileSync(new URL('./app.js', import.meta.url),'utf8');
const html=readFileSync(new URL('./index.html', import.meta.url),'utf8');
assert.match(app,/PATCH_WORKER_INACTIVITY_WATCHDOG_MS\s*=\s*60_000/);
assert.match(app,/function cancelPatchGeneration/);
assert.match(app,/hideSelectionOverlay\(\{ discardPreview: true \}\)/);
assert.match(html,/id="cancelPatchGeneration"/);

console.log(JSON.stringify({ok:true,capitalAudit:capitals,outsideRiverSeaPointsPreserved:beforeRiver.length,seam:result.seamDiagnostics?.status||null},null,2));