import 'dart:async'; import 'package:flutter/material.dart'; import 'package:hooks_riverpod/hooks_riverpod.dart'; import 'package:immich_mobile/shared/models/asset.dart'; import 'package:immich_mobile/shared/models/exif_info.dart'; import 'package:immich_mobile/shared/models/store.dart'; import 'package:immich_mobile/shared/providers/api.provider.dart'; import 'package:immich_mobile/shared/providers/db.provider.dart'; import 'package:immich_mobile/shared/services/api.service.dart'; import 'package:immich_mobile/shared/services/sync.service.dart'; import 'package:immich_mobile/utils/openapi_extensions.dart'; import 'package:immich_mobile/utils/tuple.dart'; import 'package:isar/isar.dart'; import 'package:logging/logging.dart'; import 'package:openapi/api.dart'; final assetServiceProvider = Provider( (ref) => AssetService( ref.watch(apiServiceProvider), ref.watch(syncServiceProvider), ref.watch(dbProvider), ), ); class AssetService { final ApiService _apiService; final SyncService _syncService; final log = Logger('AssetService'); final Isar _db; AssetService( this._apiService, this._syncService, this._db, ); /// Checks the server for updated assets and updates the local database if /// required. Returns `true` if there were any changes. Future refreshRemoteAssets() async { final Stopwatch sw = Stopwatch()..start(); final int numOwnedRemoteAssets = await _db.assets .where() .remoteIdIsNotNull() .filter() .ownerIdEqualTo(Store.get(StoreKey.currentUser).isarId) .count(); final bool changes = await _syncService.syncRemoteAssetsToDb( () async => (await _getRemoteAssets(hasCache: numOwnedRemoteAssets > 0)) ?.map(Asset.remote) .toList(), ); debugPrint("refreshRemoteAssets full took ${sw.elapsedMilliseconds}ms"); return changes; } /// Returns `null` if the server state did not change, else list of assets Future?> _getRemoteAssets({ required bool hasCache, }) async { try { final etag = hasCache ? Store.tryGet(StoreKey.assetETag) : null; final Pair, String?>? remote = await _apiService.assetApi.getAllAssetsWithETag(eTag: etag); if (remote == null) { return null; } if (remote.second != null && remote.second != etag) { Store.put(StoreKey.assetETag, remote.second); } return remote.first; } catch (e, stack) { log.severe('Error while getting remote assets', e, stack); return null; } } Future?> deleteAssets( Iterable deleteAssets, ) async { try { final List payload = []; for (final asset in deleteAssets) { payload.add(asset.remoteId!); } return await _apiService.assetApi .deleteAsset(DeleteAssetDto(ids: payload)); } catch (e) { debugPrint("Error getAllAsset ${e.toString()}"); return null; } } /// Loads the exif information from the database. If there is none, loads /// the exif info from the server (remote assets only) Future loadExif(Asset a) async { a.exifInfo ??= await _db.exifInfos.get(a.id); // fileSize is always filled on the server but not set on client if (a.exifInfo?.fileSize == null) { if (a.isRemote) { final dto = await _apiService.assetApi.getAssetById(a.remoteId!); if (dto != null && dto.exifInfo != null) { final newExif = Asset.remote(dto).exifInfo!.copyWith(id: a.id); if (newExif != a.exifInfo) { if (a.isInDb) { _db.writeTxn(() => a.put(_db)); } else { debugPrint("[loadExif] parameter Asset is not from DB!"); } } } } else { // TODO implement local exif info parsing } } return a; } Future> updateAssets( List assets, UpdateAssetDto updateAssetDto, ) async { final List dtos = await Future.wait( assets.map( (a) => _apiService.assetApi.updateAsset(a.remoteId!, updateAssetDto), ), ); bool allInDb = true; for (int i = 0; i < assets.length; i++) { final dto = dtos[i], old = assets[i]; if (dto != null) { final remote = Asset.remote(dto); if (old.canUpdate(remote)) { assets[i] = old.updatedCopy(remote); } allInDb &= assets[i].isInDb; } } final toUpdate = allInDb ? assets : assets.where((e) => e.isInDb).toList(); await _syncService.upsertAssetsWithExif(toUpdate); return assets; } Future> changeFavoriteStatus( List assets, bool isFavorite, ) { return updateAssets(assets, UpdateAssetDto(isFavorite: isFavorite)); } Future> changeArchiveStatus(List assets, bool isArchive) { return updateAssets(assets, UpdateAssetDto(isArchived: isArchive)); } }