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 ModuleKit
import StatsKit

public class Sensors: Module {
    private var sensorsReader: SensorsReader
    private let popupView: Popup = Popup()
    private var settingsView: Settings
    
    public init(_ store: UnsafePointer<Store>?, _ smc: UnsafePointer<SMCService>) {
        self.sensorsReader = SensorsReader(smc)
        self.settingsView = Settings("Disk", store: store!, list: &self.sensorsReader.list)
        
        super.init(
            store: store,
            popup: self.popupView,
            settings: self.settingsView
        )

        self.checkIfNoSensorsEnabled()
        self.popupView.setup(self.sensorsReader.list)
        
        self.settingsView.callback = { [unowned self] in
            self.checkIfNoSensorsEnabled()
            self.sensorsReader.read()
        }
        
        self.sensorsReader.readyCallback = { [unowned self] in
            self.readyHandler()
        }
        self.sensorsReader.callbackHandler = { [unowned self] value in
            self.usageCallback(value)
        }
        
        self.addReader(self.sensorsReader)
    }
    
    private func checkIfNoSensorsEnabled() {
        if self.sensorsReader.list.filter({ $0.state }).count == 0 {
            NotificationCenter.default.post(name: .toggleModule, object: nil, userInfo: ["module": self.config.name, "state": false])
        }
    }
    
    private func usageCallback(_ value: [Sensor_t]?) {
        if value == nil {
            return
        }
        
        self.popupView.usageCallback(value!)
        if let widget = self.widget as? SensorsWidget {
            widget.setValues(value?.filter{ $0.state }.map{ $0.formattedMiniValue } ?? [])
        }
    }
}