macos-stats/Stats/Modules/CPU/CPU.swift

//
//  CPU.swift
//  Stats
//
//  Created by Serhiy Mytrovtsiy on 01.06.2019.
//  Copyright © 2019 Serhiy Mytrovtsiy. All rights reserved.
//

import Cocoa
import Charts
import Repeat

class CPU: Module {
    public var name: String = "CPU"
    public var updateInterval: Double = 1
    
    public var enabled: Bool = true
    public var available: Bool = true
    
    public var readers: [Reader] = []
    public var task: Repeater?
    
    public var widget: ModuleWidget = ModuleWidget()
    public var popup: ModulePopup = ModulePopup(true)
    public var menu: NSMenuItem = NSMenuItem()
    
    internal let defaults = UserDefaults.standard
    internal var submenu: NSMenu = NSMenu()
    
    internal var systemValue: NSTextField = NSTextField()
    internal var userValue: NSTextField = NSTextField()
    internal var idleValue: NSTextField = NSTextField()
    internal var processViewList: [NSStackView] = []
    internal var chart: LineChartView = LineChartView()
    
    init() {
        if !self.available { return }
        
        self.enabled = defaults.object(forKey: name) != nil ? defaults.bool(forKey: name) : true
        self.updateInterval = defaults.object(forKey: "\(name)_interval") != nil ? defaults.double(forKey: "\(name)_interval") : self.updateInterval
        self.widget.type = defaults.object(forKey: "\(name)_widget") != nil ? defaults.float(forKey: "\(name)_widget") : Widgets.Mini
        
        self.initWidget()
        self.initMenu()
        self.initPopup()
        
        readers.append(CPULoadReader(self.name, self.loadUpdater, self.chartUpdater, true))
        readers.append(CPUUsageReader(self.usageUpdater))
        readers.append(CPUProcessReader(self.processesUpdater))
        
        self.task = Repeater.init(interval: .seconds(self.updateInterval), observer: { _ in
            self.readers.forEach { reader in
                reader.read()
            }
        })
    }
    
    public func start() {
        if self.task != nil && self.task!.state.isRunning == false {
            self.task!.start()
        }
    }
    
    public func stop() {
        if self.task!.state.isRunning {
            self.task?.pause()
        }
    }
    
    public func restart () {
        self.stop()
        self.start()
    }
    
    private func loadUpdater(value: [Double]) {
        if !value.isEmpty && self.widget.view is Widget {
            (self.widget.view as! Widget).setValue(data: value)
        }
    }
}