optimized apt experience

lib/data/provider/apt.dart

@@ -1,25 +1,41 @@
+import 'dart:async';
+import 'dart:convert';
+import 'dart:typed_data';
+
 import 'package:dartssh2/dartssh2.dart';
+import 'package:logging/logging.dart';
 import 'package:toolbox/core/extension/uint8list.dart';
 import 'package:toolbox/core/provider_base.dart';
 import 'package:toolbox/data/model/apt/upgrade_pkg_info.dart';
 import 'package:toolbox/data/model/distribution.dart';
 
+typedef PwdRequestFunc = Future<String> Function(String? userName);
+final pwdRequestWithUserReg = RegExp(r'\[sudo\] password for (.+):');
+
 class AptProvider extends BusyProvider {
+  final logger = Logger('AptProvider');
+
   SSHClient? client;
   Distribution? dist;
+  Function()? onUpgrade;
+  Function()? onUpdate;
+  PwdRequestFunc? onPasswordRequest;
+
   String? whoami;
   List<UpgradePkgInfo>? upgradeable;
   String? error;
+  String? upgradeLog;
   String? updateLog;
-  Function()? onUpgrade;
+  String? savedPwd;
 
   AptProvider();
 
-  Future<void> init(
-      SSHClient client, Distribution dist, Function() onUpgrade) async {
+  Future<void> init(SSHClient client, Distribution dist, Function() onUpgrade,
+      Function() onUpdate, PwdRequestFunc onPasswordRequest) async {
     this.client = client;
     this.dist = dist;
     this.onUpgrade = onUpgrade;
+    this.onPasswordRequest = onPasswordRequest;
     whoami = (await client.run('whoami').string).trim();
   }
 
@@ -30,8 +46,12 @@ class AptProvider extends BusyProvider {
     dist = null;
     upgradeable = null;
     error = null;
-    updateLog = null;
-    whoami = null;
+    upgradeLog = null;
+    updateLog = whoami = null;
+    savedPwd = null;
+    onUpgrade = null;
+    onUpdate = null;
+    onPasswordRequest = null;
   }
 
   Future<void> refreshInstalled() async {
@@ -44,38 +64,43 @@ class AptProvider extends BusyProvider {
     try {
       getUpgradeableList(result);
     } catch (e) {
-      error = e.toString();
+      error = '[Server Raw]:\n$result\n[App Error]:\n$e';
     } finally {
       notifyListeners();
     }
   }
 
-  void getUpgradeableList(String raw) {
+  void getUpgradeableList(String? raw) {
+    if (raw == null) return;
+    var list = raw.split('\n');
     switch (dist) {
       case Distribution.rehl:
-        var list = raw.split('\n').sublist(2);
+        list = list.sublist(2);
         list.removeWhere((element) => element.isEmpty);
         final endLine = list.lastIndexWhere(
             (element) => element.contains('Obsoleting Packages'));
         list = list.sublist(0, endLine);
-        upgradeable = list.map((e) => UpgradePkgInfo(e, dist!)).toList();
         break;
-      case Distribution.debian:
-      case Distribution.unknown:
       default:
-        final list = raw.split('\n').sublist(4);
+        list = list.sublist(4);
         list.removeWhere((element) => element.isEmpty);
-        upgradeable = list.map((e) => UpgradePkgInfo(e, dist!)).toList();
     }
+    upgradeable = list.map((e) => UpgradePkgInfo(e, dist!)).toList();
   }
 
   Future<String> _update() async {
     switch (dist) {
       case Distribution.rehl:
-        return await client!.run('yum check-update').string;
-      case Distribution.debian:
+        return await client!.run(_wrap('yum check-update')).string;
       default:
-        await client!.run('apt update');
+        final session = await client!.execute(_wrap('apt update'));
+        session.stderr.listen((event) => _onPwd(event, session.stdin));
+        session.stdout.listen((event) {
+          updateLog = (updateLog ?? '') + event.string;
+          notifyListeners();
+          onUpdate!();
+        });
+        await session.done;
         return await client!.run('apt list --upgradeable').string;
     }
   }
@@ -85,24 +110,50 @@ class AptProvider extends BusyProvider {
       error = 'No client';
       return;
     }
-    updateLog = null;
 
-    final session = await client!.execute(upgradeCmd);
-    session.stdout.listen((data) {
-      updateLog = (updateLog ?? '') + data.string;
+    final upgradeCmd = () {
+      switch (dist) {
+        case Distribution.rehl:
+          return 'yum upgrade -y';
+        default:
+          return 'apt upgrade -y';
+      }
+    }();
+
+    final session = await client!.execute(_wrap(upgradeCmd));
+    session.stderr.listen((e) => _onPwd(e, session.stdin));
+
+    session.stdout.listen((data) async {
+      upgradeLog = (upgradeLog ?? '') + data.string;
       notifyListeners();
       onUpgrade!();
     });
+
+    upgradeLog = null;
+    await session.done;
     refreshInstalled();
   }
 
-  String get upgradeCmd {
-    switch (dist) {
-      case Distribution.rehl:
-        return 'yum upgrade -y';
-      case Distribution.debian:
-      default:
-        return 'apt upgrade -y';
+  Future<void> _onPwd(Uint8List e, StreamSink<Uint8List> stdin) async {
+    final event = e.string;
+    if (event.contains('[sudo] password for ')) {
+      final user = pwdRequestWithUserReg.firstMatch(event)?.group(1);
+      logger.info('sudo password request for $user');
+      final pwd = await () async {
+        if (savedPwd != null) return savedPwd!;
+        final inputPwd = await (onPasswordRequest ?? (_) async => '')(user);
+        if (inputPwd.isNotEmpty) {
+          savedPwd = inputPwd;
+        }
+        return inputPwd;
+      }();
+      if (pwd.isEmpty) {
+        logger.info('sudo password request cancelled');
+      }
+      stdin.add(Uint8List.fromList(utf8.encode(pwd + '\n')));
     }
   }
+
+  String _wrap(String cmd) =>
+      'export LANG=en_US.utf-8 && ${isSU ? "" : "sudo -S "}$cmd';
 }