365 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Dart
		
	
	
	
	
	
			
		
		
	
	
			365 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Dart
		
	
	
	
	
	
import 'dart:convert';
 | 
						|
import 'dart:io';
 | 
						|
import 'dart:ui' as ui;
 | 
						|
 | 
						|
import 'package:desktop_multi_window/desktop_multi_window.dart';
 | 
						|
import 'package:flutter/material.dart';
 | 
						|
import 'package:flutter/services.dart';
 | 
						|
import 'package:flutter_hbb/common.dart';
 | 
						|
import 'package:flutter_hbb/common/shared_state.dart';
 | 
						|
import 'package:flutter_hbb/consts.dart';
 | 
						|
import 'package:flutter_hbb/models/state_model.dart';
 | 
						|
import 'package:flutter_hbb/desktop/pages/remote_page.dart';
 | 
						|
import 'package:flutter_hbb/desktop/widgets/remote_toolbar.dart';
 | 
						|
import 'package:flutter_hbb/desktop/widgets/tabbar_widget.dart';
 | 
						|
import 'package:flutter_hbb/desktop/widgets/material_mod_popup_menu.dart'
 | 
						|
    as mod_menu;
 | 
						|
import 'package:flutter_hbb/desktop/widgets/popup_menu.dart';
 | 
						|
import 'package:flutter_hbb/utils/multi_window_manager.dart';
 | 
						|
import 'package:flutter_svg/flutter_svg.dart';
 | 
						|
import 'package:get/get.dart';
 | 
						|
import 'package:bot_toast/bot_toast.dart';
 | 
						|
 | 
						|
import '../../models/platform_model.dart';
 | 
						|
 | 
						|
class _MenuTheme {
 | 
						|
  static const Color blueColor = MyTheme.button;
 | 
						|
  // kMinInteractiveDimension
 | 
						|
  static const double height = 20.0;
 | 
						|
  static const double dividerHeight = 12.0;
 | 
						|
}
 | 
						|
 | 
						|
class ConnectionTabPage extends StatefulWidget {
 | 
						|
  final Map<String, dynamic> params;
 | 
						|
 | 
						|
  const ConnectionTabPage({Key? key, required this.params}) : super(key: key);
 | 
						|
 | 
						|
  @override
 | 
						|
  State<ConnectionTabPage> createState() => _ConnectionTabPageState(params);
 | 
						|
}
 | 
						|
 | 
						|
class _ConnectionTabPageState extends State<ConnectionTabPage> {
 | 
						|
  final tabController =
 | 
						|
      Get.put(DesktopTabController(tabType: DesktopTabType.remoteScreen));
 | 
						|
  final contentKey = UniqueKey();
 | 
						|
  static const IconData selectedIcon = Icons.desktop_windows_sharp;
 | 
						|
  static const IconData unselectedIcon = Icons.desktop_windows_outlined;
 | 
						|
 | 
						|
  late MenubarState _menubarState;
 | 
						|
 | 
						|
  var connectionMap = RxList<Widget>.empty(growable: true);
 | 
						|
 | 
						|
  _ConnectionTabPageState(Map<String, dynamic> params) {
 | 
						|
    _menubarState = MenubarState();
 | 
						|
    RemoteCountState.init();
 | 
						|
    final peerId = params['id'];
 | 
						|
    if (peerId != null) {
 | 
						|
      ConnectionTypeState.init(peerId);
 | 
						|
      tabController.onSelected = (_, id) {
 | 
						|
        bind.setCurSessionId(id: id);
 | 
						|
        WindowController.fromWindowId(windowId())
 | 
						|
            .setTitle(getWindowNameWithId(id));
 | 
						|
      };
 | 
						|
      tabController.add(TabInfo(
 | 
						|
        key: peerId,
 | 
						|
        label: peerId,
 | 
						|
        selectedIcon: selectedIcon,
 | 
						|
        unselectedIcon: unselectedIcon,
 | 
						|
        onTabCloseButton: () => tabController.closeBy(peerId),
 | 
						|
        page: RemotePage(
 | 
						|
          key: ValueKey(peerId),
 | 
						|
          id: peerId,
 | 
						|
          password: params['password'],
 | 
						|
          menubarState: _menubarState,
 | 
						|
          switchUuid: params['switch_uuid'],
 | 
						|
          forceRelay: params['forceRelay'],
 | 
						|
        ),
 | 
						|
      ));
 | 
						|
      _update_remote_count();
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  @override
 | 
						|
  void initState() {
 | 
						|
    super.initState();
 | 
						|
 | 
						|
    tabController.onRemoved = (_, id) => onRemoveId(id);
 | 
						|
 | 
						|
    rustDeskWinManager.setMethodHandler((call, fromWindowId) async {
 | 
						|
      print(
 | 
						|
          "[Remote Page] call ${call.method} with args ${call.arguments} from window $fromWindowId");
 | 
						|
 | 
						|
      // for simplify, just replace connectionId
 | 
						|
      if (call.method == "new_remote_desktop") {
 | 
						|
        final args = jsonDecode(call.arguments);
 | 
						|
        final id = args['id'];
 | 
						|
        final switchUuid = args['switch_uuid'];
 | 
						|
        window_on_top(windowId());
 | 
						|
        ConnectionTypeState.init(id);
 | 
						|
        tabController.add(TabInfo(
 | 
						|
          key: id,
 | 
						|
          label: id,
 | 
						|
          selectedIcon: selectedIcon,
 | 
						|
          unselectedIcon: unselectedIcon,
 | 
						|
          onTabCloseButton: () => tabController.closeBy(id),
 | 
						|
          page: RemotePage(
 | 
						|
            key: ValueKey(id),
 | 
						|
            id: id,
 | 
						|
            password: args['password'],
 | 
						|
            menubarState: _menubarState,
 | 
						|
            switchUuid: switchUuid,
 | 
						|
            forceRelay: args['forceRelay'],
 | 
						|
          ),
 | 
						|
        ));
 | 
						|
      } else if (call.method == kWindowDisableGrabKeyboard) {
 | 
						|
        stateGlobal.grabKeyboard = false;
 | 
						|
      } else if (call.method == "onDestroy") {
 | 
						|
        tabController.clear();
 | 
						|
      } else if (call.method == kWindowActionRebuild) {
 | 
						|
        reloadCurrentWindow();
 | 
						|
      }
 | 
						|
      _update_remote_count();
 | 
						|
    });
 | 
						|
    Future.delayed(Duration.zero, () {
 | 
						|
      restoreWindowPosition(WindowType.RemoteDesktop, windowId: windowId());
 | 
						|
    });
 | 
						|
  }
 | 
						|
 | 
						|
  @override
 | 
						|
  void dispose() {
 | 
						|
    super.dispose();
 | 
						|
    _menubarState.save();
 | 
						|
  }
 | 
						|
 | 
						|
  @override
 | 
						|
  Widget build(BuildContext context) {
 | 
						|
    final tabWidget = Obx(
 | 
						|
      () => Container(
 | 
						|
        decoration: BoxDecoration(
 | 
						|
          border: Border.all(
 | 
						|
              color: MyTheme.color(context).border!,
 | 
						|
              width: stateGlobal.windowBorderWidth.value),
 | 
						|
        ),
 | 
						|
        child: Scaffold(
 | 
						|
          backgroundColor: Theme.of(context).colorScheme.background,
 | 
						|
          body: DesktopTab(
 | 
						|
            controller: tabController,
 | 
						|
            onWindowCloseButton: handleWindowCloseButton,
 | 
						|
            tail: const AddButton().paddingOnly(left: 10),
 | 
						|
            pageViewBuilder: (pageView) => pageView,
 | 
						|
            labelGetter: DesktopTab.labelGetterAlias,
 | 
						|
            tabBuilder: (key, icon, label, themeConf) => Obx(() {
 | 
						|
              final connectionType = ConnectionTypeState.find(key);
 | 
						|
              if (!connectionType.isValid()) {
 | 
						|
                return Row(
 | 
						|
                  mainAxisAlignment: MainAxisAlignment.center,
 | 
						|
                  children: [
 | 
						|
                    icon,
 | 
						|
                    label,
 | 
						|
                  ],
 | 
						|
                );
 | 
						|
              } else {
 | 
						|
                bool secure =
 | 
						|
                    connectionType.secure.value == ConnectionType.strSecure;
 | 
						|
                bool direct =
 | 
						|
                    connectionType.direct.value == ConnectionType.strDirect;
 | 
						|
                var msgConn;
 | 
						|
                if (secure && direct) {
 | 
						|
                  msgConn = translate("Direct and encrypted connection");
 | 
						|
                } else if (secure && !direct) {
 | 
						|
                  msgConn = translate("Relayed and encrypted connection");
 | 
						|
                } else if (!secure && direct) {
 | 
						|
                  msgConn = translate("Direct and unencrypted connection");
 | 
						|
                } else {
 | 
						|
                  msgConn = translate("Relayed and unencrypted connection");
 | 
						|
                }
 | 
						|
                var msgFingerprint = '${translate('Fingerprint')}:\n';
 | 
						|
                var fingerprint = FingerprintState.find(key).value;
 | 
						|
                if (fingerprint.length > 5 * 8) {
 | 
						|
                  var first = fingerprint.substring(0, 39);
 | 
						|
                  var second = fingerprint.substring(40);
 | 
						|
                  msgFingerprint += '$first\n$second';
 | 
						|
                } else {
 | 
						|
                  msgFingerprint += fingerprint;
 | 
						|
                }
 | 
						|
 | 
						|
                final tab = Row(
 | 
						|
                  mainAxisAlignment: MainAxisAlignment.center,
 | 
						|
                  children: [
 | 
						|
                    icon,
 | 
						|
                    Tooltip(
 | 
						|
                      message: '$msgConn\n$msgFingerprint',
 | 
						|
                      child: SvgPicture.asset(
 | 
						|
                        'assets/${connectionType.secure.value}${connectionType.direct.value}.svg',
 | 
						|
                        width: themeConf.iconSize,
 | 
						|
                        height: themeConf.iconSize,
 | 
						|
                      ).paddingOnly(right: 5),
 | 
						|
                    ),
 | 
						|
                    label,
 | 
						|
                  ],
 | 
						|
                );
 | 
						|
 | 
						|
                return Listener(
 | 
						|
                  onPointerDown: (e) {
 | 
						|
                    if (e.kind != ui.PointerDeviceKind.mouse) {
 | 
						|
                      return;
 | 
						|
                    }
 | 
						|
                    if (e.buttons == 2) {
 | 
						|
                      showRightMenu(
 | 
						|
                        (CancelFunc cancelFunc) {
 | 
						|
                          return _tabMenuBuilder(key, cancelFunc);
 | 
						|
                        },
 | 
						|
                        target: e.position,
 | 
						|
                      );
 | 
						|
                    }
 | 
						|
                  },
 | 
						|
                  child: tab,
 | 
						|
                );
 | 
						|
              }
 | 
						|
            }),
 | 
						|
          ),
 | 
						|
        ),
 | 
						|
      ),
 | 
						|
    );
 | 
						|
    return Platform.isMacOS || kUseCompatibleUiMode
 | 
						|
        ? tabWidget
 | 
						|
        : Obx(() => SubWindowDragToResizeArea(
 | 
						|
              key: contentKey,
 | 
						|
              child: tabWidget,
 | 
						|
              // Specially configured for a better resize area and remote control.
 | 
						|
              childPadding: kDragToResizeAreaPadding,
 | 
						|
              resizeEdgeSize: stateGlobal.resizeEdgeSize.value,
 | 
						|
              windowId: stateGlobal.windowId,
 | 
						|
            ));
 | 
						|
  }
 | 
						|
 | 
						|
  // Note: Some dup code to ../widgets/remote_menubar
 | 
						|
  Widget _tabMenuBuilder(String key, CancelFunc cancelFunc) {
 | 
						|
    final List<MenuEntryBase<String>> menu = [];
 | 
						|
    const EdgeInsets padding = EdgeInsets.only(left: 8.0, right: 5.0);
 | 
						|
    final remotePage = tabController.state.value.tabs
 | 
						|
        .firstWhere((tab) => tab.key == key)
 | 
						|
        .page as RemotePage;
 | 
						|
    final ffi = remotePage.ffi;
 | 
						|
    final pi = ffi.ffiModel.pi;
 | 
						|
    final perms = ffi.ffiModel.permissions;
 | 
						|
    menu.addAll([
 | 
						|
      MenuEntryButton<String>(
 | 
						|
        childBuilder: (TextStyle? style) => Text(
 | 
						|
          translate('Close'),
 | 
						|
          style: style,
 | 
						|
        ),
 | 
						|
        proc: () {
 | 
						|
          tabController.closeBy(key);
 | 
						|
          cancelFunc();
 | 
						|
        },
 | 
						|
        padding: padding,
 | 
						|
      ),
 | 
						|
      MenuEntryButton<String>(
 | 
						|
        childBuilder: (TextStyle? style) => Obx(() => Text(
 | 
						|
              translate(
 | 
						|
                  _menubarState.show.isTrue ? 'Hide Menubar' : 'Show Menubar'),
 | 
						|
              style: style,
 | 
						|
            )),
 | 
						|
        proc: () {
 | 
						|
          _menubarState.switchShow();
 | 
						|
          cancelFunc();
 | 
						|
        },
 | 
						|
        padding: padding,
 | 
						|
      ),
 | 
						|
      MenuEntryDivider<String>(),
 | 
						|
      RemoteMenuEntry.viewStyle(
 | 
						|
        key,
 | 
						|
        ffi,
 | 
						|
        padding,
 | 
						|
        dismissFunc: cancelFunc,
 | 
						|
      ),
 | 
						|
    ]);
 | 
						|
 | 
						|
    if (!ffi.canvasModel.cursorEmbedded &&
 | 
						|
        !ffi.ffiModel.viewOnly &&
 | 
						|
        !pi.is_wayland) {
 | 
						|
      menu.add(MenuEntryDivider<String>());
 | 
						|
      menu.add(RemoteMenuEntry.showRemoteCursor(
 | 
						|
        key,
 | 
						|
        padding,
 | 
						|
        dismissFunc: cancelFunc,
 | 
						|
      ));
 | 
						|
    }
 | 
						|
 | 
						|
    if (perms['keyboard'] != false && !ffi.ffiModel.viewOnly) {
 | 
						|
      if (perms['clipboard'] != false) {
 | 
						|
        menu.add(RemoteMenuEntry.disableClipboard(key, padding,
 | 
						|
            dismissFunc: cancelFunc));
 | 
						|
      }
 | 
						|
 | 
						|
      menu.add(
 | 
						|
          RemoteMenuEntry.insertLock(key, padding, dismissFunc: cancelFunc));
 | 
						|
 | 
						|
      if (pi.platform == kPeerPlatformLinux || pi.sasEnabled) {
 | 
						|
        menu.add(RemoteMenuEntry.insertCtrlAltDel(key, padding,
 | 
						|
            dismissFunc: cancelFunc));
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    menu.add(MenuEntryButton<String>(
 | 
						|
      childBuilder: (TextStyle? style) => Text(
 | 
						|
        translate('Copy Fingerprint'),
 | 
						|
        style: style,
 | 
						|
      ),
 | 
						|
      proc: () => onCopyFingerprint(FingerprintState.find(key).value),
 | 
						|
      padding: padding,
 | 
						|
      dismissOnClicked: true,
 | 
						|
      dismissCallback: cancelFunc,
 | 
						|
    ));
 | 
						|
 | 
						|
    return mod_menu.PopupMenu<String>(
 | 
						|
      items: menu
 | 
						|
          .map((entry) => entry.build(
 | 
						|
              context,
 | 
						|
              const MenuConfig(
 | 
						|
                commonColor: _MenuTheme.blueColor,
 | 
						|
                height: _MenuTheme.height,
 | 
						|
                dividerHeight: _MenuTheme.dividerHeight,
 | 
						|
              )))
 | 
						|
          .expand((i) => i)
 | 
						|
          .toList(),
 | 
						|
    );
 | 
						|
  }
 | 
						|
 | 
						|
  void onRemoveId(String id) async {
 | 
						|
    if (tabController.state.value.tabs.isEmpty) {
 | 
						|
      await WindowController.fromWindowId(windowId()).close();
 | 
						|
    }
 | 
						|
    ConnectionTypeState.delete(id);
 | 
						|
    _update_remote_count();
 | 
						|
  }
 | 
						|
 | 
						|
  int windowId() {
 | 
						|
    return widget.params["windowId"];
 | 
						|
  }
 | 
						|
 | 
						|
  Future<bool> handleWindowCloseButton() async {
 | 
						|
    final connLength = tabController.length;
 | 
						|
    if (connLength <= 1) {
 | 
						|
      tabController.clear();
 | 
						|
      return true;
 | 
						|
    } else {
 | 
						|
      final opt = "enable-confirm-closing-tabs";
 | 
						|
      final bool res;
 | 
						|
      if (!option2bool(opt, await bind.mainGetOption(key: opt))) {
 | 
						|
        res = true;
 | 
						|
      } else {
 | 
						|
        res = await closeConfirmDialog();
 | 
						|
      }
 | 
						|
      if (res) {
 | 
						|
        tabController.clear();
 | 
						|
      }
 | 
						|
      return res;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  _update_remote_count() =>
 | 
						|
      RemoteCountState.find().value = tabController.length;
 | 
						|
}
 |