feat: full local assets / album sync

2024-10-17 23:33:00 +05:30
parent a09710ec7b
commit c91a2878dc
87 changed files with 2417 additions and 366 deletions
@@ -0,0 +1,78 @@
+import 'dart:async';
+
+import 'package:drift/drift.dart';
+import 'package:immich_mobile/domain/entities/album_asset.entity.drift.dart';
+import 'package:immich_mobile/domain/interfaces/album_asset.interface.dart';
+import 'package:immich_mobile/domain/models/asset.model.dart';
+import 'package:immich_mobile/domain/repositories/database.repository.dart';
+import 'package:immich_mobile/domain/utils/drift_model_converters.dart';
+import 'package:immich_mobile/utils/mixins/log.mixin.dart';
+
+class AlbumToAssetRepository with LogMixin implements IAlbumToAssetRepository {
+  final DriftDatabaseRepository _db;
+
+  const AlbumToAssetRepository(this._db);
+
+  @override
+  FutureOr<bool> addAssetIds(int albumId, Iterable<int> assetIds) async {
+    try {
+      await _db.albumToAsset.insertAll(
+        assetIds.map((a) => AlbumToAssetCompanion.insert(
+              assetId: a,
+              albumId: albumId,
+            )),
+        onConflict: DoNothing(
+          target: [_db.albumToAsset.assetId, _db.albumToAsset.albumId],
+        ),
+      );
+      return true;
+    } catch (e, s) {
+      log.e("Error while adding assets to albumId - $albumId", e, s);
+      return false;
+    }
+  }
+
+  @override
+  FutureOr<List<int>> getAssetIdsOnlyInAlbum(int albumId) async {
+    final assetId = _db.asset.id;
+    final query = _db.asset.selectOnly()
+      ..addColumns([assetId])
+      ..join([
+        innerJoin(
+          _db.albumToAsset,
+          _db.albumToAsset.assetId.equalsExp(_db.asset.id) &
+              _db.asset.remoteId.isNull(),
+          useColumns: false,
+        ),
+      ])
+      ..groupBy(
+        [assetId],
+        having: _db.albumToAsset.albumId.count().equals(1) &
+            _db.albumToAsset.albumId.max().equals(albumId),
+      );
+
+    return await query.map((row) => row.read(assetId)!).get();
+  }
+
+  @override
+  FutureOr<List<Asset>> getAssetsForAlbum(int albumId) async {
+    final query = _db.asset.select().join([
+      innerJoin(
+        _db.albumToAsset,
+        _db.albumToAsset.assetId.equalsExp(_db.asset.id) &
+            _db.albumToAsset.albumId.equals(albumId),
+        useColumns: false,
+      ),
+    ]);
+
+    return await query
+        .map((row) =>
+            DriftModelConverters.toAssetModel(row.readTable(_db.asset)))
+        .get();
+  }
+
+  @override
+  FutureOr<void> deleteAlbumId(int albumId) async {
+    await _db.albumToAsset.deleteWhere((row) => row.albumId.equals(albumId));
+  }
+}