macos-stats/Modules/Sensors/main.swift

//
//  main.swift
//  Sensors
//
//  Created by Serhiy Mytrovtsiy on 17/06/2020.
//  Using Swift 5.0.
//  Running on macOS 10.15.
//
//  Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
//

import Cocoa
import Kit

public class Sensors: Module {
    private let sensorsReader: SensorsReader
    private let popupView: Popup
    private let settingsView: Settings
    private let notificationsView: Notifications
    
    private var fanValueState: FanValue {
        FanValue(rawValue: Store.shared.string(key: "\(self.config.name)_fanValue", defaultValue: "percentage")) ?? .percentage
    }
    
    public init() {
        self.sensorsReader = SensorsReader()
        self.settingsView = Settings("Sensors", list: self.sensorsReader.list.sensors)
        self.popupView = Popup()
        self.notificationsView = Notifications(.sensors)
        
        super.init(
            popup: self.popupView,
            settings: self.settingsView,
            notifications: self.notificationsView
        )
        guard self.available else { return }
        
        self.popupView.setup(self.sensorsReader.list.sensors)
        self.notificationsView.setup(self.sensorsReader.list.sensors)
        
        self.settingsView.callback = { [weak self] in
            self?.sensorsReader.read()
        }
        self.settingsView.setInterval = { [weak self] value in
            self?.sensorsReader.setInterval(value)
        }
        self.settingsView.HIDcallback = { [weak self] in
            DispatchQueue.global(qos: .background).async {
                self?.sensorsReader.HIDCallback()
                DispatchQueue.main.async {
                    self?.popupView.setup(self?.sensorsReader.list.sensors)
                    self?.settingsView.setList(list: self?.sensorsReader.list.sensors ?? [])
                    self?.notificationsView.setup(self?.sensorsReader.list.sensors)
                }
            }
        }
        self.settingsView.unknownCallback = { [weak self] in
            DispatchQueue.global(qos: .background).async {
                self?.sensorsReader.unknownCallback()
                DispatchQueue.main.async {
                    self?.popupView.setup(self?.sensorsReader.list.sensors)
                    self?.settingsView.setList(list: self?.sensorsReader.list.sensors ?? [])
                    self?.notificationsView.setup(self?.sensorsReader.list.sensors)
                }
            }
        }
        
        self.sensorsReader.callbackHandler = { [weak self] value in
            self?.usageCallback(value)
        }
        self.sensorsReader.readyCallback = { [weak self] in
            self?.readyHandler()
        }
        
        self.addReader(self.sensorsReader)
    }
    
    public override func willTerminate() {
        guard SMCHelper.shared.isActive() else { return }
        
        self.sensorsReader.list.sensors.filter({ $0 is Fan }).forEach { (s: Sensor_p) in
            if let f = s as? Fan, let mode = f.customMode {
                if mode != .automatic {
                    SMCHelper.shared.setFanMode(f.id, mode: FanMode.automatic.rawValue)
                }
            }
        }
    }
    
    public override func isAvailable() -> Bool {
        return !self.sensorsReader.list.sensors.isEmpty
    }
    
    private func checkIfNoSensorsEnabled() {
        if self.sensorsReader.list.sensors.filter({ $0.state }).isEmpty {
            NotificationCenter.default.post(name: .toggleModule, object: nil, userInfo: ["module": self.config.name, "state": false])
        }
    }
    
    private func usageCallback(_ raw: Sensors_List?) {
        guard let value = raw, self.enabled else { return }
        
        var list: [Stack_t] = []
        var flatList: [[ColorValue]] = []
        
        value.sensors.forEach { (s: Sensor_p) in
            if s.state {
                var value = s.formattedMiniValue
                if let f = s as? Fan {
                    flatList.append([ColorValue(((f.value*100)/f.maxSpeed)/100)])
                    if self.fanValueState == .percentage {
                        value = "\(f.percentage)%"
                    }
                }
                list.append(Stack_t(key: s.key, value: value, additional: s.name))
            }
        }
        
        self.popupView.usageCallback(value.sensors)
        self.notificationsView.usageCallback(value.sensors)
        
        self.menuBar.widgets.filter{ $0.isActive }.forEach { (w: Widget) in
            switch w.item {
            case let widget as StackWidget: widget.setValues(list)
            case let widget as BarChart: widget.setValue(flatList)
            default: break
            }
        }
    }
}