macos-stats/Kit/Widgets/State.swift

//
//  State.swift
//  Kit
//
//  Created by Serhiy Mytrovtsiy on 18/09/2022.
//  Using Swift 5.0.
//  Running on macOS 12.6.
//
//  Copyright © 2022 Serhiy Mytrovtsiy. All rights reserved.
//

import Cocoa

public class StateWidget: WidgetWrapper {
    private var activeColorState: Color = .secondGreen
    private var nonactiveColorState: Color = .secondRed
    
    private var value: Bool = false
    
    private var colors: [Color] = Color.allColors
    
    public init(title: String, config: NSDictionary?, preview: Bool = false) {
        if config != nil {
            var configuration = config!
            
            if preview {
                if let previewConfig = config!["Preview"] as? NSDictionary {
                    configuration = previewConfig
                    if let value = configuration["Value"] as? Bool {
                        self.value = value
                    }
                }
            }
        }
        
        super.init(.state, title: title, frame: CGRect(
            x: 0,
            y: Constants.Widget.margin.y,
            width: 8 + (2*Constants.Widget.margin.x),
            height: Constants.Widget.height - (2*Constants.Widget.margin.y)
        ))
        
        self.canDrawConcurrently = true
        
        if !preview {
            self.activeColorState = Color.fromString(Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_activeColor", defaultValue: self.activeColorState.key))
            self.nonactiveColorState = Color.fromString(Store.shared.string(key: "\(self.title)_\(self.type.rawValue)_nonactiveColor", defaultValue: self.nonactiveColorState.key))
        }
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    public override func draw(_ dirtyRect: NSRect) {
        super.draw(dirtyRect)
        
        let circle = NSBezierPath(ovalIn: CGRect(x: Constants.Widget.margin.x, y: (self.frame.height - 8)/2, width: 8, height: 8))
        let color = self.value ? self.activeColorState : self.nonactiveColorState
        (color.additional as? NSColor)?.set()
        circle.fill()
    }
    
    public func setValue(_ value: Bool) {
        guard self.value != value else { return }
        self.value = value
        DispatchQueue.main.async(execute: {
            self.display()
        })
    }
    
    // MARK: - Settings
    
    public override func settings() -> NSView {
        let view = SettingsContainerView()
        
        view.addArrangedSubview(selectSettingsRow(
            title: localizedString("Active state color"),
            action: #selector(self.toggleActiveColor),
            items: self.colors,
            selected: self.activeColorState.key
        ))
        
        view.addArrangedSubview(selectSettingsRow(
            title: localizedString("Nonactive state color"),
            action: #selector(self.toggleNonactiveColor),
            items: self.colors,
            selected: self.nonactiveColorState.key
        ))
        
        return view
    }
    
    @objc private func toggleActiveColor(_ sender: NSMenuItem) {
        guard let key = sender.representedObject as? String else { return }
        if let newColor = Color.allCases.first(where: { $0.key == key }) {
            self.activeColorState = newColor
        }
        Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_activeColor", value: key)
        self.display()
    }
    
    @objc private func toggleNonactiveColor(_ sender: NSMenuItem) {
        guard let key = sender.representedObject as? String else { return }
        if let newColor = Color.allCases.first(where: { $0.key == key }) {
            self.nonactiveColorState = newColor
        }
        Store.shared.set(key: "\(self.title)_\(self.type.rawValue)_nonactiveColor", value: key)
        self.display()
    }
}