#include "weapon.h"

#include "../ini/inihelper.h"
#include "../iniparser/iniparser.h"
#include "../logging.h"
#include "../util.h"

#include <stdlib.h>
#include <ctype.h>

int set_field(char field[WEAPONS_COUNT+1], const char *line, bool no9) {
	// Validate the new string
	for(int i=0; i<WEAPONS_COUNT && line[i]; i++) {
		if(line[i] < '0' || line[i] > '9' || (no9 && line[i] == '9')) {
			flib_log_e("Invalid character in weapon config string \"%.*s\", position %i", WEAPONS_COUNT, line, i);
			return -1;
		}
	}

	bool lineEnded = false;
	for(int i=0; i<WEAPONS_COUNT; i++) {
		if(!lineEnded && !line[i]) {
			flib_log_w("Incomplete weapon config line \"%s\", filling with zeroes.", line);
			lineEnded = true;
		}
		if(lineEnded) {
			field[i] = '0';
		} else {
			field[i] = line[i];
		}
	}
	field[WEAPONS_COUNT] = 0;
	return 0;
}

static flib_weaponset *flib_weaponset_create_str(const char *name, const char *loadoutStr, const char *crateProbStr, const char *crateAmmoStr, const char *delayStr) {
	flib_weaponset *result = NULL;
	if(!name || !loadoutStr || !crateProbStr || !crateAmmoStr || !delayStr) {
		flib_log_e("null parameter in flib_weaponset_create_str");
	} else {
		flib_weaponset *newSet = calloc(1, sizeof(flib_weaponset));
		char *nameCopy = flib_strdupnull(name);
		if(newSet && nameCopy) {
			newSet->name = nameCopy;
			nameCopy = NULL;
			bool error = false;
			error |= set_field(newSet->loadout, loadoutStr, false);
			error |= set_field(newSet->crateprob, crateProbStr, false);
			error |= set_field(newSet->crateammo, crateAmmoStr, false);
			error |= set_field(newSet->delay, delayStr, false);
			if(!error) {
				result = newSet;
				newSet = NULL;
			}
		}
		free(nameCopy);
		flib_weaponset_destroy(newSet);
	}
	return result;
}

void flib_weaponset_destroy(flib_weaponset *cfg) {
	if(cfg) {
		free(cfg->name);
		free(cfg);
	}
}

flib_weaponset *flib_weaponset_create(const char *name) {
	return flib_weaponset_create_str(name, AMMOLINE_DEFAULT_QT, AMMOLINE_DEFAULT_PROB, AMMOLINE_DEFAULT_CRATE, AMMOLINE_DEFAULT_DELAY);
}

flib_weaponset *flib_weaponset_from_ini(const char *filename) {
	flib_weaponset *result = NULL;
	if(!filename) {
		flib_log_e("null parameter in flib_weaponset_from_ini");
	} else {
		dictionary *settingfile = iniparser_load(filename);
		if(!settingfile) {
			flib_log_e("Error loading weapon scheme file %s", filename);
		} else {
			bool error = false;
			char *name = inihelper_getstring(settingfile, &error, "weaponset", "name");
			char *loadout = inihelper_getstring(settingfile, &error, "weaponset", "loadout");
			char *crateprob = inihelper_getstring(settingfile, &error, "weaponset", "crateprob");
			char *crateammo = inihelper_getstring(settingfile, &error, "weaponset", "crateammo");
			char *delay = inihelper_getstring(settingfile, &error, "weaponset", "delay");
			if(error) {
				flib_log_e("Missing key in weapon scheme file %s", filename);
			} else {
				result = flib_weaponset_create_str(name, loadout, crateprob, crateammo, delay);
			}
		}
		iniparser_freedict(settingfile);
	}
	return result;
}

int flib_weaponset_to_ini(const char *filename, const flib_weaponset *set) {
	int result = -1;
	if(!filename || !set) {
		flib_log_e("null parameter in flib_weaponset_to_ini");
	} else {
		dictionary *dict = dictionary_new(0);
		if(dict) {
			bool error = false;
			// Add the sections
			error |= iniparser_set(dict, "weaponset", NULL);

			// Add the values
			error |= inihelper_setstr(dict, "weaponset", "name", set->name);
			error |= inihelper_setstr(dict, "weaponset", "loadout", set->loadout);
			error |= inihelper_setstr(dict, "weaponset", "crateprob", set->crateprob);
			error |= inihelper_setstr(dict, "weaponset", "crateammo", set->crateammo);
			error |= inihelper_setstr(dict, "weaponset", "delay", set->delay);
			if(!error) {
				FILE *inifile = fopen(filename, "wb");
				if(inifile) {
					iniparser_dump_ini(dict, inifile);
					fclose(inifile);
					result = 0;
				}
			}
			dictionary_del(dict);
		}
	}
	return result;
}