immich/mobile/lib/modules/activities/providers/activity.provider.dart

import 'package:immich_mobile/modules/activities/models/activity.model.dart';
import 'package:immich_mobile/modules/activities/providers/activity_service.provider.dart';
import 'package:immich_mobile/modules/activities/providers/activity_statistics.provider.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';

part 'activity.provider.g.dart';

/// Maintains the current list of all activities for <share-album-id, asset>
@riverpod
class AlbumActivity extends _$AlbumActivity {
  @override
  Future<List<Activity>> build(String albumId, [String? assetId]) async {
    return ref
        .watch(activityServiceProvider)
        .getAllActivities(albumId, assetId: assetId);
  }

  Future<void> removeActivity(String id) async {
    if (await ref.watch(activityServiceProvider).removeActivity(id)) {
      final activities = state.valueOrNull ?? [];
      final removedActivity = activities.firstWhere((a) => a.id == id);
      activities.remove(removedActivity);
      state = AsyncData(activities);
      // Decrement activity count only for comments
      if (removedActivity.type == ActivityType.comment) {
        ref
            .watch(activityStatisticsProvider(albumId, assetId).notifier)
            .removeActivity();
      }
    }
  }

  Future<void> addLike() async {
    final activity = await ref
        .watch(activityServiceProvider)
        .addActivity(albumId, ActivityType.like, assetId: assetId);
    if (activity.hasValue) {
      final activities = state.asData?.value ?? [];
      state = AsyncData([...activities, activity.requireValue]);
    }
  }

  Future<void> addComment(String comment) async {
    final activity = await ref.watch(activityServiceProvider).addActivity(
          albumId,
          ActivityType.comment,
          assetId: assetId,
          comment: comment,
        );

    if (activity.hasValue) {
      final activities = state.valueOrNull ?? [];
      state = AsyncData([...activities, activity.requireValue]);
      ref
          .watch(activityStatisticsProvider(albumId, assetId).notifier)
          .addActivity();
      // The previous addActivity call would increase the count of an asset if assetId != null
      // To also increase the activity count of the album, calling it once again with assetId set to null
      if (assetId != null) {
        ref.watch(activityStatisticsProvider(albumId).notifier).addActivity();
      }
    }
  }
}

/// Mock class for testing
abstract class AlbumActivityInternal extends _$AlbumActivity {}