import 'package:flutter/foundation.dart'; import 'package:hooks_riverpod/hooks_riverpod.dart'; import 'package:immich_mobile/modules/home/services/asset.service.dart'; import 'package:immich_mobile/modules/home/services/asset_cache.service.dart'; import 'package:immich_mobile/shared/services/device_info.service.dart'; import 'package:collection/collection.dart'; import 'package:intl/intl.dart'; import 'package:openapi/api.dart'; import 'package:photo_manager/photo_manager.dart'; class AssetNotifier extends StateNotifier> { final AssetService _assetService; final AssetCacheService _assetCacheService; final DeviceInfoService _deviceInfoService = DeviceInfoService(); AssetNotifier(this._assetService, this._assetCacheService) : super([]); getAllAsset() async { if (_assetCacheService.isValid() && state.isEmpty) { state = await _assetCacheService.getAssetsAsync(); } var allAssets = await _assetService.getAllAsset(); if (allAssets != null) { state = allAssets; _assetCacheService.putAssets(allAssets); } } clearAllAsset() { state = []; } onNewAssetUploaded(AssetResponseDto newAsset) { state = [...state, newAsset]; } deleteAssets(Set deleteAssets) async { var deviceInfo = await _deviceInfoService.getDeviceInfo(); var deviceId = deviceInfo["deviceId"]; var deleteIdList = []; // Delete asset from device for (var asset in deleteAssets) { // Delete asset on device if present if (asset.deviceId == deviceId) { var localAsset = await AssetEntity.fromId(asset.deviceAssetId); if (localAsset != null) { deleteIdList.add(localAsset.id); } } } try { await PhotoManager.editor.deleteWithIds(deleteIdList); } catch (e) { debugPrint("Delete asset from device failed: $e"); } // Delete asset on server List? deleteAssetResult = await _assetService.deleteAssets(deleteAssets); if (deleteAssetResult == null) { return; } for (var asset in deleteAssetResult) { if (asset.status == DeleteAssetStatus.SUCCESS) { state = state.where((immichAsset) => immichAsset.id != asset.id).toList(); } } } } final assetProvider = StateNotifierProvider>((ref) { return AssetNotifier( ref.watch(assetServiceProvider), ref.watch(assetCacheServiceProvider)); }); final assetGroupByDateTimeProvider = StateProvider((ref) { var assets = ref.watch(assetProvider); assets.sortByCompare( (e) => DateTime.parse(e.createdAt), (a, b) => b.compareTo(a), ); return assets.groupListsBy( (element) => DateFormat('y-MM-dd') .format(DateTime.parse(element.createdAt).toLocal()), ); }); final assetGroupByMonthYearProvider = StateProvider((ref) { var assets = ref.watch(assetProvider); assets.sortByCompare( (e) => DateTime.parse(e.createdAt), (a, b) => b.compareTo(a), ); return assets.groupListsBy( (element) => DateFormat('MMMM, y') .format(DateTime.parse(element.createdAt).toLocal()), ); });