ClashX.Meta/ClashX/Views/StatusItemView.swift

117 lines
3.5 KiB
Swift

//
// StatusItemView.swift
// ClashX
//
// Created by CYC on 2018/6/23.
// Copyright © 2018 yichengchen. All rights reserved.
//
import AppKit
import Foundation
import RxCocoa
import RxSwift
class StatusItemView: NSView {
@IBOutlet var imageView: NSImageView!
@IBOutlet var uploadSpeedLabel: NSTextField!
@IBOutlet var downloadSpeedLabel: NSTextField!
@IBOutlet var speedContainerView: NSView!
weak var statusItem: NSStatusItem?
lazy var menuImage: NSImage = {
let customImagePath = (NSHomeDirectory() as NSString).appendingPathComponent("/.config/clash/menuImage.png")
return NSImage(contentsOfFile: customImagePath) ?? NSImage(named: "menu_icon")!
}()
static func create(statusItem: NSStatusItem?) -> StatusItemView {
var topLevelObjects: NSArray?
if Bundle.main.loadNibNamed("StatusItemView", owner: self, topLevelObjects: &topLevelObjects) {
let view = (topLevelObjects!.first(where: { $0 is NSView }) as? StatusItemView)!
view.statusItem = statusItem
view.setupView()
return view
}
return NSView() as! StatusItemView
}
func setupView() {
let fontSize: CGFloat = 9
let font: NSFont
if let fontName = UserDefaults.standard.string(forKey: "kStatusMenuFontName"),
let f = NSFont(name: fontName, size: fontSize) {
font = f
} else {
font = NSFont.menuBarFont(ofSize: fontSize)
}
uploadSpeedLabel.font = font
downloadSpeedLabel.font = font
uploadSpeedLabel.textColor = NSColor.black
downloadSpeedLabel.textColor = NSColor.black
}
func updateViewStatus(enableProxy: Bool) {
let selectedColor = NSColor.red
let unselectedColor: NSColor
if #available(OSX 10.14, *) {
unselectedColor = selectedColor.withSystemEffect(.disabled)
} else {
unselectedColor = selectedColor.withAlphaComponent(0.5)
}
imageView.image = menuImage.tint(color: enableProxy ? selectedColor : unselectedColor)
updateStatusItemView()
}
func updateSpeedLabel(up: Int, down: Int) {
guard !speedContainerView.isHidden else { return }
let kbup = up / 1024
let kbdown = down / 1024
var finalUpStr: String
var finalDownStr: String
if kbup < 1024 {
finalUpStr = "\(kbup)KB/s"
} else {
finalUpStr = String(format: "%.2fMB/s", Double(kbup) / 1024.0)
}
if kbdown < 1024 {
finalDownStr = "\(kbdown)KB/s"
} else {
finalDownStr = String(format: "%.2fMB/s", Double(kbdown) / 1024.0)
}
if downloadSpeedLabel.stringValue == finalDownStr && uploadSpeedLabel.stringValue == finalUpStr {
return
}
downloadSpeedLabel.stringValue = finalDownStr
uploadSpeedLabel.stringValue = finalUpStr
updateStatusItemView()
}
func showSpeedContainer(show: Bool) {
speedContainerView.isHidden = !show
updateStatusItemView()
}
func updateStatusItemView() {
statusItem?.updateImage(withView: self)
}
}
extension NSStatusItem {
func updateImage(withView view: NSView) {
if let rep = view.bitmapImageRepForCachingDisplay(in: view.bounds) {
view.cacheDisplay(in: view.bounds, to: rep)
let img = NSImage(size: view.bounds.size)
img.addRepresentation(rep)
img.isTemplate = true
image = img
}
}
}