Clean/lib/module/analysis/analysis_controller.dart

import 'dart:math';

import 'package:classify_photo/classify_photo.dart';
import 'package:clean/base/base_controller.dart';
import 'package:clean/module/analysis/analysis_state.dart';
import 'package:clean/utils/expand.dart';
import 'package:flutter/cupertino.dart';
import 'package:flutter_screenutil/flutter_screenutil.dart';
import 'package:get/get.dart';
import 'package:get/get_core/src/get_main.dart';
import 'package:get/get_rx/src/rx_types/rx_types.dart';
import 'package:permission_handler/permission_handler.dart';
import 'package:wechat_camera_picker/wechat_camera_picker.dart';

import '../../model/asset_info.dart';
import '../../utils/file_utils.dart';
import '../../utils/image_util.dart';
import '../../utils/toast_util.dart';
import '../image_picker/image_picker_assets.dart';

class AnalysisController extends BaseController {
  // 是否为编辑状态
  RxBool isEdit = false.obs;

  // 使用共享状态
  RxList<AssetInfo> imageList = AnalysisState.imageList;

  RxMap<String, List<AssetInfo>> assetsByMonth = AnalysisState.assetsByMonth;

  // 获取月份数量
  int get monthCount => assetsByMonth.length;

  // 获取总图片数量
  int get totalAssetCount =>
      assetsByMonth.values.fold(0, (sum, list) => sum + list.length);

  // 存储选中的图片ID
  final RxSet<String> selectedAssets = <String>{}.obs;

  // 是否全选
  RxBool isAllSelected = false.obs;

  // 选中图片的总容量（字节）
  final RxInt selectedTotalSize = 0.obs;

  @override
  void onInit() {
    // TODO: implement onInit
    super.onInit();

    loadAssets();
  }

  // 加载并分组图片
  Future<void> loadAssets() async {
    var newImageList = <AssetInfo>[];
    newImageList = await FileUtils.getAllAssets(FileType.analysis);
    AnalysisState.imageList.value = newImageList;
    AnalysisState.updateMonthlyAssets();
    if (newImageList.isEmpty) {
      isEdit.value = false;
      return;
    }

    // 清空现有数据
    assetsByMonth.clear();

    // 按月份分组
    for (var asset in newImageList) {
      final monthKey = ImageUtil.getMonthKey(asset.createDateTime);
      if (!assetsByMonth.containsKey(monthKey)) {
        assetsByMonth[monthKey] = [];
      }
      assetsByMonth[monthKey]!.add(asset);
      asset.dateTitle = ImageUtil.formatMonthKey(monthKey);
    }

    newImageList.sort((a, b) => b.createDateTime.compareTo(a.createDateTime));

    AnalysisState.imageList.value = newImageList;
    AnalysisState.updateMonthlyAssets();

    // 对每个月份内的图片按时间排序（新的在前）
    assetsByMonth.forEach((key, list) {
      list.sort((a, b) => b.createDateTime.compareTo(a.createDateTime));
    });

    // 打印分组结果
    assetsByMonth.forEach((key, assets) {
      print('${ImageUtil.formatMonthKey(key)}: ${assets.length} photos');
    });
  }

  // 上传按钮点击
  void uploadBtnClick() {
    showCupertinoModalPopup(
      context: Get.context!,
      builder: (context) {
        return CupertinoActionSheet(
          actions: <Widget>[
            //操作按钮集合
            CupertinoActionSheetAction(
              onPressed: () {
                Navigator.pop(context);
                openGallery();
              },
              child: Text(
                'Upload from Gallery',
                style: TextStyle(
                  color: "#007AFF".color,
                  fontWeight: FontWeight.w500,
                  fontSize: 16.sp,
                ),
              ),
            ),
            CupertinoActionSheetAction(
              onPressed: () {
                Navigator.pop(context);
                openCamera();
              },
              child: Text(
                'Take and Upload',
                style: TextStyle(
                  color: "#007AFF".color,
                  fontWeight: FontWeight.w500,
                  fontSize: 16.sp,
                ),
              ),
            ),
          ],
          cancelButton: CupertinoActionSheetAction(
            //取消按钮
            onPressed: () {
              Navigator.pop(context);
            },
            child: Text(
              'Cancel',
              style: TextStyle(
                color: "#007AFF".color,
                fontWeight: FontWeight.w500,
                fontSize: 16.sp,
              ),
            ),
          ),
        );
      },
    );
  }

// 保存并刷新图片列表
  Future<void> saveAndRefreshAssets(List<AssetEntity> assets) async {
    for (var asset in assets) {
      await FileUtils.saveAsset(FileType.analysis, asset);
    }
    // 重新加载图片列表
    loadAssets();
  }

  // 选择/取消选择图片
  void toggleSelectAsset(String assetId) {
    final asset = imageList.firstWhere((asset) => asset.id == assetId);

    if (selectedAssets.contains(assetId)) {
      selectedAssets.remove(assetId);
      if (asset.size != null) {
        selectedTotalSize.value -= asset.size!;
      }
    } else {
      selectedAssets.add(assetId);

      if (asset.size != null) {
        selectedTotalSize.value += asset.size!;
      }
    }

    // 更新全选状态
    isAllSelected.value = selectedAssets.length == imageList.length;
  }

  // 全选/取消全选
  void toggleSelectAll() {
    if (isAllSelected.value) {
      selectedAssets.clear();
      selectedTotalSize.value = 0;
    } else {
      selectedAssets.addAll(imageList.map((asset) => asset.id));
      selectedTotalSize.value = imageList.fold(
          0, (sum, asset) => sum + (asset.size != null ? asset.size! : 0));
    }
    isAllSelected.value = !isAllSelected.value;
  }

  // 退出编辑模式时清空选择
  void exitEditMode() {
    isEdit.value = false;
    selectedAssets.clear();
    isAllSelected.value = false;
    selectedTotalSize.value = 0;
  }

  // 删除文件
  void deleteBtnClick() {
    // 获取要删除的资产
    final assetsToDelete =
        imageList.where((asset) => selectedAssets.contains(asset.id)).toList();

    for (var asset in assetsToDelete) {
      FileUtils.deleteAsset(FileType.analysis, asset.id.substring(0, 36));
    }

    selectedTotalSize.value = 0;
    loadAssets();
  }

  // 格式化文件大小显示
  String formatFileSize(int bytes) {
    if (bytes <= 0) return "Delete";

    final units = ['B', 'KB', 'MB', 'GB'];
    int digitGroups = (log(bytes) / log(1024)).floor();

    if (bytes == 0) {
      return "Delete";
    } else {
      return "Delete(${(bytes / pow(1024, digitGroups)).toStringAsFixed(1)} ${units[digitGroups]})";
    }
  }

// 开启图库
  Future<void> openGallery() async {
    var status = await Permission.photos.status;
    if (status == PermissionStatus.granted) {
      List<AssetEntity> assets = <AssetEntity>[];
      for (var asset in imageList) {
        var newAsset = await asset.toAssetEntity();
        if (newAsset != null) {
          assets.add(newAsset);
        }
      }
      List<AssetEntity>? pickList = await ImagePickAssets.pick();
      if (pickList != null && pickList.isNotEmpty) {
        await saveAndRefreshAssets(pickList);
        final asset = pickList.first;
        final file = await asset.originFile;
        // final exifInfo = await ClassifyPhoto().getPhotoExif(file!.path);
        // print(exifInfo);
      }
    } else {
      ToastUtil.show("请先开启权限");
    }
  }

// 开启相机
  Future<void> openCamera() async {
    final entity = await CameraPicker.pickFromCamera(
      Get.context!,
      pickerConfig: const CameraPickerConfig(),
    );
    if (entity != null) {
      await saveAndRefreshAssets([entity]);
    }
  }
}