immich/mobile/lib/pages/common/change_experience.page.dart

import 'dart:async';

import 'package:auto_route/auto_route.dart';
import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/domain/models/store.model.dart';
import 'package:immich_mobile/entities/store.entity.dart';
import 'package:immich_mobile/extensions/build_context_extensions.dart';
import 'package:immich_mobile/infrastructure/repositories/store.repository.dart';
import 'package:immich_mobile/providers/album/album.provider.dart';
import 'package:immich_mobile/providers/asset.provider.dart';
import 'package:immich_mobile/providers/background_sync.provider.dart';
import 'package:immich_mobile/providers/backup/backup.provider.dart';
import 'package:immich_mobile/providers/backup/manual_upload.provider.dart';
import 'package:immich_mobile/providers/gallery_permission.provider.dart';
import 'package:immich_mobile/providers/infrastructure/db.provider.dart';
import 'package:immich_mobile/providers/infrastructure/platform.provider.dart';
import 'package:immich_mobile/providers/infrastructure/readonly_mode.provider.dart';
import 'package:immich_mobile/providers/websocket.provider.dart';
import 'package:immich_mobile/services/background.service.dart';
import 'package:immich_mobile/utils/migration.dart';
import 'package:logging/logging.dart';
import 'package:permission_handler/permission_handler.dart';

@RoutePage()
class ChangeExperiencePage extends ConsumerStatefulWidget {
  final bool switchingToBeta;

  const ChangeExperiencePage({super.key, required this.switchingToBeta});

  @override
  ConsumerState createState() => _ChangeExperiencePageState();
}

class _ChangeExperiencePageState extends ConsumerState<ChangeExperiencePage> {
  AsyncValue<bool> hasMigrated = const AsyncValue.loading();

  @override
  void initState() {
    super.initState();
    WidgetsBinding.instance.addPostFrameCallback((_) => _handleMigration());
  }

  Future<void> _handleMigration() async {
    try {
      await _performMigrationLogic().timeout(
        const Duration(minutes: 3),
        onTimeout: () async {
          await IsarStoreRepository(ref.read(isarProvider)).upsert(StoreKey.betaTimeline, widget.switchingToBeta);
          await DriftStoreRepository(ref.read(driftProvider)).upsert(StoreKey.betaTimeline, widget.switchingToBeta);
        },
      );

      if (mounted) {
        setState(() {
          HapticFeedback.heavyImpact();
          hasMigrated = const AsyncValue.data(true);
        });
      }
    } catch (e, s) {
      Logger("ChangeExperiencePage").severe("Error during migration", e, s);
      if (mounted) {
        setState(() {
          hasMigrated = AsyncValue.error(e, s);
        });
      }
    }
  }

  Future<void> _performMigrationLogic() async {
    if (widget.switchingToBeta) {
      final assetNotifier = ref.read(assetProvider.notifier);
      if (assetNotifier.mounted) {
        assetNotifier.dispose();
      }
      final albumNotifier = ref.read(albumProvider.notifier);
      if (albumNotifier.mounted) {
        albumNotifier.dispose();
      }

      // Cancel uploads
      await Store.put(StoreKey.backgroundBackup, false);
      ref
          .read(backupProvider.notifier)
          .configureBackgroundBackup(enabled: false, onBatteryInfo: () {}, onError: (_) {});
      ref.read(backupProvider.notifier).setAutoBackup(false);
      ref.read(backupProvider.notifier).cancelBackup();
      ref.read(manualUploadProvider.notifier).cancelBackup();
      // Start listening to new websocket events
      ref.read(websocketProvider.notifier).stopListenToOldEvents();
      ref.read(websocketProvider.notifier).startListeningToBetaEvents();

      await ref.read(driftProvider).reset();
      await Store.put(StoreKey.shouldResetSync, true);
      final permission = await ref.read(galleryPermissionNotifier.notifier).requestGalleryPermission();

      if (permission.isGranted) {
        await ref.read(backgroundSyncProvider).syncLocal(full: true);
        await migrateDeviceAssetToSqlite(ref.read(isarProvider), ref.read(driftProvider));
        await migrateBackupAlbumsToSqlite(ref.read(isarProvider), ref.read(driftProvider));
        await migrateStoreToSqlite(ref.read(isarProvider), ref.read(driftProvider));
        await ref.read(backgroundServiceProvider).disableService();
      }
    } else {
      await ref.read(backgroundSyncProvider).cancel();
      ref.read(websocketProvider.notifier).stopListeningToBetaEvents();
      ref.read(websocketProvider.notifier).startListeningToOldEvents();
      ref.read(readonlyModeProvider.notifier).setReadonlyMode(false);
      await migrateStoreToIsar(ref.read(isarProvider), ref.read(driftProvider));
      await ref.read(backgroundServiceProvider).resumeServiceIfEnabled();
      await ref.read(backgroundWorkerFgServiceProvider).disable();
    }

    await IsarStoreRepository(ref.read(isarProvider)).upsert(StoreKey.betaTimeline, widget.switchingToBeta);
    await DriftStoreRepository(ref.read(driftProvider)).upsert(StoreKey.betaTimeline, widget.switchingToBeta);
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
      body: Center(
        child: Column(
          mainAxisSize: MainAxisSize.min,
          children: [
            AnimatedSwitcher(
              duration: Durations.long4,
              child: hasMigrated.when(
                data: (data) => const Icon(Icons.check_circle_rounded, color: Colors.green, size: 48.0),
                error: (error, stackTrace) => const Icon(Icons.error, color: Colors.red, size: 48.0),
                loading: () => const SizedBox(width: 50.0, height: 50.0, child: CircularProgressIndicator()),
              ),
            ),
            const SizedBox(height: 16.0),
            SizedBox(
              width: 300.0,
              child: AnimatedSwitcher(
                duration: Durations.long4,
                child: hasMigrated.when(
                  data: (data) => Text(
                    "Migration success!\nPlease close and reopen the app to apply changes",
                    style: context.textTheme.titleMedium,
                    textAlign: TextAlign.center,
                  ),
                  error: (error, stackTrace) => Text(
                    "Migration failed!\nError: $error",
                    style: context.textTheme.titleMedium,
                    textAlign: TextAlign.center,
                  ),
                  loading: () => Text(
                    "Data migration in progress...\nPlease wait and don't close this page",
                    style: context.textTheme.titleMedium,
                    textAlign: TextAlign.center,
                  ),
                ),
              ),
            ),
          ],
        ),
      ),
    );
  }
}