mirror of https://github.com/immich-app/immich.git
100 lines
4.5 KiB
Dart
100 lines
4.5 KiB
Dart
import 'package:drift/drift.dart';
|
|
import 'package:hooks_riverpod/hooks_riverpod.dart';
|
|
import 'package:immich_mobile/constants/constants.dart';
|
|
import 'package:immich_mobile/domain/models/asset/asset_metadata.model.dart';
|
|
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
|
|
import 'package:immich_mobile/extensions/drift_extensions.dart';
|
|
import 'package:immich_mobile/infrastructure/entities/local_asset.entity.dart';
|
|
import 'package:immich_mobile/infrastructure/repositories/db.repository.dart';
|
|
import 'package:immich_mobile/infrastructure/repositories/local_album.repository.dart';
|
|
import 'package:immich_mobile/platform/native_sync_api.g.dart';
|
|
import 'package:immich_mobile/providers/api.provider.dart';
|
|
import 'package:immich_mobile/providers/infrastructure/db.provider.dart';
|
|
import 'package:immich_mobile/providers/infrastructure/sync.provider.dart';
|
|
import 'package:immich_mobile/providers/user.provider.dart';
|
|
import 'package:logging/logging.dart';
|
|
// ignore: import_rule_openapi
|
|
import 'package:openapi/api.dart';
|
|
|
|
Future<void> syncCloudIds(ProviderContainer ref) async {
|
|
final db = ref.read(driftProvider);
|
|
// Populate cloud IDs for local assets that don't have one yet
|
|
await _populateCloudIds(db);
|
|
|
|
// Wait for remote sync to complete, so we have up-to-date asset metadata entries
|
|
await ref.read(syncStreamServiceProvider).sync();
|
|
|
|
// Fetch the mapping for backed up assets that have a cloud ID locally but do not have a cloud ID on the server
|
|
final currentUser = ref.read(currentUserProvider);
|
|
if (currentUser == null) {
|
|
Logger('migrateCloudIds').warning('Current user is null. Aborting cloudId migration.');
|
|
return;
|
|
}
|
|
|
|
final mappingsToUpdate = await _fetchCloudIdMappings(db, currentUser.id);
|
|
final assetApi = ref.read(apiServiceProvider).assetsApi;
|
|
for (final mapping in mappingsToUpdate) {
|
|
final mobileMeta = AssetMetadataUpsertItemDto(
|
|
key: AssetMetadataKey.mobileApp,
|
|
value: RemoteAssetMobileAppMetadata(cloudId: mapping.localAsset.cloudId, eTag: mapping.localAsset.eTag),
|
|
);
|
|
try {
|
|
await assetApi.updateAssetMetadata(mapping.remoteAssetId, AssetMetadataUpsertDto(items: [mobileMeta]));
|
|
} catch (error, stack) {
|
|
Logger('migrateCloudIds').warning('Failed to update metadata for asset ${mapping.remoteAssetId}', error, stack);
|
|
}
|
|
}
|
|
}
|
|
|
|
Future<void> _populateCloudIds(Drift drift) async {
|
|
final query = drift.localAssetEntity.selectOnly()
|
|
..addColumns([drift.localAssetEntity.id])
|
|
..where(drift.localAssetEntity.iCloudId.isNull());
|
|
final ids = await query.map((row) => row.read(drift.localAssetEntity.id)!).get();
|
|
final cloudMapping = await NativeSyncApi().getCloudIdForAssetIds(ids);
|
|
await DriftLocalAlbumRepository(drift).updateCloudMapping(cloudMapping);
|
|
}
|
|
|
|
typedef _CloudIdMapping = ({String remoteAssetId, LocalAsset localAsset});
|
|
|
|
Future<List<_CloudIdMapping>> _fetchCloudIdMappings(Drift drift, String userId) async {
|
|
final createdAt = drift.localAssetEntity.createdAt.strftime('%s');
|
|
final adjustmentTime = coalesce([drift.localAssetEntity.adjustmentTime.strftime('%s'), const Constant('0')]);
|
|
final latitude = coalesce([
|
|
drift.localAssetEntity.latitude,
|
|
const Constant(0.0),
|
|
]).truncateTo(2).cast(DriftSqlType.string);
|
|
final longitude = coalesce([
|
|
drift.localAssetEntity.longitude,
|
|
const Constant(0.0),
|
|
]).truncateTo(2).cast(DriftSqlType.string);
|
|
|
|
final delimiter = const Constant(kUploadETagDelimiter);
|
|
final eTag = createdAt + delimiter + adjustmentTime + delimiter + latitude + delimiter + longitude;
|
|
|
|
final query =
|
|
drift.remoteAssetEntity.select().join([
|
|
leftOuterJoin(
|
|
drift.localAssetEntity,
|
|
drift.localAssetEntity.checksum.equalsExp(drift.remoteAssetEntity.checksum),
|
|
),
|
|
leftOuterJoin(
|
|
drift.remoteAssetCloudIdEntity,
|
|
drift.localAssetEntity.iCloudId.equalsExp(drift.remoteAssetCloudIdEntity.cloudId),
|
|
useColumns: false,
|
|
),
|
|
])..where(
|
|
// Only select assets that have a local cloud ID but either no remote cloud ID or a mismatched eTag
|
|
drift.localAssetEntity.id.isNotNull() &
|
|
drift.localAssetEntity.iCloudId.isNotNull() &
|
|
drift.remoteAssetEntity.ownerId.equals(userId) &
|
|
(drift.remoteAssetCloudIdEntity.cloudId.isNull() | drift.remoteAssetCloudIdEntity.eTag.isNotExp(eTag)),
|
|
);
|
|
return query.map((row) {
|
|
return (
|
|
remoteAssetId: row.read(drift.remoteAssetEntity.id)!,
|
|
localAsset: row.readTable(drift.localAssetEntity).toDto(),
|
|
);
|
|
}).get();
|
|
}
|