ClashX.Meta/ClashX/Views/ProxyGroupSpeedTestMenuItem.swift

//
//  ProxyGroupSpeedTestMenuItem.swift
//  ClashX
//
//  Created by yicheng on 2019/10/15.
//  Copyright © 2019 west2online. All rights reserved.
//

import Carbon
import Cocoa

class ProxyGroupSpeedTestMenuItem: NSMenuItem {
    var proxyGroup: ClashProxy
    var testType: TestType

    init(group: ClashProxy) {
        proxyGroup = group
        switch group.type {
        case .urltest, .fallback:
            testType = .reTest
        case .select:
            testType = .benchmark
        default:
            testType = .unknown
        }

        super.init(title: NSLocalizedString("Benchmark", comment: ""), action: nil, keyEquivalent: "")

        switch testType {
        case .benchmark:
            view = ProxyGroupSpeedTestMenuItemView(testType.title)
        case .reTest:
            title = testType.title
        case .unknown:
            assertionFailure()
        }
    }

    required init(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

fileprivate class ProxyGroupSpeedTestMenuItemView: NSView {
    private let label: NSTextField
    private let font = NSFont.menuFont(ofSize: 14)
    private var isMouseInsideView = false
    private var eventHandler: EventHandlerRef?

    init(_ title: String) {
        label = NSTextField(labelWithString: title)
        label.font = font
        label.sizeToFit()
        super.init(frame: NSRect(x: 0, y: 0, width: label.bounds.width + 40, height: 20))
        translatesAutoresizingMaskIntoConstraints = false
        heightAnchor.constraint(equalToConstant: 20).isActive = true
        addSubview(label)
        label.frame = NSRect(x: 20, y: 0, width: label.bounds.width, height: 20)
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    private func startBenchmark() {
        guard let group = (enclosingMenuItem as? ProxyGroupSpeedTestMenuItem)?.proxyGroup
        else { return }
        let testGroup = DispatchGroup()
        label.stringValue = NSLocalizedString("Testing", comment: "")
        enclosingMenuItem?.isEnabled = false
        setNeedsDisplay(bounds)
        for proxyName in group.speedtestAble {
            testGroup.enter()
            ApiRequest.getProxyDelay(proxyName: proxyName) { delay in
                let delayStr = delay == 0 ? "fail" : "\(delay) ms"
                NotificationCenter.default.post(name: kSpeedTestFinishForProxy,
                                                object: nil,
                                                userInfo: ["proxyName": proxyName, "delay": delayStr])
                testGroup.leave()
            }
        }

        testGroup.notify(queue: .main) {
            [weak self] in
            guard let self = self, let menu = self.enclosingMenuItem else { return }
            self.label.stringValue = menu.title
            self.label.textColor = NSColor.labelColor
            menu.isEnabled = true
            self.setNeedsDisplay(self.bounds)
        }
    }

    override func updateTrackingAreas() {
        super.updateTrackingAreas()
        if #available(macOS 10.15.1, *) {
            trackingAreas.forEach { removeTrackingArea($0) }
            addTrackingArea(NSTrackingArea(rect: bounds, options: [.mouseEnteredAndExited, .activeAlways], owner: self, userInfo: nil))
            addTrackingArea(NSTrackingArea(rect: bounds, options: [.mouseMoved, .activeAlways], owner: self, userInfo: nil))
        }
    }

    override func viewDidMoveToWindow() {
        super.viewDidMoveToWindow()
        if #available(macOS 10.15.1, *) {
            setupCarbon()
        }
    }

//    https://gist.github.com/p0deje/da5e5cfda6be8cb87c2e7caad3a3df63
//    https://stackoverflow.com/questions/53273191/custom-carbon-key-event-handler-fails-after-mouse-events
    @available(macOS 10.15.1, *)
    private func setupCarbon() {
        if window != nil {
            if let dispatcher = GetEventDispatcherTarget() {
                let eventHandlerCallback: EventHandlerUPP = { eventHandlerCallRef, eventRef, userData in
                    guard let userData = userData else { return 0 }
                    let itemView: ProxyGroupSpeedTestMenuItemView = bridge(ptr: userData)
                    itemView.startBenchmark()
                    return 0
                }

                let eventSpecs = [EventTypeSpec(eventClass: OSType(kEventClassMouse), eventKind: UInt32(kEventMouseUp))]

                InstallEventHandler(dispatcher, eventHandlerCallback, 1, eventSpecs, bridge(obj: self), &eventHandler)
            }
        } else {
            RemoveEventHandler(eventHandler)
        }
    }

    override func mouseEntered(with event: NSEvent) {
        if #available(macOS 10.15.1, *) {
            isMouseInsideView = true
            setNeedsDisplay(bounds)
        }
    }

    override func mouseExited(with event: NSEvent) {
        if #available(macOS 10.15.1, *) {
            isMouseInsideView = false
            setNeedsDisplay(bounds)
        }
    }

    override func hitTest(_ point: NSPoint) -> NSView? {
        if bounds.contains(point) {
            return label
        }
        return super.hitTest(point)
    }

    override func mouseUp(with event: NSEvent) {
        if #available(macOS 10.15.1, *) {} else {
            startBenchmark()
        }
    }

    override func draw(_ dirtyRect: NSRect) {
        super.draw(dirtyRect)
        guard let menu = enclosingMenuItem else { return }

        let isHighlighted: Bool
        if #available(macOS 10.15.1, *) {
            isHighlighted = isMouseInsideView
        } else {
            isHighlighted = menu.isHighlighted
        }
        if isHighlighted && menu.isEnabled {
            NSColor.selectedMenuItemColor.setFill()
            label.textColor = NSColor.white
        } else {
            NSColor.clear.setFill()
            if enclosingMenuItem?.isEnabled ?? true {
                label.textColor = NSColor.labelColor
            } else {
                label.textColor = NSColor.secondaryLabelColor
            }
        }
        dirtyRect.fill()
    }
}

extension ProxyGroupSpeedTestMenuItem {
    enum TestType {
        case benchmark
        case reTest
        case unknown

        var title: String {
            switch self {
            case .benchmark: return NSLocalizedString("Benchmark", comment: "")
            case .reTest: return NSLocalizedString("ReTest", comment: "")
            case .unknown: return ""
            }
        }
    }
}