flutter_server_box/lib/view/widget/server_func_btns.dart

import 'dart:io';

import 'package:flutter/material.dart';
import 'package:flutter_gen/gen_l10n/l10n.dart';
import 'package:toolbox/core/extension/context/common.dart';
import 'package:toolbox/core/extension/context/dialog.dart';
import 'package:toolbox/core/extension/context/snackbar.dart';
import 'package:toolbox/core/extension/ssh_client.dart';
import 'package:toolbox/core/extension/uint8list.dart';
import 'package:toolbox/data/model/pkg/manager.dart';
import 'package:toolbox/data/model/server/dist.dart';
import 'package:toolbox/data/res/provider.dart';

import '../../core/route.dart';
import '../../core/utils/misc.dart';
import '../../core/utils/platform.dart';
import '../../core/utils/server.dart';
import '../../data/model/app/menu.dart';
import '../../data/model/pkg/upgrade_info.dart';
import '../../data/model/server/server_private_info.dart';
import '../../data/model/server/snippet.dart';
import 'popup_menu.dart';
import 'tag.dart';

class ServerFuncBtnsTopRight extends StatelessWidget {
  final ServerPrivateInfo spi;
  final S s;

  const ServerFuncBtnsTopRight({
    super.key,
    required this.spi,
    required this.s,
  });

  @override
  Widget build(BuildContext context) {
    return PopupMenu<ServerTabMenuType>(
      items: ServerTabMenuType.values
          .map((e) => PopupMenuItem<ServerTabMenuType>(
                value: e,
                child: Row(
                  children: [
                    Icon(e.icon),
                    const SizedBox(
                      width: 10,
                    ),
                    Text(e.text(s)),
                  ],
                ),
              ))
          .toList(),
      padding: const EdgeInsets.symmetric(horizontal: 10),
      onSelected: (val) => _onTapMoreBtns(val, spi, context, s),
    );
  }
}

class ServerFuncBtns extends StatelessWidget {
  const ServerFuncBtns({
    super.key,
    required this.spi,
    required this.s,
    this.iconSize,
  });

  final ServerPrivateInfo spi;
  final S s;
  final double? iconSize;

  @override
  Widget build(BuildContext context) {
    return Row(
      mainAxisAlignment: MainAxisAlignment.spaceAround,
      children: ServerTabMenuType.values
          .map((e) => IconButton(
                onPressed: () => _onTapMoreBtns(e, spi, context, s),
                padding: EdgeInsets.zero,
                tooltip: e.name,
                icon: Icon(e.icon, size: iconSize ?? 15),
              ))
          .toList(),
    );
  }
}

void _onTapMoreBtns(
  ServerTabMenuType value,
  ServerPrivateInfo spi,
  BuildContext context,
  S s,
) async {
  switch (value) {
    case ServerTabMenuType.pkg:
      _onPkg(context, s, spi);
      break;
    case ServerTabMenuType.sftp:
      AppRoute.sftp(spi: spi).checkGo(
        context: context,
        check: () => _checkClient(context, spi.id, s.waitConnection),
      );
      break;
    case ServerTabMenuType.snippet:
      final snippets = await showDialog<List<Snippet>>(
        context: context,
        builder: (_) => TagPicker<Snippet>(
          items: Providers.snippet.snippets,
          tags: Providers.server.tags.toSet(),
        ),
      );
      if (snippets == null) {
        return;
      }
      final result = await Providers.server.runSnippets(spi.id, snippets);
      if (result != null && result.isNotEmpty) {
        context.showRoundDialog(
          title: Text(s.result),
          child: Text(result),
          actions: [
            TextButton(
              onPressed: () => copy2Clipboard(result),
              child: Text(s.copy),
            )
          ],
        );
      }
      break;
    case ServerTabMenuType.docker:
      AppRoute.docker(spi: spi).checkGo(
        context: context,
        check: () => _checkClient(context, spi.id, s.waitConnection),
      );
      break;
    case ServerTabMenuType.process:
      AppRoute.process(spi: spi).checkGo(
        context: context,
        check: () => _checkClient(context, spi.id, s.waitConnection),
      );
      break;
    case ServerTabMenuType.terminal:
      _gotoSSH(spi, context);
      break;
  }
}

Future<void> _gotoSSH(
  ServerPrivateInfo spi,
  BuildContext context,
) async {
  // as a `Mobile first` app -> handle mobile first
  //
  // run built-in ssh on macOS due to incompatibility
  if (!isDesktop || isMacOS) {
    AppRoute.ssh(spi: spi).go(context);
    return;
  }
  final extraArgs = <String>[];
  if (spi.port != 22) {
    extraArgs.addAll(['-p', '${spi.port}']);
  }

  final path = () {
    final tempKeyFileName = 'srvbox_pk_${spi.pubKeyId}';
    return joinPath(Directory.systemTemp.path, tempKeyFileName);
  }();
  final file = File(path);
  final shouldGenKey = spi.pubKeyId != null;
  if (shouldGenKey) {
    if (await file.exists()) {
      await file.delete();
    }
    await file.writeAsString(getPrivateKey(spi.pubKeyId!));
    extraArgs.addAll(["-i", path]);
  }

  List<String> sshCommand = ["ssh", "${spi.user}@${spi.ip}"] + extraArgs;
  final system = Platform.operatingSystem;
  switch (system) {
    case "windows":
      await Process.start("cmd", ["/c", "start"] + sshCommand);
      break;
    case "linux":
      await Process.start("x-terminal-emulator", ["-e"] + sshCommand);
      break;
    default:
      context.showSnackBar('Mismatch system: $system');
  }
  // For security reason, delete the private key file after use
  if (shouldGenKey) {
    if (!await file.exists()) return;
    await Future.delayed(const Duration(seconds: 2), file.delete);
  }
}

bool _checkClient(BuildContext context, String id, String msg) {
  final server = Providers.server.servers[id];
  if (server == null || server.client == null) {
    context.showSnackBar(msg);
    return false;
  }
  return true;
}

Future<void> _onPkg(BuildContext context, S s, ServerPrivateInfo spi) async {
  final server = spi.findServer;
  if (server == null) {
    context.showSnackBar(s.noClient);
    return;
  }
  final sys = server.status.sysVer;
  final pkg = PkgManager.fromDist(sys.dist);

  // Update pkg list
  context.showLoadingDialog();
  final updateCmd = pkg?.update;
  if (updateCmd != null) {
    await server.client!.execWithPwd(
      updateCmd,
      context: context,
    );
  }
  context.pop();

  final listCmd = pkg?.listUpdate;
  if (listCmd == null) {
    context.showSnackBar('Unsupported dist: $sys');
    return;
  }

  // Get upgrade list
  context.showLoadingDialog();
  final result = await server.client?.run(listCmd).string;
  context.pop();
  if (result == null) {
    context.showSnackBar(s.noResult);
    return;
  }
  final list = pkg?.updateListRemoveUnused(result.split('\n'));
  final upgradeable = list?.map((e) => UpgradePkgInfo(e, pkg)).toList();
  if (upgradeable == null || upgradeable.isEmpty) {
    context.showSnackBar(s.noUpdateAvailable);
    return;
  }
  final args = upgradeable.map((e) => e.package).join(' ');
  final isSU = server.spi.user == 'root';
  final upgradeCmd = isSU ? pkg?.upgrade(args) : 'sudo ${pkg?.upgrade(args)}';

  // Confirm upgrade
  final gotoUpgrade = await context.showRoundDialog<bool>(
    title: Text(s.attention),
    child: SingleChildScrollView(
      child: Text('${s.foundNUpdate(upgradeable.length)}\n\n$upgradeCmd'),
    ),
    actions: [
      TextButton(
        onPressed: () => context.pop(true),
        child: Text(s.update),
      ),
    ],
  );

  if (gotoUpgrade != true) return;

  AppRoute.ssh(spi: spi, initCmd: upgradeCmd).checkGo(
    context: context,
    check: () => _checkClient(context, spi.id, s.waitConnection),
  );
}