//
//  boe.menus.mm
//  BoE
//
//  Created by Celtic Minstrel on 14-03-25.
//
//

#include "boe.menus.h"

#include <Cocoa/Cocoa.h>
#include "universe.h"
#include "scenario.h"
#include "boe.party.h"
#include "boe.infodlg.h"
#include "boe.consts.h"

extern short on_spell_menu[2][62];
extern short spell_level[62];
extern short spell_cost[2][62];
extern short on_monst_menu[256];
extern const char* mage_s_name[62];
extern const char* priest_s_name[62];
extern bool party_in_memory;
extern short current_pc;
extern cUniverse univ;
extern cScenario scenario;
extern eGameMode overall_mode;

#ifndef __APPLE__
#error boe.menus.mm is Mac-specific code; try compiling boe.menus.win.cpp instead
#endif

using MenuHandle = NSMenu*;

MenuHandle menu_bar_handle;
MenuHandle apple_menu,file_menu,extra_menu,help_menu,monster_info_menu,library_menu;
MenuHandle actions_menu,music_menu,mage_spells_menu,priest_spells_menu;

@interface MenuHandler : NSObject
-(void) appMenu:(id) sender;
-(void) fileMenu:(id) sender;
-(void) optMenu:(id) sender;
-(void) actMenu:(id) sender;
-(void) monstMenu:(id) sender;
-(void) spellMenu:(id) sender;
-(void) libMenu:(id) sender;
-(void) helpMenu:(id) sender;
-(void) onlineHelp:(id) sender;
-(void) menuHelp:(id) sender; // The "About This Menu" options
@end

@interface MonsterWrapper : NSObject
@property cMonster* monst;
+(id) withMonster: (cMonster&) theMonster;
@end

@interface SpellWrapper : NSObject
@property int type, num;
+(id) withSpell:(int) num ofType:(int) type;
@end

void hideMenuBar() {
	[NSMenu setMenuBarVisible:NO];
}

void showMenuBar() {
	[NSMenu setMenuBarVisible:YES];
}

void adjust_monst_menu() {
	short i,monst_pos = 0;
	NSMenu* monst_menu;
	
	if(overall_mode == MODE_STARTUP) return;
	
	monst_menu = [[menu_bar_handle itemWithTitle: @"Monsters"] submenu];
	id targ = [[monst_menu itemAtIndex: 0] target];
	for(i = 0; i < 256; i++) {
		on_monst_menu[i] = -1;
	}
	for(i = 1; i < 256; i++) {
		if ((i == 1) || (univ.party.m_noted[i] > 0)) {
			on_monst_menu[monst_pos] = i;
			monst_pos++;
		}
	}
	
	while([monst_menu numberOfItems] > 2) {
		[monst_menu removeItemAtIndex: 2];
	}
	for(i = 0; i < 256; i++) {
		if(on_monst_menu[i] >= 0) {
			std::string monst_name = scenario.scen_monsters[on_monst_menu[i]].m_name;
			NSString* str = [NSString stringWithUTF8String: monst_name.c_str()];
			NSMenuItem* newItem = [monst_menu addItemWithTitle: str action: @selector(monstMenu:) keyEquivalent: @""];
			[newItem setTarget: targ];
			[newItem setRepresentedObject: [MonsterWrapper withMonster: scenario.scen_monsters[on_monst_menu[i]]]];
		}
	}
}

static void setMenuCallback(NSMenuItem* item, id targ, SEL selector, int num) {
	[item setTarget: targ];
	[item setAction: selector];
	[item setRepresentedObject: [[NSNumber numberWithInt: num] retain]];
}

void init_menubar() {
	NSApplication* app = [NSApplication sharedApplication];
	[NSBundle loadNibNamed: @"menu" owner: app];
	menu_bar_handle = [app mainMenu];
	
	apple_menu = [[menu_bar_handle itemWithTitle: @"Blades of Exile"] submenu];
	file_menu = [[menu_bar_handle itemWithTitle: @"File"] submenu];
	extra_menu = [[menu_bar_handle itemWithTitle: @"Options"] submenu];
	help_menu = [[menu_bar_handle itemWithTitle: @"Help"] submenu];
	library_menu = [[menu_bar_handle itemWithTitle: @"Library"] submenu];
	actions_menu = [[menu_bar_handle itemWithTitle: @"Actions"] submenu];
	
	MenuHandler* handler = [[[MenuHandler alloc] init] retain];
	setMenuCallback([help_menu itemAtIndex: 0], handler, @selector(onlineHelp:), 0);
	// TODO: Do away with "handle_apple_menu" and "appMenu" in favour of pushing "About" to the help menu
	setMenuCallback([apple_menu itemWithTitle: @"About Blades of Exile"], handler, @selector(appMenu:), 1);
	setMenuCallback([apple_menu itemWithTitle: @"Preferences…"], handler, @selector(fileMenu:), 6);
	setMenuCallback([apple_menu itemWithTitle: @"Quit Blades of Exile"], handler, @selector(fileMenu:), 8);
	// TODO: Check to make sure that Cocoa 0-indexes its menus
	// TODO: Need to verify arrangement of the File menu
	setMenuCallback([file_menu itemWithTitle: @"New Game…"], handler, @selector(fileMenu:), 4);
	setMenuCallback([file_menu itemWithTitle: @"Save"], handler, @selector(fileMenu:), 2);
	setMenuCallback([file_menu itemWithTitle: @"Save As…"], handler, @selector(fileMenu:), 3);
	setMenuCallback([file_menu itemWithTitle: @"Open…"], handler, @selector(fileMenu:), 1);
	setMenuCallback([file_menu itemWithTitle: @"Abort"], handler, @selector(fileMenu:), 0);
//	for(int i = 0; i < [file_menu numberOfItems]; i++)
//		setMenuCallback([file_menu itemAtIndex: i], handler, @selector(fileMenu:), i + 1);
	for(int i = 0; i < [extra_menu numberOfItems]; i++)
		setMenuCallback([extra_menu itemAtIndex: i], handler, @selector(optMenu:), i + 1);
	for(int i = 1; i < [help_menu numberOfItems]; i++)
		setMenuCallback([help_menu itemAtIndex: i], handler, @selector(helpMenu:), i - 1);
	for(int i = 0; i < [library_menu numberOfItems]; i++)
		setMenuCallback([library_menu itemAtIndex: i], handler, @selector(libMenu:), i + 1);
	for(int i = 0; i < [actions_menu numberOfItems]; i++)
		setMenuCallback([actions_menu itemAtIndex: i], handler, @selector(actMenu:), i + 1);
	// TODO: Spell menus and monster menu
	NSMenu* mage_menu = [[menu_bar_handle itemWithTitle: @"Mage Spells"] submenu];
	setMenuCallback([mage_menu itemAtIndex: 0], handler, @selector(menuHelp:), 0);
	NSMenu* priest_menu = [[menu_bar_handle itemWithTitle: @"Priest Spells"] submenu];
	setMenuCallback([priest_menu itemAtIndex: 0], handler, @selector(menuHelp:), 1);
	NSMenu* monst_menu = [[menu_bar_handle itemWithTitle: @"Monsters"] submenu];
	setMenuCallback([monst_menu itemAtIndex: 0], handler, @selector(menuHelp:), 2);
	
	menu_activate();
}

void init_spell_menus()
{
	short i,j;
	
	for (i = 0; i < 2; i++)
		for (j = 0; j < 62; j++)
			on_spell_menu[i][j] = -1;
	
}



void adjust_spell_menus()
{
	short i,j,spell_pos = 0;
	NSMenu* spell_menu;
	char spell_name[256];
	short old_on_spell_menu[2][62];
	bool need_menu_change = false;
	
	if(overall_mode == MODE_STARTUP || current_pc == 6)
		return;
	
	for (i = 0; i < 2; i++)
		for (j = 0; j < 62; j++)
			old_on_spell_menu[i][j] = on_spell_menu[i][j];
	
	spell_menu = [[menu_bar_handle itemWithTitle: @"Cast Mage"] submenu];
	id targ = [[spell_menu itemAtIndex: 0] target];
	
	for (i = 0; i < 62; i++) {
		on_spell_menu[0][i] = -1;
	}
	for (i = 0; i < 62; i++)
		if (pc_can_cast_spell(current_pc,0,i)) {
			on_spell_menu[0][spell_pos] = i;
			spell_pos++;
		}
	for (i = 0; i < 62; i++)
		if (on_spell_menu[0][i] != old_on_spell_menu[0][i])
			need_menu_change = true;
	if (need_menu_change) {
		while([spell_menu numberOfItems] > 2) {
			[spell_menu removeItemAtIndex: 2];
		}
		for (i = 0; i < 62; i++)
			if (on_spell_menu[0][i] >= 0) {
				if (spell_cost[0][on_spell_menu[0][i]] > 0)
					sprintf(spell_name," L%d - %s, C %d",spell_level[on_spell_menu[0][i]],
							mage_s_name[on_spell_menu[0][i]],spell_cost[0][on_spell_menu[0][i]]);
				else sprintf(spell_name," L%d - %s, C ?",spell_level[on_spell_menu[0][i]],
							 mage_s_name[on_spell_menu[0][i]]);
				spell_name[0] = strlen((char *) spell_name);
				//strcpy((char *) (spell_name + 1),mage_s_name[on_spell_menu[0][i]]);
				NSString* str = [NSString stringWithUTF8String: spell_name];
				NSMenuItem* newItem = [spell_menu addItemWithTitle: str action: @selector(spellMenu:) keyEquivalent: @""];
				[newItem setTarget: targ];
				[newItem setRepresentedObject: [SpellWrapper withSpell: i ofType: 0]];
			}
	}
	
	need_menu_change = false;
	spell_pos = 0;
	
	spell_menu = [[menu_bar_handle itemWithTitle: @"Cast Priest"] submenu];
	
	for (i = 0; i < 62; i++) {
		on_spell_menu[1][i] = -1;
	}
	for (i = 0; i < 62; i++)
		if (pc_can_cast_spell(current_pc,1,i)) {
			on_spell_menu[1][spell_pos] = i;
			spell_pos++;
		}
	for (i = 0; i < 62; i++)
		if (on_spell_menu[1][i] != old_on_spell_menu[1][i])
			need_menu_change = true;
	if (need_menu_change) {
		while([spell_menu numberOfItems] > 2) {
			[spell_menu removeItemAtIndex: 2];
		}
		for (i = 0; i < 62; i++)
			if (on_spell_menu[1][i] >= 0) {
				//spell_name[0] = strlen((char *) priest_s_name[on_spell_menu[1][i]]);
				//strcpy((char *) (spell_name + 1),priest_s_name[on_spell_menu[1][i]]);
				if (spell_cost[1][on_spell_menu[1][i]] > 0)
					sprintf(spell_name," L%d - %s, C %d",spell_level[on_spell_menu[1][i]],
							priest_s_name[on_spell_menu[1][i]],spell_cost[1][on_spell_menu[1][i]]);
				else sprintf(spell_name," L%d - %s, C ?",spell_level[i],
							 priest_s_name[on_spell_menu[1][i]]);
				spell_name[0] = strlen((char *) spell_name);
				// TODO: Figure out what to put for the action (which should be an Objective-C selector)
				NSString* str = [NSString stringWithUTF8String: spell_name];
				NSMenuItem* newItem = [spell_menu addItemWithTitle: str action: @selector(spellMenu:) keyEquivalent: @""];
				[newItem setTarget: targ];
				[newItem setRepresentedObject: [SpellWrapper withSpell: i ofType: 1]];
			}
	}
	
	
}


void menu_activate() {
	if(overall_mode == MODE_STARTUP) {
		[[menu_bar_handle itemWithTitle: @"Options"] setEnabled: NO];
		[[menu_bar_handle itemWithTitle: @"Monsters"] setEnabled: NO];
		[[menu_bar_handle itemWithTitle: @"Actions"] setEnabled: NO];
		[[menu_bar_handle itemWithTitle: @"Cast Mage"] setEnabled: NO];
		[[menu_bar_handle itemWithTitle: @"Cast Priest"] setEnabled: NO];
		
		[[file_menu itemWithTitle: @"Save"] setEnabled: NO];
		if(!party_in_memory) {
			[[file_menu itemWithTitle: @"Save As…"] setEnabled: NO];
		} else {
			[[file_menu itemWithTitle: @"Save As…"] setEnabled: YES];
		}
		return;
	}
	
	[[menu_bar_handle itemWithTitle: @"Options"] setEnabled: YES];
	[[menu_bar_handle itemWithTitle: @"Monsters"] setEnabled: YES];
	[[menu_bar_handle itemWithTitle: @"Actions"] setEnabled: YES];
	[[menu_bar_handle itemWithTitle: @"Cast Mage"] setEnabled: YES];
	[[menu_bar_handle itemWithTitle: @"Cast Priest"] setEnabled: YES];
	
	[[file_menu itemWithTitle: @"Save"] setEnabled: YES];
	[[file_menu itemWithTitle: @"Save As…"] setEnabled: YES];
}

void handle_apple_menu(int item_hit);
void handle_file_menu(int item_hit);
void handle_options_menu(int item_hit);
void handle_help_menu(int item_hit);
void handle_library_menu(int item_hit);
void handle_actions_menu(int item_hit);
void handle_monster_info_menu(int item_hit);
void handle_mage_spells_menu(int item_hit);
void handle_priest_spells_menu(int item_hit);

@implementation MenuHandler
-(void) appMenu:(id) sender {
	handle_apple_menu([[sender representedObject] intValue]);
}

-(void) fileMenu:(id) sender {
	handle_file_menu([[sender representedObject] intValue]);
}

-(void) optMenu:(id) sender {
	handle_options_menu([[sender representedObject] intValue]);
}

-(void) actMenu:(id) sender {
	handle_actions_menu([[sender representedObject] intValue]);
}

-(void) monstMenu:(id) sender {
	handle_monster_info_menu(1);
}

-(void) spellMenu:(id) sender {
	SpellWrapper* spell = [sender representedObject];
	if([spell type] == 0) handle_mage_spells_menu([spell num]);
	else if([spell type] == 1) handle_priest_spells_menu([spell num]);
}

-(void) libMenu:(id) sender {
	handle_library_menu([[sender representedObject] intValue]);
}

-(void) helpMenu:(id) sender {
	handle_help_menu([[sender representedObject] intValue]);
}

-(void) onlineHelp:(id) sender {
	[[NSWorkspace sharedWorkspace] openURL: [NSURL URLWithString: @"https://calref.net/~sylae/boe-doc/game/Contents.html"]];
}

-(void) menuHelp:(id) sender {
	// TODO: Do something for monster menu as well
	if([[sender representedObject] intValue] < 2)
		give_help(209,0);
}
@end

@implementation MonsterWrapper
@synthesize monst;

+(id) withMonster:(cMonster&) theMonster {
	MonsterWrapper* wrapper = [[MonsterWrapper alloc] init];
	[wrapper setMonst: &theMonster];
	return [wrapper retain];
}
@end

@implementation SpellWrapper
@synthesize type, num;

+(id) withSpell:(int) num ofType:(int) type {
	SpellWrapper* wrapper = [[SpellWrapper alloc] init];
	[wrapper setType: type];
	[wrapper setNum: num];
	return [wrapper retain];
}
@end