feat: systemd management (#532)

lib/data/provider/systemd.dart

@@ -0,0 +1,167 @@
+import 'package:dartssh2/dartssh2.dart';
+import 'package:fl_lib/fl_lib.dart';
+import 'package:server_box/core/extension/ssh_client.dart';
+import 'package:server_box/data/model/app/shell_func.dart';
+import 'package:server_box/data/model/server/server_private_info.dart';
+import 'package:server_box/data/model/server/systemd.dart';
+import 'package:server_box/data/res/provider.dart';
+
+final class SystemdProvider {
+  late final SSHClient _client;
+
+  SystemdProvider.init(ServerPrivateInfo spi) {
+    _client = Pros.server.pick(spi: spi)!.client!;
+    getUnits();
+  }
+
+  final isBusy = false.vn;
+  final isRoot = false.vn;
+  final units = <SystemdUnit>[].vn;
+
+  Future<void> getUnits() async {
+    isBusy.value = true;
+
+    try {
+      final result = await _client.runScriptIn(_getUnitsCmd);
+      final units = result.split('\n');
+      final isRootRaw = units.firstOrNull;
+      isRoot.value = isRootRaw == '0';
+
+      final userUnits = <String>[];
+      final systemUnits = <String>[];
+      for (final unit in units.skip(1)) {
+        if (unit.startsWith('/etc/systemd/system')) {
+          systemUnits.add(unit);
+        } else if (unit.startsWith('~/.config/systemd/user')) {
+          userUnits.add(unit);
+        } else if (unit.trim().isNotEmpty) {
+          Loggers.app.warning('Unknown unit: $unit');
+        }
+      }
+
+      final parsedUserUnits =
+          await _parseUnitObj(userUnits, SystemdUnitScope.user);
+      final parsedSystemUnits =
+          await _parseUnitObj(systemUnits, SystemdUnitScope.system);
+      this.units.value = [...parsedUserUnits, ...parsedSystemUnits];
+    } catch (e, s) {
+      Loggers.app.warning('Parse systemd', e, s);
+    }
+
+    isBusy.value = false;
+  }
+
+  Future<List<SystemdUnit>> _parseUnitObj(
+    List<String> unitNames,
+    SystemdUnitScope scope,
+  ) async {
+    final unitNames_ = unitNames
+        .map((e) => e.trim().split('/').last.split('.').first)
+        .toList();
+    final script = '''
+for unit in ${unitNames_.join(' ')}; do
+  state=\$(systemctl show --no-pager \$unit)
+  echo -n "${ShellFunc.seperator}\n\$state"
+done
+''';
+    final result = await _client.runScriptIn(script);
+    final units = result.split(ShellFunc.seperator);
+
+    final parsedUnits = <SystemdUnit>[];
+    for (final unit in units) {
+      final parts = unit.split('\n');
+      var name = '';
+      var type = '';
+      var state = '';
+      String? description;
+      for (final part in parts) {
+        if (part.startsWith('Id=')) {
+          final val = _getIniVal(part).split('.');
+          name = val.first;
+          type = val.last;
+          continue;
+        }
+        if (part.startsWith('ActiveState=')) {
+          state = _getIniVal(part);
+          continue;
+        }
+        if (part.startsWith('Description=')) {
+          description = _getIniVal(part);
+          continue;
+        }
+      }
+
+      final unitType = SystemdUnitType.fromString(type);
+      if (unitType == null) {
+        Loggers.app.warning('Unit type: $type');
+        continue;
+      }
+      final unitState = SystemdUnitState.fromString(state);
+      if (unitState == null) {
+        Loggers.app.warning('Unit state: $state');
+        continue;
+      }
+
+      parsedUnits.add(SystemdUnit(
+        name: name,
+        type: unitType,
+        scope: scope,
+        state: unitState,
+        description: description,
+      ));
+    }
+
+    parsedUnits.sort((a, b) {
+      // user units first
+      if (a.scope != b.scope) {
+        return a.scope == SystemdUnitScope.user ? -1 : 1;
+      }
+      // active units first
+      if (a.state != b.state) {
+        return a.state == SystemdUnitState.active ? -1 : 1;
+      }
+      return a.name.compareTo(b.name);
+    });
+    return parsedUnits;
+  }
+}
+
+String _getIniVal(String line) {
+  return line.split('=').last;
+}
+
+const _getUnitsCmd = '''
+# If root, get system & user units, otherwise get user units
+uid=\$(id -u)
+echo \$uid
+
+get_files() {
+  unit_type=\$1
+  base_dir=\$2
+
+  # If base_dir is not a directory, return
+  if [ ! -d "\$base_dir" ]; then
+    return
+  fi
+
+  find "\$base_dir" -type f -name "*.\$unit_type" -print | sort
+}
+
+get_type_files() {
+    unit_type=\$1
+
+    base_dir=""
+    if [ "\$uid" -eq 0 ]; then
+        get_files \$unit_type /etc/systemd/system
+        get_files \$unit_type ~/.config/systemd/user
+    else
+        get_files \$unit_type ~/.config/systemd/user
+    fi
+}
+
+types="service socket mount timer"
+
+for type in \$types; do
+    get_type_files \$type
+done
+''';