Source Code added

mobile/lib/repositories/album_media.repository.dart

@@ -0,0 +1,99 @@
+import 'package:hooks_riverpod/hooks_riverpod.dart';
+import 'package:immich_mobile/domain/models/store.model.dart';
+import 'package:immich_mobile/entities/album.entity.dart';
+import 'package:immich_mobile/entities/asset.entity.dart';
+import 'package:immich_mobile/entities/store.entity.dart';
+import 'package:immich_mobile/infrastructure/entities/user.entity.dart';
+import 'package:immich_mobile/repositories/asset_media.repository.dart';
+import 'package:photo_manager/photo_manager.dart' hide AssetType;
+
+final albumMediaRepositoryProvider = Provider((ref) => const AlbumMediaRepository());
+
+class AlbumMediaRepository {
+  const AlbumMediaRepository();
+
+  bool get useCustomFilter => Store.get(StoreKey.photoManagerCustomFilter, true);
+
+  FilterOptionGroup? _getAlbumFilter({
+    DateTimeCond? updateTimeCond,
+    bool? containsPathModified,
+    List<OrderOption>? orderBy,
+  }) => useCustomFilter
+      ? FilterOptionGroup(
+          imageOption: const FilterOption(needTitle: true, sizeConstraint: SizeConstraint(ignoreSize: true)),
+          videoOption: const FilterOption(
+            needTitle: true,
+            sizeConstraint: SizeConstraint(ignoreSize: true),
+            durationConstraint: DurationConstraint(allowNullable: true),
+          ),
+          containsPathModified: containsPathModified ?? false,
+          createTimeCond: DateTimeCond.def().copyWith(ignore: true),
+          updateTimeCond: updateTimeCond ?? DateTimeCond.def().copyWith(ignore: true),
+          orders: orderBy ?? [],
+        )
+      : null;
+
+  Future<List<Album>> getAll() async {
+    final filter = useCustomFilter
+        ? CustomFilter.sql(where: '${CustomColumns.base.width} > 0')
+        : FilterOptionGroup(containsPathModified: true);
+
+    final List<AssetPathEntity> assetPathEntities = await PhotoManager.getAssetPathList(
+      hasAll: true,
+      filterOption: filter,
+    );
+    return assetPathEntities.map(_toAlbum).toList();
+  }
+
+  Future<List<String>> getAssetIds(String albumId) async {
+    final album = await AssetPathEntity.fromId(albumId, filterOption: _getAlbumFilter());
+    final List<AssetEntity> assets = await album.getAssetListRange(start: 0, end: 0x7fffffffffffffff);
+    return assets.map((e) => e.id).toList();
+  }
+
+  Future<int> getAssetCount(String albumId) async {
+    final album = await AssetPathEntity.fromId(albumId, filterOption: _getAlbumFilter());
+    return album.assetCountAsync;
+  }
+
+  Future<List<Asset>> getAssets(
+    String albumId, {
+    int start = 0,
+    int end = 0x7fffffffffffffff,
+    DateTime? modifiedFrom,
+    DateTime? modifiedUntil,
+    bool orderByModificationDate = false,
+  }) async {
+    final onDevice = await AssetPathEntity.fromId(
+      albumId,
+      filterOption: _getAlbumFilter(
+        updateTimeCond: modifiedFrom == null && modifiedUntil == null
+            ? null
+            : DateTimeCond(min: modifiedFrom ?? DateTime.utc(-271820), max: modifiedUntil ?? DateTime.utc(275760)),
+        orderBy: orderByModificationDate ? [const OrderOption(type: OrderOptionType.updateDate)] : [],
+      ),
+    );
+
+    final List<AssetEntity> assets = await onDevice.getAssetListRange(start: start, end: end);
+    return assets.map(AssetMediaRepository.toAsset).toList().cast();
+  }
+
+  Future<Album> get(String id) async {
+    final assetPathEntity = await AssetPathEntity.fromId(id, filterOption: _getAlbumFilter(containsPathModified: true));
+    return _toAlbum(assetPathEntity);
+  }
+
+  static Album _toAlbum(AssetPathEntity assetPathEntity) {
+    final Album album = Album(
+      name: assetPathEntity.name,
+      createdAt: assetPathEntity.lastModified?.toUtc() ?? DateTime.now().toUtc(),
+      modifiedAt: assetPathEntity.lastModified?.toUtc() ?? DateTime.now().toUtc(),
+      shared: false,
+      activityEnabled: false,
+    );
+    album.owner.value = User.fromDto(Store.get(StoreKey.currentUser));
+    album.localId = assetPathEntity.id;
+    album.isAll = assetPathEntity.isAll;
+    return album;
+  }
+}