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,
} 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') });

// 扫地机 S10:**与水浸同款添加流程**(串口配对 + WiFi 配网)。
// 入口品类「Floor Cleaning Robot S10」→ 长按 ch17+18 5s 进添加模式 → Connect Device → Configure Wi-Fi 配网。
const CATEGORY = process.env.ROBOT_S10_CATEGORY || 'Floor Cleaning Robot S10';
const deviceName = process.env.ROBOT_S10_DEVICE || 'S10';
const NAME_PATTERN = process.env.ROBOT_S10_PATTERN || 'S10[^"]*\\s*[0-9A-Za-z]{0,4}';
const ADD_ANCHOR = '[P0][ONES:78369]'; // 扫地机 S10 添加(Robot S10)

describe('Robot S10 Connect - 添加扫地机S10(串口配对 + WiFi配网,同水浸)', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;

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

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

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

      const result = await addDeviceWithSerialPairing(driver, {
        categoryName: CATEGORY,
        categoryScrollHint: 0,
        deviceKeyword: deviceName,
        relayDevice: 'robot_s10',      // ch17+18 同时长按 5s(config/relay.config)
        registryCategory: 'robot',
        namePattern: NAME_PATTERN,
        // 比水浸多一步向导:选品类后是「Step 1: Power on」→ 点 Next 到「Step 2: Enter pairing mode」,helper 在此按 ch17+18。
        beforePairing: async (d) => {
          for (let i = 0; i < 8; i++) {
            if (/Step 2|Enter pairing mode|Connect Device/i.test(await d.getSource())) break; // 已到第二步
            const n = await d.findElementRaw('-android uiautomator', 'new UiSelector().text("Next")').catch(() => null);
            if (n) { await d.tapElement(n); await sleep(2000); } else break;
          }
        },
        // 连接后(同水浸):① 点 "Connect Device" 启动连接 ② 进 Configure Wi-Fi 页 → configureHubWifi 填 SSID/密码
        postConnectSteps: async (d) => {
          const btn = await d.findElementRaw('-android uiautomator', 'new UiSelector().text("Connect Device")')
            || await d.findElementRaw('-android uiautomator', 'new UiSelector().text("Connect device")');
          if (btn) { await d.tapElement(btn); await sleep(2500); }
          await configureHubWifi(d, wifi);
        },
        connectionKeywords: [
          'Connect Device', 'Configure Wi-Fi', 'Connecting', 'Discover device',
          '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} 添加扫地机S10`, 'PASS', Date.now() - start, `添加成功(WiFi:${wifi.ssid}), 耗时${elapsed}s`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record(`${ADD_ANCHOR} 添加扫地机S10`, 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });
});