immich/mobile/lib/infrastructure/repositories/sync_stream.repository.dart

import 'dart:convert';

import 'package:collection/collection.dart';
import 'package:drift/drift.dart';
import 'package:immich_mobile/constants/constants.dart';
import 'package:immich_mobile/domain/models/album/album.model.dart';
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
import 'package:immich_mobile/domain/models/asset_edit.model.dart';
import 'package:immich_mobile/domain/models/memory.model.dart';
import 'package:immich_mobile/domain/models/user.model.dart';
import 'package:immich_mobile/domain/models/user_metadata.model.dart';
import 'package:immich_mobile/extensions/string_extensions.dart';
import 'package:immich_mobile/infrastructure/entities/asset_edit.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/asset_face.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/auth_user.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/exif.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/local_album.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/memory.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/memory_asset.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/partner.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/person.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/remote_album.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/remote_album_asset.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/remote_album_user.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/remote_asset.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/remote_asset_cloud_id.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/stack.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/user.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/user_metadata.entity.drift.dart';
import 'package:immich_mobile/infrastructure/repositories/db.repository.dart';
import 'package:immich_mobile/infrastructure/utils/exif.converter.dart';
import 'package:logging/logging.dart';
import 'package:openapi/api.dart' as api show AssetVisibility, AlbumUserRole, UserMetadataKey, AssetEditAction;
import 'package:openapi/api.dart' hide UserMetadataKey, AssetEditAction, AssetVisibility, AlbumUserRole;

class SyncStreamRepository extends DriftDatabaseRepository {
  final Logger _logger = Logger('DriftSyncStreamRepository');
  final Drift _db;

  SyncStreamRepository(super.db) : _db = db;

  Future<void> reset() async {
    _logger.fine("SyncResetV1 received. Resetting remote entities");
    try {
      await _db.exclusively(() async {
        // foreign_keys PRAGMA is no-op within transactions
        // https://www.sqlite.org/pragma.html#pragma_foreign_keys
        await _db.customStatement('PRAGMA foreign_keys = OFF');
        try {
          await transaction(() async {
            // FK cascade (ON DELETE SET NULL) does not fire while foreign_keys = OFF,
            // so null linkedRemoteAlbumId manually to avoid dangling pointers in local_album_entity.
            await _db.localAlbumEntity.update().write(
              const LocalAlbumEntityCompanion(linkedRemoteAlbumId: Value(null)),
            );
            await _db.assetFaceEntity.deleteAll();
            await _db.memoryAssetEntity.deleteAll();
            await _db.memoryEntity.deleteAll();
            await _db.partnerEntity.deleteAll();
            await _db.personEntity.deleteAll();
            await _db.remoteAlbumAssetEntity.deleteAll();
            await _db.remoteAlbumEntity.deleteAll();
            await _db.remoteAlbumUserEntity.deleteAll();
            await _db.remoteAssetEntity.deleteAll();
            await _db.remoteExifEntity.deleteAll();
            await _db.stackEntity.deleteAll();
            await _db.authUserEntity.deleteAll();
            await _db.userEntity.deleteAll();
            await _db.userMetadataEntity.deleteAll();
            await _db.remoteAssetCloudIdEntity.deleteAll();
            await _db.assetEditEntity.deleteAll();
          });
        } finally {
          // re-enable FK even if the transaction throws, otherwise the connection
          // would be left with foreign_keys = OFF, silently disabling cascades.
          await _db.customStatement('PRAGMA foreign_keys = ON');
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncResetV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAuthUsersV1(Iterable<SyncAuthUserV1> data) async {
    try {
      await _db.batch((batch) {
        for (final user in data) {
          final companion = AuthUserEntityCompanion(
            name: Value(user.name),
            email: Value(user.email),
            hasProfileImage: Value(user.hasProfileImage),
            profileChangedAt: Value(user.profileChangedAt),
            avatarColor: Value(user.avatarColor?.toAvatarColor() ?? AvatarColor.primary),
            isAdmin: Value(user.isAdmin),
            pinCode: Value(user.pinCode),
            quotaSizeInBytes: Value(user.quotaSizeInBytes ?? 0),
            quotaUsageInBytes: Value(user.quotaUsageInBytes),
          );

          batch.insert(
            _db.authUserEntity,
            companion.copyWith(id: Value(user.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncAuthUserV1', error, stack);
      rethrow;
    }
  }

  Future<void> deleteUsersV1(Iterable<SyncUserDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final user in data) {
          batch.deleteWhere(_db.userEntity, (row) => row.id.equals(user.userId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncUserDeleteV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateUsersV1(Iterable<SyncUserV1> data) async {
    try {
      await _db.batch((batch) {
        for (final user in data) {
          final companion = UserEntityCompanion(
            name: Value(user.name),
            email: Value(user.email),
            hasProfileImage: Value(user.hasProfileImage),
            profileChangedAt: Value(user.profileChangedAt),
            avatarColor: Value(user.avatarColor?.toAvatarColor() ?? AvatarColor.primary),
          );

          batch.insert(_db.userEntity, companion.copyWith(id: Value(user.id)), onConflict: DoUpdate((_) => companion));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncUserV1', error, stack);
      rethrow;
    }
  }

  Future<void> deletePartnerV1(Iterable<SyncPartnerDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final partner in data) {
          batch.delete(
            _db.partnerEntity,
            PartnerEntityCompanion(sharedById: Value(partner.sharedById), sharedWithId: Value(partner.sharedWithId)),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncPartnerDeleteV1', error, stack);
      rethrow;
    }
  }

  Future<void> updatePartnerV1(Iterable<SyncPartnerV1> data) async {
    try {
      await _db.batch((batch) {
        for (final partner in data) {
          final companion = PartnerEntityCompanion(inTimeline: Value(partner.inTimeline));

          batch.insert(
            _db.partnerEntity,
            companion.copyWith(sharedById: Value(partner.sharedById), sharedWithId: Value(partner.sharedWithId)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: SyncPartnerV1', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAssetsV1(Iterable<SyncAssetDeleteV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final asset in data) {
          batch.deleteWhere(_db.remoteAssetEntity, (row) => row.id.equals(asset.assetId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAssetsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetsV1(Iterable<SyncAssetV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final asset in data) {
          final companion = RemoteAssetEntityCompanion(
            name: Value(asset.originalFileName),
            type: Value(asset.type.toAssetType()),
            createdAt: Value.absentIfNull(asset.fileCreatedAt),
            updatedAt: Value.absentIfNull(asset.fileModifiedAt),
            uploadedAt: Value(asset.createdAt),
            durationMs: Value(asset.duration?.toDuration()?.inMilliseconds ?? 0),
            checksum: Value(asset.checksum),
            isFavorite: Value(asset.isFavorite),
            ownerId: Value(asset.ownerId),
            localDateTime: Value(asset.localDateTime),
            thumbHash: Value(asset.thumbhash),
            deletedAt: Value(asset.deletedAt),
            visibility: Value(asset.visibility.toAssetVisibility()),
            livePhotoVideoId: Value(asset.livePhotoVideoId),
            stackId: Value(asset.stackId),
            libraryId: Value(asset.libraryId),
            width: Value(asset.width),
            height: Value(asset.height),
            isEdited: Value(asset.isEdited),
          );

          batch.insert(
            _db.remoteAssetEntity,
            companion.copyWith(id: Value(asset.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetsV2(Iterable<SyncAssetV2> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final asset in data) {
          final companion = RemoteAssetEntityCompanion(
            name: Value(asset.originalFileName),
            type: Value(asset.type.toAssetType()),
            createdAt: Value.absentIfNull(asset.fileCreatedAt),
            updatedAt: Value.absentIfNull(asset.fileModifiedAt),
            uploadedAt: Value(asset.createdAt),
            durationMs: Value(asset.duration),
            checksum: Value(asset.checksum),
            isFavorite: Value(asset.isFavorite),
            ownerId: Value(asset.ownerId),
            localDateTime: Value(asset.localDateTime),
            thumbHash: Value(asset.thumbhash),
            deletedAt: Value(asset.deletedAt),
            visibility: Value(asset.visibility.toAssetVisibility()),
            livePhotoVideoId: Value(asset.livePhotoVideoId),
            stackId: Value(asset.stackId),
            libraryId: Value(asset.libraryId),
            width: Value(asset.width),
            height: Value(asset.height),
            isEdited: Value(asset.isEdited),
          );

          batch.insert(
            _db.remoteAssetEntity,
            companion.copyWith(id: Value(asset.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetsV2 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetsExifV1(Iterable<SyncAssetExifV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final exif in data) {
          final companion = RemoteExifEntityCompanion(
            city: Value(exif.city),
            state: Value(exif.state),
            country: Value(exif.country),
            dateTimeOriginal: Value(exif.dateTimeOriginal),
            description: Value(exif.description),
            exposureTime: Value(exif.exposureTime),
            fNumber: Value(exif.fNumber),
            fileSize: Value(exif.fileSizeInByte),
            focalLength: Value(exif.focalLength),
            latitude: Value(exif.latitude?.toDouble()),
            longitude: Value(exif.longitude?.toDouble()),
            iso: Value(exif.iso),
            make: Value(exif.make),
            model: Value(exif.model),
            orientation: Value(exif.orientation),
            timeZone: Value(exif.timeZone),
            rating: Value(exif.rating),
            projectionType: Value(exif.projectionType),
            lens: Value(exif.lensModel),
            width: Value(exif.exifImageWidth),
            height: Value(exif.exifImageHeight),
          );

          batch.insert(
            _db.remoteExifEntity,
            companion.copyWith(assetId: Value(exif.assetId)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });

      await _db.batch((batch) {
        for (final exif in data) {
          int? width;
          int? height;

          if (ExifDtoConverter.isOrientationFlipped(exif.orientation)) {
            width = exif.exifImageHeight;
            height = exif.exifImageWidth;
          } else {
            width = exif.exifImageWidth;
            height = exif.exifImageHeight;
          }

          batch.update(
            _db.remoteAssetEntity,
            RemoteAssetEntityCompanion(width: Value(width), height: Value(height)),
            where: (row) => row.id.equals(exif.assetId) & row.width.isNull() & row.height.isNull(),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetsExifV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAssetsMetadataV1(Iterable<SyncAssetMetadataDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final metadata in data) {
          if (metadata.key == kMobileMetadataKey) {
            batch.deleteWhere(_db.remoteAssetCloudIdEntity, (row) => row.assetId.equals(metadata.assetId));
          }
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAssetsMetadataV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetsMetadataV1(Iterable<SyncAssetMetadataV1> data) async {
    try {
      await _db.batch((batch) {
        for (final metadata in data) {
          if (metadata.key == kMobileMetadataKey) {
            final map = metadata.value as Map<String, Object?>;
            final companion = RemoteAssetCloudIdEntityCompanion(
              cloudId: Value(map['iCloudId']?.toString()),
              createdAt: Value(map['createdAt'] != null ? DateTime.parse(map['createdAt'] as String) : null),
              adjustmentTime: Value(
                map['adjustmentTime'] != null ? DateTime.parse(map['adjustmentTime'] as String) : null,
              ),
              latitude: Value(map['latitude'] != null ? (double.tryParse(map['latitude'] as String)) : null),
              longitude: Value(map['longitude'] != null ? (double.tryParse(map['longitude'] as String)) : null),
            );
            batch.insert(
              _db.remoteAssetCloudIdEntity,
              companion.copyWith(assetId: Value(metadata.assetId)),
              onConflict: DoUpdate((_) => companion),
            );
          }
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetsMetadataV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetEditsV1(Iterable<SyncAssetEditV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final edit in data) {
          final companion = AssetEditEntityCompanion(
            id: Value(edit.id),
            assetId: Value(edit.assetId),
            action: Value(edit.action.toAssetEditAction()),
            parameters: Value(edit.parameters as Map<String, Object?>),
            sequence: Value(edit.sequence),
          );

          batch.insert(_db.assetEditEntity, companion, onConflict: DoUpdate((_) => companion));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetEditsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> replaceAssetEditsV1(String assetId, Iterable<SyncAssetEditV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        batch.deleteWhere(_db.assetEditEntity, (row) => row.assetId.equals(assetId));

        for (final edit in data) {
          final companion = AssetEditEntityCompanion(
            id: Value(edit.id),
            assetId: Value(edit.assetId),
            action: Value(edit.action.toAssetEditAction()),
            parameters: Value(edit.parameters as Map<String, Object?>),
            sequence: Value(edit.sequence),
          );

          batch.insert(_db.assetEditEntity, companion);
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: replaceAssetEditsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAssetEditsV1(Iterable<SyncAssetEditDeleteV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final edit in data) {
          batch.deleteWhere(_db.assetEditEntity, (row) => row.id.equals(edit.editId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAssetEditsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAlbumsV1(Iterable<SyncAlbumDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          batch.deleteWhere(_db.remoteAlbumEntity, (row) => row.id.equals(album.albumId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAlbumsV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAlbumsV1(Iterable<SyncAlbumV1> data) async {
    try {
      await _db.transaction(() async {
        await _db.batch((batch) {
          for (final album in data) {
            final companion = RemoteAlbumEntityCompanion(
              name: Value(album.name),
              description: Value(album.description),
              isActivityEnabled: Value(album.isActivityEnabled),
              order: Value(album.order.toAlbumAssetOrder()),
              thumbnailAssetId: Value(album.thumbnailAssetId),
              createdAt: Value(album.createdAt),
              updatedAt: Value(album.updatedAt),
            );

            batch.insert(
              _db.remoteAlbumEntity,
              companion.copyWith(id: Value(album.id)),
              onConflict: DoUpdate((_) => companion),
            );
          }
        });

        await _db.batch((batch) {
          for (final album in data) {
            final companion = RemoteAlbumUserEntityCompanion(
              albumId: Value(album.id),
              userId: Value(album.ownerId),
              role: const Value(AlbumUserRole.owner),
            );

            batch.insert(_db.remoteAlbumUserEntity, companion, onConflict: DoUpdate((_) => companion));
          }
        });
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAlbumsV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAlbumsV2(Iterable<SyncAlbumV2> data) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          final companion = RemoteAlbumEntityCompanion(
            name: Value(album.name),
            description: Value(album.description),
            isActivityEnabled: Value(album.isActivityEnabled),
            order: Value(album.order.toAlbumAssetOrder()),
            thumbnailAssetId: Value(album.thumbnailAssetId),
            createdAt: Value(album.createdAt),
            updatedAt: Value(album.updatedAt),
          );

          batch.insert(
            _db.remoteAlbumEntity,
            companion.copyWith(id: Value(album.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAlbumsV2', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAlbumUsersV1(Iterable<SyncAlbumUserDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          batch.delete(
            _db.remoteAlbumUserEntity,
            RemoteAlbumUserEntityCompanion(albumId: Value(album.albumId), userId: Value(album.userId)),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAlbumUsersV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAlbumUsersV1(Iterable<SyncAlbumUserV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          final companion = RemoteAlbumUserEntityCompanion(role: Value(album.role.toAlbumUserRole()));

          batch.insert(
            _db.remoteAlbumUserEntity,
            companion.copyWith(albumId: Value(album.albumId), userId: Value(album.userId)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAlbumUsersV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAlbumToAssetsV1(Iterable<SyncAlbumToAssetDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          batch.delete(
            _db.remoteAlbumAssetEntity,
            RemoteAlbumAssetEntityCompanion(albumId: Value(album.albumId), assetId: Value(album.assetId)),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAlbumToAssetsV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAlbumToAssetsV1(Iterable<SyncAlbumToAssetV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final album in data) {
          final companion = RemoteAlbumAssetEntityCompanion(
            albumId: Value(album.albumId),
            assetId: Value(album.assetId),
          );

          batch.insert(_db.remoteAlbumAssetEntity, companion, onConflict: DoNothing());
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAlbumToAssetsV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> updateMemoriesV1(Iterable<SyncMemoryV1> data) async {
    try {
      await _db.batch((batch) {
        for (final memory in data) {
          final companion = MemoryEntityCompanion(
            createdAt: Value(memory.createdAt),
            deletedAt: Value(memory.deletedAt),
            ownerId: Value(memory.ownerId),
            type: Value(memory.type.toMemoryType()),
            data: Value(jsonEncode(memory.data)),
            isSaved: Value(memory.isSaved),
            memoryAt: Value(memory.memoryAt),
            seenAt: Value.absentIfNull(memory.seenAt),
            showAt: Value.absentIfNull(memory.showAt),
            hideAt: Value.absentIfNull(memory.hideAt),
          );

          batch.insert(
            _db.memoryEntity,
            companion.copyWith(id: Value(memory.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateMemoriesV1', error, stack);
      rethrow;
    }
  }

  Future<void> deleteMemoriesV1(Iterable<SyncMemoryDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final memory in data) {
          batch.deleteWhere(_db.memoryEntity, (row) => row.id.equals(memory.memoryId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteMemoriesV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateMemoryAssetsV1(Iterable<SyncMemoryAssetV1> data) async {
    try {
      await _db.batch((batch) {
        for (final asset in data) {
          final companion = MemoryAssetEntityCompanion(memoryId: Value(asset.memoryId), assetId: Value(asset.assetId));

          batch.insert(_db.memoryAssetEntity, companion, onConflict: DoNothing());
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateMemoryAssetsV1', error, stack);
      rethrow;
    }
  }

  Future<void> deleteMemoryAssetsV1(Iterable<SyncMemoryAssetDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final asset in data) {
          batch.delete(
            _db.memoryAssetEntity,
            MemoryAssetEntityCompanion(memoryId: Value(asset.memoryId), assetId: Value(asset.assetId)),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteMemoryAssetsV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateStacksV1(Iterable<SyncStackV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final stack in data) {
          final companion = StackEntityCompanion(
            createdAt: Value(stack.createdAt),
            updatedAt: Value(stack.updatedAt),
            ownerId: Value(stack.ownerId),
            primaryAssetId: Value(stack.primaryAssetId),
          );

          batch.insert(
            _db.stackEntity,
            companion.copyWith(id: Value(stack.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateStacksV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> deleteStacksV1(Iterable<SyncStackDeleteV1> data, {String debugLabel = 'user'}) async {
    try {
      await _db.batch((batch) {
        for (final stack in data) {
          batch.deleteWhere(_db.stackEntity, (row) => row.id.equals(stack.stackId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteStacksV1 - $debugLabel', error, stack);
      rethrow;
    }
  }

  Future<void> updateUserMetadatasV1(Iterable<SyncUserMetadataV1> data) async {
    try {
      await _db.batch((batch) {
        for (final userMetadata in data) {
          final companion = UserMetadataEntityCompanion(value: Value(userMetadata.value as Map<String, Object?>));

          batch.insert(
            _db.userMetadataEntity,
            companion.copyWith(userId: Value(userMetadata.userId), key: Value(userMetadata.key.toUserMetadataKey())),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteUserMetadatasV1', error, stack);
      rethrow;
    }
  }

  Future<void> deleteUserMetadatasV1(Iterable<SyncUserMetadataDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final userMetadata in data) {
          batch.delete(
            _db.userMetadataEntity,
            UserMetadataEntityCompanion(
              userId: Value(userMetadata.userId),
              key: Value(userMetadata.key.toUserMetadataKey()),
            ),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteUserMetadatasV1', error, stack);
      rethrow;
    }
  }

  Future<void> updatePeopleV1(Iterable<SyncPersonV1> data) async {
    try {
      await _db.batch((batch) {
        for (final person in data) {
          final companion = PersonEntityCompanion(
            createdAt: Value(person.createdAt),
            updatedAt: Value(person.updatedAt),
            ownerId: Value(person.ownerId),
            name: Value(person.name),
            faceAssetId: Value(person.faceAssetId),
            isFavorite: Value(person.isFavorite),
            isHidden: Value(person.isHidden),
            color: Value(person.color),
            birthDate: Value(person.birthDate),
          );

          batch.insert(
            _db.personEntity,
            companion.copyWith(id: Value(person.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updatePeopleV1', error, stack);
      rethrow;
    }
  }

  Future<void> deletePeopleV1(Iterable<SyncPersonDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final person in data) {
          batch.deleteWhere(_db.personEntity, (row) => row.id.equals(person.personId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deletePeopleV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetFacesV1(Iterable<SyncAssetFaceV1> data) async {
    try {
      await _db.batch((batch) {
        for (final assetFace in data) {
          final companion = AssetFaceEntityCompanion(
            assetId: Value(assetFace.assetId),
            personId: Value(assetFace.personId),
            imageWidth: Value(assetFace.imageWidth),
            imageHeight: Value(assetFace.imageHeight),
            boundingBoxX1: Value(assetFace.boundingBoxX1),
            boundingBoxY1: Value(assetFace.boundingBoxY1),
            boundingBoxX2: Value(assetFace.boundingBoxX2),
            boundingBoxY2: Value(assetFace.boundingBoxY2),
            sourceType: Value(assetFace.sourceType),
          );

          batch.insert(
            _db.assetFaceEntity,
            companion.copyWith(id: Value(assetFace.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetFacesV1', error, stack);
      rethrow;
    }
  }

  Future<void> updateAssetFacesV2(Iterable<SyncAssetFaceV2> data) async {
    try {
      await _db.batch((batch) {
        for (final assetFace in data) {
          final companion = AssetFaceEntityCompanion(
            assetId: Value(assetFace.assetId),
            personId: Value(assetFace.personId),
            imageWidth: Value(assetFace.imageWidth),
            imageHeight: Value(assetFace.imageHeight),
            boundingBoxX1: Value(assetFace.boundingBoxX1),
            boundingBoxY1: Value(assetFace.boundingBoxY1),
            boundingBoxX2: Value(assetFace.boundingBoxX2),
            boundingBoxY2: Value(assetFace.boundingBoxY2),
            sourceType: Value(assetFace.sourceType),
            deletedAt: Value(assetFace.deletedAt),
            isVisible: Value(assetFace.isVisible),
          );

          batch.insert(
            _db.assetFaceEntity,
            companion.copyWith(id: Value(assetFace.id)),
            onConflict: DoUpdate((_) => companion),
          );
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: updateAssetFacesV2', error, stack);
      rethrow;
    }
  }

  Future<void> deleteAssetFacesV1(Iterable<SyncAssetFaceDeleteV1> data) async {
    try {
      await _db.batch((batch) {
        for (final assetFace in data) {
          batch.deleteWhere(_db.assetFaceEntity, (row) => row.id.equals(assetFace.assetFaceId));
        }
      });
    } catch (error, stack) {
      _logger.severe('Error: deleteAssetFacesV1', error, stack);
      rethrow;
    }
  }

  Future<void> pruneAssets() async {
    try {
      await _db.transaction(() async {
        final authQuery = _db.authUserEntity.selectOnly()
          ..addColumns([_db.authUserEntity.id])
          ..limit(1);
        final currentUserId = await authQuery.map((row) => row.read(_db.authUserEntity.id)).getSingleOrNull();
        if (currentUserId == null) {
          _logger.warning('No authenticated user found during pruneAssets. Skipping asset pruning.');
          return;
        }

        final partnerQuery = _db.partnerEntity.selectOnly()
          ..addColumns([_db.partnerEntity.sharedById])
          ..where(_db.partnerEntity.sharedWithId.equals(currentUserId));
        final partnerIds = await partnerQuery.map((row) => row.read(_db.partnerEntity.sharedById)).get();

        final validUsers = {currentUserId, ...partnerIds.nonNulls};

        // Asset is not owned by the current user or any of their partners and is not part of any (shared) album
        // Likely a stale asset that was previously shared but has been removed
        await _db.remoteAssetEntity.deleteWhere((asset) {
          return asset.ownerId.isNotIn(validUsers) &
              asset.id.isNotInQuery(
                _db.remoteAlbumAssetEntity.selectOnly()..addColumns([_db.remoteAlbumAssetEntity.assetId]),
              );
        });
      });
    } catch (error, stack) {
      _logger.severe('Error: pruneAssets', error, stack);
      // We do not rethrow here as this is a client-only cleanup and should not affect the sync process
    }
  }
}

extension on AssetTypeEnum {
  AssetType toAssetType() => switch (this) {
    AssetTypeEnum.IMAGE => AssetType.image,
    AssetTypeEnum.VIDEO => AssetType.video,
    AssetTypeEnum.AUDIO => AssetType.audio,
    AssetTypeEnum.OTHER => AssetType.other,
    _ => throw Exception('Unknown AssetType value: $this'),
  };
}

extension on AssetOrder {
  AlbumAssetOrder toAlbumAssetOrder() => switch (this) {
    AssetOrder.asc => AlbumAssetOrder.asc,
    AssetOrder.desc => AlbumAssetOrder.desc,
    _ => throw Exception('Unknown AssetOrder value: $this'),
  };
}

extension on MemoryType {
  MemoryTypeEnum toMemoryType() => switch (this) {
    MemoryType.onThisDay => MemoryTypeEnum.onThisDay,
    _ => throw Exception('Unknown MemoryType value: $this'),
  };
}

extension on api.AlbumUserRole {
  AlbumUserRole toAlbumUserRole() => switch (this) {
    api.AlbumUserRole.editor => AlbumUserRole.editor,
    api.AlbumUserRole.viewer => AlbumUserRole.viewer,
    api.AlbumUserRole.owner => AlbumUserRole.owner,
    _ => throw Exception('Unknown AlbumUserRole value: $this'),
  };
}

extension on api.AssetVisibility {
  AssetVisibility toAssetVisibility() => switch (this) {
    api.AssetVisibility.timeline => AssetVisibility.timeline,
    api.AssetVisibility.hidden => AssetVisibility.hidden,
    api.AssetVisibility.archive => AssetVisibility.archive,
    api.AssetVisibility.locked => AssetVisibility.locked,
    _ => throw Exception('Unknown AssetVisibility value: $this'),
  };
}

extension on api.UserMetadataKey {
  UserMetadataKey toUserMetadataKey() => switch (this) {
    api.UserMetadataKey.onboarding => UserMetadataKey.onboarding,
    api.UserMetadataKey.preferences => UserMetadataKey.preferences,
    api.UserMetadataKey.license => UserMetadataKey.license,
    _ => throw Exception('Unknown UserMetadataKey value: $this'),
  };
}

extension on UserAvatarColor {
  AvatarColor? toAvatarColor() => AvatarColor.values.firstWhereOrNull((c) => c.name == value);
}

extension on api.AssetEditAction {
  AssetEditAction toAssetEditAction() => switch (this) {
    api.AssetEditAction.crop => AssetEditAction.crop,
    api.AssetEditAction.rotate => AssetEditAction.rotate,
    api.AssetEditAction.mirror => AssetEditAction.mirror,
    _ => AssetEditAction.other,
  };
}