#include "schemelist.h"

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

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

static void flib_schemelist_destroy(flib_schemelist *list) {
	if(list) {
		for(int i=0; i<list->schemeCount; i++) {
			flib_cfg_release(list->schemes[i]);
		}
		free(list);
	}
}

static char *makePrefixedName(int schemeIndex, const char *settingName) {
	return flib_asprintf("%i\\%s", schemeIndex, settingName);
}

static int readSettingsFromIni(flib_ini *ini, flib_cfg *scheme, int index) {
	flib_cfg_meta *meta = scheme->meta;
	bool error = false;
	for(int i=0; i<meta->settingCount && !error; i++) {
		char *key = makePrefixedName(index, meta->settings[i].name);
		if(!key) {
			error = true;
		} else if(flib_ini_get_int_opt(ini, &scheme->settings[i], key, meta->settings[i].def)) {
			flib_log_e("Error reading setting %s in schemes file.", key);
			error = true;
		}
		free(key);
	}
	return error;
}

static int readModsFromIni(flib_ini *ini, flib_cfg *scheme, int index) {
	flib_cfg_meta *meta = scheme->meta;
	bool error = false;
	for(int i=0; i<meta->modCount && !error; i++) {
		char *key = makePrefixedName(index, meta->mods[i].name);
		if(!key) {
			error = true;
		} else if(flib_ini_get_bool_opt(ini, &scheme->mods[i], key, false)) {
			flib_log_e("Error reading mod %s in schemes file.", key);
			error = true;
		}
		free(key);
	}
	return error;
}

static flib_cfg *readSchemeFromIni(flib_cfg_meta *meta, flib_ini *ini, int index) {
	flib_cfg *result = NULL;
	char *schemeNameKey = makePrefixedName(index+1, "name");
	if(schemeNameKey) {
		char *schemeName = NULL;
		if(!flib_ini_get_str_opt(ini, &schemeName, schemeNameKey, "Unnamed")) {
			flib_cfg *scheme = flib_cfg_create(meta, schemeName);
			if(scheme) {
				if(!readSettingsFromIni(ini, scheme, index) && !readModsFromIni(ini, scheme, index)) {
					result = flib_cfg_retain(scheme);
				}
			}
			flib_cfg_release(scheme);
		}
		free(schemeName);
	}
	free(schemeNameKey);
	return result;
}

static flib_schemelist *fromIniHandleError(flib_schemelist *result, flib_ini *ini) {
	flib_ini_destroy(ini);
	flib_schemelist_destroy(result);
	return NULL;
}

flib_schemelist *flib_schemelist_from_ini(flib_cfg_meta *meta, const char *filename) {
	flib_schemelist *list = NULL;
	if(!meta || !filename) {
		flib_log_e("null parameter in flib_schemelist_from_ini");
		return NULL;
	}
	flib_ini *ini = flib_ini_load(filename);
	if(!ini || flib_ini_enter_section(ini, "schemes")) {
		flib_log_e("Missing file or missing section \"schemes\" in file %s.", filename);
		return fromIniHandleError(list, ini);
	}

	list = flib_schemelist_create();
	if(!list) {
		return fromIniHandleError(list, ini);
	}

	int schemeCount = 0;
	if(flib_ini_get_int(ini, &schemeCount, "size")) {
		flib_log_e("Missing or malformed scheme count in config file %s.", filename);
		return fromIniHandleError(list, ini);
	}

	for(int i=0; i<schemeCount; i++) {
		flib_cfg *scheme = readSchemeFromIni(meta, ini, i);
		if(!scheme || flib_schemelist_insert(list, scheme, i)) {
			flib_cfg_release(scheme);
			flib_log_e("Error reading scheme %i from config file %s.", i, filename);
			return fromIniHandleError(list, ini);
		}
		flib_cfg_release(scheme);
	}


	flib_ini_destroy(ini);
	return list;
}

static int writeSchemeToIni(flib_cfg *scheme, flib_ini *ini, int index) {
	flib_cfg_meta *meta = scheme->meta;
	bool error = false;

	char *key = makePrefixedName(index+1, "name");
	error |= !key || flib_ini_set_str(ini, key, scheme->schemeName);
	free(key);

	for(int i=0; i<meta->modCount && !error; i++) {
		char *key = makePrefixedName(index+1, meta->mods[i].name);
		error |= !key || flib_ini_set_bool(ini, key, scheme->mods[i]);
		free(key);
	}

	for(int i=0; i<meta->settingCount && !error; i++) {
		char *key = makePrefixedName(index+1, meta->settings[i].name);
		error |= !key || flib_ini_set_int(ini, key, scheme->settings[i]);
		free(key);
	}
	return error;
}

int flib_schemelist_to_ini(const char *filename, const flib_schemelist *schemes) {
	int result = -1;
	if(!filename || !schemes) {
		flib_log_e("null parameter in flib_schemelist_to_ini");
	} else {
		flib_ini *ini = flib_ini_create(NULL);
		if(ini && !flib_ini_create_section(ini, "schemes")) {
			bool error = false;
			error |= flib_ini_set_int(ini, "size", schemes->schemeCount);
			for(int i=0; i<schemes->schemeCount && !error; i++) {
				error |= writeSchemeToIni(schemes->schemes[i], ini, i);
			}

			if(!error) {
				result = flib_ini_save(ini, filename);
			}
		}
		flib_ini_destroy(ini);
	}
	return result;
}

flib_schemelist *flib_schemelist_create() {
	return flib_schemelist_retain(flib_calloc(1, sizeof(flib_schemelist)));
}

flib_schemelist *flib_schemelist_retain(flib_schemelist *list) {
	if(list) {
		flib_retain(&list->_referenceCount, "flib_schemelist");
	}
	return list;
}

void flib_schemelist_release(flib_schemelist *list) {
	if(list && flib_release(&list->_referenceCount, "flib_schemelist")) {
		flib_schemelist_destroy(list);
	}
}

flib_cfg *flib_schemelist_find(flib_schemelist *list, const char *name) {
	if(list && name) {
		for(int i=0; i<list->schemeCount; i++) {
			if(!strcmp(name, list->schemes[i]->schemeName)) {
				return list->schemes[i];
			}
		}
	}
	return NULL;
}

int flib_schemelist_insert(flib_schemelist *list, flib_cfg *cfg, int pos) {
	int result = -1;
	if(!list || !cfg || pos < 0 || pos > list->schemeCount) {
		flib_log_e("Invalid parameter in flib_schemelist_insert");
	} else {
		flib_cfg **newSchemes = flib_realloc(list->schemes, (list->schemeCount+1)*sizeof(*list->schemes));
		if(newSchemes) {
			list->schemes = newSchemes;
			memmove(list->schemes+pos+1, list->schemes+pos, (list->schemeCount-pos)*sizeof(*list->schemes));
			list->schemes[pos] = flib_cfg_retain(cfg);
			list->schemeCount++;
			result = 0;
		}
	}
	return result;
}

int flib_schemelist_delete(flib_schemelist *list, int pos) {
	int result = -1;
	if(!list || pos < 0 || pos >= list->schemeCount) {
		flib_log_e("Invalid parameter in flib_schemelist_delete");
	} else {
		flib_cfg_release(list->schemes[pos]);
		memmove(list->schemes+pos, list->schemes+pos+1, (list->schemeCount-(pos+1))*sizeof(*list->schemes));
		list->schemes[list->schemeCount-1] = NULL;
		list->schemeCount--;

		// If the realloc fails, just keep using the old buffer...
		flib_cfg **newSchemes = flib_realloc(list->schemes, list->schemeCount*sizeof(*list->schemes));
		if(newSchemes || list->schemeCount==1) {
			list->schemes = newSchemes;
		}
		result = 0;
	}
	return result;
}