#include "files.h"

#include "net.h"

#include <string.h>

#include <stdlib.h>

#define MP_CONFIG_DIR "mp_config"

#define MP_CONFIG_WILDCARD "*.cfg"

#define SKIRMISH_CONFIG_WILDCARD "*.skirmish_cfg"

#define IP_ADDRESS_DIR "ip_address/"

#define IP_ADDRESS_WILDCARD "*.IP Address"

extern char MP_SessionName[];

extern char MP_Config_Description[];

extern int CloseGameDirectory(void *dir);

extern int CreateGameDirectory(const char *dirname);

extern const struct Map MultiplayerMaps[];

extern const struct Map CoopMaps[];

static int custom_multiplayer_levels = 0;

static int custom_coop_skirmish_levels = 0;

int NumMultiplayerLevels = 0;

int NumCoopLevels = MAX_NO_OF_COOPERATIVE_EPISODES;

const char** MultiplayerLevelNames = NULL;

const char** CoopLevelNames = NULL;

char *LastDescriptionFile = NULL;

char* LastDescriptionText = NULL;

void SetDefaultNetStruct()

{

    int i;

    netGameData.myCharacterSubType = NGSCT_General;

    netGameData.myCharacterType = NGCT_Marine;

    netGameData.myNextCharacterType= NGCT_Marine;

    netGameData.gameType = NGT_Coop;

    netGameData.myGameState = NGS_StartUp;

    netGameData.LMS_AlienIndex = -1;

    netGameData.LMS_RestartTimer = 0;

    netGameData.myLastScream = -1;

    netGameData.GameTimeElapsed = 0;

    netGameData.myStartFlag = 0;

    netGameData.levelNumber = 0;

    netGameData.sendDecals = 1;

    netGameData.sendTimer = 0;

    netGameData.sendFrequency = 0;

    netGameData.pointsForRespawn = 0;

    netGameData.lastPointsBasedRespawn = 0;

    netGameData.gameDescriptionTimeDelay = 0;

    netGameData.needGameDescription = 0;

    netGameData.stateCheckTimeDelay = 0;

    netGameData.myStrategyCheckSum = 0;

    netGameData.joiningGameStatus = 0;

    netGameData.teamScores[0] = 0;

    netGameData.teamScores[1] = 0;

    netGameData.teamScores[2] = 0;

    netGameData.numDeaths[0] = 0;

    netGameData.numDeaths[1] = 0;

    netGameData.numDeaths[2] = 0;

    netGameData.landingNoise = 0;

    for(i=0; i < NET_MAXPLAYERS; i++)

    {

        int j;

        netGameData.playerData[i].playerId = 0;        

        for(j=0; j < NET_PLAYERNAMELENGTH; j++)

            netGameData.playerData[i].name[j] = '\0';

        netGameData.playerData[i].characterType = NGCT_Marine;

        netGameData.playerData[i].characterSubType = NGSCT_General;

        for(j=0; j < NET_MAXPLAYERS; j++)

            netGameData.playerData[i].playerFrags[j] = 0;

        netGameData.playerData[i].aliensKilled[0] = 0;

        netGameData.playerData[i].aliensKilled[1] = 0;

        netGameData.playerData[i].aliensKilled[2] = 0;

        netGameData.playerData[i].deathsFromAI = 0;

        netGameData.playerData[i].playerScore = 0;

        netGameData.playerData[i].playerScoreAgainst = 0;

        netGameData.playerData[i].lastKnownPosition.vx = 0;

        netGameData.playerData[i].lastKnownPosition.vy = 0;

        netGameData.playerData[i].lastKnownPosition.vz = 0;

        netGameData.playerData[i].timer = 0;

        netGameData.playerData[i].startFlag = 0;

        netGameData.playerData[i].playerAlive = 1;

        netGameData.playerData[i].playerHasLives = 1;

    }

}

void SetDefaultMultiplayerConfig()

{

    netGameData.timeForRespawn = 40;

    netGameData.invulnerableTime = 5;

    netGameData.useSharedLives = 0;

    netGameData.maxLives = 0;

    netGameData.scoreLimit = 1000000;    

    netGameData.timeLimit = 255;

    netGameData.invulnerableTime = 5;

    netGameData.characterKillValues[0] = 100;

    netGameData.characterKillValues[1] = 150;

    netGameData.characterKillValues[2] = 75;

    netGameData.baseKillValue = 100;

    netGameData.useDynamicScoring = 1;

    netGameData.useCharacterKillValues = 1;

    netGameData.aiKillValues[0] = 75;

    netGameData.aiKillValues[1] = 100;

    netGameData.aiKillValues[2] = 150;

    netGameData.maxPredator = 8;

    netGameData.maxAlien = 8;

    netGameData.maxMarine = 8;

    netGameData.maxMarineGeneral = 8;

    netGameData.maxMarinePulseRifle = 8;

    netGameData.maxMarineSmartgun = 8;

    netGameData.maxMarineFlamer = 8;

    netGameData.maxMarineSadar = 8;

    netGameData.maxMarineGrenade = 8;

    netGameData.maxMarineMinigun = 8;

    netGameData.maxMarineSmartDisc = 8;

    netGameData.maxMarinePistols = 8;

    netGameData.allowSmartgun = 1;

    netGameData.allowFlamer = 1;

    netGameData.allowSadar = 1;

    netGameData.allowGrenadeLauncher = 1;

    netGameData.allowMinigun = 1;

    netGameData.allowDisc = 1;

    netGameData.allowPistol = 1;

    netGameData.allowPlasmaCaster = 1;

    netGameData.allowSpeargun = 1;

    netGameData.allowMedicomp = 1;

    netGameData.allowSmartDisc = 1;

    netGameData.allowPistols = 1;

    netGameData.preDestroyLights = 0;

    netGameData.disableFriendlyFire = 0;

    netGameData.fallingDamage = 1;

    netGameData.pistolInfiniteAmmo = 1;

    netGameData.specialistPistols = 1;

    netGameData.maxLives = 0;

    netGameData.useSharedLives = 0;

    netGameData.pointsForRespawn = 0;

    netGameData.timeForRespawn = 40;

    netGameData.gameSpeed = NETGAMESPEED_100PERCENT;

}

void LoadMultiplayerConfiguration(const char* name)

{

    char filename[200];

    SetDefaultMultiplayerConfig(); //set defaults first , in case there are entries not set by this file

    sprintf(filename,"%s%c%s.%scfg", MP_CONFIG_DIR, DIR_SEPARATOR, name, ((Skirmish == AvP.PlayMode) ? "skirmish_" : "" ));

    FILE* file = OpenGameFile(filename, 1, FILETYPE_CONFIG);

    if(NULL != file)

    {

        int i, length;

        fread(&netGameData, sizeof(netGameData), 1, file);

        fread(&MP_SessionName[0], sizeof(char), 13, file);

        length = 0;

        fread(&length, sizeof(int), 1, file);

        fread(MP_Config_Description, 1, length, file);

        length = 0;

        fread(&length, sizeof(int), 1, file);

        fread(AvP.LevelName, 1, (length < 40 ? length : 39), file);

        netGameData.levelNumber = -1;

        if(NGT_Coop == netGameData.gameType)

        {

            for(i=0; i < MAX_NO_OF_COOPERATIVE_EPISODES; i++)

            {

                if(!strcmp(AvP.LevelName, CoopMaps[i].filename))

                {

                    netGameData.levelNumber = i;

                    break;

                }

            }

            for(; (-1 == netGameData.levelNumber) && (i < NumCoopLevels); i++)

            {

                if(!strcmp(AvP.LevelName + 7, CoopLevelNames[i])) // +7 = remove 'custom/'

                {

                    netGameData.levelNumber = i;

                }

            }

        }

        else

        {

            for(i=0; i < MAX_NO_OF_MULTIPLAYER_EPISODES; i++)

            {

                if(!strcmp(AvP.LevelName, MultiplayerMaps[i].filename))

                {

                    netGameData.levelNumber = i;

                    break;

                }

            }

            for(; (-1 == netGameData.levelNumber) && (i < NumMultiplayerLevels); i++)

            {

                if(!strcmp(AvP.LevelName + 7, MultiplayerLevelNames[i]))

                {

                    netGameData.levelNumber = i;

                    break;

                }

            }

        }

        if(netGameData.levelNumber < 0)

            netGameData.levelNumber = 0; //we don't have the level

        AvP.level = netGameData.levelNumber;

        fclose(file);

    }

}

void SaveMultiplayerConfiguration(const char* name)

{

    char filename[200];

    sprintf(filename,"%s%c%s.%scfg", MP_CONFIG_DIR, DIR_SEPARATOR, name, ((Skirmish == AvP.PlayMode) ? "skirmish_" : "" ));

    FILE* file = OpenGameFile(filename, 0, FILETYPE_CONFIG);

    if (file == NULL)

    {

        CreateGameDirectory(MP_CONFIG_DIR); /* try again */

        file = OpenGameFile(filename, 0, FILETYPE_CONFIG);

        if (file == NULL)

            return;

    }

    fwrite(&netGameData, sizeof(netGameData), 1, file);

    fwrite(&MP_SessionName[0], sizeof(char), 13, file);

    //write the description of the config

    int length = strlen(MP_Config_Description)+1;

    fwrite(&length, sizeof(int), 1, file);

    fwrite(MP_Config_Description, 1, length, file);

    length = strlen(AvP.LevelName) + 1;

    fwrite(&length, sizeof(int), 1, file);

    fwrite(AvP.LevelName, 1, length, file);

    fclose(file);

    //clear the last description stuff

    free(LastDescriptionFile);

    LastDescriptionFile = NULL;

    free(LastDescriptionText);

    LastDescriptionText = NULL;

}

void SaveIPAddress(const char* name, const char* address)

{

    if(!name || !address || !strlen(name) || !strlen(address))

        return;

    char filename[200];

    sprintf(filename,"%s/%s.IP Address",IP_ADDRESS_DIR,name);

    FILE* file = OpenGameFile(filename, 0, FILETYPE_CONFIG);

    if (file == NULL)

    {

        CreateGameDirectory(IP_ADDRESS_DIR); /* try again */

        file = OpenGameFile(filename, 0, FILETYPE_CONFIG);

        if (file == NULL)

             return;

    }

    fwrite(address,1,strlen(address)+1,file);

    fclose(file);

}

void LoadIPAddress(const char* name)

{

    extern char IPAddressString[]; 

    if(!name)

        return;

    FILE* file;

    char filename[200];

    sprintf(filename,"%s/%s.IP Address",IP_ADDRESS_DIR,name);

    file = OpenGameFile(filename, 1, FILETYPE_CONFIG);

    if(!file)

        return;

    fread(IPAddressString,1,16,file);

    IPAddressString[15]=0;

    fclose(file);

}

void cleanup_strings()

{

    int i;

    for(i=0; i < custom_coop_skirmish_levels; i++)

        free((void*)CoopLevelNames[MAX_NO_OF_COOPERATIVE_EPISODES + i]);

    for(i=0; i < custom_multiplayer_levels; i++)

        free((void*)MultiplayerLevelNames[MAX_NO_OF_MULTIPLAYER_EPISODES + i]);

    free(MultiplayerLevelNames);

    free(CoopLevelNames);

}

extern void *OpenGameDirectory(const char *dirname, const char *pattern, int type);

extern GameDirectoryFile *ScanGameDirectory(void *dir);

void BuildMultiplayerLevelNameArray()

{

    //first do a search for custom level rifs

    // allow a wildcard search

    void *gd;

    char ** CustomCoopSkirmishLevelNameList = malloc(sizeof(char*)* 14); // 14 = all avp gold map packs installed

    char ** CustomMultiPlayerLevelNameList = malloc(sizeof(char*)* 14); // 14 = all avp gold map packs installed

    /* TODO: Have to use PERM until the load_rif code can handle CONFIG */

    if ((gd = OpenGameDirectory("avp_rifs/custom/", "*.rif", FILETYPE_GAMEDATA)) != NULL)

    {

        GameDirectoryFile *gdf;

        while ((gdf = ScanGameDirectory(gd)) != NULL)

        {

            if ((gdf->attr & FILEATTR_DIRECTORY) || !(gdf->attr & FILEATTR_READABLE))

                continue;

            char * level_rif = strstr(gdf->filename, ".rif");

            if(NULL != level_rif)

            {

                int custom_filename_len = strlen(gdf->filename);

                char *custom_map = malloc(custom_filename_len + 1);

                if(NULL == custom_map)

                    break;

                strncpy(custom_map, gdf->filename, custom_filename_len + 1);

                if(NULL != strstr(gdf->filename, "(c)"))

                {

                    if(custom_coop_skirmish_levels > 13)

                    {

                        char **ptr = (char**)realloc(CustomCoopSkirmishLevelNameList,(sizeof(char**) * (14 + custom_coop_skirmish_levels + 1)));

                        if(NULL != ptr)

                        {

                            CustomCoopSkirmishLevelNameList = ptr;

                        }

                        else

                        break;

                    }

                    CustomCoopSkirmishLevelNameList[custom_coop_skirmish_levels++] = custom_map;

                }

                else

                {

                    if(custom_multiplayer_levels > 13)

                    {

                        char **ptr= (char**)realloc(CustomMultiPlayerLevelNameList, (sizeof(char**) * (14 + custom_multiplayer_levels + 1)));

                        if(NULL != ptr)

                        {

                            CustomMultiPlayerLevelNameList = ptr;

                        }

                        else

                        break;

                    }

                    CustomMultiPlayerLevelNameList[custom_multiplayer_levels++] = custom_map;

                }

            }

        }

        CloseGameDirectory(gd);

    }

    else

    {

        CreateGameDirectory("custom"); // maybe it didn't exist..

    }

    NumMultiplayerLevels = MAX_NO_OF_MULTIPLAYER_EPISODES + custom_multiplayer_levels;

    NumCoopLevels += custom_coop_skirmish_levels;

    MultiplayerLevelNames = malloc(sizeof(char*) * NumMultiplayerLevels);

    int i=0;

    //first the standard multiplayer levels

    for(i=0; i < MAX_NO_OF_MULTIPLAYER_EPISODES; i++)

        MultiplayerLevelNames[i] = MultiplayerMaps[i].pretty_name;

    CoopLevelNames = malloc(sizeof(char*) * NumCoopLevels);

    //and the standard coop levels

    for(i=0; i < MAX_NO_OF_COOPERATIVE_EPISODES; i++)

        CoopLevelNames[i] = CoopMaps[i].pretty_name;

    for(i=0; i < custom_coop_skirmish_levels; i++)

        CoopLevelNames[MAX_NO_OF_COOPERATIVE_EPISODES + i] = CustomCoopSkirmishLevelNameList[i];

    for(i=0; i < custom_multiplayer_levels; i++)

        MultiplayerLevelNames[MAX_NO_OF_MULTIPLAYER_EPISODES + i] = CustomMultiPlayerLevelNameList[i];

    free(CustomCoopSkirmishLevelNameList);

    free(CustomMultiPlayerLevelNameList);

}