/*

 * Hedgewars-iOS, a Hedgewars port for iOS devices

 * Copyright (c) 2009-2010 Vittorio Giovara <vittorio.giovara@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

 *

 * File created on 10/01/2010.

 */

#import "GameSetup.h"

#import "SDL_uikitappdelegate.h"

#import "SDL_net.h"

#import "PascalImports.h"

#import "CommodityFunctions.h"

#import "OverlayViewController.h"

#define BUFFER_SIZE 255     // like in original frontend

@implementation GameSetup

@synthesize systemSettings, gameConfig, savePath, menuStyle;

-(id) initWithDictionary:(NSDictionary *)gameDictionary {

    if (self = [super init]) {

        ipcPort = randomPort();

        // the general settings file + menu style (read by the overlay)

        NSDictionary *dictSett = [[NSDictionary alloc] initWithContentsOfFile:SETTINGS_FILE()];

        self.menuStyle = [[dictSett objectForKey:@"menu"] boolValue];

        self.systemSettings = dictSett;

        [dictSett release];

        // this game run settings

        self.gameConfig = [gameDictionary objectForKey:@"game_dictionary"];

        // is it a netgame?

        isNetGame = [[gameDictionary objectForKey:@"netgame"] boolValue];

        // is it a Save?

        NSString *path = [gameDictionary objectForKey:@"savefile"];

        // if path is empty it means that you have to create a new file, otherwise read from that file

        if ([path isEqualToString:@""] == YES) {

            NSDateFormatter *outputFormatter = [[NSDateFormatter alloc] init];

            [outputFormatter setDateFormat:@"yyyy-MM-dd '@' HH.mm"];

            NSString *newDateString = [outputFormatter stringFromDate:[NSDate date]];

            self.savePath = [SAVES_DIRECTORY() stringByAppendingFormat:@"%@.hws", newDateString];

            [outputFormatter release];

        } else

            self.savePath = path;

    }

    return self;

}

-(void) dealloc {

    [gameConfig release];

    [systemSettings release];

    [savePath release];

    [super dealloc];

}

#pragma mark -

#pragma mark Provider functions

// unpacks team data from the selected team.plist to a sequence of engine commands

-(void) provideTeamData:(NSString *)teamName forHogs:(NSInteger) numberOfPlayingHogs withHealth:(NSInteger) initialHealth ofColor:(NSNumber *)teamColor {

    /*

     addteam <32charsMD5hash> <color> <team name>

     addhh <level> <health> <hedgehog name>

     <level> is 0 for human, 1-5 for bots (5 is the most stupid)

    */

    NSString *teamFile = [[NSString alloc] initWithFormat:@"%@/%@", TEAMS_DIRECTORY(), teamName];

    NSDictionary *teamData = [[NSDictionary alloc] initWithContentsOfFile:teamFile];

    [teamFile release];

    NSString *teamHashColorAndName = [[NSString alloc] initWithFormat:@"eaddteam %@ %@ %@",

                                      [teamData objectForKey:@"hash"], [teamColor stringValue], [teamName stringByDeletingPathExtension]];

    [self sendToEngine: teamHashColorAndName];

    [teamHashColorAndName release];

    NSString *grave = [[NSString alloc] initWithFormat:@"egrave %@", [teamData objectForKey:@"grave"]];

    [self sendToEngine: grave];

    [grave release];

    NSString *fort = [[NSString alloc] initWithFormat:@"efort %@", [teamData objectForKey:@"fort"]];

    [self sendToEngine: fort];

    [fort release];

    NSString *voicepack = [[NSString alloc] initWithFormat:@"evoicepack %@", [teamData objectForKey:@"voicepack"]];

    [self sendToEngine: voicepack];

    [voicepack release];

    NSString *flag = [[NSString alloc] initWithFormat:@"eflag %@", [teamData objectForKey:@"flag"]];

    [self sendToEngine: flag];

    [flag release];

    NSArray *hogs = [teamData objectForKey:@"hedgehogs"];

    for (int i = 0; i < numberOfPlayingHogs; i++) {

        NSDictionary *hog = [hogs objectAtIndex:i];

        NSString *hogLevelHealthAndName = [[NSString alloc] initWithFormat:@"eaddhh %@ %d %@",

                                           [hog objectForKey:@"level"], initialHealth, [hog objectForKey:@"hogname"]];

        [self sendToEngine: hogLevelHealthAndName];

        [hogLevelHealthAndName release];

        NSString *hogHat = [[NSString alloc] initWithFormat:@"ehat %@", [hog objectForKey:@"hat"]];

        [self sendToEngine: hogHat];

        [hogHat release];

    }

    [teamData release];

}

// unpacks ammostore data from the selected ammo.plist to a sequence of engine commands

-(void) provideAmmoData:(NSString *)ammostoreName forPlayingTeams:(NSInteger) numberOfTeams {

    NSString *weaponPath = [[NSString alloc] initWithFormat:@"%@/%@",WEAPONS_DIRECTORY(),ammostoreName];

    NSDictionary *ammoData = [[NSDictionary alloc] initWithContentsOfFile:weaponPath];

    [weaponPath release];

    // if we're loading an older version of ammos fill the engine message with 0s

    int diff = HW_getNumberOfWeapons() - [[ammoData objectForKey:@"ammostore_initialqt"] length];

    NSString *update = @"";

    while ([update length] < diff)

        update = [update stringByAppendingString:@"0"];

    NSString *ammloadt = [[NSString alloc] initWithFormat:@"eammloadt %@%@", [ammoData objectForKey:@"ammostore_initialqt"], update];

    [self sendToEngine: ammloadt];

    [ammloadt release];

    NSString *ammprob = [[NSString alloc] initWithFormat:@"eammprob %@%@", [ammoData objectForKey:@"ammostore_probability"], update];

    [self sendToEngine: ammprob];

    [ammprob release];

    NSString *ammdelay = [[NSString alloc] initWithFormat:@"eammdelay %@%@", [ammoData objectForKey:@"ammostore_delay"], update];

    [self sendToEngine: ammdelay];

    [ammdelay release];

    NSString *ammreinf = [[NSString alloc] initWithFormat:@"eammreinf %@%@", [ammoData objectForKey:@"ammostore_crate"], update];

    [self sendToEngine: ammreinf];

    [ammreinf release];

    // send this for each team so it applies the same ammostore to all teams

    NSString *ammstore = [[NSString alloc] initWithString:@"eammstore"];

    for (int i = 0; i < numberOfTeams; i++)

        [self sendToEngine: ammstore];

    [ammstore release];

    [ammoData release];

}

// unpacks scheme data from the selected scheme.plist to a sequence of engine commands

-(NSInteger) provideScheme:(NSString *)schemeName {

    NSString *schemePath = [[NSString alloc] initWithFormat:@"%@/%@",SCHEMES_DIRECTORY(),schemeName];

    NSDictionary *schemeDictionary = [[NSDictionary alloc] initWithContentsOfFile:schemePath];

    [schemePath release];

    NSArray *basicArray = [schemeDictionary objectForKey:@"basic"];

    NSArray *gamemodArray = [schemeDictionary objectForKey:@"gamemod"];

    int result = 0;

    int mask = 0x00000004;

    // pack the gameflags in a single var and send it

    for (NSNumber *value in gamemodArray) {

        if ([value boolValue] == YES)

            result |= mask;

        mask <<= 1;

    }

    NSString *flags = [[NSString alloc] initWithFormat:@"e$gmflags %d",result];

    [self sendToEngine:flags];

    [flags release];

    /* basic game flags */

    NSString *path = [[NSString alloc] initWithFormat:@"%@/basicFlags_en.plist",IFRONTEND_DIRECTORY()];

    NSArray *mods = [[NSArray alloc] initWithContentsOfFile:path];

    [path release];

    int i = 0;

    // initial health

    result = [[basicArray objectAtIndex:i++] intValue];

    // turn time

    NSInteger tentativeTurntime = [[basicArray objectAtIndex:i++] intValue];

    if (tentativeTurntime >= 100)

        tentativeTurntime = 9999;

    NSString *turnTime = [[NSString alloc] initWithFormat:@"e$turntime %d",tentativeTurntime * 1000];

    [self sendToEngine:turnTime];

    [turnTime release];

    NSString *minesTime = [[NSString alloc] initWithFormat:@"e$turntime %d",[[basicArray objectAtIndex:i++] intValue] * 1000];

    [self sendToEngine:minesTime];

    [minesTime release];

    for (; i < [basicArray count]; i++) {

        NSDictionary *basicDict = [mods objectAtIndex:i];

        NSString *command = [basicDict objectForKey:@"command"];

        NSInteger value = [[basicArray objectAtIndex:i] intValue];

        if ([basicDict objectForKey:@"checkOverMax"] && value >= [[basicDict objectForKey:@"max"] intValue])

            value = 9999;

        NSString *strToSend = [[NSString alloc] initWithFormat:@"%@ %d",command,value];

        [self sendToEngine:strToSend];

        [strToSend release];

    }

    [mods release];

    [schemeDictionary release];

    return result;

}

#pragma mark -

#pragma mark Thread/Network relevant code

// select one of GameSetup method and execute it in a seprate thread

-(void) startThread:(NSString *)selector {

    SEL usage = NSSelectorFromString(selector);

    [NSThread detachNewThreadSelector:usage toTarget:self withObject:nil];

}

-(void) dumpRawData:(const uint8_t*)buffer ofSize:(uint8_t) length {

    // is it performant to reopen the stream every time?

    NSOutputStream *os = [[NSOutputStream alloc] initToFileAtPath:self.savePath append:YES];

    [os open];

    [os write:&length maxLength:1];

    [os write:buffer maxLength:length];

    [os close];

    [os release];

}

// wrapper that computes the length of the message and then sends the command string, saving the command on a file

-(int) sendToEngine:(NSString *)string {

    uint8_t length = [string length];

    [self dumpRawData:(const uint8_t *)[string UTF8String] ofSize:length];

    SDLNet_TCP_Send(csd, &length, 1);

    return SDLNet_TCP_Send(csd, [string UTF8String], length);

}

// wrapper that computes the length of the message and then sends the command string, skipping file writing

-(int) sendToEngineNoSave:(NSString *)string {

    uint8_t length = [string length];

    SDLNet_TCP_Send(csd, &length, 1);

    return SDLNet_TCP_Send(csd, [string UTF8String], length);

}

-(int) sendToServer:(NSString *)command withArgument:(NSString *)argument {

    NSString *message = [[NSString alloc] initWithFormat:@"%@\n%@\n\n",command,argument];

    int result = SDLNet_TCP_Send(esd, [message UTF8String], [message length]);

    [message release];

    return result;

}

-(int) sendToServer:(NSString *)command {

    NSString *message = [[NSString alloc] initWithFormat:@"%@\n\n",command];

    int result = SDLNet_TCP_Send(esd, [message UTF8String], [message length]);

    [message release];

    return result;

}

-(void) serverProtocol {

    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

    IPaddress ip;

    BOOL clientQuit = NO;

    char *buffer = (char *)malloc(sizeof(char)*BUFFER_SIZE);

    int dim = BUFFER_SIZE;

    uint8_t msgSize;

    NSString *arg = nil;

    if (SDLNet_Init() < 0) {

        DLog(@"SDLNet_Init: %s", SDLNet_GetError());

        clientQuit = YES;

    }

    // Resolving the host using NULL make network interface to listen

    if (SDLNet_ResolveHost(&ip, "netserver.hedgewars.org", DEFAULT_NETGAME_PORT) < 0 && !clientQuit) {

        DLog(@"SDLNet_ResolveHost: %s\n", SDLNet_GetError());

        clientQuit = YES;

    }

    // Open a connection with the IP provided (listen on the host's port)

    if (!(esd = SDLNet_TCP_Open(&ip)) && !clientQuit) {

        DLog(@"SDLNet_TCP_Open: %s %\n", SDLNet_GetError(), DEFAULT_NETGAME_PORT);

        clientQuit = YES;

    }

    DLog(@"Found server on port %d", DEFAULT_NETGAME_PORT);

    while (!clientQuit) {

        int index = 0;

        BOOL exitBufferLoop = NO;

        memset(buffer, '\0', dim);

        while (exitBufferLoop != YES) {

            msgSize = SDLNet_TCP_Recv(esd, &buffer[index], 2);

            // exit in case of error

            if (msgSize <= 0) {

                DLog(@"SDLNet_TCP_Recv: %s", SDLNet_GetError());

                clientQuit = YES;

                break;

            }

            // update index position and check for End-Of-Message

            index += msgSize;

            if (strncmp(&buffer[index-2], "\n\n", 2) == 0) {

                exitBufferLoop = YES;

            }

            // if message is too big allocate new space

            if (index >= dim) {

                dim += BUFFER_SIZE;

                buffer = (char *)realloc(buffer, dim);

                if (buffer == NULL) {

                    clientQuit = YES;

                    break;

                }

            }

        }

        NSString *bufferedMessage = [[NSString alloc] initWithBytes:buffer length:index-2 encoding:NSASCIIStringEncoding];

        NSArray *listOfCommands = [bufferedMessage componentsSeparatedByString:@"\n"];

        [bufferedMessage release];

        NSString *command = [listOfCommands objectAtIndex:0];

        DLog(@"size = %d, %@", index-2, listOfCommands);

        if ([command isEqualToString:@"PING"]) {

            if ([listOfCommands count] > 1)

                [self sendToServer:@"PONG" withArgument:[listOfCommands objectAtIndex:1]];

            else

                [self sendToServer:@"PONG"];

            [arg release];

        }

        else if ([command isEqualToString:@"NICK"]) {

            //TODO: what is this for?

        }

        else if ([command isEqualToString:@"PROTO"]) {

            if ([[listOfCommands objectAtIndex:1] intValue] == 34) {

                //TODO: unused

            }

        }

        else if ([command isEqualToString:@"ROOM"]) {

            //TODO: stub

        }

        else if ([command isEqualToString:@"LOBBY:LEFT"]) {

            //TODO: stub

        }

        else if ([command isEqualToString:@"LOBBY:JOINED"]) {

            //TODO: stub

        }

        else if ([command isEqualToString:@"ASKPASSWORD"]) {

            //TODO: store hashed password in settings.plist (nil here will vouluntary trigger an exception)

            [self sendToServer:@"PASSWORD" withArgument:nil];

        }

        else if ([command isEqualToString:@"CONNECTED"]) {

            [self sendToServer:@"NICK" withArgument:@"koda"];

            [self sendToServer:@"PROTO" withArgument:@"34"];

        }

        else if ([command isEqualToString:@"SERVER_MESSAGE"]) {

            DLog(@"%@", [listOfCommands objectAtIndex:1]);

        }

        else if ([command isEqualToString:@"WARNING"]) {

            if ([listOfCommands count] > 1)

                DLog(@"Server warning - %@", [listOfCommands objectAtIndex:1]);

            else

                DLog(@"Server warning - unknown");

        }

        else if ([command isEqualToString:@"ERROR"]) {

            DLog(@"Server error - %@", [listOfCommands objectAtIndex:1]);

        }

        else if ([command isEqualToString:@"BYE"]) {

            //TODO: handle "Reconnected too fast"

            DLog(@"Server disconnected, reason: %@", [listOfCommands objectAtIndex:1]);

            clientQuit = YES;

        }

        else {

            DLog(@"Unknown/Unsupported message received: %@", command);

        }

    }

    DLog(@"Server exited, ending thread");

    free(buffer);

    // Close the client socket

    SDLNet_TCP_Close(esd);

    SDLNet_Quit();

    [pool release];

}

// method that handles net setup with engine and keeps connection alive

-(void) engineProtocol {

    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

    TCPsocket sd;

    IPaddress ip;

    int eProto;

    BOOL clientQuit;

    uint8_t buffer[BUFFER_SIZE];

    uint8_t msgSize;

    clientQuit = NO;

    csd = NULL;

    if (SDLNet_Init() < 0) {

        DLog(@"SDLNet_Init: %s", SDLNet_GetError());

        clientQuit = YES;

    }

    // Resolving the host using NULL make network interface to listen

    if (SDLNet_ResolveHost(&ip, NULL, ipcPort) < 0 && !clientQuit) {

        DLog(@"SDLNet_ResolveHost: %s\n", SDLNet_GetError());

        clientQuit = YES;

    }

    // Open a connection with the IP provided (listen on the host's port)

    if (!(sd = SDLNet_TCP_Open(&ip)) && !clientQuit) {

        DLog(@"SDLNet_TCP_Open: %s %\n", SDLNet_GetError(), ipcPort);

        clientQuit = YES;

    }

    DLog(@"Waiting for a client on port %d", ipcPort);

    while (csd == NULL)

        csd = SDLNet_TCP_Accept(sd);

    SDLNet_TCP_Close(sd);

    while (!clientQuit) {

        msgSize = 0;

        memset(buffer, '\0', BUFFER_SIZE);

        if (SDLNet_TCP_Recv(csd, &msgSize, sizeof(uint8_t)) <= 0)

            break;

        if (SDLNet_TCP_Recv(csd, buffer, msgSize) <=0)

            break;

        switch (buffer[0]) {

            case 'C':

                DLog(@"sending game config...\n%@",self.gameConfig);

                if (isNetGame == YES)

                    [self sendToEngineNoSave:@"TN"];

                else

                    [self sendToEngineNoSave:@"TL"];

                NSString *saveHeader = @"TS";

                [self dumpRawData:(const uint8_t *)[saveHeader UTF8String] ofSize:[saveHeader length]];

                // seed info

                [self sendToEngine:[self.gameConfig objectForKey:@"seed_command"]];

                // dimension of the map

                [self sendToEngine:[self.gameConfig objectForKey:@"templatefilter_command"]];

                [self sendToEngine:[self.gameConfig objectForKey:@"mapgen_command"]];

                [self sendToEngine:[self.gameConfig objectForKey:@"mazesize_command"]];

                // static land (if set)

                NSString *staticMap = [self.gameConfig objectForKey:@"staticmap_command"];

                if ([staticMap length] != 0)

                    [self sendToEngine:staticMap];

                // lua script (if set)

                NSString *script = [self.gameConfig objectForKey:@"mission_command"];

                if ([script length] != 0)

                    [self sendToEngine:script];

                // theme info

                [self sendToEngine:[self.gameConfig objectForKey:@"theme_command"]];

                // scheme (returns initial health)

                NSInteger health = [self provideScheme:[self.gameConfig objectForKey:@"scheme"]];

                // send an ammostore for each team

                NSArray *teamsConfig = [self.gameConfig objectForKey:@"teams_list"];

                [self provideAmmoData:[self.gameConfig objectForKey:@"weapon"] forPlayingTeams:[teamsConfig count]];

                // finally add hogs

                for (NSDictionary *teamData in teamsConfig) {

                    [self provideTeamData:[teamData objectForKey:@"team"]

                                  forHogs:[[teamData objectForKey:@"number"] intValue]

                               withHealth:health

                                  ofColor:[teamData objectForKey:@"color"]];

                }