/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2014 Olaf Wintermann. 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 HOLDER 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 <stdio.h>
#import <stdlib.h>
#import <string.h>
#import <stdarg.h>
#import <objc/runtime.h>

#import "menu.h"
#import "window.h"
#import "EventData.h"

#pragma GCC diagnostic ignored "-Wundeclared-selector"
#pragma clang diagnostic ignored "-Wundeclared-selector"

// holds all items that need bindings
// value type: MenuItem*
static NSMutableArray *bindingItems;

@implementation MenuItem

- (MenuItem*)init:(int)itId {
    self.itemId = [[NSString alloc] initWithFormat:@"item%d", itId];
    return self;
}

- (void)handleToggleEvent:(id)sender {
    NSMenuItem *item = (NSMenuItem*)sender;
    if(!_state) {
        item.state = NSControlStateValueOn;
    } else {
        item.state = NSControlStateValueOff;
    }
    _state = !_state;
    
    if(_callback) {
        UiEvent event;
        event.obj = _obj;
        event.window = event.obj->window;
        event.document = event.obj->ctx->document;
        event.eventdata = NULL;
        event.eventdatatype = 0;
        event.intval = _state;
        event.set = ui_get_setop();
        _callback(&event, _userdata);
    }
}

@end

static ui_menu_add_f createMenuItem[] = {
    /* UI_MENU                 */ add_menu_widget,
    /* UI_MENU_ITEM            */ add_menuitem_widget,
    /* UI_MENU_CHECK_ITEM      */ add_checkitem_widget,
    /* UI_MENU_RADIO_ITEM      */ add_radioitem_widget,
    /* UI_MENU_ITEM_LIST       */ add_menuitem_list_widget,
    /* UI_MENU_CHECKITEM_LIST  */ add_menuitem_list_widget,
    /* UI_MENU_RADIOITEM_LIST  */ add_menuitem_list_widget,
    /* UI_MENU_SEPARATOR       */ add_menuseparator_widget
};

void ui_add_menu_items(UiObject *obj, NSMenu *parent, int i, UiMenu *menu) {
    UiMenuItemI *it = menu->items_begin;
    int index = 0;
    while(it) {
        createMenuItem[it->type](obj, parent, index, it);
        it = it->next;
        index++;
    }
}

void add_menu_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    UiMenu *it = (UiMenu*)item;
    NSString *str = [[NSString alloc] initWithUTF8String:it->label];
    NSMenu *menu = [[NSMenu alloc] initWithTitle: str];
    NSMenuItem *menuItem = [parent addItemWithTitle:str action:nil keyEquivalent:@""];
    [parent setSubmenu:menu forItem:menuItem];
    
    ui_add_menu_items(obj, menu, i, it);
}

void add_menuitem_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    UiMenuItem *it = (UiMenuItem*)item;
    
    NSString *str = [[NSString alloc] initWithUTF8String:it->label];
    NSMenuItem *menuItem = [parent addItemWithTitle:str action:@selector(menuItemAction) keyEquivalent:@""];
    
    if(it->callback) {
        EventData *event = [[EventData alloc] init:it->callback userdata:it->userdata action:it->action];
        if(obj) {
            event.obj = obj;
            menuItem.target = event;
            menuItem.action = @selector(handleEvent:);
        }
        objc_setAssociatedObject(menuItem, "eventdata", event, OBJC_ASSOCIATION_RETAIN);
    }
}

void add_menuseparator_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    NSMenuItem *menuItem = [NSMenuItem separatorItem];
    [parent addItem:menuItem];
}

static int nItem = 0;

void add_checkitem_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    UiMenuCheckItem *it = (UiMenuCheckItem*)item;
    
    NSString *str = [[NSString alloc] initWithUTF8String:it->label];
    NSMenuItem *menuItem = [parent addItemWithTitle:str action:@selector(menuCheckItemAction:) keyEquivalent:@""];
    
    MenuItem *mItem = [[MenuItem alloc] init:nItem++];
    mItem.menuItem = menuItem;
    mItem.obj = obj;
    mItem.callback = it->callback;
    mItem.userdata = it->userdata;
    mItem.checkItem = it;
    if(it->varname) {
        mItem.varname = [[NSString alloc] initWithUTF8String:it->varname];
    }
    
    objc_setAssociatedObject(menuItem, "menuitem", mItem, OBJC_ASSOCIATION_RETAIN);
    
    if(!obj) {
        [bindingItems addObject:mItem];
    } else {
        mItem.var = uic_widget_var(obj->ctx, obj->ctx, NULL, it->varname, UI_VAR_INTEGER);
        if(mItem.var) {
            UiInteger *i = mItem.var->value;
            i->obj = (__bridge void*)mItem;
            i->get = ui_menu_toggleitem_get;
            i->set = ui_menu_toggleitem_set;
        }
        menuItem.target = mItem;
        menuItem.action = @selector(handleToggleEvent:);
    }
}

void add_radioitem_widget(UiObject *obj, NSMenu *parent, int index, UiMenuItemI *item) {
    UiMenuRadioItem *it = (UiMenuRadioItem*)item;
    
    NSString *str = [[NSString alloc] initWithUTF8String:it->label];
    NSMenuItem *menuItem = [parent addItemWithTitle:str action:@selector(menuRadioItemAction:) keyEquivalent:@""];
    
    MenuItem *mItem = [[MenuItem alloc] init:nItem++];
    mItem.callback = it->callback;
    mItem.userdata = it->userdata;
    mItem.radioItem = it;
    
    objc_setAssociatedObject(menuItem, "menuitem", mItem, OBJC_ASSOCIATION_RETAIN);
    [bindingItems addObject:mItem];
}

void add_checkitemnv_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    
}

void add_menuitem_list_widget(UiObject *obj, NSMenu *parent, int i, UiMenuItemI *item) {
    
}


static BOOL menu_is_initialized = NO;

void ui_menu_init(void) {
    if(menu_is_initialized) {
        return;
    }
    menu_is_initialized = YES;
    
    bindingItems = [[NSMutableArray alloc] init];
    
    UiMenu *menus_begin = uic_get_menu_list();
    UiMenu *ls = menus_begin;
    int index = 1;
    while(ls) {
        if(ls->item.type == UI_MENU) {
            NSString *str = [[NSString alloc] initWithUTF8String:ls->label];
            NSMenu *menu = [[NSMenu alloc] initWithTitle: str];
            NSMenuItem *menuItem = [[NSApp mainMenu] insertItemWithTitle:str action:nil keyEquivalent:@"" atIndex:index];
            [[NSApp mainMenu] setSubmenu:menu forItem:menuItem];
            
            ui_add_menu_items(NULL, menu, 0, ls);
        }
        ls = (UiMenu*)ls->item.next;
        index++;
    }
}

NSArray* ui_get_binding_items(void) {
    return bindingItems;
}


int64_t ui_menu_toggleitem_get(UiInteger *i) {
    MenuItem *item = (__bridge MenuItem*)i->obj;
    i->value = item.state;
    return i->value;
}

void ui_menu_toggleitem_set(UiInteger *i, int64_t value) {
    MenuItem *item = (__bridge MenuItem*)i->obj;
    i->value = value;
    if(value != 0) {
        item.menuItem.state = NSControlStateValueOn;
    } else {
        item.menuItem.state = NSControlStateValueOff;
    }
}
