immich/mobile/lib/modules/favorite/providers/favorite_provider.dart

import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/shared/models/asset.dart';
import 'package:immich_mobile/shared/providers/asset.provider.dart';

class FavoriteSelectionNotifier extends StateNotifier<Set<int>> {
  FavoriteSelectionNotifier(this.assetsState, this.assetNotifier) : super({}) {
    state = assetsState.allAssets
        .where((asset) => asset.isFavorite)
        .map((asset) => asset.id)
        .toSet();
  }

  final AssetsState assetsState;
  final AssetNotifier assetNotifier;

  void _setFavoriteForAssetId(int id, bool favorite) {
    if (!favorite) {
      state = state.difference({id});
    } else {
      state = state.union({id});
    }
  }

  bool _isFavorite(int id) {
    return state.contains(id);
  }

  Future<void> toggleFavorite(Asset asset) async {
    // TODO support local favorite assets
    if (asset.storage == AssetState.local) return;
    _setFavoriteForAssetId(asset.id, !_isFavorite(asset.id));

    await assetNotifier.toggleFavorite(
      asset,
      state.contains(asset.id),
    );
  }

  Future<void> addToFavorites(Iterable<Asset> assets) {
    state = state.union(assets.map((a) => a.id).toSet());
    final futures = assets.map(
      (a) => assetNotifier.toggleFavorite(
        a,
        true,
      ),
    );

    return Future.wait(futures);
  }
}

final favoriteProvider =
    StateNotifierProvider<FavoriteSelectionNotifier, Set<int>>((ref) {
  return FavoriteSelectionNotifier(
    ref.watch(assetProvider),
    ref.watch(assetProvider.notifier),
  );
});

final favoriteAssetProvider = StateProvider((ref) {
  final favorites = ref.watch(favoriteProvider);

  return ref
      .watch(assetProvider)
      .allAssets
      .where((element) => favorites.contains(element.id))
      .toList();
});