//
//  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;
@synthesize serial;

- (void)applicationDidFinishLaunching:(NSNotification *)aNotification {
    serial = [[Serial alloc] init];
    
    // 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
    }
    
    JSKSystemMonitor *systemMonitor = [JSKSystemMonitor systemMonitor];
    
#ifdef DEBUG
    JSKMCPUUsageInfo cpuUsageInfo = systemMonitor.cpuUsageInfo;
    NSLog(@"CPU Usage: %.3f%%\n", cpuUsageInfo.usage);
    
    JSKMMemoryUsageInfo memoryUsageInfo = systemMonitor.memoryUsageInfo;
    NSLog(@"Memory Usage: %.2fGB Free, %.2fGB Used, %.2fGB Active, %.2fGB Inactive, %.2fGB Compressed, %.2fGB Wired\n", memoryUsageInfo.freeMemory / (1024.0 * 1024.0 * 1024.0), memoryUsageInfo.usedMemory / (1024.0 * 1024.0 * 1024.0), memoryUsageInfo.activeMemory / (1024.0 * 1024.0 * 1024.0), memoryUsageInfo.inactiveMemory / (1024.0 * 1024.0 * 1024.0), memoryUsageInfo.compressedMemory / (1024.0 * 1024.0 * 1024.0), memoryUsageInfo.wiredMemory / (1024.0 * 1024.0 * 1024.0));
#endif
    
    // 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
    
    // 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];
        
#ifdef DEBUG
        NSString *name = [smc humanReadableNameForKey:key];
        NSLog(@"Sensor \"%@\": \"%@\"\n", key, name);
#endif
        
        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
        }
        
        if ([key isEqualToString:@"TG0D"]) {
            NSMenuItem *item = [[NSMenuItem alloc] initWithTitle:@"GPU Temperature" action:@selector(selectedGPUVisualization:) keyEquivalent:@""];
            [menuVisualizations addItem:item];
            
            // TODO enable item if it was last used
        }
    }
    
    // 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];
                
                // Try to open serial port
                [serial setPortName:savedPort];
                if ([serial openPort]) {
                    // Unselect it when an error occured opening the port
                    [[menuPorts itemAtIndex:i] setState:NSOffState];
                }
            }
        }
    }
    
    // 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 {
    // Close serial port, if it was opened
    if ([serial isOpen]) {
        [serial closePort];
    }
}
- (IBAction)relistSerialPorts:(id)sender {
    // Refill port list
    NSArray *ports = [Serial listSerialPorts];
    [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];
        
        // Mark it if it is currently open
        if ([serial isOpen]) {
            if ([[ports objectAtIndex:i] isEqualToString:[serial portName]]) {
                [[menuPorts itemAtIndex:i] setState:NSOnState];
            }
        }
    }
}

- (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 if ([sender.title isEqualToString:@"GPU Temperature"]) {
        // 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];
    
    // Close previously opened port, if any
    if ([serial isOpen]) {
        [serial closePort];
    }
    
    // Try to open selected port
    [serial setPortName:[source title]];
    if ([serial openPort] != 0) {
        [source setState:NSOffState];
    } else {
        // TODO Restore the current configuration
    }
}

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

@end