new: gpu status (#201)

lib/data/model/app/shell_func.dart

@@ -33,7 +33,8 @@ enum ShellFunc {
 
   /// Issue #168
   /// Use `sh` for compatibility
-  static final installShellCmd = """
+  static final installShellCmd =
+      """
 mkdir -p $_homeVar/$_srvBoxDir
 cat << 'EOF' > $_installShellPath
 ${ShellFunc.allScript}
@@ -134,7 +135,8 @@ fi''';
 
   static final String allScript = () {
     final sb = StringBuffer();
-    sb.write('''
+    sb.write(
+        '''
 #!/bin/sh
 # Script for ServerBox app v1.0.${BuildData.build}
 # DO NOT delete this file while app is running
@@ -153,7 +155,8 @@ userId=\$(id -u)
 ''');
     // Write each func
     for (final func in values) {
-      sb.write('''
+      sb.write(
+          '''
 ${func.name}() {
 ${func._cmd.split('\n').map((e) => '\t$e').join('\n')}
 }
@@ -199,7 +202,9 @@ enum StatusCmdType {
   tempType,
   tempVal,
   host,
-  diskio;
+  diskio,
+  nvdia,
+  ;
 }
 
 /// Cmds for linux server
@@ -217,6 +222,7 @@ const _statusCmds = [
   'cat /sys/class/thermal/thermal_zone*/temp',
   'hostname',
   'cat /proc/diskstats',
+  'nvidia-smi -q -x',
 ];
 
 enum DockerCmdType {

lib/data/model/server/nvdia.dart

@@ -0,0 +1,132 @@
+import 'package:xml/xml.dart';
+
+/// [
+///   {
+///     "name": "GeForce RTX 3090",
+///     "temp": 40,
+///     "power": "30W / 350W",
+///     "memory": {
+///       "total": 24268,
+///       "used": 240,
+///       "unit": "MiB",
+///       "processes": [
+///         {
+///           "pid": 1456,
+///           "name": "/usr/lib/xorg/Xorg",
+///           "memory": 40
+///         },
+///       ]
+///     },
+///   }
+/// ]
+///
+
+class NvdiaSmi {
+  static List<NvdiaSmiItem> fromXml(String raw) {
+    final xmlData = XmlDocument.parse(raw);
+    final gpus = xmlData.findAllElements('gpu');
+    final result = List<NvdiaSmiItem?>.generate(gpus.length, (index) {
+      final gpu = gpus.elementAt(index);
+      final name = gpu.findElements('product_name').firstOrNull?.innerText;
+      final temp = gpu
+          .findElements('temperature')
+          .firstOrNull
+          ?.findElements('gpu_temp')
+          .firstOrNull
+          ?.innerText;
+      final power = gpu.findElements('gpu_power_readings').firstOrNull;
+      final powerDraw =
+          power?.findElements('power_draw').firstOrNull?.innerText;
+      final powerLimit =
+          power?.findElements('current_power_limit').firstOrNull?.innerText;
+      final memory = gpu.findElements('fb_memory_usage').firstOrNull;
+      final memoryUsed = memory?.findElements('used').firstOrNull?.innerText;
+      final memoryTotal = memory?.findElements('total').firstOrNull?.innerText;
+      final processes = gpu
+          .findElements('processes')
+          .firstOrNull
+          ?.findElements('process_info');
+      final memoryProcesses =
+          List<NvdiaSmiMemProcess?>.generate(processes?.length ?? 0, (index) {
+        final process = processes?.elementAt(index);
+        final pid = process?.findElements('pid').firstOrNull?.innerText;
+        final name =
+            process?.findElements('process_name').firstOrNull?.innerText;
+        final memory =
+            process?.findElements('used_memory').firstOrNull?.innerText;
+        if (pid != null && name != null && memory != null) {
+          return NvdiaSmiMemProcess(
+            int.parse(pid),
+            name,
+            int.parse(
+              memory.split(' ').firstOrNull ?? '0',
+            ),
+          );
+        }
+        return null;
+      });
+      memoryProcesses.removeWhere((element) => element == null);
+      if (name != null &&
+          temp != null &&
+          powerDraw != null &&
+          powerLimit != null &&
+          memory != null) {
+        return NvdiaSmiItem(
+          name,
+          int.parse(temp.split(' ').firstOrNull ?? '0'),
+          '$powerDraw / $powerLimit',
+          NvdiaSmiMem(
+            int.parse(memoryTotal?.split(' ').firstOrNull ?? '0'),
+            int.parse(memoryUsed?.split(' ').firstOrNull ?? '0'),
+            'MiB',
+            List.from(memoryProcesses),
+          ),
+        );
+      }
+      return null;
+    });
+    result.removeWhere((element) => element == null);
+    return List.from(result);
+  }
+}
+
+class NvdiaSmiItem {
+  final String name;
+  final int temp;
+  final String power;
+  final NvdiaSmiMem memory;
+
+  const NvdiaSmiItem(this.name, this.temp, this.power, this.memory);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiItem{name: $name, temp: $temp, power: $power, memory: $memory}';
+  }
+}
+
+class NvdiaSmiMem {
+  final int total;
+  final int used;
+  final String unit;
+  final List<NvdiaSmiMemProcess> processes;
+
+  const NvdiaSmiMem(this.total, this.used, this.unit, this.processes);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiMem{total: $total, used: $used, unit: $unit, processes: $processes}';
+  }
+}
+
+class NvdiaSmiMemProcess {
+  final int pid;
+  final String name;
+  final int memory;
+
+  const NvdiaSmiMemProcess(this.pid, this.name, this.memory);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiMemProcess{pid: $pid, name: $name, memory: $memory}';
+  }
+}

lib/data/model/server/server.dart

@@ -4,6 +4,7 @@ import 'package:toolbox/data/model/server/cpu.dart';
 import 'package:toolbox/data/model/server/disk.dart';
 import 'package:toolbox/data/model/server/memory.dart';
 import 'package:toolbox/data/model/server/net_speed.dart';
+import 'package:toolbox/data/model/server/nvdia.dart';
 import 'package:toolbox/data/model/server/server_private_info.dart';
 import 'package:toolbox/data/model/server/system.dart';
 import 'package:toolbox/data/model/server/temp.dart';
@@ -55,6 +56,7 @@ class ServerStatus {
   SystemType system;
   String? err;
   DiskIO diskIO;
+  List<NvdiaSmiItem>? nvdia;
 
   /// Whether is connectting, parsing and etc.
   bool _isBusy = false;
@@ -72,6 +74,7 @@ class ServerStatus {
     required this.system,
     required this.diskIO,
     this.err,
+    this.nvdia,
   });
 }
 

lib/data/model/server/server_status_update_req.dart

@@ -1,3 +1,4 @@
+import 'package:toolbox/data/model/server/nvdia.dart';
 import 'package:toolbox/data/model/server/server.dart';
 import 'package:toolbox/data/model/server/system.dart';
 import 'package:toolbox/data/res/logger.dart';
@@ -110,6 +111,13 @@ Future<ServerStatus> _getLinuxStatus(ServerStatusUpdateReq req) async {
   } catch (e, s) {
     Loggers.parse.warning(e, s);
   }
+
+  try {
+    final nvdia = NvdiaSmi.fromXml(StatusCmdType.nvdia.find(segments));
+    req.ss.nvdia = nvdia;
+  } catch (e, s) {
+    Loggers.parse.warning(e, s);
+  }
   return req.ss;
 }
 

lib/data/res/default.dart

@@ -11,6 +11,7 @@ class Defaults {
     'cpu',
     'mem',
     'swap',
+    'gpu',
     'disk',
     'net',
     'temp'