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_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 {
    if (selectedFilesSize.value > 1024) {
      return "${(selectedFilesSize.value / 1024).toStringAsFixed(2)}MB";
    } else if (selectedFilesSize.value > 1024 * 1024) {
      return "${(selectedFilesSize.value / 1024 / 1024).toStringAsFixed(2)}GB";
    } else {
      return "${(selectedFilesSize.value).toStringAsFixed(2)}KB";
    }
  }

  // 获取选中的文件大小
  Future<void> updateSelectedFilesSize() async {
    // 如果没有选中的文件，直接返回
    if (selectedFileCount.value == 0) {
      selectedFilesSize.value = 0;
      return;
    }
    double totalSize = 0;
    for (var group in photoGroups) {
      for (int i = 0; i < group.images.length; i++) {
        if (group.selectedImages[i]) {
          // 检查缓存
          final assetId = group.images[i].id;
          if (ImagePickerUtil.fileSizeCache.containsKey(assetId)) {
            totalSize += ImagePickerUtil.fileSizeCache[assetId]!;
          } else {
            final file = await group.images[i].file;
            if (file != null) {
              final size = (await file.length()) / 1024; // 转换为KB
              totalSize += size;
              ImagePickerUtil.fileSizeCache[assetId] = size;
            }
          }
        }
      }
    }
    selectedFilesSize.value = totalSize;
    PhotoManager.clearFileCache();
  }

  // 切换图片选中状态
  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();

          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();
  }
}