macos-stats/Modules/CPU/main.swift

//
//  main.swift
//  CPU
//
//  Created by Serhiy Mytrovtsiy on 09/04/2020.
//  Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
//

import Cocoa
import ModuleKit
import StatsKit

public struct CPU_Load: value_t {
    var totalUsage: Double = 0
    var usagePerCore: [Double] = []
    
    var systemLoad: Double = 0
    var userLoad: Double = 0
    var idleLoad: Double = 0
    
    public var widget_value: Double {
        get {
            return self.totalUsage
        }
    }
}

public struct TopProcess {
    var pid: Int = 0
    var command: String = ""
    var usage: Double = 0
}

public class CPU: Module {
    private let popupView: Popup = Popup()
    private var settingsView: Settings
    
    private var loadReader: LoadReader? = nil
    private let smc: UnsafePointer<SMCService>?
    
    public init(_ store: UnsafePointer<Store>, _ smc: UnsafePointer<SMCService>) {
        self.smc = smc
        self.settingsView = Settings("CPU", store: store)
        
        super.init(
            store: store,
            popup: self.popupView,
            settings: self.settingsView
        )
        guard self.available else { return }
        
        self.loadReader = LoadReader()
        self.loadReader?.store = store
        
        self.settingsView.callback = { [unowned self] in
            self.loadReader?.read()
        }
        self.settingsView.setInterval = { [unowned self] value in
            self.loadReader?.setInterval(value)
        }
        
        self.loadReader?.readyCallback = { [unowned self] in
            self.readyHandler()
        }
        self.loadReader?.callbackHandler = { [unowned self] value in
            self.loadCallback(value)
        }
        
        if let reader = self.loadReader {
            self.addReader(reader)
        }
    }
    
    private func loadCallback(_ value: CPU_Load?) {
        if value == nil {
            return
        }
        
        let temperature = self.smc?.pointee.getValue("TC0C") ?? self.smc?.pointee.getValue("TC0D") ?? self.smc?.pointee.getValue("TC0E")
        self.popupView.loadCallback(value!, tempValue: temperature)
        
        if let widget = self.widget as? Mini {
            widget.setValue(value!.totalUsage, sufix: "%")
        }
        if let widget = self.widget as? LineChart {
            widget.setValue(value!.totalUsage)
        }
        if let widget = self.widget as? BarChart {
            widget.setValue(value!.usagePerCore)
        }
    }
}