feat: home grid

2024-09-10 01:14:05 +05:30
parent 80009a77ec
commit 419d3669a2
33 changed files with 808 additions and 429 deletions
@@ -0,0 +1,137 @@
+import 'dart:async';
+
+import 'package:drift/drift.dart';
+import 'package:immich_mobile/domain/entities/asset.entity.drift.dart';
+import 'package:immich_mobile/domain/interfaces/asset.interface.dart';
+import 'package:immich_mobile/domain/models/asset.model.dart';
+import 'package:immich_mobile/domain/models/render_list.model.dart';
+import 'package:immich_mobile/domain/models/render_list_element.model.dart';
+import 'package:immich_mobile/domain/repositories/database.repository.dart';
+import 'package:immich_mobile/utils/extensions/drift.extension.dart';
+import 'package:immich_mobile/utils/mixins/log_context.mixin.dart';
+import 'package:intl/intl.dart';
+
+class RemoteAssetDriftRepository with LogContext implements IAssetRepository {
+  final DriftDatabaseRepository _db;
+
+  const RemoteAssetDriftRepository(this._db);
+
+  @override
+  Future<bool> addAll(Iterable<Asset> assets) async {
+    try {
+      await _db.batch((batch) => batch.insertAllOnConflictUpdate(
+            _db.asset,
+            assets.map(_toEntity),
+          ));
+
+      return true;
+    } catch (e, s) {
+      log.severe("Cannot insert remote assets into table", e, s);
+      return false;
+    }
+  }
+
+  @override
+  Future<bool> clearAll() async {
+    try {
+      await _db.asset.deleteAll();
+      return true;
+    } catch (e, s) {
+      log.severe("Cannot clear remote assets", e, s);
+      return false;
+    }
+  }
+
+  @override
+  Future<List<Asset>> fetchAssets({int? offset, int? limit}) async {
+    final query = _db.asset.select()
+      ..orderBy([(asset) => OrderingTerm.desc(asset.createdTime)]);
+
+    if (limit != null) {
+      query.limit(limit, offset: offset);
+    }
+
+    return (await query.get()).map(_toModel).toList();
+  }
+
+  @override
+  Stream<RenderList> getRenderList() {
+    final assetCountExp = _db.asset.id.count();
+    final createdTimeExp = _db.asset.createdTime;
+    final monthYearExp = _db.asset.createdTime.strftime('%m-%Y');
+
+    final query = _db.asset.selectOnly()
+      ..addColumns([assetCountExp, createdTimeExp])
+      ..groupBy([monthYearExp])
+      ..orderBy([OrderingTerm.desc(createdTimeExp)]);
+
+    int lastAssetOffset = 0;
+    final monthFormatter = DateFormat.yMMMM();
+
+    return query
+        .expand((row) {
+          final createdTime = row.read<DateTime>(createdTimeExp)!;
+          final assetCount = row.read(assetCountExp)!;
+          final assetOffset = lastAssetOffset;
+          lastAssetOffset += assetCount;
+
+          return [
+            RenderListMonthHeaderElement(
+              header: monthFormatter.format(createdTime),
+            ),
+            RenderListAssetElement(
+              date: createdTime,
+              assetCount: assetCount,
+              assetOffset: assetOffset,
+            ),
+          ];
+        })
+        .watch()
+        .map((elements) {
+          final int totalCount;
+          final lastAssetElement =
+              elements.whereType<RenderListAssetElement>().lastOrNull;
+          if (lastAssetElement == null) {
+            totalCount = 0;
+          } else {
+            totalCount =
+                lastAssetElement.assetCount + lastAssetElement.assetOffset;
+          }
+
+          return RenderList(elements: elements, totalCount: totalCount);
+        });
+  }
+}
+
+AssetCompanion _toEntity(Asset asset) {
+  return AssetCompanion.insert(
+    localId: Value(asset.localId),
+    remoteId: Value(asset.remoteId),
+    name: asset.name,
+    checksum: asset.checksum,
+    height: Value(asset.height),
+    width: Value(asset.width),
+    type: asset.type,
+    createdTime: asset.createdTime,
+    duration: Value(asset.duration),
+    modifiedTime: Value(asset.modifiedTime),
+    livePhotoVideoId: Value(asset.livePhotoVideoId),
+  );
+}
+
+Asset _toModel(AssetData asset) {
+  return Asset(
+    id: asset.id,
+    localId: asset.localId,
+    remoteId: asset.remoteId,
+    name: asset.name,
+    type: asset.type,
+    checksum: asset.checksum,
+    createdTime: asset.createdTime,
+    modifiedTime: asset.modifiedTime,
+    height: asset.height,
+    width: asset.width,
+    livePhotoVideoId: asset.livePhotoVideoId,
+    duration: asset.duration,
+  );
+}