immich/mobile/lib/repositories/album_media.repository.dart

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;
  }
}