#include "ipcprotocol.h"

#include "../util/util.h"

#include "../util/logging.h"

#include <stdio.h>

#include <stdbool.h>

#include <string.h>

#include <inttypes.h>

#include <stdlib.h>

int flib_ipc_append_message(flib_vector *vec, const char *fmt, ...) {

	int result = -1;

	if(!vec || !fmt) {

		flib_log_e("null parameter in flib_ipc_appendmessage");

	} else {

		// 1 byte size prefix, 255 bytes max message length, 1 0-byte for vsnprintf

		char msgbuffer[257];

		// Format the message, leaving one byte at the start for the length

		va_list argp;

		va_start(argp, fmt);

		int msgSize = vsnprintf(msgbuffer+1, 256, fmt, argp);

		va_end(argp);

		if(msgSize > 255) {

			flib_log_e("Message too long (%u bytes) in flib_ipc_appendmessage", (unsigned)msgSize);

		} else if(msgSize<0) {

			flib_log_e("printf error in flib_ipc_appendmessage");

		} else {

			// Add the length prefix

			((uint8_t*)msgbuffer)[0] = msgSize;

			// Append it to the vector

			if(flib_vector_append(vec, msgbuffer, msgSize+1) == msgSize+1) {

				result = 0;

			}

		}

	}

	return result;

}

int flib_ipc_append_mapconf(flib_vector *vec, const flib_map *map, bool mappreview) {

	int result = -1;

	flib_vector *tempvector = flib_vector_create();

	if(!vec || !map) {

		flib_log_e("null parameter in flib_ipc_append_mapconf");

	} else if(tempvector) {

		bool error = false;

		if(map->mapgen == MAPGEN_NAMED) {

			error |= flib_ipc_append_message(tempvector, "emap %s", map->name);

		}

		if(map->theme && !mappreview) {

			error |= flib_ipc_append_message(tempvector, "etheme %s", map->theme);

		}

		error |= flib_ipc_append_message(tempvector, "e$template_filter %i", map->templateFilter);

		error |= flib_ipc_append_message(tempvector, "e$mapgen %i", map->mapgen);

		if(map->mapgen == MAPGEN_MAZE) {

			error |= flib_ipc_append_message(tempvector, "e$maze_size %i", map->mazeSize);

		}

		if(map->mapgen == MAPGEN_DRAWN) {

			/*

			 * We have to split the drawn map data into several edraw messages here because

			 * it can be longer than the maximum message size.

			 */

			const char *edraw = "edraw ";

			int edrawlen = strlen(edraw);

			for(int offset=0; offset<map->drawDataSize; offset+=200) {

				int bytesRemaining = map->drawDataSize-offset;

				int fragmentsize = bytesRemaining < 200 ? bytesRemaining : 200;

				uint8_t messagesize = edrawlen + fragmentsize;

				error |= (flib_vector_append(tempvector, &messagesize, 1) != 1);

				error |= (flib_vector_append(tempvector, edraw, edrawlen) != edrawlen);

				error |= (flib_vector_append(tempvector, map->drawData+offset, fragmentsize) != fragmentsize);

			}

		}

		if(!error) {

			// Message created, now we can copy everything.

			flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);

			if(flib_vector_append(vec, constbuf.data, constbuf.size) == constbuf.size) {

				result = 0;

			}

		}

	}

	flib_vector_destroy(tempvector);

	return result;

}

int flib_ipc_append_seed(flib_vector *vec, const char *seed) {

	if(!vec || !seed) {

		flib_log_e("null parameter in flib_ipc_append_seed");

		return -1;

	} else {

		return flib_ipc_append_message(vec, "eseed %s", seed);

	}

}

int flib_ipc_append_gamescheme(flib_vector *vec, const flib_cfg *scheme) {

	int result = -1;

	flib_vector *tempvector = flib_vector_create();

	if(!vec || !scheme) {

		flib_log_e("null parameter in flib_ipc_append_gamescheme");

	} else if(tempvector) {

		const flib_cfg_meta *meta = scheme->meta;

		bool error = false;

		uint32_t gamemods = 0;

		for(int i=0; i<meta->modCount; i++) {

			if(scheme->mods[i]) {

				gamemods |= (1<<meta->mods[i].bitmaskIndex);

			}

		}

		error |= flib_ipc_append_message(tempvector, "e$gmflags %"PRIu32, gamemods);

		for(int i=0; i<meta->settingCount; i++) {

			int value = scheme->settings[i];

			if(meta->settings[i].maxMeansInfinity) {

				value = value>=meta->settings[i].max ? 9999 : value;

			}

			if(meta->settings[i].times1000) {

				value *= 1000;

			}

			error |= flib_ipc_append_message(tempvector, "%s %i", meta->settings[i].engineCommand, value);

		}

		if(!error) {

			// Message created, now we can copy everything.

			flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);

			if(flib_vector_append(vec, constbuf.data, constbuf.size) == constbuf.size) {

				result = 0;

			}

		}

	}

	flib_vector_destroy(tempvector);

	return result;

}

static int appendWeaponSet(flib_vector *vec, flib_weaponset *set) {

	return flib_ipc_append_message(vec, "eammloadt %s", set->loadout)

		|| flib_ipc_append_message(vec, "eammprob %s", set->crateprob)

		|| flib_ipc_append_message(vec, "eammdelay %s", set->delay)

		|| flib_ipc_append_message(vec, "eammreinf %s", set->crateammo);

}

int flib_ipc_append_addteam(flib_vector *vec, const flib_team *team, bool perHogAmmo, bool noAmmoStore) {

	int result = -1;

	flib_vector *tempvector = flib_vector_create();

	if(!vec || !team) {

		flib_log_e("invalid parameter in flib_ipc_append_addteam");

	} else if(tempvector) {

		bool error = false;

		if(!perHogAmmo && !noAmmoStore) {

			error |= appendWeaponSet(tempvector, team->hogs[0].weaponset);

			error |= flib_ipc_append_message(tempvector, "eammstore");

		}

		char *hash = team->hash ? team->hash : "00000000000000000000000000000000";

		error |= flib_ipc_append_message(tempvector, "eaddteam %s %"PRIu32" %s", hash, team->color, team->name);

		if(team->remoteDriven) {

			error |= flib_ipc_append_message(tempvector, "erdriven");

		}

		error |= flib_ipc_append_message(tempvector, "egrave %s", team->grave);

		error |= flib_ipc_append_message(tempvector, "efort %s", team->fort);

		error |= flib_ipc_append_message(tempvector, "evoicepack %s", team->voicepack);

		error |= flib_ipc_append_message(tempvector, "eflag %s", team->flag);

		for(int i=0; i<team->bindingCount; i++) {

			error |= flib_ipc_append_message(tempvector, "ebind %s %s", team->bindings[i].binding, team->bindings[i].action);

		}

		for(int i=0; i<team->hogsInGame; i++) {

			if(perHogAmmo && !noAmmoStore) {

				error |= appendWeaponSet(tempvector, team->hogs[i].weaponset);

			}

			error |= flib_ipc_append_message(tempvector, "eaddhh %i %i %s", team->hogs[i].difficulty, team->hogs[i].initialHealth, team->hogs[i].name);

			error |= flib_ipc_append_message(tempvector, "ehat %s", team->hogs[i].hat);

		}

		if(!error) {

			// Message created, now we can copy everything.

			flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);

			if(flib_vector_append(vec, constbuf.data, constbuf.size) == constbuf.size) {

				result = 0;

			}

		}

	}

	flib_vector_destroy(tempvector);

	return result;

}

static bool getGameMod(const flib_cfg *conf, int maskbit) {

	for(int i=0; i<conf->meta->modCount; i++) {

		if(conf->meta->mods[i].bitmaskIndex == maskbit) {

			return conf->mods[i];

		}

	}

	flib_log_e("Unable to find game mod with mask bit %i", maskbit);

	return false;

}

int flib_ipc_append_fullconfig(flib_vector *vec, const flib_gamesetup *setup, bool netgame) {

	int result = -1;

	flib_vector *tempvector = flib_vector_create();

	if(!vec || !setup) {

		flib_log_e("null parameter in flib_ipc_append_fullconfig");

	} else if(tempvector) {

		bool error = false;

		bool perHogAmmo = false;

		bool sharedAmmo = false;

		error |= flib_ipc_append_message(vec, netgame ? "TN" : "TL");

		error |= flib_ipc_append_seed(vec, setup->seed);

		if(setup->map) {

			error |= flib_ipc_append_mapconf(tempvector, setup->map, false);

		}

		if(setup->script) {

			error |= flib_ipc_append_message(tempvector, "escript %s", setup->script);

		}

		if(setup->gamescheme) {

			error |= flib_ipc_append_gamescheme(tempvector, setup->gamescheme);

			sharedAmmo = getGameMod(setup->gamescheme, GAMEMOD_SHAREDAMMO_MASKBIT);

			// Shared ammo has priority over per-hog ammo

			perHogAmmo = !sharedAmmo && getGameMod(setup->gamescheme, GAMEMOD_PERHOGAMMO_MASKBIT);

		}

		if(setup->teams && setup->teamcount>0) {

			uint32_t *clanColors = flib_calloc(setup->teamcount, sizeof(uint32_t));

			if(!clanColors) {

				error = true;

			} else {

				int clanCount = 0;

				for(int i=0; i<setup->teamcount; i++) {

					flib_team *team = setup->teams[i];

					bool newClan = false;

					// Find the clan index of this team (clans are identified by color).

					// The upper 8 bits (alpha) are ignored in the engine as well.

					uint32_t color = team->color&UINT32_C(0x00ffffff);

					int clan = 0;

					while(clan<clanCount && clanColors[clan] != color) {

						clan++;

					}

					if(clan==clanCount) {

						newClan = true;

						clanCount++;

						clanColors[clan] = color;

					}

					// If shared ammo is active, only add an ammo store for the first team in each clan.

					bool noAmmoStore = sharedAmmo&&!newClan;

					error |= flib_ipc_append_addteam(tempvector, setup->teams[i], perHogAmmo, noAmmoStore);

				}

			}

			free(clanColors);

		}

		error |= flib_ipc_append_message(tempvector, "!");

		if(!error) {

			// Message created, now we can copy everything.

			flib_constbuffer constbuf = flib_vector_as_constbuffer(tempvector);

			if(flib_vector_append(vec, constbuf.data, constbuf.size) == constbuf.size) {

				result = 0;

			}

		}

	}

	return result;

}