Babylonia-terminal/babylonia_terminal_launcher/lib/models/proton.dart

import 'package:fixnum/fixnum.dart';
import 'package:flutter/material.dart';

import './../messages/steps/proton.pb.dart';
import './../providers/providers.dart';

enum ProtonInstallationState {
  idle,
  downloading,
  decompressing,
}

class Proton with ChangeNotifier {
  ProtonInstallationState protonState = ProtonInstallationState.idle;

  Int64 currentProgress = Int64(0);
  Int64 maxProgress = Int64(0);

  Future startInstallation(
      GameStateProvider gameStateProvider, String protonVersion) async {
    StartProtonInstallation(protonVersion: protonVersion).sendSignalToRust();
    final progressStream = ProtonDownloadProgress.rustSignalStream;
    await for (final rustSignal in progressStream) {
      currentProgress = rustSignal.message.current;
      maxProgress = rustSignal.message.max;

      if (protonState == ProtonInstallationState.idle) {
        protonState = ProtonInstallationState.downloading;
      }

      notifyListeners();

      if (currentProgress >= maxProgress) {
        break;
      }
    }

    final notificationDecompressingStream =
        NotifyProtonStartDecompressing.rustSignalStream;
    await for (final _ in notificationDecompressingStream) {
      protonState = ProtonInstallationState.decompressing;
      notifyListeners();
      break;
    }

    final notificationInstalledStream =
        NotifyProtonSuccessfullyInstalled.rustSignalStream;
    await for (final _ in notificationInstalledStream) {
      gameStateProvider.updateGameState();
      break;
    }
  }
}