new: shutdown | reboot on rootless user

lib/data/model/app/shell_func.dart

@@ -11,6 +11,8 @@ enum AppShellFuncType {
   status,
   docker,
   process,
+  shutdown,
+  reboot,
   ;
 
   String get flag {
@@ -21,6 +23,10 @@ enum AppShellFuncType {
         return 'd';
       case AppShellFuncType.process:
         return 'p';
+      case AppShellFuncType.shutdown:
+        return 'sd';
+      case AppShellFuncType.reboot:
+        return 'r';
     }
   }
 
@@ -36,6 +42,10 @@ enum AppShellFuncType {
         return 'dockeR';
       case AppShellFuncType.process:
         return 'process';
+      case AppShellFuncType.shutdown:
+        return 'ShutDown';
+      case AppShellFuncType.reboot:
+        return 'Reboot';
     }
   }
 
@@ -68,10 +78,24 @@ else
 \tps -ax
 fi
 ''';
+      case AppShellFuncType.shutdown:
+        return '''
+if [ "\$userId" = "0" ]; then
+\tshutdown -h now
+else
+\tsudo -S shutdown -h now
+fi''';
+      case AppShellFuncType.reboot:
+        return '''
+if [ "\$userId" = "0" ]; then
+\treboot
+else
+\tsudo -S reboot
+fi''';
     }
   }
 
-  static String get shellScript {
+  static final String shellScript = () {
     final sb = StringBuffer();
     // Write each func
     for (final func in values) {
@@ -98,7 +122,7 @@ ${func.cmd}
     ;;
 esac''');
     return sb.toString();
-  }
+  }();
 }
 
 extension EnumX on Enum {
@@ -196,6 +220,7 @@ export LANG=en_US.UTF-8
 isLinux=\$(uname 2>&1 | grep "Linux")
 # Link /bin/sh to busybox?
 isBusybox=\$(ls -l /bin/sh | grep "busybox")
+userId=\$(id -u)
 
 ${AppShellFuncType.shellScript}
 """;

lib/data/provider/docker.dart

@@ -1,12 +1,10 @@
 import 'dart:async';
-import 'dart:typed_data';
 
 import 'package:dartssh2/dartssh2.dart';
 import 'package:flutter/material.dart';
 import 'package:logging/logging.dart';
 import 'package:toolbox/core/extension/ssh_client.dart';
 import 'package:toolbox/core/extension/stringx.dart';
-import 'package:toolbox/core/utils/ui.dart';
 import 'package:toolbox/data/model/app/shell_func.dart';
 import 'package:toolbox/data/model/docker/image.dart';
 import 'package:toolbox/data/model/docker/ps.dart';
@@ -34,34 +32,33 @@ class DockerProvider extends ChangeNotifier {
   String? version;
   String? edition;
   DockerErr? error;
-  PwdRequestFunc? onPwdReq;
   String? hostId;
   String? runLog;
-  bool isRequestingPwd = false;
+  BuildContext? context;
 
   void init(
     SSHClient client,
     String userName,
     PwdRequestFunc onPwdReq,
     String hostId,
+    BuildContext context,
   ) {
     this.client = client;
     this.userName = userName;
-    this.onPwdReq = onPwdReq;
+    this.context = context;
     this.hostId = hostId;
   }
 
   void clear() {
-    client = userName = error = items = version = edition = onPwdReq = null;
-    isRequestingPwd = false;
+    client = userName = error = items = version = edition = context = null;
     hostId = runLog = images = null;
   }
 
   Future<void> refresh() async {
     var raw = '';
-    await client!.exec(
+    await client!.execWithPwd(
       AppShellFuncType.docker.exec,
-      onStderr: _onPwd,
+      context: context,
       onStdout: (data, _) => raw = '$raw$data',
     );
 
@@ -145,13 +142,6 @@ class DockerProvider extends ChangeNotifier {
     // }
   }
 
-  Future<void> _onPwd(String event, StreamSink<Uint8List> stdin) async {
-    if (isRequestingPwd) return;
-    isRequestingPwd = true;
-    await onPwd(event, stdin, onPwdReq);
-    isRequestingPwd = false;
-  }
-
   Future<DockerErr?> stop(String id) async => await run('docker stop $id');
 
   Future<DockerErr?> start(String id) async => await run('docker start $id');
@@ -168,16 +158,14 @@ class DockerProvider extends ChangeNotifier {
 
     runLog = '';
     final errs = <String>[];
-    final code = await client!.exec(
+    final code = await client!.execWithPwd(
       _wrap(cmd),
-      onStderr: (data, sink) {
-        _onPwd(data, sink);
-        errs.add(data);
-      },
+      context: context,
       onStdout: (data, _) {
         runLog = '$runLog$data';
         notifyListeners();
       },
+      onStderr: (data, _) => errs.add(data),
     );
     runLog = null;
     notifyListeners();

lib/data/provider/pkg.dart

@@ -1,12 +1,10 @@
 import 'dart:async';
-import 'dart:typed_data';
 
 import 'package:dartssh2/dartssh2.dart';
 import 'package:flutter/material.dart';
 import 'package:logging/logging.dart';
 import 'package:toolbox/core/extension/ssh_client.dart';
 import 'package:toolbox/core/extension/uint8list.dart';
-import 'package:toolbox/core/utils/ui.dart';
 import 'package:toolbox/data/model/pkg/manager.dart';
 import 'package:toolbox/data/model/pkg/upgrade_info.dart';
 import 'package:toolbox/data/model/server/dist.dart';
@@ -19,7 +17,7 @@ class PkgProvider extends ChangeNotifier {
   PkgManager? type;
   Function()? onUpgrade;
   Function()? onUpdate;
-  PwdRequestFunc? onPasswordRequest;
+  BuildContext? context;
 
   String? whoami;
   List<UpgradePkgInfo>? upgradeable;
@@ -27,20 +25,18 @@ class PkgProvider extends ChangeNotifier {
   String? upgradeLog;
   String? updateLog;
   String lastLog = '';
-  bool isRequestingPwd = false;
 
   Future<void> init(
     SSHClient client,
     Dist? dist,
     Function() onUpgrade,
     Function() onUpdate,
-    PwdRequestFunc onPasswordRequest,
     String user,
+    BuildContext context,
   ) async {
     this.client = client;
     this.dist = dist;
     this.onUpgrade = onUpgrade;
-    this.onPasswordRequest = onPasswordRequest;
     whoami = user;
 
     type = fromDist(dist);
@@ -52,9 +48,8 @@ class PkgProvider extends ChangeNotifier {
   bool get isSU => whoami == 'root';
 
   void clear() {
-    client = dist = updateLog = upgradeLog = upgradeable =
-        error = whoami = onUpdate = onUpgrade = onPasswordRequest = null;
-    isRequestingPwd = false;
+    client = dist = updateLog = upgradeLog =
+        upgradeable = error = whoami = onUpdate = onUpgrade = context = null;
   }
 
   Future<void> refresh() async {
@@ -78,9 +73,9 @@ class PkgProvider extends ChangeNotifier {
   Future<String?> _update() async {
     final updateCmd = type?.update;
     if (updateCmd != null) {
-      await client!.exec(
+      await client!.execWithPwd(
         _wrap(updateCmd),
-        onStderr: _onPwd,
+        context: context,
         onStdout: (data, sink) {
           updateLog = (updateLog ?? '') + data;
           if (onUpdate != null) onUpdate!();
@@ -105,9 +100,9 @@ class PkgProvider extends ChangeNotifier {
       return;
     }
 
-    await client!.exec(
+    await client!.execWithPwd(
       _wrap(upgradeCmd),
-      onStderr: _onPwd,
+      context: context,
       onStdout: (log, sink) {
         if (lastLog == log.trim()) return;
         upgradeLog = (upgradeLog ?? '') + log;
@@ -121,13 +116,6 @@ class PkgProvider extends ChangeNotifier {
     refresh();
   }
 
-  Future<void> _onPwd(String event, StreamSink<Uint8List> stdin) async {
-    if (isRequestingPwd) return;
-    isRequestingPwd = true;
-    await onPwd(event, stdin, onPasswordRequest);
-    isRequestingPwd = false;
-  }
-
   String _wrap(String cmd) =>
       'export LANG=en_US.utf-8 && ${isSU ? "" : "sudo -S "}$cmd';
 }