Source Code added

2026-02-02 15:06:40 +01:00 · 2026-02-02 15:06:40 +01:00 · 9efa9bc6dd
commit 9efa9bc6dd
parent 800376eafd
3912 changed files with 754770 additions and 2 deletions
--- a/mobile/lib/infrastructure/repositories/backup.repository.dart
+++ b/mobile/lib/infrastructure/repositories/backup.repository.dart
@ -0,0 +1,117 @@
+import 'dart:async';
+
+import 'package:drift/drift.dart';
+import 'package:hooks_riverpod/hooks_riverpod.dart';
+import 'package:immich_mobile/domain/models/album/local_album.model.dart';
+import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
+import 'package:immich_mobile/infrastructure/entities/local_asset.entity.dart';
+import 'package:immich_mobile/infrastructure/repositories/db.repository.dart';
+import 'package:immich_mobile/providers/infrastructure/db.provider.dart';
+
+final backupRepositoryProvider = Provider<DriftBackupRepository>(
+  (ref) => DriftBackupRepository(ref.watch(driftProvider)),
+);
+
+class DriftBackupRepository extends DriftDatabaseRepository {
+  final Drift _db;
+  const DriftBackupRepository(this._db) : super(_db);
+
+  _getExcludedSubquery() {
+    return _db.localAlbumAssetEntity.selectOnly()
+      ..addColumns([_db.localAlbumAssetEntity.assetId])
+      ..join([
+        innerJoin(
+          _db.localAlbumEntity,
+          _db.localAlbumAssetEntity.albumId.equalsExp(_db.localAlbumEntity.id),
+          useColumns: false,
+        ),
+      ])
+      ..where(_db.localAlbumEntity.backupSelection.equalsValue(BackupSelection.excluded));
+  }
+
+  /// Returns all backup-related counts in a single query.
+  ///
+  /// - total:     number of distinct assets in selected albums, excluding those that are also in excluded albums
+  /// - backup:    number of those assets that already exist on the server for [userId]
+  /// - remainder: number of those assets that do not yet exist on the server for [userId]
+  ///              (includes processing)
+  /// - processing: number of those assets that are still preparing/have a null checksum
+  Future<({int total, int remainder, int processing})> getAllCounts(String userId) async {
+    const sql = '''
+        SELECT
+        COUNT(*) AS total_count,
+        COUNT(*) FILTER (WHERE lae.checksum IS NULL) AS processing_count,
+        COUNT(*) FILTER (WHERE rae.id IS NULL) AS remainder_count
+        FROM local_asset_entity lae
+        LEFT JOIN main.remote_asset_entity rae
+            ON lae.checksum = rae.checksum AND rae.owner_id = ?1
+        WHERE EXISTS (
+            SELECT 1
+            FROM local_album_asset_entity laa
+            INNER JOIN main.local_album_entity la on laa.album_id = la.id
+            WHERE laa.asset_id = lae.id
+                AND la.backup_selection = ?2
+        )
+        AND NOT EXISTS (
+            SELECT 1
+            FROM local_album_asset_entity laa
+            INNER JOIN main.local_album_entity la on laa.album_id = la.id
+            WHERE laa.asset_id = lae.id
+                AND la.backup_selection = ?3
+        );
+      ''';
+
+    final row = await _db
+        .customSelect(
+          sql,
+          variables: [
+            Variable.withString(userId),
+            Variable.withInt(BackupSelection.selected.index),
+            Variable.withInt(BackupSelection.excluded.index),
+          ],
+          readsFrom: {_db.localAlbumAssetEntity, _db.localAlbumEntity, _db.localAssetEntity, _db.remoteAssetEntity},
+        )
+        .getSingle();
+
+    final data = row.data;
+    return (
+      total: (data['total_count'] as int?) ?? 0,
+      remainder: (data['remainder_count'] as int?) ?? 0,
+      processing: (data['processing_count'] as int?) ?? 0,
+    );
+  }
+
+  Future<List<LocalAsset>> getCandidates(String userId, {bool onlyHashed = true}) async {
+    final selectedAlbumIds = _db.localAlbumEntity.selectOnly(distinct: true)
+      ..addColumns([_db.localAlbumEntity.id])
+      ..where(_db.localAlbumEntity.backupSelection.equalsValue(BackupSelection.selected));
+
+    final query = _db.localAssetEntity.select()
+      ..where(
+        (lae) =>
+            existsQuery(
+              _db.localAlbumAssetEntity.selectOnly()
+                ..addColumns([_db.localAlbumAssetEntity.assetId])
+                ..where(
+                  _db.localAlbumAssetEntity.albumId.isInQuery(selectedAlbumIds) &
+                      _db.localAlbumAssetEntity.assetId.equalsExp(lae.id),
+                ),
+            ) &
+            notExistsQuery(
+              _db.remoteAssetEntity.selectOnly()
+                ..addColumns([_db.remoteAssetEntity.checksum])
+                ..where(
+                  _db.remoteAssetEntity.checksum.equalsExp(lae.checksum) & _db.remoteAssetEntity.ownerId.equals(userId),
+                ),
+            ) &
+            lae.id.isNotInQuery(_getExcludedSubquery()),
+      )
+      ..orderBy([(localAsset) => OrderingTerm.desc(localAsset.createdAt)]);
+
+    if (onlyHashed) {
+      query.where((lae) => lae.checksum.isNotNull());
+    }
+
+    return query.map((localAsset) => localAsset.toDto()).get();
+  }
+}