Riverpod Mutation Provider

mutation_provider.dart

import 'package:flutter_riverpod/flutter_riverpod.dart';

class MutationProvider {
  static AutoDisposeNotifierProvider<MutationNotifier<Arg, ReturnType>,
      AsyncValue<ReturnType?>> autoDispose<Arg, ReturnType>({
    required Future<ReturnType> Function(Ref, Arg) mutationFn,
  }) {
    return NotifierProvider.autoDispose(
      () => MutationNotifier(
        mutationFn: mutationFn,
      ),
    );
  }
}

class MutationNotifier<Arg, ReturnType>
    extends AutoDisposeNotifier<AsyncValue<ReturnType?>> {
  MutationNotifier({
    required Future<ReturnType> Function(Ref ref, Arg arg) mutationFn,
  }) : _mutationFn = mutationFn;

  final Future<ReturnType> Function(Ref ref, Arg arg) _mutationFn;

  @override
  AsyncValue<ReturnType?> build() {
    return const AsyncValue.data(null);
  }

  Future<AsyncValue<ReturnType>> mutate(Arg arg) async {
    state = AsyncValue.loading();
    state = await AsyncValue.guard(() async {
      return await _mutationFn(ref, arg);
    });
    return state as AsyncValue<ReturnType>;
  }
}