/*
 * Copyright (c) 2014, Peter Rutenbar <pruten@gmail.com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#import "shoeApplication.h"
#import "shoeScreenWindow.h"
#import "shoeScreenWindowController.h"
#import "shoePreferencesWindowController.h"
#include <ctype.h>

@implementation shoeApplication

#define mapkeymod(u, a, m) do { \
    assert((a >> 7) == 0); \
    uint16_t value = ((m) << 8)| (a); \
    rb_insert(keymap, u, &value, NULL); \
} while (0) \

#define mapkey(_u, a) mapkeymod(_u, a, 0)

- (void)initKeyboardMap
{
    keymap = rb_new(p_new_pool(NULL), sizeof(uint16_t));
    
    // Letters
    mapkey('a', 0x00);
    mapkey('b', 0x0b);
    mapkey('c', 0x08);
    mapkey('d', 0x02);
    mapkey('e', 0x0e);
    mapkey('f', 0x03);
    mapkey('g', 0x05);
    mapkey('h', 0x04);
    mapkey('i', 0x22);
    mapkey('j', 0x26);
    mapkey('k', 0x28);
    mapkey('l', 0x25);
    mapkey('m', 0x2e);
    mapkey('n', 0x2d);
    mapkey('o', 0x1f);
    mapkey('p', 0x23);
    mapkey('q', 0x0c);
    mapkey('r', 0x0f);
    mapkey('s', 0x01);
    mapkey('t', 0x11);
    mapkey('u', 0x20);
    mapkey('v', 0x09);
    mapkey('w', 0x0d);
    mapkey('x', 0x07);
    mapkey('y', 0x10);
    mapkey('z', 0x06);
    
    // Numbers
    mapkey('0', 0x1d);
    mapkey('1', 0x12);
    mapkey('2', 0x13);
    mapkey('3', 0x14);
    mapkey('4', 0x15);
    mapkey('5', 0x17);
    mapkey('6', 0x16);
    mapkey('7', 0x1a);
    mapkey('8', 0x1c);
    mapkey('9', 0x19);
    
    // Top row symbols
    mapkeymod(')', 0x1d, modShift);
    mapkeymod('!', 0x12, modShift);
    mapkeymod('@', 0x13, modShift);
    mapkeymod('#', 0x14, modShift);
    mapkeymod('$', 0x15, modShift);
    mapkeymod('%', 0x17, modShift);
    mapkeymod('^', 0x16, modShift);
    mapkeymod('&', 0x1a, modShift);
    mapkeymod('*', 0x1c, modShift);
    mapkeymod('(', 0x19, modShift);
    
    // Other symbols (no shift)
    mapkeymod('`', 0x32, 0);
    mapkeymod('-', 0x1b, 0);
    mapkeymod('=', 0x18, 0);
    mapkeymod('[', 0x21, 0);
    mapkeymod(']', 0x1e, 0);
    mapkeymod('\\', 0x2a, 0);
    mapkeymod(';', 0x29, 0);
    mapkeymod('\'', 0x27, 0);
    mapkeymod(',', 0x2b, 0);
    mapkeymod('.', 0x2f, 0);
    mapkeymod('/', 0x2c, 0);
    
    // Other symbols (with shift)
    mapkeymod('~', 0x32, modShift);
    mapkeymod('_', 0x1b, modShift);
    mapkeymod('+', 0x18, modShift);
    mapkeymod('{', 0x21, modShift);
    mapkeymod('}', 0x1e, modShift);
    mapkeymod('|', 0x2a, modShift);
    mapkeymod(':', 0x29, modShift);
    mapkeymod('"', 0x27, modShift);
    mapkeymod('<', 0x2b, modShift);
    mapkeymod('>', 0x2f, modShift);
    mapkeymod('?', 0x2c, modShift);
    
    // Function keys
    mapkey(NSF1FunctionKey, 0x7a);
    mapkey(NSF2FunctionKey, 0x78);
    mapkey(NSF3FunctionKey, 0x63);
    mapkey(NSF4FunctionKey, 0x76);
    mapkey(NSF5FunctionKey, 0x60);
    mapkey(NSF6FunctionKey, 0x61);
    mapkey(NSF7FunctionKey, 0x62);
    mapkey(NSF8FunctionKey, 0x64);
    mapkey(NSF9FunctionKey, 0x65);
    mapkey(NSF10FunctionKey, 0x6d);
    mapkey(NSF11FunctionKey, 0x67);
    mapkey(NSF12FunctionKey, 0x6f);
    mapkey(NSF13FunctionKey, 0x69);
    mapkey(NSF14FunctionKey, 0x6b);
    mapkey(NSF15FunctionKey, 0x71);
    
    // Arrows
    mapkey(NSUpArrowFunctionKey, 0x3e);
    mapkey(NSDownArrowFunctionKey, 0x3d);
    mapkey(NSRightArrowFunctionKey, 0x3c);
    mapkey(NSLeftArrowFunctionKey, 0x3b);
    
    // Delete
    mapkey(NSDeleteFunctionKey, 0x75);
    mapkey(NSBackspaceCharacter, 0x33);
    mapkey(NSDeleteCharacter, 0x33);
    
    // Enter, NL, CR
    mapkey(NSCarriageReturnCharacter, 0x24);
    mapkey(NSNewlineCharacter, 0x24);
    mapkey(NSEnterCharacter, 0x24);
    
    // Other keys
    mapkey(0x1b, 0x35); // escape
    mapkey(' ', 0x31); // space
    mapkey(NSTabCharacter, 0x30); // tab
}

- (void)sendEvent:(NSEvent *)event
{
    if (doCaptureKeys) {
        assert(isRunning);
        NSEventType type = [event type];
        if (type == NSFlagsChanged) {
            NSUInteger modifierFlags = [event modifierFlags];
            shoebill_key_modifier(modifierFlags >> 16);
            
            // Block any key-related event while the command key is down
            if (modifierFlags & NSCommandKeyMask)
                return ;
            
            [super sendEvent:event];
        }
        else if (type == NSKeyDown || type == NSKeyUp) {
            NSString *chars = [[event charactersIgnoringModifiers] lowercaseString];
            NSUInteger modifierFlags = [event modifierFlags];
            unichar c = [chars characterAtIndex:0];
            uint16_t value;
            
            if (keymap == NULL)
                [self initKeyboardMap];
            
            if (rb_find(keymap, c, &value)) {
                shoebill_key_modifier((value >> 8) | (modifierFlags >> 16));
                shoebill_key((type == NSKeyDown), value & 0xff);
                
            }
            
            // Block any key-related event while the command key is down
            if (modifierFlags & NSCommandKeyMask)
                return ;
            
            [super sendEvent:event];
            
        }
    }
    
    [super sendEvent:event];
}


- (void) complain:(NSString*)str
{
    NSAlert *theAlert = [NSAlert
                         alertWithMessageText:nil
                         defaultButton:nil
                         alternateButton:nil
                         otherButton:nil
                         informativeTextWithFormat:@"%@", str
                         ];
    [theAlert runModal];
}


- (BOOL) fetchUserDefaults
{
    uint32_t i;
    NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
    
    NSString *rootKernelPathStr = [defaults objectForKey:@"rootKernelPath"];
    NSString *romPathStr = [defaults objectForKey:@"romPath"];
    NSInteger verboseState = [defaults integerForKey:@"verboseState"];
    NSInteger memsize = [defaults integerForKey:@"memorySize"];
    NSData *pramData = [defaults objectForKey:@"pramData"];
    
    if (rootKernelPathStr == Nil || [rootKernelPathStr length]==0) {
        [self complain:@"Kernel path invalid!"];
        return NO;
    }
    
    else if (romPathStr == Nil || [romPathStr length] == 0) {
        [self complain:@"ROM path invalid!"];
        return NO;
    }
    
    if ((memsize < 1) || (memsize > 1024))
        memsize = 8;
    
    for (i=0; i<4; i++) {
        NSInteger height = [defaults integerForKey:[NSString stringWithFormat:@"screenHeight%u", i]];
        NSInteger width = [defaults integerForKey:[NSString stringWithFormat:@"screenWidth%u", i]];
        NSInteger enabled = [defaults integerForKey:[NSString stringWithFormat:@"screenEnabled%u", i]];
        
    
        if ((height < 342) || (height > 0xffff))
            height = 480;
        
        if ((width < 512) || (width > 0xffff))
            width = 640;
        
        screens[i].width = (uint16_t)width;
        screens[i].height = (uint16_t)height;
        screens[i].enabled = (uint16_t)enabled;
    }
    
    
    for (i=0; i<7; i++) {
        NSString *str = [defaults objectForKey:[NSString stringWithFormat:@"scsiPath%u", i]];
        if (str == nil || [str length] == 0)
            config.scsi_devices[i].path = NULL;
        else
            config.scsi_devices[i].path = strdup([str UTF8String]);
        
    }
    
    char *rootKernelPathCString = strdup([rootKernelPathStr UTF8String]);
    char *romPathCString = strdup([romPathStr UTF8String]);
    
    config.aux_verbose = (verboseState == NSOnState);
    config.ram_size = (uint32_t)memsize * 1024 * 1024;
    config.aux_kernel_path = rootKernelPathCString;
    config.rom_path = romPathCString;
    config.debug_mode = 0;
    
    [pramData getBytes:config.pram length:256];
    
    /* 
     * If the pram is corrupt, zap it.
     * A/UX will apparently never zap corrupted pram,
     * probably because it expects the bootloader/MacOS to do it.
     */
    if (memcmp(config.pram+0xc, "NuMc", 4) != 0)
        [self zapPram:defaults ptr:config.pram];
    
    NSString *defaultTapPath = [defaults objectForKey:@"tapPathE"];
    NSString *defaultMacAddr = [defaults objectForKey:@"macAddressE"];
    ethEnabled = [defaults integerForKey:@"ethernetEnabledE"];
    if (ethEnabled) {
        if (tap_fd_valid && strcmp(tapPath, [defaultTapPath UTF8String]) != 0) {
            close(tap_fd);
            tap_fd_valid = 0;
        }
        if (tapPath)
            free(tapPath);
        tapPath = strdup([defaultTapPath UTF8String]);
        if (!(parseMACAddr([defaultMacAddr UTF8String], mac))) {
            [self complain:@"Bad MAC addr"];
            ethEnabled = 0;
            return NO;
        }
    }
    
    return YES;
}

- (void) createScreenWindow:(uint8_t)slotnum
                     height:(uint16_t)height
                      width:(uint16_t)width
{
    shoebill_install_video_card(&config,
                                slotnum,
                                width,
                                height);
    
    windowController[slotnum] = [[shoeScreenWindowController alloc]
                                 initWithWindowNibName:@"shoeScreenView"
                                 slotnum:slotnum];
}

- (void) zapPram:(NSUserDefaults*)defaults ptr:(uint8_t*)ptr
{
    uint8_t init[256];
    
    shoebill_validate_or_zap_pram(init, 1);
    
    if (ptr)
        memcpy(ptr, init, 256);
    
    NSData *data = [NSData dataWithBytes:init length:256];
    [defaults setObject:data forKey:@"pramData"];
    [defaults synchronize];
    
    assert("zapPram" && (memcmp(init+0xc, "NuMc", 4) == 0));
}

void pram_callback (void *param, const uint8_t addr, const uint8_t byte)
{
    struct shoe_app_pram_data_t *pram = (struct shoe_app_pram_data_t*)param;
    pram->pram[addr] = byte;
    pram->updated = 1;
    //printf("pram_callback: set pram[0x%x] = 0x%02x (%c)\n", addr, byte, isprint(byte)?byte:'.');
}

- (void) flushPram
{
    uint8_t copy[256];
    if (pram->updated) {
        pram->updated = 0;
        memcpy(copy, pram->pram, 256);
        
        NSData* data = [NSData dataWithBytes:copy length:256];
        NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
        
        [defaults setObject:data forKey:@"pramData"];
        [defaults synchronize];
    }
}

- (void) pramFlushTimerFired:(NSTimer *)timer
{
    [self flushPram];
}

- (void) startEmulator
{
    if (isRunning)
        return;
    
    uint32_t i;
    
    bzero(&config, sizeof(shoebill_config_t));
    
    if (![self fetchUserDefaults])
        return ;
    
    self->pram = calloc(1, sizeof(struct shoe_app_pram_data_t));
    memcpy(self->pram, config.pram, 256);
    pram_flush_timer = [NSTimer
                        scheduledTimerWithTimeInterval:1.0
                        target:self
                        selector:@selector(pramFlushTimerFired:)
                        userInfo:nil
                        repeats:YES];
    config.pram_callback = pram_callback;
    config.pram_callback_param = (void*)self->pram;
    
    uint32_t result = shoebill_initialize(&config);
    
    if (!result) {
        [self complain:[NSString stringWithFormat:@"%s", config.error_msg]];
        return ;
    }
    
    for (i=0; i<4; i++) {
        if (screens[i].enabled)
            [self createScreenWindow:(9+i) height:screens[i].height width:screens[i].width];
    }
    
    if (ethEnabled) {
        if (!tap_fd_valid) {
            tap_fd = open(tapPath, O_RDWR | O_NOFOLLOW);
            if (tap_fd == -1) {
                NSAlert *theAlert = [NSAlert
                                     alertWithMessageText:nil
                                     defaultButton:nil
                                     alternateButton:nil
                                     otherButton:nil
                                     informativeTextWithFormat:@"Couldn't open tap device (errno = %d)", errno
                                     ];
                [theAlert runModal];
                return ;
            }
            tap_fd_valid = 1;
        }
        if (!shoebill_install_ethernet_card(&config, 13, mac, tap_fd)) {
            [self complain:[NSString stringWithFormat:@"%s", config.error_msg]];
            return ;
        }
            
    }
    
    shoebill_start();
    
    isRunning = true;
    
    for (i=0; i<16; i++) {
        if (windowController[i]) {
            shoeScreenWindow *win = (shoeScreenWindow*)[windowController[i] window];
            [win reevaluateKeyWindowness];
        }
    }
    
    [run_stop_menu_item setTitle: @"Stop"];
    [run_stop_menu_item setKeyEquivalent:@""];
}

- (void) stopEmulator
{
    uint32_t i;
    
    for (i=0; i<16; i++) {
        if (windowController[i]) {
            [windowController[i] close];
            windowController[i] = NULL;
        }
    }
    doCaptureKeys = false;
    doCaptureMouse = false;
    isRunning = false;
    
    shoebill_stop();
    
    [pram_flush_timer invalidate];
    pram_flush_timer = nil;
    [self flushPram];
    free(self->pram);
    
    if (config.aux_kernel_path)
        free((void*)config.aux_kernel_path);
    if (config.rom_path)
        free((void*)config.rom_path);
    
    [run_stop_menu_item setTitle: @"Run"];
    [run_stop_menu_item setKeyEquivalent:@"r"];
}

- (IBAction)runMenuItem:(id)sender
{
    if (isRunning)
        [self stopEmulator];
    else
        [self startEmulator];
}


@end