Clean/lib/base/base_photo_controller.dart

import 'package:clean/base/base_controller.dart';
import 'package:clean/data/repositories/user_repository.dart';
import 'package:clean/dialog/photo_delete_finish_dialog.dart';
import 'package:clean/dialog/photo_deleting_dialog.dart';
import 'package:clean/module/store/store_view.dart';
import 'package:flutter/cupertino.dart';
import 'package:flutter_smart_dialog/flutter_smart_dialog.dart';
import 'package:get/get.dart';
import 'package:wechat_assets_picker/wechat_assets_picker.dart';
import 'package:clean/module/image_picker/image_picker_util.dart';
import 'package:clean/module/people_photo/photo_group.dart';
import 'package:clean/utils/toast_util.dart';
import 'package:clean/data/bean/photos_type.dart';
import 'package:clean/module/photo_preview/photo_preview_view.dart';

abstract class BasePhotoController extends BaseController {
  final RxList<PhotoGroup> photoGroups = <PhotoGroup>[].obs;
  final RxDouble selectedFilesSize = 0.0.obs;
  RxInt selectedFileCount = 0.obs;
  final RxSet<String> selectedPhotosIds = <String>{}.obs;


  // 将selectedFilesSize转成String类型,然后单位转换,如果超过1MB,则转成MB,超过1GB,则转成GB,否则KB
  String get selectedFilesSizeString {
    print("BasePhotoController  selectedFilesSize.value ${selectedFilesSize.value}");
    final double sizeInKB = selectedFilesSize.value;

    if (sizeInKB >= 1024 * 1024) {  // 先检查最大单位（GB）
      return "${(sizeInKB / (1024 * 1024)).toStringAsFixed(2)}GB";
    } else if (sizeInKB >= 1024) {  // 然后检查MB
      return "${(sizeInKB / 1024).toStringAsFixed(2)}MB";
    } else {  // 最后是KB
      return "${sizeInKB.toStringAsFixed(2)}KB";
    }
  }

  Future<void> updateSelectedFilesSize() async {


    if (selectedPhotosIds.isEmpty) {

      selectedFilesSize.value = 0;

      print("BasePhotoController updateSelectedFilesSize isEmpty");

      return;
    }

    print("BasePhotoController updateSelectedFilesSize");

    double totalSize = 0;

    // 创建一个 Future 列表，用于并行获取文件大小
    List<Future<double>> sizeFutures =
        selectedPhotosIds.map((id) => getFileSize(id)).toList();

    // 等待所有文件大小计算完成
    List<double> sizes = await Future.wait(sizeFutures);
    // 计算总大小
    totalSize = sizes.fold(0, (sum, size) => sum + size);
    if (selectedPhotosIds.isEmpty) {
      selectedFilesSize.value = 0;
      return;
    }
    selectedFilesSize.value = totalSize; // 更新总大小
    PhotoManager.clearFileCache();
  }

  Future<double> getFileSize(String assetId) async {
    // 先检查缓存
    if (ImagePickerUtil.fileSizeCache.containsKey(assetId)) {
      return ImagePickerUtil.fileSizeCache[assetId]!;
    }

    final entity = await AssetEntity.fromId(assetId);
    if (entity != null) {
      final file = await entity.file;
      if (file != null) {
        double size = await file.length() / 1024; // 转换为 KB
        ImagePickerUtil.fileSizeCache[assetId] = size; // 缓存文件大小
        file.delete();
        return size;
      }
    }
    return 0;
  }


  // 切换图片选中状态
  Future<void> toggleImageSelection(
      List<AssetEntity> groupTitle, int imageIndex) async {
    print("BasePhotoController toggleImageSelection");
    final group = getGroupByImages(groupTitle);
    final image = group.images[imageIndex];
    final selected = !group.selectedImages[imageIndex];

    group.selectedImages[imageIndex] = selected;
    updateSelectedPhotosIds(image.id, selected);

    group.isSelected.value = group.selectedImages.every((selected) => selected);

    selectedFileCount.value = selectedPhotosIds.length;

    // 更新文件大小
    if (selected) {
      // 如果选中，增加文件大小
      if (ImagePickerUtil.fileSizeCache.containsKey(image.id)) {
        selectedFilesSize.value += ImagePickerUtil.fileSizeCache[image.id]!;
      } else {
        final file = await image.file;
        if (file != null) {
          selectedFilesSize.value += (await file.length()) / 1024; // 转换为KB
        }
      }
    } else {
      // 如果取消选中，减少文件大小
      if (ImagePickerUtil.fileSizeCache.containsKey(image.id)) {
        selectedFilesSize.value -= ImagePickerUtil.fileSizeCache[image.id]!;
      } else {
        final file = await image.file;
        if (file != null) {
          selectedFilesSize.value -= (await file.length()) / 1024; // 转换为KB
        }
      }
    }

    PhotoManager.clearFileCache();
  }

  void clickImage(List<AssetEntity> images, int imageIndex, PhotosType type) {
    print("BasePhotoController clickImage");
    final group = getGroupByImages(images);
    final image = group.images[imageIndex];
    PhotoPreviewPage.start(type, image.id);
  }

  // 切换图片组选中状态
  void toggleGroupSelection(List<AssetEntity> imagesList) {
    final group = getGroupByImages(imagesList);
    final newValue = !group.isSelected.value;
    group.toggleSelectAll(newValue);

    for (var image in group.images) {
      updateSelectedPhotosIds(image.id, newValue);
    }
    selectedFileCount.value = selectedPhotosIds.length;

    updateSelectedFilesSize();
  }

  PhotoGroup getGroupByImages(List<AssetEntity> images) {
    final imageIds = images.map((img) => img.id).toSet();
    return photoGroups.firstWhere(
        (group) => group.images.every((image) => imageIds.contains(image.id)));
  }

  void updateSelectedPhotosIds(String photoId, bool isSelected) {
    if (isSelected) {
      selectedPhotosIds.add(photoId);
    } else {
      selectedPhotosIds.remove(photoId);
    }
  }

  // 恢复选中状态
  void restoreSelections() async {
    final selectedIds = selectedPhotosIds.toSet();
    for (var group in photoGroups) {
      for (int i = 0; i < group.images.length; i++) {
        group.selectedImages[i] = selectedIds.contains(group.images[i].id);
      }
      group.isSelected.value =
          group.selectedImages.every((selected) => selected);
    }
    selectedFileCount.value = selectedIds.length;
    if (selectedIds.isEmpty) {
      selectedFilesSize.value = 0;
      return;
    }
    await updateSelectedFilesSize();
  }

  @override
  void onInit() {
    print("BasePhotoController onInit");
    super.onInit();
    loadPhotos();
    restoreSelections();
  }

  void loadPhotos();

  // 点击删除
  void clickDelete() async {
    if (userRepository.isVip()) {
      if (selectedPhotosIds.isNotEmpty) {
        photoDeletingDialog();
        final assetsToDelete = photoGroups
            .expand((group) => group.images
                .where((asset) => selectedPhotosIds.contains(asset.id)))
            .toList();

        final List<String> result = await PhotoManager.editor.deleteWithIds(
          assetsToDelete.map((e) => e.id).toList(),
        );

        if (result.length == selectedPhotosIds.length) {
          for (var group in photoGroups) {
            group.images.removeWhere(
                (element) => selectedPhotosIds.contains(element.id));
          }

          ImagePickerUtil.updatePhotoGroupDate(
              getPhotosType(), selectedPhotosIds);

          selectedPhotosIds.clear();
          selectedFileCount.value = selectedPhotosIds.length;
          print(
              "BasePhotoController clickDelete selectedPhotosIds $selectedPhotosIds");
          updateSelectedFilesSize();

          ToastUtil.show("Delete success");
          Future.delayed(Duration(seconds: 2), () {
            SmartDialog.dismiss(tag: 'photoDeletingDialog');
            photoDeleteFinishDialog();
          });
        } else {
          SmartDialog.dismiss(tag: 'photoDeletingDialog');
          ToastUtil.show("Delete failed");
        }
      }
    } else {
      StorePage.start();
    }
  }

  void updateSelections(Set<String> selectedIds) {
    print(
        "BasePhotoController updateSelections selectedIds $selectedIds, selectedPhotosIds $selectedPhotosIds getPhotosType() ${getPhotosType()}");
    // selectedId如果是selectedPhotosIds对象，那么selectedPhotosIds.assignAll(selectedIds)会清空
    // selectedPhotosIds.assignAll(selectedIds);

    selectedPhotosIds.assignAll(Set.from(selectedIds));
    print(
        "BasePhotoController updateSelections selectedIds $selectedIds, selectedPhotosIds $selectedPhotosIds getPhotosType() ${getPhotosType()}");

    switch (getPhotosType()) {
      case PhotosType.peoplePhotos:
        ImagePickerUtil.selectedPeoplePhotosIds.assignAll(selectedPhotosIds);
        break;
      case PhotosType.screenshots:
        ImagePickerUtil.selectedScreenshotPhotosIds
            .assignAll(selectedPhotosIds);
        break;
      case PhotosType.similarPhotos:
        ImagePickerUtil.selectedSimilarPhotosIds.assignAll(selectedPhotosIds);
        break;
      case PhotosType.locationPhotos:
        ImagePickerUtil.selectedLocationPhotosIds.assignAll(selectedPhotosIds);
        break;
      case PhotosType.blurryPhotos:
        ImagePickerUtil.selectedBlurryPhotosIds.assignAll(selectedPhotosIds);
        break;
    }
  }

  PhotosType getPhotosType();

  // 将photoGroups中所有的图片返回
  List<AssetEntity> getAllPhotos() {
    return photoGroups.expand((group) => group.images).toList();
  }

  @override
  void onClose() {
    super.onClose();
    PhotoManager.clearFileCache();
  }
}