MWClash/lib/widgets/tray_container.dart

import 'dart:io';

import 'package:fl_clash/common/common.dart';
import 'package:fl_clash/enum/enum.dart';
import 'package:fl_clash/models/models.dart';
import 'package:flutter/material.dart';
import 'package:intl/intl.dart';
import 'package:provider/provider.dart';
import 'package:tray_manager/tray_manager.dart';

class TrayContainer extends StatefulWidget {
  final Widget child;

  const TrayContainer({
    super.key,
    required this.child,
  });

  @override
  State<TrayContainer> createState() => _TrayContainerState();
}

class _TrayContainerState extends State<TrayContainer> with TrayListener {
  var isTrayInit = false;

  @override
  void initState() {
    super.initState();
    trayManager.addListener(this);
  }

  _updateOtherTray() async {
    if (isTrayInit == false) {
      await trayManager.setIcon(
        Other.getTrayIconPath(),
      );
      isTrayInit = true;
    }
  }

  _updateLinuxTray() async {
    await trayManager.destroy();
    await trayManager.setIcon(
      Other.getTrayIconPath(),
    );
  }

  updateMenu(TrayContainerSelectorState state) async {
    if (!Platform.isLinux) {
      _updateOtherTray();
    }
    List<MenuItem> menuItems = [];
    final showMenuItem = MenuItem(
      label: appLocalizations.show,
      onClick: (_) {
        window?.show();
      },
    );
    menuItems.add(showMenuItem);
    menuItems.add(MenuItem.separator());
    // for (final group in state.groups) {
    //   List<MenuItem> subMenuItems = [];
    //   final isCurrentGroup = group.name == state.currentGroupName;
    //   for (final proxy in group.all) {
    //     final isCurrentProxy = proxy.name == state.currentProxyName;
    //     subMenuItems.add(
    //       MenuItem.checkbox(
    //           label: proxy.name,
    //           checked: isCurrentGroup && isCurrentProxy,
    //           onClick: (_) {
    //             final config = context.appController.config;
    //             config.currentProfile?.groupName = group.name;
    //             config.currentProfile?.proxyName = proxy.name;
    //             config.update();
    //             context.appController.changeProxy();
    //           }),
    //     );
    //   }
    //   menuItems.add(
    //     MenuItem.submenu(
    //       label: group.name,
    //       submenu: Menu(
    //         items: subMenuItems,
    //       ),
    //     ),
    //   );
    // }
    // if (state.groups.isNotEmpty) {
    //   menuItems.add(MenuItem.separator());
    // }
    for (final mode in Mode.values) {
      menuItems.add(
        MenuItem.checkbox(
          label: Intl.message(mode.name),
          onClick: (_) {
            context.appController.clashConfig.mode = mode;
          },
          checked: mode == state.mode,
        ),
      );
    }
    menuItems.add(MenuItem.separator());
    final proxyMenuItem = MenuItem.checkbox(
      label: appLocalizations.systemProxy,
      onClick: (_) async {
        context.appController.updateSystemProxy(!state.isRun);
      },
      checked: state.isRun,
    );
    menuItems.add(proxyMenuItem);
    final autoStartMenuItem = MenuItem.checkbox(
      label: appLocalizations.autoLaunch,
      onClick: (_) async {
        context.appController.config.autoLaunch =
            !context.appController.config.autoLaunch;
      },
      checked: state.autoLaunch,
    );
    menuItems.add(autoStartMenuItem);
    menuItems.add(MenuItem.separator());
    final exitMenuItem = MenuItem(
      label: appLocalizations.exit,
      onClick: (_) async {
        await context.appController.handleExit();
      },
    );
    menuItems.add(exitMenuItem);
    final menu = Menu();
    menu.items = menuItems;
    trayManager.setContextMenu(menu);
    if (Platform.isLinux) {
      _updateLinuxTray();
    }
  }

  @override
  Widget build(BuildContext context) {
    return Selector3<AppState, Config, ClashConfig, TrayContainerSelectorState>(
      selector: (_, appState, config, clashConfig) =>
          TrayContainerSelectorState(
        mode: clashConfig.mode,
        autoLaunch: config.autoLaunch,
        isRun: appState.runTime != null,
        locale: config.locale,
      ),
      builder: (_, state, child) {
        WidgetsBinding.instance.addPostFrameCallback((timeStamp) {
          updateMenu(state);
        });
        return child!;
      },
      child: widget.child,
    );
  }

  @override
  void onTrayIconRightMouseDown() {
    trayManager.popUpContextMenu();
  }

  @override
  onTrayIconMouseDown() {
    window?.show();
  }

  @override
  dispose() {
    trayManager.removeListener(this);
    super.dispose();
  }
}