import 'dart:convert';
import 'dart:io';

import 'package:crypto/crypto.dart' show sha256;
import 'package:flutter/material.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:http/http.dart' as http;
import 'package:http/testing.dart';
import 'package:observer/camera/model_catalog_section.dart';
import 'package:observer/models/model_manager.dart';

const _modelBytes = [1, 2, 3, 4, 5, 6, 7, 8];
String _shaHex(List<int> bytes) => sha256.convert(bytes).toString();

/// 分两段到达的下载流（中间 500ms 停顿：足够「取消」按钮出现并点击，
/// 且第二次分块到达前取消标志必然已置位——每轮 pump 只推进 400ms fake 时钟）
Stream<List<int>> _delayedChunks() async* {
  yield [1, 2, 3, 4];
  await Future<void>.delayed(const Duration(milliseconds: 500));
  yield [5, 6, 7, 8];
}

Map<String, dynamic> _item({
  int datasetId = 7,
  String variant = kVariantS,
  String name = '数据集A',
}) =>
    {
      'datasetId': datasetId,
      'datasetName': name,
      if (variant.isNotEmpty) 'variant': variant,
      'version': 'v1.0.0',
      'labels': ['target', 'suspect'],
      'sizeBytes': _modelBytes.length,
      'sha256': _shaHex(_modelBytes),
      'downloadUrl': '/download/models/$datasetId/$variant.tflite',
      'coverUrl': '/api/v1/app/cover?namePrefix=DS001',
    };

Map<String, dynamic> _catalog(List<Map<String, dynamic>> models) => {
      'code': 0,
      'message': 'ok',
      'data': {'models': models}
    };

/// 真实流式下载客户端：send 立即返回分块流（MockClient 的 Response.fromStream
/// 会先消费完整流再返回，导致进度永远停在 0）。
class _StreamingClient extends http.BaseClient {
  @override
  Future<http.StreamedResponse> send(http.BaseRequest request) async {
    if (request.url.path == '/api/v1/app/update') {
      return http.StreamedResponse(
          http.ByteStream.fromBytes(utf8.encode(
              jsonEncode(_catalog([_item()])))),
          200);
    }
    return http.StreamedResponse(_delayedChunks(), 200,
        contentLength: _modelBytes.length);
  }
}

void main() {
  late Directory root;

  setUp(() async {
    root = await Directory.systemTemp.createTemp('catalog-section-test');
  });

  tearDown(() => root.delete(recursive: true));

  ModelManager manager(List<Map<String, dynamic>> models) => ModelManager(
        baseUrl: 'http://test.local',
        client: MockClient((req) async {
          if (req.url.path == '/api/v1/app/update') {
            return http.Response.bytes(
                utf8.encode(jsonEncode(_catalog(models))), 200);
          }
          if (req.url.path.startsWith('/download/models/')) {
            return http.Response.bytes(_modelBytes, 200);
          }
          return http.Response('not found', 404);
        }),
        rootDir: () async => root,
      );

  /// 推进真实异步链：pump 冲刷 fake 微任务队列（并推进 fake 时钟触发 fake 定时器），
  /// 再用 runAsync 让真实事件循环转动，送达文件 IO 完成事件。
  /// testWidgets 的 fake-async 区中发起的真实 IO 必须这样交替驱动，否则异步链卡死。
  Future<void> pumpRealIo(WidgetTester tester,
      {Duration fake = Duration.zero}) async {
    await tester.pump(fake);
    await tester.runAsync(
        () => Future<void>.delayed(const Duration(milliseconds: 5)));
    await tester.pump();
  }

  /// 交替 pump/真实等待，直至 finder 命中（UI 呈现目标状态）。
  Future<void> pumpUntilFound(WidgetTester tester, Finder finder,
      {Duration fake = Duration.zero}) async {
    for (var i = 0; i < 100 && finder.evaluate().isEmpty; i++) {
      await pumpRealIo(tester, fake: fake);
    }
  }

  Future<void> pumpSection(WidgetTester tester, ModelManager m) async {
    // 与设置弹层一致：区块置于可滚动容器（区块含识别模式行后超出测试视口，
    // 点按前需 ensureVisible 滚动到目标）
    await tester.pumpWidget(MaterialApp(
        home: Scaffold(
            body: SingleChildScrollView(
                child: ModelCatalogSection(manager: m)))));
    await tester.pumpAndSettle(const Duration(milliseconds: 50),
        EnginePhase.sendSemanticsUpdate, const Duration(seconds: 5));
  }

  /// 滚动到目标可见后点按（区块可超出视口）
  Future<void> tapVisible(WidgetTester tester, Finder finder) async {
    await tester.ensureVisible(finder);
    await tester.pump();
    await tester.tap(finder);
  }

  testWidgets('未下载：显示下载按钮，点击后下载完成自动变为使用中', (tester) async {
    final m = manager([_item()]);
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    expect(find.text('数据集A'), findsOneWidget);
    expect(find.text('下载'), findsOneWidget);

    await tapVisible(tester, find.text('下载'));
    await tester.pump(); // 下载启动，进度条出现
    // 驱动下载 + 自动激活的 IO 链走完，直至 UI 呈现「使用中」
    await pumpUntilFound(tester, find.text('使用中'));
    await tester.pumpAndSettle();

    expect(m.isDownloaded(7, kVariantS), isTrue);
    expect(m.isActive(7, kVariantS), isTrue);
    expect(find.text('使用中'), findsOneWidget);
  });

  testWidgets('已激活：点击使用中取消使用', (tester) async {
    final m = manager([_item()]);
    await tester.runAsync(() => m.refresh());
    await tester.runAsync(() => m.downloadModel(m.catalog.first));
    await pumpSection(tester, m);

    expect(find.text('使用中'), findsOneWidget);
    await tapVisible(tester, find.text('使用中'));
    // 排空 setActive 的激活集重载与落盘 IO，直至 UI 呈现「使用」
    await pumpUntilFound(tester, find.text('使用'));
    await tester.pumpAndSettle();

    expect(m.isActive(7, kVariantS), isFalse);
    expect(find.text('使用'), findsOneWidget);
  });

  testWidgets('目录为空：显示暂无已发布模型', (tester) async {
    final m = ModelManager(
      baseUrl: 'http://test.local',
      client: MockClient((_) async => http.Response.bytes(
          utf8.encode(jsonEncode(_catalog([]))),
          200)),
      rootDir: () async => root,
    );
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    expect(find.text('暂无已发布模型'), findsOneWidget);
  });

  testWidgets('下载中：显示进度和取消按钮，取消后恢复下载', (tester) async {
    final m = ModelManager(
      baseUrl: 'http://test.local',
      client: _StreamingClient(),
      rootDir: () async => root,
    );
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    await tapVisible(tester, find.text('下载'));
    // 首个分块到达，进度条与取消按钮出现（IO 链需交替驱动）
    await pumpUntilFound(tester, find.text('取消'),
        fake: const Duration(milliseconds: 400));
    expect(find.text('取消'), findsOneWidget);

    await tapVisible(tester, find.text('取消'));
    // 推进 fake 时钟触发延迟分块 → 取消分支清场（.part 删除等真实 IO），直至恢复「下载」
    await pumpUntilFound(tester, find.text('下载'),
        fake: const Duration(milliseconds: 400));
    await tester.pumpAndSettle();

    expect(m.isDownloaded(7, kVariantS), isFalse);
    expect(m.isActive(7, kVariantS), isFalse);
    expect(find.text('下载'), findsOneWidget);
  });

  testWidgets('双档位：同物种合并一张卡，下载只启用目标档，使用即停旧档', (tester) async {
    final m = manager([_item(), _item(variant: kVariantN)]);
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    // 合并卡：同一数据集只一张卡，无双卡/无档位角标；双档行标注两档
    expect(find.text('数据集A'), findsOneWidget);
    expect(find.text('下载'), findsOneWidget, reason: '合并卡只有一个主按钮');
    expect(find.textContaining('高精度 v1.0.0'), findsOneWidget);
    expect(find.textContaining('高性能 v1.0.0'), findsOneWidget);

    // 一次下载取回双档：同物种只启用目标档（默认高精度 s），n 档备好不启用。
    // 双档下载收尾链交错，等 models 真正重载完成再断言
    await tapVisible(tester, find.text('下载'));
    await pumpUntilFound(tester, find.text('使用中'));
    for (var i = 0;
        i < 50 &&
            !(m.isDownloaded(7, kVariantS) &&
                m.isDownloaded(7, kVariantN) &&
                m.isActive(7, kVariantS) &&
                !m.isActive(7, kVariantN) &&
                m.models.length == 1 &&
                m.models.first.variant == kVariantS);
        i++) {
      await pumpRealIo(tester);
    }
    await tester.pumpAndSettle();

    expect(m.isDownloaded(7, kVariantS), isTrue);
    expect(m.isDownloaded(7, kVariantN), isTrue);
    expect(m.isActive(7, kVariantS), isTrue);
    expect(m.isActive(7, kVariantN), isFalse,
        reason: '同物种同时只运行目标档，另一档备好不叠加启用');
    expect(m.mode, kVariantS, reason: '下载双档不应改变目标档');
    expect(m.models.length, 1);
    expect(m.models.first.variant, kVariantS);
    expect(find.text('使用中'), findsOneWidget);

    // 目标档切到高性能：运行中的 s 不受影响；n 已下载未启用 → 主按钮「使用」
    // （保存目标档偏好是真实 IO，等按钮呈现才算 UI 落定；2026-09-03 简化：
    // 不再叫「改用X」、不显示「当前使用 X」提示）
    await tapVisible(tester, find.text('高性能'));
    await pumpUntilFound(tester, find.text('使用'));
    await tester.pumpAndSettle();
    expect(m.mode, kVariantN);
    expect(m.isActive(7, kVariantS), isTrue,
        reason: '切换目标档不应改变运行中的模型');
    expect(find.text('使用'), findsOneWidget);
    expect(find.textContaining('当前使用'), findsNothing);

    // 点「使用」：同物种 s 自动停用，n 热启用（setActive 尾部的落盘
    // 与 notify 是真实 IO，等 UI 呈现「使用中」才算整链落定）
    await tapVisible(tester, find.text('使用'));
    await pumpUntilFound(tester, find.text('使用中'));
    await tester.pumpAndSettle();
    expect(m.isActive(7, kVariantS), isFalse, reason: '启用目标档自动停用另一档');
    expect(m.isActive(7, kVariantN), isTrue);
    expect(m.models.length, 1);
    expect(m.models.first.variant, kVariantN);
    expect(find.text('使用中'), findsOneWidget, reason: '主按钮跟随当前档');
  });

  testWidgets('不同动物可用不同档位：使用只作用于点击的动物', (tester) async {
    final m = manager([
      _item(datasetId: 7, name: '数据集A'),
      _item(datasetId: 7, variant: kVariantN, name: '数据集A'),
      _item(datasetId: 8, name: '数据集B'),
      _item(datasetId: 8, variant: kVariantN, name: '数据集B'),
    ]);
    await tester.runAsync(() => m.refresh());
    // 两个物种都下载并运行默认档（高精度 s），各自 n 档备好
    for (final it in m.catalog) {
      await tester.runAsync(() => m.downloadModel(it));
    }
    await pumpSection(tester, m);

    expect(m.models.length, 2);
    expect(find.text('使用中'), findsNWidgets(2));

    // 目标档切高性能：两卡主按钮都变「使用」（各自 s 仍在运行、n 已备好）
    await tapVisible(tester, find.text('高性能'));
    await pumpUntilFound(tester, find.text('使用'));
    expect(find.text('使用'), findsNWidgets(2));

    // 只点第一张卡（数据集 7，目录序在前）：该动物切到高性能，另一动物保持高精度
    await tapVisible(tester, find.text('使用').first);
    await pumpUntilFound(tester, find.text('使用中')); // 数据集 7 的卡先翻为使用中
    for (var i = 0;
        i < 50 &&
            !(m.models.length == 2 &&
                m.models.firstWhere((b) => b.datasetId == 7).variant ==
                    kVariantN &&
                m.models.firstWhere((b) => b.datasetId == 8).variant ==
                    kVariantS);
        i++) {
      await pumpRealIo(tester);
    }
    await tester.pumpAndSettle();

    expect(m.isActive(7, kVariantN), isTrue);
    expect(m.isActive(7, kVariantS), isFalse, reason: '启用目标档自动停用同动物另一档');
    expect(m.isActive(8, kVariantS), isTrue, reason: '未点击的动物保持原档位');
    expect(m.isActive(8, kVariantN), isFalse);
    expect(m.models.length, 2, reason: '两个动物仍同时识别');
    expect(m.models.firstWhere((b) => b.datasetId == 7).variant, kVariantN);
    expect(m.models.firstWhere((b) => b.datasetId == 8).variant, kVariantS);
    expect(find.text('使用中'), findsOneWidget, reason: '只有数据集 7 的卡使用中');
    expect(find.text('使用'), findsOneWidget,
        reason: '数据集 8 目标档已备好：按钮就是使用（不叫改用）');
    expect(find.textContaining('当前使用'), findsNothing);
  });

  testWidgets('识别模式分段控件：只改目标档偏好，不影响已在运行的模型', (tester) async {
    final m = manager([_item(), _item(variant: kVariantN)]);
    await tester.runAsync(() => m.refresh());
    await tester.runAsync(() => m.downloadModel(m.catalog.first)); // s 启用
    await pumpSection(tester, m);

    expect(find.text('高精度'), findsOneWidget, reason: '分段控件选中默认档');
    expect(find.text('高性能'), findsOneWidget);
    expect(find.text('使用中'), findsOneWidget);

    await tapVisible(tester, find.text('高性能'));
    // 目标档 n 尚未下载（之前只下了 s）→ 主按钮变为「下载」：仍不切运行模型
    await pumpUntilFound(tester, find.text('下载'));
    await tester.pumpAndSettle();
    expect(m.mode, kVariantN, reason: '点击分段控件应改目标档');
    expect(m.isActive(7, kVariantS), isTrue, reason: '只改偏好不切运行模型');
    expect(m.models.length, 1);
    expect(m.models.first.variant, kVariantS, reason: '没有热加载/切换');
    expect(find.text('下载'), findsOneWidget, reason: '卡片按钮面向新目标档');
    expect(find.textContaining('当前使用'), findsNothing);
    // 目标档偏好持久化由 model_manager_test「识别目标档位偏好持久化」覆盖；
    // widget 测试内 tap 触发的文件写与真实 IO 交错不可控，不在本层断言重启
  });
}