import { describe, it, beforeAll, afterAll, expect } from 'vitest';
import { DeviceDriver } from '../../drivers/types';
import { createDriver } from '../../drivers/factory';
import { TestReporter } from '../../utils/test-reporter';
import {
  sleep,
  addDeviceWithSerialPairing,
  findDeviceNameOnHomepage,
  saveAddedDevice,
  configureHubWifi,
  onesAdd,
} from '../../utils/common';
import { getWifiCredentials } from '../../config/wifi.config';
import * as dotenv from 'dotenv';
import * as path from 'path';

dotenv.config({ path: path.resolve(__dirname, '../../.env') });

// Plug Mini 是 WiFi 产品:relay 按键配对 + WiFi 配网。ch1=JP(默认)/ch2=US。
const PLUG_MINI_CATEGORY = process.env.PLUG_MINI_CATEGORY || 'Plug Mini (JP)';
const RELAY_DEVICE = process.env.PLUG_MINI_RELAY || 'plug_mini'; // JP=plug_mini(ch1) / US=plug_mini_us(ch2)
const deviceName = process.env.PLUG_MINI_DEVICE || PLUG_MINI_CATEGORY; // 首页名形如 "Plug Mini (EU) 2A" = 品类名+后缀,**含区域**
// 区域专属校验:从品类名派生(转义括号),避免 JP/US/EU 互相串台误判(之前用通用 "Plug Mini XX" 会匹配到任意一个 → 假通过)
const PLUG_MINI_PATTERN = process.env.PLUG_MINI_PATTERN || `${PLUG_MINI_CATEGORY.replace(/[()]/g, '\\$&')}\\s*[0-9A-Za-z]{1,2}`;
const ADD_ANCHOR = process.env.PLUG_MINI_ONES ? `[P0][ONES:${process.env.PLUG_MINI_ONES}]` : onesAdd('plug', 'PlugMini 5E'); // EU 传 PLUG_MINI_ONES=332918

describe('Plug Mini Connect - 添加Plug Mini设备(串口配对 + WiFi配网)', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;

  beforeAll(async () => {
    driver = createDriver();
    await driver.createSession();
    reporter = new TestReporter('PlugMini_Connect', driver.platform.toUpperCase());
  });

  afterAll(async () => {
    reporter.generate();
    await driver.destroySession();
  });

  it(`${ADD_ANCHOR} 通过BLE+WiFi添加Plug Mini(${PLUG_MINI_CATEGORY})`, { timeout: 360000 }, async () => {
    const start = Date.now();
    const wifi = getWifiCredentials();
    try {
      const existing = await findDeviceNameOnHomepage(driver, PLUG_MINI_PATTERN);
      if (existing) {
        console.log(`${existing}已在首页，跳过重新添加`);
        saveAddedDevice('plug', existing);
        reporter.record(`${ADD_ANCHOR} 添加${PLUG_MINI_CATEGORY}`, 'SKIP', Date.now() - start, `${existing}已存在`);
        return;
      }

      const result = await addDeviceWithSerialPairing(driver, {
        categoryName: PLUG_MINI_CATEGORY,
        categoryScrollHint: 0,
        deviceKeyword: deviceName,
        relayDevice: RELAY_DEVICE,    // ch1(JP)/ch2(US),长按 2s
        registryCategory: 'plug',
        namePattern: PLUG_MINI_PATTERN,
        // plug 第一次按键+Next 即连上去 WiFi 页;重按会把它弄出连接 → 不重按,只延长等待
        repressOnRetry: false,
        tapOtherPairing: process.env.PLUG_OTHER_PAIRING === '1', // JP 自动配对不行,需点"其他配对方式"(PLUG_OTHER_PAIRING=1 启用)
        pairRetries: 2,
        pairPollIters: 16,
        // 选品类后是 "Before Using" 免责页(非 Disclaimer 关键字),beforePairing 里点 Next 过掉 → 到"长按按钮"引导页
        beforePairing: async (d) => {
          const n = await d.findElementRaw('-android uiautomator', 'new UiSelector().text("Next")');
          if (n) { await d.tapElement(n); await sleep(2000); }
        },
        // 连接后:进 Wi-Fi Settings 页填 SSID/密码(configureHubWifi 已兼容标题/输入框检测)
        postConnectSteps: async (d) => { await configureHubWifi(d, wifi); },
        connectionKeywords: [
          'Configure Wi-Fi', 'Connecting', 'Pick a room', 'Done',
          'added successfully', 'Added successfully', 'Initial Setup',
        ],
      });
      expect(result).toBe(true);

      const elapsed = ((Date.now() - start) / 1000).toFixed(1);
      reporter.record(`${ADD_ANCHOR} 添加${PLUG_MINI_CATEGORY}`, 'PASS', Date.now() - start, `${deviceName}添加成功(WiFi:${wifi.ssid}), 耗时${elapsed}s`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record(`${ADD_ANCHOR} 添加${PLUG_MINI_CATEGORY}`, 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });
});