immich/mobile/lib/presentation/actions/upload.action.dart

import 'dart:async';

import 'package:flutter/material.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/constants/enums.dart';
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
import 'package:immich_mobile/generated/translations.g.dart';
import 'package:immich_mobile/presentation/actions/action.dart';
import 'package:immich_mobile/providers/asset_upload_coordinator.provider.dart';
import 'package:immich_mobile/providers/backup/asset_upload_progress.provider.dart';
import 'package:immich_mobile/providers/infrastructure/toast.provider.dart';
import 'package:immich_mobile/services/foreground_upload.service.dart';
import 'package:immich_mobile/utils/error_handler.dart';
import 'package:immich_ui/immich_ui.dart';

final _stateProvider = Provider.family.autoDispose<List<LocalAsset>?, ActionSource>((ref, source) {
  final assets = ref.watch(assetsActionProvider(source));
  final local = assets.backedUp(isBackedUp: false).local().toList(growable: false);
  return local.isEmpty ? null : local;
}, dependencies: [assetsActionProvider]);

class UploadAction extends AssetActionBuilder {
  final bool showProgress;

  const UploadAction({required super.source, this.showProgress = false});

  @override
  ActionItem? create(BuildContext context, WidgetRef ref) {
    final assets = ref.watch(_stateProvider(source));
    if (assets == null) {
      return null;
    }

    return .new(icon: Icons.backup_outlined, label: context.t.upload, onAction: () => _upload(context, ref, assets));
  }

  Future<void> _upload(BuildContext context, WidgetRef ref, List<LocalAsset> assets) async {
    try {
      if (!showProgress) {
        await uploadAssets(context, ref, assets, source: source);
        return;
      }

      // The dialog is not awaited: it stays up while the upload runs and is
      // dismissed below, unless the user cancelled it themselves first
      var isDialogOpen = true;
      unawaited(
        showDialog<void>(
          context: context,
          barrierDismissible: false,
          builder: (_) => const _UploadProgressDialog(),
        ).whenComplete(() => isDialogOpen = false),
      );

      await uploadAssets(context, ref, assets, source: source);

      if (isDialogOpen && context.mounted) {
        Navigator.of(context, rootNavigator: true).pop();
      }
    } catch (error, stack) {
      handleError(error, stack: stack, description: "Failed to upload the assets");
    }
  }
}

@visibleForTesting
Future<void> uploadAssets(
  BuildContext context,
  WidgetRef ref,
  List<LocalAsset> assets, {
  required ActionSource source,
}) async {
  final progress = ref.read(assetUploadProgressProvider.notifier);
  final uploads = ref.read(assetUploadCoordinatorProvider);
  final toastService = ref.read(toastServiceProvider);
  final errorMessage = context.t.scaffold_body_error_occurred;

  final cancelToken = Completer<void>();
  ref.read(manualUploadCancelTokenProvider.notifier).state = cancelToken;

  final uploaded = <String>{};
  final failed = <String>{};
  for (final asset in assets) {
    progress.setProgress(asset.id, 0.0);
  }

  try {
    await uploads.upload(
      source: source,
      assets: assets,
      cancelToken: cancelToken,
      callbacks: UploadCallbacks(
        onProgress: (id, _, bytes, total) => progress.setProgress(id, total > 0 ? bytes / total : 0.0),
        onSuccess: (id, _) {
          uploaded.add(id);
          progress.remove(id);
        },
        onError: (id, _) {
          failed.add(id);
          progress.setError(id);
        },
      ),
    );
  } finally {
    ref.read(manualUploadCancelTokenProvider.notifier).state = null;
  }

  final uploadedCount = uploaded.difference(failed).length;
  if (!cancelToken.isCompleted && (uploadedCount != assets.length || failed.isNotEmpty)) {
    toastService.error(errorMessage);
  }

  unawaited(Future.delayed(const Duration(seconds: 2), progress.clear));
}

class _UploadProgressDialog extends ConsumerWidget {
  const _UploadProgressDialog();

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    final progressMap = ref.watch(assetUploadProgressProvider);

    final values = progressMap.values.where((value) => value >= 0).toList(growable: false);
    final progress = values.isEmpty ? 0.0 : values.reduce((a, b) => a + b) / values.length;
    final hasError = progressMap.values.any((value) => value < 0);

    return AlertDialog(
      title: Text(context.t.uploading),
      content: Column(
        mainAxisSize: .min,
        children: [
          if (hasError)
            const Icon(Icons.error_outline, color: Colors.red, size: 48)
          else
            CircularProgressIndicator(value: progress > 0 ? progress : null),
          const SizedBox(height: 16),
          Text(hasError ? context.t.scaffold_body_error_occurred : '${(progress * 100).toInt()}%'),
        ],
      ),
      actions: [
        ImmichTextButton(
          onPressed: () {
            ref.read(manualUploadCancelTokenProvider)?.complete();
            ref.read(manualUploadCancelTokenProvider.notifier).state = null;
            Navigator.of(context, rootNavigator: true).pop();
          },
          labelText: context.t.cancel,
        ),
      ],
    );
  }
}