Files
macos-stats/Kit/Widgets/Speed.swift

585 lines
23 KiB
Swift

//
// Speed.swift
// Kit
//
// Created by Serhiy Mytrovtsiy on 24/05/2020.
// Using Swift 5.0.
// Running on macOS 10.15.
//
// Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
//
import Cocoa
public class SpeedWidget: WidgetWrapper {
private var icon: String = "dots"
private var state: Bool = false
private var valueState: Bool = true
private var baseValue: String = "byte"
private var unitsState: Bool = true
private var monochromeState: Bool = false
private var valueColorState: Bool = false
private var transparentIconsState: Bool = false
private var valueAlignmentState: String = "right"
private var modeState: String = "twoRows"
private var downloadColorState: Color = .secondBlue
private var uploadColorState: Color = .secondRed
private var symbols: [String] = ["U", "D"]
private var uploadField: NSTextField? = nil
private var downloadField: NSTextField? = nil
private var uploadValue: Int64 = 0
private var downloadValue: Int64 = 0
private var width: CGFloat = 58
private var valueColorView: NSView? = nil
private var transparentIconView: NSView? = nil
private var valueAlignmentView: NSView? = nil
private var downloadColor: NSColor {
self.monochromeState ? MonochromeColor.blue : (self.downloadColorState.additional as? NSColor ?? NSColor.systemBlue)
}
private var uploadColor: NSColor {
self.monochromeState ? MonochromeColor.red : (self.uploadColorState.additional as? NSColor ?? NSColor.red)
}
private var noActivityColor: NSColor {
self.transparentIconsState ? NSColor.clear : NSColor.textColor
}
private var valueAlignment: NSTextAlignment {
get {
if let alignmentPair = Alignments.first(where: { $0.key == self.valueAlignmentState }) {
return alignmentPair.additional as? NSTextAlignment ?? .left
}
return .left
}
}
public init(title: String, config: NSDictionary?, preview: Bool = false) {
let widgetTitle: String = title
if config != nil {
if let symbols = config!["Symbols"] as? [String] {
self.symbols = symbols
}
if let icon = config!["Icon"] as? String {
self.icon = icon
}
}
super.init(.speed, title: widgetTitle, frame: CGRect(
x: 0,
y: Constants.Widget.margin.y,
width: width,
height: Constants.Widget.height - (2*Constants.Widget.margin.y)
))
self.canDrawConcurrently = true
if !preview {
self.valueState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_value", defaultValue: self.valueState)
self.icon = Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_icon", defaultValue: self.baseValue)
self.baseValue = Store.shared.string(key: "\(self.title)_base", defaultValue: self.baseValue)
self.unitsState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_units", defaultValue: self.unitsState)
self.monochromeState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_monochrome", defaultValue: self.monochromeState)
self.valueColorState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_valueColor", defaultValue: self.valueColorState)
self.downloadColorState = Color.fromString(Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_downloadColor", defaultValue: self.downloadColorState.key))
self.uploadColorState = Color.fromString(Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_uploadColor", defaultValue: self.uploadColorState.key))
self.transparentIconsState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_transparentIcons", defaultValue: self.transparentIconsState)
self.valueAlignmentState = Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_valueAlignment", defaultValue: self.valueAlignmentState)
self.modeState = Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_mode", defaultValue: self.modeState)
}
if self.valueState && self.icon != "none" {
self.state = true
}
if preview {
self.downloadValue = 8947141
self.uploadValue = 478678
}
}
required init?(coder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
public override func draw(_ dirtyRect: NSRect) {
super.draw(dirtyRect)
var width: CGFloat = 0
switch self.modeState {
case "oneRow":
width = self.drawOneRow(dirtyRect)
case "twoRows":
width = self.drawTwoRows(dirtyRect)
default:
width = 0
}
self.setWidth(width)
}
// MARK: - one row
private func drawOneRow(_ dirtyRect: NSRect) -> CGFloat {
var width: CGFloat = Constants.Widget.margin.x
let downloadIconColor = self.downloadValue >= 1_024 ? self.downloadColor : self.noActivityColor
let uploadIconColor = self.uploadValue >= 1_024 ? self.uploadColor : self.noActivityColor
switch self.icon {
case "dots":
width += self.drawDot(CGPoint(x: width, y: 0), color: uploadIconColor)
case "arrows":
width += self.drawArrow(CGPoint(x: width, y: 0), symbol: "U", color: uploadIconColor)
case "chars":
width += self.drawChar(CGPoint(x: width, y: 0), symbol: self.symbols[0], color: uploadIconColor)
default: break
}
width += self.valueState && self.icon != "none" ? 2 : 0
if self.valueState {
width += self.drawValue(self.uploadValue,
offset: CGPoint(x: width, y: 0),
color: self.valueColorState && self.uploadValue >= 1_024 ? self.uploadColor : NSColor.textColor
)
}
width += 4
switch self.icon {
case "dots":
width += self.drawDot(CGPoint(x: width, y: 0), color: downloadIconColor)
case "arrows":
width += self.drawArrow(CGPoint(x: width, y: 0), symbol: "D", color: downloadIconColor)
case "chars":
width += self.drawChar(CGPoint(x: width, y: 0), symbol: self.symbols[1], color: downloadIconColor)
default: break
}
width += self.valueState && self.icon != "none" ? 2 : 0
if self.valueState {
width += self.drawValue(self.downloadValue,
offset: CGPoint(x: width, y: 0),
color: self.valueColorState && self.downloadValue >= 1_024 ? self.downloadColor : NSColor.textColor
)
}
return width + Constants.Widget.margin.x
}
private func drawValue(_ value: Int64, offset: CGPoint, color: NSColor) -> CGFloat {
let rowWidth: CGFloat = self.unitsState ? 58 : 32
let height: CGFloat = self.frame.height
let style = NSMutableParagraphStyle()
style.alignment = self.valueAlignment
let size: CGFloat = 10
let downloadStringAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 11, weight: .regular),
NSAttributedString.Key.foregroundColor: color,
NSAttributedString.Key.paragraphStyle: style
]
let base: DataSizeBase = DataSizeBase(rawValue: self.baseValue) ?? .byte
let rect = CGRect(x: offset.x, y: (height-size)/2 + offset.y + 1, width: rowWidth - (Constants.Widget.margin.x*2), height: size)
let value = NSAttributedString.init(
string: Units(bytes: value).getReadableSpeed(base: base, omitUnits: !self.unitsState),
attributes: downloadStringAttributes
)
value.draw(with: rect)
return rowWidth
}
private func drawDot(_ offset: CGPoint, color: NSColor) -> CGFloat {
var size: CGFloat = 8
var height: CGFloat = self.frame.height
if self.modeState == "twoRows" {
size = 6
height /= 2
}
var circle = NSBezierPath()
circle = NSBezierPath(ovalIn: CGRect(x: offset.x, y: (height-size)/2 + offset.y, width: size, height: size))
color.set()
circle.fill()
return size
}
private func drawArrow(_ offset: CGPoint, symbol: String, color: NSColor) -> CGFloat {
let height = self.frame.height
let size = height * 0.8
let scaleFactor = NSScreen.main?.backingScaleFactor ?? 1
let lineWidth: CGFloat = 1
let arrowSize: CGFloat = 3 + (scaleFactor/2)
let x = arrowSize + (lineWidth / 2)
let y = (height - size)/2
var start: CGPoint = CGPoint()
var end: CGPoint = CGPoint()
if symbol == "D" {
start = CGPoint(x: offset.x + x, y: size + y)
end = CGPoint(x: offset.x + x, y: y)
} else if symbol == "U" {
start = CGPoint(x: offset.x + x, y: y)
end = CGPoint(x: offset.x + x, y: size + y)
}
let arrow = NSBezierPath()
arrow.addArrow(
start: start,
end: end,
pointerLineLength: arrowSize,
arrowAngle: CGFloat(Double.pi / 5)
)
color.set()
arrow.lineWidth = lineWidth
arrow.stroke()
arrow.close()
return arrowSize
}
private func drawChar(_ offset: CGPoint, symbol: String, color: NSColor) -> CGFloat {
let rowHeight: CGFloat = self.frame.height
let height: CGFloat = 10
let downloadAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 12, weight: .regular),
NSAttributedString.Key.foregroundColor: color,
NSAttributedString.Key.paragraphStyle: NSMutableParagraphStyle()
]
let rect = CGRect(x: offset.x, y: offset.y + ((rowHeight-height)/2) + 1, width: 10, height: height)
let str = NSAttributedString.init(string: symbol, attributes: downloadAttributes)
str.draw(with: rect)
return 10
}
// MARK: - two rows
private func drawTwoRows(_ dirtyRect: NSRect) -> CGFloat {
var width: CGFloat = 7
var x: CGFloat = 7
switch self.icon {
case "dots":
self.drawDots(dirtyRect)
case "arrows":
self.drawArrows(dirtyRect)
case "chars":
self.drawChars(dirtyRect)
default:
x = 0
width = 0
}
if self.valueState {
let rowWidth: CGFloat = self.unitsState ? 48 : 30
let rowHeight: CGFloat = self.frame.height / 2
let style = NSMutableParagraphStyle()
style.alignment = self.valueAlignment
let downloadStringAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 9, weight: .light),
NSAttributedString.Key.foregroundColor: self.valueColorState && self.downloadValue >= 1_024 ? self.downloadColor : NSColor.textColor,
NSAttributedString.Key.paragraphStyle: style
]
let uploadStringAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 9, weight: .light),
NSAttributedString.Key.foregroundColor: self.valueColorState && self.uploadValue >= 1_024 ? self.uploadColor : NSColor.textColor,
NSAttributedString.Key.paragraphStyle: style
]
let base: DataSizeBase = DataSizeBase(rawValue: self.baseValue) ?? .byte
var rect = CGRect(x: Constants.Widget.margin.x + x, y: 1, width: rowWidth - (Constants.Widget.margin.x*2), height: rowHeight)
let download = NSAttributedString.init(
string: Units(bytes: self.downloadValue).getReadableSpeed(base: base, omitUnits: !self.unitsState),
attributes: downloadStringAttributes
)
download.draw(with: rect)
rect = CGRect(x: Constants.Widget.margin.x + x, y: rect.height+1, width: rowWidth - (Constants.Widget.margin.x*2), height: rowHeight)
let upload = NSAttributedString.init(
string: Units(bytes: self.uploadValue).getReadableSpeed(base: base, omitUnits: !self.unitsState),
attributes: uploadStringAttributes
)
upload.draw(with: rect)
width += rowWidth
}
return width
}
private func drawDots(_ dirtyRect: NSRect) {
let rowHeight: CGFloat = self.frame.height / 2
let size: CGFloat = 6
let y: CGFloat = (rowHeight-size)/2
var downloadCircle = NSBezierPath()
downloadCircle = NSBezierPath(ovalIn: CGRect(x: Constants.Widget.margin.x, y: y-0.2, width: size, height: size))
(self.downloadValue >= 1_024 ? self.downloadColor : self.noActivityColor).set()
downloadCircle.fill()
var uploadCircle = NSBezierPath()
uploadCircle = NSBezierPath(ovalIn: CGRect(x: Constants.Widget.margin.x, y: 10.5, width: size, height: size))
(self.uploadValue >= 1_024 ? self.uploadColor : self.noActivityColor).set()
uploadCircle.fill()
}
private func drawArrows(_ dirtyRect: NSRect) {
let arrowAngle = CGFloat(Double.pi / 5)
let half = self.frame.size.height / 2
let scaleFactor = NSScreen.main?.backingScaleFactor ?? 1
let lineWidth: CGFloat = 1
let arrowSize: CGFloat = 3 + (scaleFactor/2)
let x = Constants.Widget.margin.x + arrowSize + (lineWidth / 2)
let downloadArrow = NSBezierPath()
downloadArrow.addArrow(
start: CGPoint(x: x, y: half - Constants.Widget.spacing/2),
end: CGPoint(x: x, y: 0),
pointerLineLength: arrowSize,
arrowAngle: arrowAngle
)
(self.downloadValue >= 1_024 ? self.downloadColor : self.noActivityColor).set()
downloadArrow.lineWidth = lineWidth
downloadArrow.stroke()
downloadArrow.close()
let uploadArrow = NSBezierPath()
uploadArrow.addArrow(
start: CGPoint(x: x, y: half + Constants.Widget.spacing/2),
end: CGPoint(x: x, y: self.frame.size.height),
pointerLineLength: arrowSize,
arrowAngle: arrowAngle
)
(self.uploadValue >= 1_024 ? self.uploadColor : self.noActivityColor).set()
uploadArrow.lineWidth = lineWidth
uploadArrow.stroke()
uploadArrow.close()
}
private func drawChars(_ dirtyRect: NSRect) {
let rowHeight: CGFloat = self.frame.height / 2
if self.symbols.count > 1 {
let downloadAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 9, weight: .regular),
NSAttributedString.Key.foregroundColor: self.downloadValue >= 1_024 ? self.downloadColor : self.noActivityColor,
NSAttributedString.Key.paragraphStyle: NSMutableParagraphStyle()
]
let rect = CGRect(x: Constants.Widget.margin.x, y: 1, width: 8, height: rowHeight)
let str = NSAttributedString.init(string: self.symbols[1], attributes: downloadAttributes)
str.draw(with: rect)
}
if !self.symbols.isEmpty {
let uploadAttributes = [
NSAttributedString.Key.font: NSFont.systemFont(ofSize: 9, weight: .regular),
NSAttributedString.Key.foregroundColor: self.uploadValue >= 1_024 ? self.uploadColor : self.noActivityColor,
NSAttributedString.Key.paragraphStyle: NSMutableParagraphStyle()
]
let rect = CGRect(x: Constants.Widget.margin.x, y: rowHeight+1, width: 8, height: rowHeight)
let str = NSAttributedString.init(string: self.symbols[0], attributes: uploadAttributes)
str.draw(with: rect)
}
}
// MARK: - settings
public override func settings() -> NSView {
let view = SettingsContainerView()
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Display mode"),
action: #selector(changeDisplayMode),
items: SensorsWidgetMode.filter({ $0.key == "oneRow" || $0.key == "twoRows"}),
selected: self.modeState
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Pictogram"),
action: #selector(toggleIcon),
items: SpeedPictogram,
selected: self.icon
))
self.transparentIconView = toggleSettingRow(
title: localizedString("Transparent pictogram when no activity"),
action: #selector(toggleTransparentIcons),
state: self.transparentIconsState
)
view.addArrangedSubview(self.transparentIconView!)
findAndToggleEnableNSControlState(self.transparentIconView!, state: self.icon != "none")
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Base"),
action: #selector(toggleBase),
items: SpeedBase,
selected: self.baseValue
))
view.addArrangedSubview(toggleSettingRow(
title: localizedString("Value"),
action: #selector(toggleValue),
state: self.valueState
))
self.valueColorView = toggleSettingRow(
title: localizedString("Colorize value"),
action: #selector(toggleValueColor),
state: self.valueColorState
)
view.addArrangedSubview(self.valueColorView!)
findAndToggleEnableNSControlState(self.valueColorView, state: self.valueState)
self.valueAlignmentView = selectSettingsRow(
title: localizedString("Alignment"),
action: #selector(toggleValueAlignment),
items: Alignments,
selected: self.valueAlignmentState
)
view.addArrangedSubview(self.valueAlignmentView!)
findAndToggleEnableNSControlState(self.valueAlignmentView, state: self.valueState)
view.addArrangedSubview(toggleSettingRow(
title: localizedString("Units"),
action: #selector(toggleUnits),
state: self.unitsState
))
view.addArrangedSubview(toggleSettingRow(
title: localizedString("Monochrome accent"),
action: #selector(toggleMonochrome),
state: self.monochromeState
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Color of upload"),
action: #selector(toggleUploadColor),
items: Color.allColors,
selected: self.uploadColorState.key
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Color of download"),
action: #selector(toggleDownloadColor),
items: Color.allColors,
selected: self.downloadColorState.key
))
return view
}
@objc private func changeDisplayMode(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String else { return }
self.modeState = key
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_mode", value: key)
self.display()
}
@objc private func toggleValue(_ sender: NSControl) {
self.valueState = controlState(sender)
findAndToggleEnableNSControlState(self.valueColorView, state: self.valueState)
findAndToggleEnableNSControlState(self.valueAlignmentView, state: self.valueState)
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_value", value: self.valueState)
self.display()
}
@objc private func toggleUnits(_ sender: NSControl) {
self.unitsState = controlState(sender)
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_units", value: self.unitsState)
self.display()
}
@objc private func toggleIcon(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String else { return }
self.icon = key
findAndToggleEnableNSControlState(self.transparentIconView, state: self.icon != "none")
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_icon", value: key)
self.display()
}
@objc private func toggleBase(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String else { return }
self.baseValue = key
Store.shared.set(key: "\(self.title)_base", value: self.baseValue)
}
@objc private func toggleMonochrome(_ sender: NSControl) {
self.monochromeState = controlState(sender)
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_monochrome", value: self.monochromeState)
self.display()
}
@objc private func toggleValueColor(_ sender: NSControl) {
self.valueColorState = controlState(sender)
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_valueColor", value: self.valueColorState)
self.display()
}
@objc private func toggleUploadColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.uploadColorState = newValue
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_uploadColor", value: key)
}
@objc private func toggleDownloadColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.downloadColorState = newValue
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_downloadColor", value: key)
}
public func setValue(upload: Int64, download: Int64) {
var updated: Bool = false
if self.downloadValue != download {
self.downloadValue = abs(download)
updated = true
}
if self.uploadValue != upload {
self.uploadValue = abs(upload)
updated = true
}
if updated {
DispatchQueue.main.async(execute: {
self.display()
})
}
}
@objc private func toggleTransparentIcons(_ sender: NSControl) {
self.transparentIconsState = controlState(sender)
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_transparentIcons", value: self.transparentIconsState)
self.display()
}
@objc private func toggleValueAlignment(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String else { return }
if let newAlignment = Alignments.first(where: { $0.key == key }) {
self.valueAlignmentState = newAlignment.key
}
Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_valueAlignment", value: key)
self.display()
}
}