Clean/lib/module/image_picker/image_picker_util.dart

import 'dart:io';
import 'dart:math';

import 'package:classify_photo/classify_photo.dart';
import 'package:clean/data/bean/photos_type.dart';
import 'package:clean/model/asset_group.dart';
import 'package:clean/module/locations_photo/locations_photo_controller.dart';
import 'package:clean/module/locations_photo/locations_single_photo_controller.dart';
import 'package:clean/module/people_photo/people_photo_controller.dart';
import 'package:clean/module/people_photo/photo_group.dart';
import 'package:clean/module/photo_preview/photo_preview_controller.dart';
import 'package:clean/module/screenshots_blurry/screenshots_controller.dart';
import 'package:clean/module/similar_photo/similar_photo_controller.dart';
import 'package:flutter/widgets.dart';
import 'package:get/get.dart';
import 'package:photo_classifier/models.dart';
import 'package:photo_manager/photo_manager.dart';

import '../../handler/photo_scan_handler.dart';
import '../calendar/calendar_state.dart';
import '../home/home_controller.dart';

class ImagePickerUtil {
  ImagePickerUtil._();

  static const RequestType permissionType = RequestType.image;

  static final RxInt similarPhotoCount = 0.obs;

  // 全局存储不同类型的照片
  // 截图图片
  static final RxList<AssetEntity> screenshotPhotos = <AssetEntity>[].obs;

  // 模糊图片
  static final RxList<AssetEntity> blurryPhotos = <AssetEntity>[].obs;

  // 相似图片
  static final RxList<List<AssetEntity>> similarPhotos =
      <List<AssetEntity>>[].obs;

  // 地点图片
  static final RxMap<String, List<AssetEntity>> locationPhotos =
      <String, List<AssetEntity>>{}.obs;

  // 人物图片
  static final RxList<AssetEntity> peoplePhotos = <AssetEntity>[].obs;

  // android测试

  static late List<AssetEntity> androidPhotos = <AssetEntity>[];

  static final RxSet<String> selectedScreenshotPhotosIds = <String>{}.obs;
  static final RxSet<String> selectedSimilarPhotosIds = <String>{}.obs;
  static final RxSet<String> selectedLocationPhotosIds = <String>{}.obs;
  static final RxSet<String> selectedPeoplePhotosIds = <String>{}.obs;
  static final RxSet<String> selectedBlurryPhotosIds = <String>{}.obs;

  // 添加大小信息的变量
  static final Rx<int> screenshotsSize = 0.obs;
  static final Rx<int> blurrySize = 0.obs;
  static final Rx<int> locationsSize = 0.obs;
  static final Rx<int> peopleSize = 0.obs;
  static final Rx<int> similarPhotosSize = 0.obs;

  static var similarResult = <ClassifiedImageGroup>[].obs;
  static var peopleResult = <ClassifiedImage>[].obs;
  static var screenshotResult = <ClassifiedImage>[].obs;
  static var blurryResult = <ClassifiedImage>[].obs;

  // 清除所有照片数据
  static void clearAllPhotos() {
    screenshotPhotos.clear();
    blurryPhotos.clear();
    similarPhotos.clear();
    locationPhotos.clear();
    peoplePhotos.clear();
  }

  static loadData(ClassificationResult result) async {
    if (result.similarGroups != null) {
      similarResult.assignAll(result.similarGroups ?? []);
      await newUpdatePhotoGroups("similar", similarResult);
    }

    if (result.peopleImages != null) {
      peopleResult.addAll(result.peopleImages ?? []);
      await newUpdatePhotoGroup("people", peopleResult);
    }

    if (result.screenshotImages != null) {
      screenshotResult.addAll(result.screenshotImages ?? []);
      await newUpdatePhotoGroup("screenshots", screenshotResult);
    }
    if (result.blurryImages != null) {
      blurryResult.addAll(result.blurryImages ?? []);
      await newUpdatePhotoGroup("blurry", blurryResult);
    }
  }

  static Future<void> updatePhotoGroupDate(
      PhotosType photosType, Set<String> selectedPhotosIds1) async {
    // // 1. 统一移除相关的照片列表
    debugPrint(
        "updatePhotoGroupDate1 ScreenshotsController screenshotController $selectedPhotosIds1");
    // 2. 移除控制器中的数据
    Set<String> selectedPhotosIds = Set.from(selectedPhotosIds1);
    updatePhotoData(selectedPhotosIds);
    //
    debugPrint(
        "updatePhotoGroupDate2 ScreenshotsController screenshotController $selectedPhotosIds");
    // 3. 根据photosType来处理不同的控制器和photoGroups
    switch (photosType) {
      case PhotosType.screenshots:
        // ScreenShotsController screenshotController = Get.find<ScreenShotsController>();
        // removeImagesAndEmptyGroups(screenshotController, selectedPhotosIds);
        // selectedScreenshotPhotosIds.removeWhere((element) => selectedPhotosIds.contains(element));
        // screenshotController.restoreSelections();
        break;

      case PhotosType.similarPhotos:
        SimilarPhotoController similarController =
            Get.find<SimilarPhotoController>();
        removeImagesAndEmptyGroups(similarController, selectedPhotosIds);
        selectedSimilarPhotosIds
            .removeWhere((element) => selectedPhotosIds.contains(element));
        similarController.restoreSelections();
        break;

      case PhotosType.locationPhotos:
        // LocationsSinglePhotoController locationsSingleController = Get.find<LocationsSinglePhotoController>();
        // LocationsPhotoController locationsPhotoController = Get.find<LocationsPhotoController>();
        // removeImagesAndEmptyGroups(locationsSingleController, selectedPhotosIds);
        // removeImagesAndEmptyGroups(locationsPhotoController, selectedPhotosIds);
        // selectedLocationPhotosIds.removeWhere((element) => selectedPhotosIds.contains(element));
        // locationsSingleController.restoreSelections();
        // locationsPhotoController.restoreSelections();
        break;

      case PhotosType.peoplePhotos:
        // PeoplePhotoController peoplePhotoController = Get.find<PeoplePhotoController>();
        // removeImagesAndEmptyGroups(peoplePhotoController, selectedPhotosIds);
        // selectedPeoplePhotosIds.removeWhere((element) => selectedPhotosIds.contains(element));
        // peoplePhotoController.restoreSelections();
        break;

      case PhotosType.blurryPhotos:
        // ScreenShotsController blurryController = Get.find<ScreenShotsController>();
        // removeImagesAndEmptyGroups(blurryController, selectedPhotosIds);
        // selectedBlurryPhotosIds.removeWhere((element) => selectedPhotosIds.contains(element));
        // blurryController.restoreSelections();
        break;
    }
  }

  // 移除每组图片里面对应的元素并删除空的组
  static void removeImagesAndEmptyGroups(
      dynamic controller, Set<String> selectedPhotosIds) {
    // 创建photoGroups的副本进行遍历，避免遍历过程中修改photoGroups
    for (var group in List.from(controller.photoGroups)) {
      // 移除所有与selectedPhotosIds匹配的图片
      group.images.removeWhere((image) => selectedPhotosIds.contains(image.id));

      // 如果当前组内没有图片了，删除该组
      if (group.images.isEmpty) {
        controller.photoGroups.remove(group);
      }
    }
  }

  static void updatePhotoData(Set<String> selectedPhotosIds1) {
    debugPrint("ImagePickerUtil1 updatePhotoData  $selectedPhotosIds1");
    Set<String> selectedPhotosIds = Set.from(selectedPhotosIds1);
    screenshotPhotos
        .removeWhere((element) => selectedPhotosIds.contains(element.id));
    for (var group in similarPhotos) {
      group.removeWhere((element) => selectedPhotosIds.contains(element.id));
    }
    locationPhotos.forEach((key, group) =>
        group.removeWhere((element) => selectedPhotosIds.contains(element.id)));
    peoplePhotos
        .removeWhere((element) => selectedPhotosIds.contains(element.id));
    blurryPhotos
        .removeWhere((element) => selectedPhotosIds.contains(element.id));
    // 2. 移除空的集合
    similarPhotos.removeWhere((element) => element.isEmpty);
    locationPhotos.removeWhere((key, value) => value.isEmpty);

    debugPrint("ImagePickerUtil2 updatePhotoData  $selectedPhotosIds");
    selectedScreenshotPhotosIds
        .removeWhere((element) => selectedPhotosIds.contains(element));
    selectedSimilarPhotosIds
        .removeWhere((element) => selectedPhotosIds.contains(element));
    selectedLocationPhotosIds
        .removeWhere((element) => selectedPhotosIds.contains(element));
    selectedPeoplePhotosIds
        .removeWhere((element) => selectedPhotosIds.contains(element));
    selectedBlurryPhotosIds
        .removeWhere((element) => selectedPhotosIds.contains(element));

    if (PhotoScanHandler.locationPhoto.value != null &&
        selectedPhotosIds.contains(PhotoScanHandler.locationPhoto.value!.id)) {
      PhotoScanHandler.locationPhoto.value = null;
      if (locationPhotos.isNotEmpty) {
        var asset = locationPhotos.values.first.first;
        PhotoScanHandler.locationPhoto.value = asset;
      }
    }
    if (PhotoScanHandler.screenshotPhoto.value != null &&
        selectedPhotosIds
            .contains(PhotoScanHandler.screenshotPhoto.value!.id)) {
      PhotoScanHandler.screenshotPhoto.value = null;
      if (screenshotPhotos.isNotEmpty) {
        var asset = screenshotPhotos.first;
        PhotoScanHandler.screenshotPhoto.value = asset;
      }
    }
    if (PhotoScanHandler.blurryPhoto.value != null &&
        selectedPhotosIds.contains(PhotoScanHandler.blurryPhoto.value!.id)) {
      PhotoScanHandler.blurryPhoto.value = null;
      if (blurryPhotos.isNotEmpty) {
        var asset = blurryPhotos.first;
        PhotoScanHandler.blurryPhoto.value = asset;
      }
    }
    PhotoScanHandler.similarPhotos
        .removeWhere((element) => selectedPhotosIds.contains(element.id));
    if (PhotoScanHandler.locationPhoto.value != null &&
        selectedPhotosIds.contains(PhotoScanHandler.locationPhoto.value!.id)) {
      PhotoScanHandler.locationPhoto.value = null;
    }
    if (similarPhotos.isNotEmpty) {
      for (var group in ImagePickerUtil.similarPhotos) {
        for (var asset in group) {
          PhotoScanHandler.similarPhotos.add(asset);
          if (PhotoScanHandler.similarPhotos.length == 4) {
            break;
          }
        }
      }
    }

    PhotoScanHandler.peoplePhotos
        .removeWhere((element) => selectedPhotosIds.contains(element.id));
    if (peoplePhotos.isNotEmpty) {
      for (var personPhotos in peoplePhotos) {
        PhotoScanHandler.peoplePhotos.add(personPhotos);
        if (PhotoScanHandler.peoplePhotos.length == 2) {
          break;
        }
      }
    }

    peopleResult
        .removeWhere((element) => selectedPhotosIds.contains(element.assetsId));
    peopleSize.value =
        peopleResult.fold(0, (sum, image) => sum + image.fileSize);
    screenshotResult
        .removeWhere((element) => selectedPhotosIds.contains(element.assetsId));
    screenshotsSize.value =
        screenshotResult.fold(0, (sum, image) => sum + image.fileSize);
    blurryResult
        .removeWhere((element) => selectedPhotosIds.contains(element.assetsId));
    blurrySize.value =
        blurryResult.fold(0, (sum, image) => sum + image.fileSize);
    similarResult.map((group) => group.images.removeWhere(
        (element) => selectedPhotosIds.contains(element.assetsId)));
    similarPhotoCount.value =
        similarResult.fold(0, (sum, group) => sum + group.groupImageCount);
    similarPhotosSize.value =
        similarResult.fold(0, (sum, group) => sum + group.groupFileSize);

    debugPrint("ImagePickerUtil3  updatePhotoData $selectedPhotosIds");
    CalendarState.removePhotosData(selectedPhotosIds);
  }

  // 更新照片数据
  static Future<void> updatePhotos(
      List<Map<String, dynamic>> photoGroups) async {
    // clearAllPhotos();

    for (var group in photoGroups) {
      String type = group['type'] as String;
      Map<dynamic, dynamic> groupData = group['group'] as Map<dynamic, dynamic>;
      List<dynamic> photos = groupData['photos'] as List<dynamic>;
      int totalSize = groupData['totalSize'] as int;
      int count = groupData['count'] as int;

      switch (type) {
        case 'screenshots':
          screenshotPhotos.clear();
          screenshotPhotos.value = await _convertToAssetEntities(photos);
          screenshotsSize.value = totalSize;
          break;
        case 'similar':
          similarPhotos.add(await _convertToAssetEntities(photos));
          similarPhotosSize.value = totalSize;
          similarPhotoCount.value = count;
          break;
        case 'location':
          String location = group['name'] as String;
          locationPhotos[location] = await _convertToAssetEntities(photos);
          locationsSize.value = totalSize;
          break;
        case 'people':
          peoplePhotos.clear();
          peoplePhotos.value = await _convertToAssetEntities(photos);
          peopleSize.value = totalSize;
          break;
        case 'blurry':
          blurryPhotos.clear();
          blurryPhotos.value = await _convertToAssetEntities(photos);
          blurrySize.value = totalSize;
          break;
      }
    }
  }

  static Future<void> newUpdatePhotoGroups(
      String type, List<ClassifiedImageGroup> photoGroup) async {
    if (type == 'similar') {
      // similarPhotos.clear();
      for (var group in photoGroup) {
        similarPhotos.add(await _newConvertToAssetEntities(group.images));
        similarPhotosSize.value += group.groupFileSize;
        similarPhotoCount.value += group.groupImageCount;
      }
    }
  }

  static Future<void> newUpdatePhotoGroup(
      String type, List<ClassifiedImage> photoGroup) async {
    switch (type) {
      case 'screenshots':
        screenshotPhotos.clear();
        screenshotPhotos.value = await _newConvertToAssetEntities(photoGroup);
        screenshotsSize.value =
            photoGroup.fold(0, (sum, image) => sum + image.fileSize);

        break;

        // case 'location':
        //   String location = group['name'] as String;
        //   locationPhotos[location] = await _convertToAssetEntities(photos);
        //   locationsSize.value = totalSize;
        break;
      case 'people':
        peoplePhotos.clear();
        peoplePhotos.assignAll(await _newConvertToAssetEntities(photoGroup));
        peopleSize.value =
            photoGroup.fold(0, (sum, image) => sum + image.fileSize);
        break;
      case 'blurry':
        blurryPhotos.clear();
        blurryPhotos.value = await _newConvertToAssetEntities(photoGroup);
        blurrySize.value =
            photoGroup.fold(0, (sum, image) => sum + image.fileSize);
        break;
    }
  }

  // 将原始照片数据转换为 AssetEntity 列表
  static Future<List<AssetEntity>> _convertToAssetEntities(
      List<dynamic> photos) async {
    List<AssetEntity> entities = [];
    for (var photo in photos) {
      final entity = await AssetEntity.fromId(photo['id'] as String);
      if (entity != null) {
        entities.add(entity);
      }
    }
    return entities;
  }

  // 新版将原始照片数据转换为AssetsEntity列表
  static Future<List<AssetEntity>> _newConvertToAssetEntities(
      List<ClassifiedImage> photos) async {
    List<AssetEntity> entities = [];
    for (var photo in photos) {
      final entity = await AssetEntity.fromId(photo.assetsId);
      if (entity != null) {
        entities.add(entity);
      }
    }
    return entities;
  }

  //申请权限
  static Future<bool> requestPermissionExtend() async {
    final PermissionState ps = await PhotoManager.requestPermissionExtend(
        requestOption: const PermissionRequestOption(
            androidPermission: AndroidPermission(
      type: permissionType,
      mediaLocation: false,
    )));
    return ps.hasAccess;
  }

  //判断是否有权限
  static Future<bool> hasPermission() async {
    final PermissionState ps = await PhotoManager.getPermissionState(
        requestOption: const PermissionRequestOption(
            androidPermission: AndroidPermission(
      type: permissionType,
      mediaLocation: false,
    )));
    return ps.hasAccess;
  }

  static String formatFileSize(int bytes, {int decimals = 2}) {
    if (bytes <= 0) return '0 B';

    const suffixes = ['B', 'KB', 'MB', 'GB', 'TB'];
    var i = (log(bytes) / log(1024)).floor();

    // 确保不超过数组范围
    i = i < suffixes.length ? i : suffixes.length - 1;

    // 计算实际大小
    final size = bytes / pow(1024, i);

    // 格式化数字，处理小数点位数
    String sizeStr;
    if (size >= 100) {
      // 大于100的只保留整数
      sizeStr = size.round().toString();
    } else {
      // 小于100的保留指定小数位
      sizeStr = size.toStringAsFixed(decimals);
    }

    // 移除末尾的0和不必要的小数点
    sizeStr = sizeStr.replaceAll(RegExp(r'\.?0+$'), '');

    return '$sizeStr ${suffixes[i]}';
  }

  /// 直接转换为 GB 单位
  static String formatToGB(int bytes, {int decimals = 2}) {
    if (bytes <= 0) return '0 GB';

    final gb = bytes / pow(1024, 3); // 直接转换为 GB

    // 格式化数字，处理小数点位数
    String sizeStr;
    if (gb >= 100) {
      // 大于100的只保留整数
      sizeStr = gb.round().toString();
    } else {
      // 小于100的保留指定小数位
      sizeStr = gb.toStringAsFixed(decimals);
    }

    // 移除末尾的0和不必要的小数点
    sizeStr = sizeStr.replaceAll(RegExp(r'\.?0+$'), '');

    return '$sizeStr GB';
  }

  /// 格式化存储大小
  static String formatSize(double sizeInGB) {
    if (sizeInGB < 0.001) {
      // 小于 1MB
      return '${(sizeInGB * 1024 * 1024).toStringAsFixed(1)} ';
    } else if (sizeInGB < 1) {
      // 小于 1GB
      return '${(sizeInGB * 1024).toStringAsFixed(1)} ';
    } else if (sizeInGB < 1024) {
      // 小于 1TB
      return '${sizeInGB.toStringAsFixed(1)} ';
    } else {
      // 大于等于 1TB
      return '${(sizeInGB / 1024).toStringAsFixed(1)} ';
    }
  }

  static Future<List<AssetEntity>> loadAssetsPaged({
    int page = 0,
    int pageSize = 9,
    bool sortByDate = true,
  }) async {
    final PermissionState result = await PhotoManager.requestPermissionExtend();
    if (!result.hasAccess) return [];

    // 获取相册列表（这里只取第一个相册）
    final List<AssetPathEntity> albums = await PhotoManager.getAssetPathList(
      type: RequestType.common,
      filterOption: FilterOptionGroup(
        orders: [
          OrderOption(
            type: OrderOptionType.createDate,
            asc: !sortByDate,
          ),
        ],
      ),
    );

    if (albums.isEmpty) return [];

    final AssetPathEntity album = albums.first;

    // 分页加载当前页的照片
    final List<AssetEntity> assets = await album.getAssetListPaged(
      page: page,
      size: pageSize,
    );

    return assets;
  }

  // 加载图片资源
  static Future<List<AssetEntity>> loadAssets({bool sortByDate = true}) async {
    final PermissionState result = await PhotoManager.requestPermissionExtend();
    if (!result.hasAccess) return [];

    if (androidPhotos.isNotEmpty) return androidPhotos;
    // 选择相册
    final List<AssetPathEntity> albums = await PhotoManager.getAssetPathList(
      type: RequestType.common,
      filterOption: FilterOptionGroup(
        orders: [
          // 根据创建日期排序，降序（最新的在前）
          OrderOption(
            type: OrderOptionType.createDate,
            asc: !sortByDate, // 是否升序排列
          ),
        ],
      ),
    );

    int assetCount = await albums.first.assetCountAsync;

    debugPrint("ImagePickerUtil loadAssets assetCount $assetCount");
    androidPhotos = await albums.first.getAssetListPaged(
      page: 0,
      size: assetCount,
    );

    return androidPhotos;
  }
}