import puppeteer from 'puppeteer-core'
import path from 'path'
const browser = await puppeteer.launch({
  executablePath: 'C:/Program Files (x86)/Microsoft/Edge/Application/msedge.exe',
  headless: 'new', args: ['--window-size=1440,900', '--enable-unsafe-swiftshader'],
  defaultViewport: { width: 1440, height: 900 },
})
const page = await browser.newPage()
const errs = []
page.on('pageerror', e => errs.push(String(e).slice(0, 120)))
const token = process.argv[2]
const sketch = process.argv[3]
await page.goto(`http://localhost:8000/ai_render?token=${encodeURIComponent(token)}`, { waitUntil: 'networkidle2' })
await new Promise(r => setTimeout(r, 2000))
await page.evaluate(() => {
  [...document.querySelectorAll('.el-radio-button')].find(b => b.textContent.includes('工程大样图'))?.querySelector('input')?.click()
})
await new Promise(r => setTimeout(r, 800))
const input = await page.$('.el-upload input[type=file]')
await input.uploadFile(path.resolve(sketch))
await new Promise(r => setTimeout(r, 1200))
await page.evaluate(() => { [...document.querySelectorAll('button')].find(b => b.textContent.includes('AI 生成大样图'))?.click() })
let done = false
for (let i = 0; i < 110; i++) {
  await new Promise(r => setTimeout(r, 2000))
  done = await page.evaluate(() => !!document.querySelector('.bp-compare'))
  if (done) break
}
// 打开生成记录抽屉
await page.evaluate(() => { [...document.querySelectorAll('button')].find(b => b.textContent.includes('生成记录'))?.click() })
await new Promise(r => setTimeout(r, 2000))
await page.screenshot({ path: '../shots/rec-drawer.png' })
console.log(done ? '生成+记录抽屉 OK' : '生成超时', '| 异常:', errs.length ? errs.join('|') : '无')
await browser.close()