//
//  AppDelegate.m
//  CaseLights
//
//  Created by Thomas Buck on 21.12.15.
//  Copyright © 2015 xythobuz. All rights reserved.
//

#import "AppDelegate.h"
#import "Serial.h"
#import "GPUStats.h"

#import <SystemInfoKit/SystemInfoKit.h>

#define PREF_SERIAL_PORT @"SerialPort"
#define PREF_LIGHTS_STATE @"LightState"

@interface AppDelegate ()

@end

@implementation AppDelegate

@synthesize statusMenu, application;
@synthesize menuColors, menuAnimations, menuVisualizations, menuPorts;
@synthesize buttonOff, buttonLights;
@synthesize statusItem, statusImage;
@synthesize staticColors;

- (void)applicationDidFinishLaunching:(NSNotification *)aNotification {
    // Prepare status bar menu
    statusImage = [NSImage imageNamed:@"MenuIcon"];
    [statusImage setTemplate:YES];
    statusItem = [[NSStatusBar systemStatusBar] statusItemWithLength:NSSquareStatusItemLength];
    [statusItem setImage:statusImage];
    [statusItem setMenu:statusMenu];
    
    // Set default configuration values, load existing ones
    NSUserDefaults *store = [NSUserDefaults standardUserDefaults];
    NSMutableDictionary *appDefaults = [NSMutableDictionary dictionaryWithObject:@"" forKey:PREF_SERIAL_PORT];
    [appDefaults setObject:[NSNumber numberWithBool:NO] forKey:PREF_LIGHTS_STATE];
    [store registerDefaults:appDefaults];
    [store synchronize];
    NSString *savedPort = [store stringForKey:PREF_SERIAL_PORT];
    BOOL turnOnLights = [store boolForKey:PREF_LIGHTS_STATE];
    
    // Prepare static colors menu
    staticColors = [NSDictionary dictionaryWithObjectsAndKeys:
                    [NSColor colorWithCalibratedRed:1.0f green:0.0f blue:0.0f alpha:0.0f], @"Red",
                    [NSColor colorWithCalibratedRed:0.0f green:1.0f blue:0.0f alpha:0.0f], @"Green",
                    [NSColor colorWithCalibratedRed:0.0f green:0.0f blue:1.0f alpha:0.0f], @"Blue",
                    [NSColor colorWithCalibratedRed:0.0f green:1.0f blue:1.0f alpha:0.0f], @"Cyan",
                    [NSColor colorWithCalibratedRed:1.0f green:0.0f blue:1.0f alpha:0.0f], @"Magenta",
                    [NSColor colorWithCalibratedRed:1.0f green:1.0f blue:0.0f alpha:0.0f], @"Yellow",
                    [NSColor colorWithCalibratedRed:1.0f green:1.0f blue:1.0f alpha:0.0f], @"White",
                    nil];
    for (NSString *key in [staticColors allKeys]) {
        NSMenuItem *item = [[NSMenuItem alloc] initWithTitle:key action:@selector(selectedStaticColor:) keyEquivalent:@""];
        [menuColors addItem:item];
        
        // TODO Enable item if it was last used
    }
    
    // TODO Prepare animations menu
    
    // Check if GPU Stats are available, add menu items if so
    NSNumber *usage;
    NSNumber *freeVRAM;
    NSNumber *usedVRAM;
    if ([GPUStats getGPUUsage:&usage freeVRAM:&freeVRAM usedVRAM:&usedVRAM] != 0) {
        NSLog(@"Error reading GPU information\n");
    } else {
        NSMenuItem *itemUsage = [[NSMenuItem alloc] initWithTitle:@"GPU Usage" action:@selector(selectedGPUVisualization:) keyEquivalent:@""];
        [menuVisualizations addItem:itemUsage];
        
        NSMenuItem *itemVRAM = [[NSMenuItem alloc] initWithTitle:@"VRAM Usage" action:@selector(selectedGPUVisualization:) keyEquivalent:@""];
        [menuVisualizations addItem:itemVRAM];
        
        // TODO Enable item if it was last used
    }
    
    // Check available temperatures and add menu items
    JSKSMC *smc = [JSKSMC smc];
    for (int i = 0; i < [[smc workingTempKeys] count]; i++) {
        NSString *key = [smc.workingTempKeys objectAtIndex:i];
        //NSString *name = [smc humanReadableNameForKey:key];
        
        //NSLog(@"%@: %@\n", key, name);
        
        if ([key isEqualToString:@"TC0D"]) {
            NSMenuItem *item = [[NSMenuItem alloc] initWithTitle:@"CPU Temperature" action:@selector(selectedCPUVisualization:) keyEquivalent:@""];
            [menuVisualizations addItem:item];
            
            // TODO enable item if it was last used
        }
        
        // TODO add GPU temperature item
    }
    
    // Add CPU Usage menu item
    NSMenuItem *cpuUsageItem = [[NSMenuItem alloc] initWithTitle:@"CPU Usage" action:@selector(selectedCPUVisualization:) keyEquivalent:@""];
    [menuVisualizations addItem:cpuUsageItem];
    
    // TODO enable item if it was last used
    
    // Add Memory Usage item
    NSMenuItem *memoryUsageItem = [[NSMenuItem alloc] initWithTitle:@"RAM Usage" action:@selector(selectedMemoryVisualization:) keyEquivalent:@""];
    [menuVisualizations addItem:memoryUsageItem];
    
    // TODO enable item if it was last used
    
    JSKSystemMonitor *systemMonitor = [JSKSystemMonitor systemMonitor];
    //JSKMCPUUsageInfo cpuUsageInfo = systemMonitor.cpuUsageInfo;
    JSKMMemoryUsageInfo memoryUsageInfo = systemMonitor.memoryUsageInfo;
    NSLog(@"Memory Usage: %lld Free, %lld Used, %lld Active, %lld Inactive, %lld Compressed, %lld Wired\n", memoryUsageInfo.freeMemory / 1000000, memoryUsageInfo.usedMemory / 1000000, memoryUsageInfo.activeMemory / 1000000, memoryUsageInfo.inactiveMemory / 1000000, memoryUsageInfo.compressedMemory / 1000000, memoryUsageInfo.wiredMemory / 1000000);
    
    // Prepare serial port menu
    NSArray *ports = [Serial listSerialPorts];
    if ([ports count] > 0) {
        [menuPorts removeAllItems];
        for (int i = 0; i < [ports count]; i++) {
            // Add Menu Item for this port
            NSMenuItem *item = [[NSMenuItem alloc] initWithTitle:[ports objectAtIndex:i] action:@selector(selectedSerialPort:) keyEquivalent:@""];
            [menuPorts addItem:item];
            
            // Set Enabled if it was used the last time
            if ((savedPort != nil) && [[ports objectAtIndex:i] isEqualToString:savedPort]) {
                [[menuPorts itemAtIndex:i] setState:NSOnState];
            }
        }
    }
    
    // TODO Open serial port, if it was already specified and found
    
    // Restore previously used lights configuration
    if (turnOnLights) {
        // TODO Turn on lights
        
        [buttonLights setState:NSOnState];
    } else {
        // TODO Turn off lights
        
    }
    
    // TODO Restore previously used LED configuration
}

- (void)applicationWillTerminate:(NSNotification *)aNotification {
    // TODO Close serial port, if it was specified and opened
    
}

- (IBAction)turnLEDsOff:(NSMenuItem *)sender {
    if ([sender state] == NSOffState) {
        // Turn off all other LED menu items
        for (int i = 0; i < [menuColors numberOfItems]; i++) {
            [[menuColors itemAtIndex:i] setState:NSOffState];
        }
        for (int i = 0; i < [menuAnimations numberOfItems]; i++) {
            [[menuAnimations itemAtIndex:i] setState:NSOffState];
        }
        for (int i = 0; i < [menuVisualizations numberOfItems]; i++) {
            [[menuVisualizations itemAtIndex:i] setState:NSOffState];
        }
        
        // Turn on "off" menu item
        [sender setState:NSOnState];
        
        // TODO Send command to turn off LEDs
    } else {
        // TODO Try to restore last LED setting
    }
}

- (IBAction)toggleLights:(NSMenuItem *)sender {
    if ([sender state] == NSOffState) {
        // TODO Turn on lights
        
        [sender setState:NSOnState];
    } else {
        // TODO Turn off lights
        
        [sender setState:NSOffState];
    }
    
    // Store changed value in preferences
    NSUserDefaults *store = [NSUserDefaults standardUserDefaults];
    [store setBool:([sender state] == NSOnState) forKey:PREF_LIGHTS_STATE];
    [store synchronize];
}

- (void)selectedStaticColor:(NSMenuItem *)source {
    // Turn off all other LED menu items
    for (int i = 0; i < [menuColors numberOfItems]; i++) {
        [[menuColors itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuAnimations numberOfItems]; i++) {
        [[menuAnimations itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuVisualizations numberOfItems]; i++) {
        [[menuVisualizations itemAtIndex:i] setState:NSOffState];
    }
    [buttonOff setState:NSOffState];
    
    // Turn on "off" menu item
    [source setState:NSOnState];
    
    // TODO store new selection
    
    // TODO send command
}

- (void)selectedGPUVisualization:(NSMenuItem *)sender {
    // Turn off all other LED menu items
    for (int i = 0; i < [menuColors numberOfItems]; i++) {
        [[menuColors itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuAnimations numberOfItems]; i++) {
        [[menuAnimations itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuVisualizations numberOfItems]; i++) {
        [[menuVisualizations itemAtIndex:i] setState:NSOffState];
    }
    [buttonOff setState:NSOffState];
    
    if ([sender.title isEqualToString:@"GPU Usage"]) {
        // Turn on "off" menu item
        [sender setState:NSOnState];
        
        // TODO store new selection
        
        // TODO send command
    } else if ([sender.title isEqualToString:@"VRAM Usage"]) {
        // Turn on "off" menu item
        [sender setState:NSOnState];
        
        // TODO store new selection
        
        // TODO send command
    } else {
        NSLog(@"Unknown GPU Visualization selected!\n");
    }
}

- (void)selectedCPUVisualization:(NSMenuItem *)sender {
    // Turn off all other LED menu items
    for (int i = 0; i < [menuColors numberOfItems]; i++) {
        [[menuColors itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuAnimations numberOfItems]; i++) {
        [[menuAnimations itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuVisualizations numberOfItems]; i++) {
        [[menuVisualizations itemAtIndex:i] setState:NSOffState];
    }
    [buttonOff setState:NSOffState];
    
    if ([sender.title isEqualToString:@"CPU Usage"]) {
        // Turn on "off" menu item
        [sender setState:NSOnState];
        
        // TODO store new selection
        
        // TODO send command
    } else if ([sender.title isEqualToString:@"CPU Temperature"]) {
        // Turn on "off" menu item
        [sender setState:NSOnState];
        
        // TODO store new selection
        
        // TODO send command
    } else {
        NSLog(@"Unknown CPU Visualization selected!\n");
    }
}

- (void)selectedMemoryVisualization:(NSMenuItem *)sender {
    // Turn off all other LED menu items
    for (int i = 0; i < [menuColors numberOfItems]; i++) {
        [[menuColors itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuAnimations numberOfItems]; i++) {
        [[menuAnimations itemAtIndex:i] setState:NSOffState];
    }
    for (int i = 0; i < [menuVisualizations numberOfItems]; i++) {
        [[menuVisualizations itemAtIndex:i] setState:NSOffState];
    }
    [buttonOff setState:NSOffState];
    
    // Turn on "off" menu item
    [sender setState:NSOnState];
    
    // TODO store new selection
    
    // TODO send command
}

- (void)selectedSerialPort:(NSMenuItem *)source {
    // Store selection for next start-up
    NSUserDefaults *store = [NSUserDefaults standardUserDefaults];
    [store setObject:[source title] forKey:PREF_SERIAL_PORT];
    [store synchronize];
    
    // De-select all other ports
    for (int i = 0; i < [menuPorts numberOfItems]; i++) {
        [[menuPorts itemAtIndex:i] setState:NSOffState];
    }
    
    // Select only the current port
    [source setState:NSOnState];
    
    // TODO Close previously opened port, if any
    
    // TODO Try to open selected port
}

- (IBAction)showAbout:(id)sender {
    [NSApp activateIgnoringOtherApps:YES];
    [application orderFrontStandardAboutPanel:self];
}

@end