112 lines
4.1 KiB
Swift
112 lines
4.1 KiB
Swift
//
|
|
// ProxyMenuItemFactory.swift
|
|
// ClashX
|
|
//
|
|
// Created by CYC on 2018/8/4.
|
|
// Copyright © 2018年 yichengchen. All rights reserved.
|
|
//
|
|
|
|
import Cocoa
|
|
import SwiftyJSON
|
|
import RxCocoa
|
|
|
|
class ProxyMenuItemFactory {
|
|
static func menuItems(completionHandler:@escaping (([NSMenuItem])->())){
|
|
ApiRequest.requestProxyGroupList { (res) in
|
|
let dataDict = JSON(res)
|
|
var menuItems = [NSMenuItem]()
|
|
if (ConfigManager.shared.currentConfig?.mode == .direct) {
|
|
completionHandler(menuItems)
|
|
return
|
|
}
|
|
for proxyGroup in dataDict.dictionaryValue.sorted(by: { $0.0 < $1.0}) {
|
|
var menu:NSMenuItem?
|
|
switch proxyGroup.value["type"].stringValue {
|
|
case "Selector": menu = self.generateSelectorMenuItem(json: dataDict, key: proxyGroup.key)
|
|
case "URLTest","Fallback": menu = self.generateUrlTestMenuItem(proxyGroup: proxyGroup)
|
|
default: continue
|
|
}
|
|
if (menu != nil) {menuItems.append(menu!)}
|
|
|
|
}
|
|
completionHandler(menuItems.reversed())
|
|
}
|
|
}
|
|
|
|
static func generateSelectorMenuItem(json:JSON,key:String)->NSMenuItem? {
|
|
let proxyGroup:(key: String, value: JSON) = (key,json[key])
|
|
let isGlobalMode = ConfigManager.shared.currentConfig?.mode == .global
|
|
if (isGlobalMode) {
|
|
if proxyGroup.key != "GLOBAL" {return nil}
|
|
} else {
|
|
if proxyGroup.key == "GLOBAL" {return nil}
|
|
}
|
|
|
|
let menu = NSMenuItem(title: proxyGroup.key, action: nil, keyEquivalent: "")
|
|
let selectedName = proxyGroup.value["now"].stringValue
|
|
let submenu = NSMenu(title: proxyGroup.key)
|
|
var hasSelected = false
|
|
for proxy in proxyGroup.value["all"].arrayValue {
|
|
if isGlobalMode {
|
|
if json[proxy.stringValue]["type"] == "Selector" {
|
|
continue
|
|
}
|
|
}
|
|
var speedText = ""
|
|
let placeHolder = "\t"
|
|
if let speedInfo = SpeedDataRecorder.shared.speedDict[proxy.stringValue] {
|
|
speedText = speedInfo < Int.max ?"\(placeHolder)\(speedInfo) ms" : "\(placeHolder)fail"
|
|
}
|
|
|
|
let proxyItem = ProxyMenuItem(title: proxy.stringValue + speedText, action: #selector(ProxyMenuItemFactory.actionSelectProxy(sender:)), keyEquivalent: "")
|
|
proxyItem.proxyName = proxy.stringValue
|
|
proxyItem.target = ProxyMenuItemFactory.self
|
|
let selected = proxy.stringValue == selectedName
|
|
proxyItem.state = selected ? .on : .off
|
|
if selected {hasSelected = true}
|
|
submenu.addItem(proxyItem)
|
|
}
|
|
menu.submenu = submenu
|
|
|
|
if (!hasSelected && submenu.items.count>0) {
|
|
self.actionSelectProxy(sender: submenu.items[0] as! ProxyMenuItem)
|
|
}
|
|
return menu
|
|
}
|
|
|
|
static func generateUrlTestMenuItem(proxyGroup:(key: String, value: JSON))->NSMenuItem? {
|
|
|
|
let menu = NSMenuItem(title: proxyGroup.key, action: nil, keyEquivalent: "")
|
|
let selectedName = proxyGroup.value["now"].stringValue
|
|
let submenu = NSMenu(title: proxyGroup.key)
|
|
|
|
let nowMenuItem = NSMenuItem(title: "now:\(selectedName)", action: nil, keyEquivalent: "")
|
|
|
|
submenu.addItem(nowMenuItem)
|
|
menu.submenu = submenu
|
|
return menu
|
|
}
|
|
|
|
@objc static func actionSelectProxy(sender:ProxyMenuItem){
|
|
guard let proxyGroup = sender.menu?.title else {return}
|
|
let proxyName = sender.proxyName
|
|
|
|
ApiRequest.updateProxyGroup(group: proxyGroup, selectProxy: proxyName) { (success) in
|
|
if (success) {
|
|
for items in sender.menu?.items ?? [NSMenuItem]() {
|
|
items.state = .off
|
|
}
|
|
sender.state = .on
|
|
// remember select proxy
|
|
ConfigManager.selectedProxyMap[proxyGroup] = proxyName
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
|
|
class ProxyMenuItem:NSMenuItem {
|
|
var proxyName:String = ""
|
|
}
|