import 'package:immich_mobile/domain/models/asset/base_asset.model.dart'; import 'package:immich_mobile/domain/models/exif.model.dart'; import 'package:immich_mobile/infrastructure/repositories/local_asset.repository.dart'; import 'package:immich_mobile/infrastructure/repositories/remote_asset.repository.dart'; import 'package:immich_mobile/infrastructure/repositories/storage.repository.dart'; import 'package:immich_mobile/infrastructure/utils/exif.converter.dart'; import 'package:immich_mobile/repositories/local_files_manager.repository.dart'; import 'package:immich_mobile/services/app_settings.service.dart'; import 'package:logging/logging.dart'; import 'package:platform/platform.dart'; class AssetService { final AppSettingsService _appSettingsService; final RemoteAssetRepository _remoteAssetRepository; final DriftLocalAssetRepository _localAssetRepository; final LocalFilesManagerRepository _localFilesManager; final StorageRepository _storageRepository; final Platform _platform; final Logger _logger; const AssetService({ required AppSettingsService appSettingsService, required RemoteAssetRepository remoteAssetRepository, required DriftLocalAssetRepository localAssetRepository, required LocalFilesManagerRepository localFilesManager, required StorageRepository storageRepository, required Logger logger, }) : _appSettingsService = appSettingsService, _remoteAssetRepository = remoteAssetRepository, _localAssetRepository = localAssetRepository, _localFilesManager = localFilesManager, _storageRepository = storageRepository, _platform = const LocalPlatform(), _logger = logger; Stream watchAsset(BaseAsset asset) { final id = asset is LocalAsset ? asset.id : (asset as RemoteAsset).id; return asset is LocalAsset ? _localAssetRepository.watchAsset(id) : _remoteAssetRepository.watchAsset(id); } Future getRemoteAsset(String id) { return _remoteAssetRepository.get(id); } Future> getStack(RemoteAsset asset) async { if (asset.stackId == null) { return []; } return _remoteAssetRepository.getStackChildren(asset).then((assets) { // Include the primary asset in the stack as the first item return [asset, ...assets]; }); } Future getExif(BaseAsset asset) async { if (!asset.hasRemote) { return null; } final id = asset is LocalAsset ? asset.remoteId! : (asset as RemoteAsset).id; return _remoteAssetRepository.getExif(id); } Future getAspectRatio(BaseAsset asset) async { bool isFlipped; double? width; double? height; if (asset.hasRemote) { final exif = await getExif(asset); isFlipped = ExifDtoConverter.isOrientationFlipped(exif?.orientation); width = exif?.width ?? asset.width?.toDouble(); height = exif?.height ?? asset.height?.toDouble(); } else if (asset is LocalAsset) { isFlipped = _platform.isAndroid && (asset.orientation == 90 || asset.orientation == 270); width = asset.width?.toDouble(); height = asset.height?.toDouble(); } else { isFlipped = false; } final orientedWidth = isFlipped ? height : width; final orientedHeight = isFlipped ? width : height; if (orientedWidth != null && orientedHeight != null && orientedHeight > 0) { return orientedWidth / orientedHeight; } return 1.0; } Future> getPlaces() { return _remoteAssetRepository.getPlaces(); } Future<(int local, int remote)> getAssetCounts() async { return (await _localAssetRepository.getCount(), await _remoteAssetRepository.getCount()); } Future getLocalHashedCount() { return _localAssetRepository.getHashedCount(); } Future handleRemoteTrashChanges(Iterable<({String checksum, DateTime? deletedAt})> syncItems) async { if (_platform.isAndroid && _appSettingsService.getSetting(AppSettingsEnum.manageLocalMediaAndroid)) { final trashedItems = syncItems.where((item) => item.deletedAt != null); if (trashedItems.isNotEmpty) { final trashedAssetsChecksums = trashedItems.map((syncItem) => syncItem.checksum); final localAssetsToTrash = await _localAssetRepository.getByChecksums(trashedAssetsChecksums); if (localAssetsToTrash.isNotEmpty) { final mediaUrls = await Future.wait( localAssetsToTrash.map( (localAsset) => _storageRepository.getAssetEntityForAsset(localAsset).then((e) => e?.getMediaUrl()), ), ); _logger.fine("Moving to trash ${mediaUrls.join(", ")} assets"); await _localFilesManager.moveToTrash(mediaUrls.nonNulls.toList()); } } final modifiedItems = syncItems.where((e) => e.deletedAt == null); if (modifiedItems.isNotEmpty) { final modifiedChecksums = modifiedItems.map((syncItem) => syncItem.checksum); final remoteAssetsToRestore = await _remoteAssetRepository.getByChecksums( modifiedChecksums, isTrashed: true, ); if (remoteAssetsToRestore.isNotEmpty) { _logger.fine("Restoring from trash ${remoteAssetsToRestore.map((e) => e.name).join(", ")} assets"); for (final remoteAsset in remoteAssetsToRestore) { await _localFilesManager.restoreFromTrash(remoteAsset.name, remoteAsset.type.index); } } } } else { return Future.value(); } } }