/*
 * Hedgewars, a free turn based strategy game
 * Copyright (C) 2012 Simeon Maxein <smaxein@googlemail.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; either version 2
 * of the License, or (at your option) any later version.
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include "netprotocol.h"

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

#include "../base64/base64.h"

#include <zlib.h>

#include <string.h>
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>

static int fillTeamFromMsg(flib_team *team, char **parts) {
	team->name = flib_strdupnull(parts[0]);
	team->grave = flib_strdupnull(parts[1]);
	team->fort = flib_strdupnull(parts[2]);
	team->voicepack = flib_strdupnull(parts[3]);
	team->flag = flib_strdupnull(parts[4]);
	team->ownerName = flib_strdupnull(parts[5]);
	if(!team->name || !team->grave || !team->fort || !team->voicepack || !team->flag || !team->ownerName) {
		return -1;
	}

	errno = 0;
	long difficulty = strtol(parts[6], NULL, 10);
	if(errno) {
		return -1;
	}

	for(int i=0; i<HEDGEHOGS_PER_TEAM; i++) {
		flib_hog *hog = &team->hogs[i];
		hog->difficulty = difficulty;
		hog->name = flib_strdupnull(parts[7+2*i]);
		hog->hat = flib_strdupnull(parts[8+2*i]);
		if(!hog->name || !hog->hat) {
			return -1;
		}
	}

	// Set some default assumptions as well
	team->colorIndex = DEFAULT_COLOR_INDEX;
	team->hogsInGame = DEFAULT_HEDGEHOG_COUNT;
	team->remoteDriven = true;
	return 0;
}

flib_team *flib_team_from_netmsg(char **parts) {
	flib_team *result = NULL;
	flib_team *tmpTeam = flib_calloc(1, sizeof(flib_team));
	if(tmpTeam) {
		if(!fillTeamFromMsg(tmpTeam, parts)) {
			result = tmpTeam;
			tmpTeam = NULL;
		} else {
			flib_log_e("Error parsing team from net.");
		}
	}
	flib_team_destroy(tmpTeam);
	return result;
}

flib_scheme *flib_scheme_from_netmsg(char **parts) {
	flib_scheme *result = flib_scheme_create(parts[0]);
	if(result) {
		for(int i=0; i<flib_meta.modCount; i++) {
			result->mods[i] = !strcmp(parts[i+1], "true");
		}
		for(int i=0; i<flib_meta.settingCount; i++) {
			result->settings[i] = atoi(parts[i+flib_meta.modCount+1]);
		}
	}
	return result;
}

flib_map *flib_map_from_netmsg(char **parts) {
	flib_map *result = flib_map_create_named(parts[3], parts[0]);
	if(result) {
		result->mapgen = atoi(parts[1]);
		result->mazeSize = atoi(parts[2]);
		result->templateFilter = atoi(parts[4]);
	}
	return result;
}

int flib_drawnmapdata_from_netmsg(char *netmsg, uint8_t** outbuf, size_t *outlen) {
	int result = -1;

	// First step: base64 decoding
	char *base64decout = NULL;
	size_t base64declen;
	bool ok = base64_decode_alloc(netmsg, strlen(netmsg), &base64decout, &base64declen);
	if(ok && base64declen>3) {
		// Second step: unzip with the QCompress header. That header is just a big-endian
		// uint32 indicating the length of the uncompressed data.
		uint8_t *ubyteBuf = (uint8_t*)base64decout;
		uint32_t unzipLen =
				(((uint32_t)ubyteBuf[0])<<24)
				+ (((uint32_t)ubyteBuf[1])<<16)
				+ (((uint32_t)ubyteBuf[2])<<8)
				+ ubyteBuf[3];
		if(unzipLen==0) {
			*outbuf = NULL;
			*outlen = 0;
			result = 0;
		} else {
			uint8_t *out = flib_malloc(unzipLen);
			if(out) {
				uLongf actualUnzipLen = unzipLen;
				int resultcode = uncompress(out, &actualUnzipLen, (Bytef*)(base64decout+4), base64declen-4);
				if(resultcode == Z_OK) {
					*outbuf = out;
					*outlen = actualUnzipLen;
					out = NULL;
					result = 0;
				} else {
					flib_log_e("Uncompressing drawn map failed. Code: %i", resultcode);
				}
			}
			free(out);
		}
	} else {
		flib_log_e("base64 decoding of drawn map failed.");
	}
	free(base64decout);
	return result;
}

flib_room *flib_room_from_netmsg(char **params) {
	flib_room *result = NULL;
	flib_room *tmpRoom = flib_calloc(1, sizeof(flib_room));
	if(tmpRoom) {
		tmpRoom->inProgress = !strcmp(params[0], "True");
		tmpRoom->name = flib_strdupnull(params[1]);
		tmpRoom->playerCount = atoi(params[2]);
		tmpRoom->teamCount = atoi(params[3]);
		tmpRoom->owner = flib_strdupnull(params[4]);
		tmpRoom->map = flib_strdupnull(params[5]);
		tmpRoom->scheme = flib_strdupnull(params[6]);
		tmpRoom->weapons = flib_strdupnull(params[7]);
		if(tmpRoom->name && tmpRoom->owner && tmpRoom->map && tmpRoom->scheme && tmpRoom->weapons) {
			result = tmpRoom;
			tmpRoom = NULL;
		}
	}
	flib_room_destroy(tmpRoom);
	return result;
}

int fillRoomArray(flib_room **array, char **params, int count) {
	for(int i=0; i<count; i++) {
		array[i] = flib_room_from_netmsg(params + 8*i);
		if(!array[i]) {
			return -1;
		}
	}
	return 0;
}

flib_room **flib_room_array_from_netmsg(char **params, int count) {
	flib_room **result = flib_calloc(count, sizeof(flib_room*));
	if(result) {
		if(fillRoomArray(result, params, count)) {
			for(int i=0; i<count; i++) {
				flib_room_destroy(result[i]);
			}
			free(result);
			result = NULL;
		}
	}
	return result;
}