#include "inihelper.h"

#include "../logging.h"

#include <string.h>

#include <stdlib.h>

#include <ctype.h>

#include <limits.h>

#include <errno.h>

#include <stdarg.h>

static char *flib_asprintf(const char *fmt, ...) {

	va_list argp;

	va_start(argp, fmt);

	size_t requiredSize = vsnprintf(NULL, 0, fmt, argp)+1;

	char *buf = malloc(requiredSize);

	if(buf) {

		vsnprintf(buf, requiredSize, fmt, argp);

	}

	va_end(argp);

	return buf;

}

char *inihelper_strdupnull(const char *str) {

	if(!str) {

		return NULL;

	}

	return flib_asprintf("%s", str);

}

static bool keychar_needs_urlencoding(char c) {

	return !isalnum(c);

}

char *inihelper_urlencode(const char *inbuf) {

	if(!inbuf) {

		return NULL;

	}

	size_t insize = strlen(inbuf);

	if(insize > SIZE_MAX/4) {

		return NULL;

	}

	char *outbuf = malloc(insize*3+1);

	if(!outbuf) {

		return NULL;

	}

    size_t inpos = 0, outpos = 0;

    while(inbuf[inpos]) {

        if(!keychar_needs_urlencoding(inbuf[inpos])) {

        	outbuf[outpos++] = inbuf[inpos++];

        } else {

            sprintf(outbuf+outpos, "%%%02X", inbuf[inpos]);

            inpos++;

            outpos += 3;

        }

    }

    outbuf[outpos] = 0;

    return outbuf;

}

char *inihelper_urldecode(const char *inbuf) {

	char *outbuf = malloc(strlen(inbuf)+1);

	if(!outbuf) {

		return NULL;

	}

    size_t inpos = 0, outpos = 0;

    while(inbuf[inpos]) {

        if(inbuf[inpos] == '%' && isxdigit(inbuf[inpos+1]) && isxdigit(inbuf[inpos+2])) {

            char temp[3] = {inbuf[inpos+1],inbuf[inpos+2],0};

            outbuf[outpos++] = strtol(temp, NULL, 16);

            inpos += 3;

        } else {

        	outbuf[outpos++] = inbuf[inpos++];

        }

    }

    outbuf[outpos] = 0;

    return outbuf;

}

char *inihelper_createDictKey(const char *sectionName, const char *keyName) {

	if(!sectionName || !keyName) {

		return NULL;

	}

	return flib_asprintf("%s:%s", sectionName, keyName);

}

char *inihelper_getstring(dictionary *inifile, bool *error, const char *sectionName, const char *keyName) {

	if(!inifile || !sectionName || !keyName) {

		*error = true;

		return NULL;

	}

	char *extendedkey = inihelper_createDictKey(sectionName, keyName);

	if(!extendedkey) {

		*error = true;

		return NULL;

	}

	char *result = iniparser_getstring(inifile, extendedkey, NULL);

	free(extendedkey);

	if(!result) {

		flib_log_i("Missing ini setting: %s/%s", sectionName, keyName);

		*error = true;

	}

	return result;

}

char *inihelper_getstringdup(dictionary *inifile, bool *error, const char *sectionName, const char *keyName) {

	return inihelper_strdupnull(inihelper_getstring(inifile, error, sectionName, keyName));

}

int inihelper_getint(dictionary *inifile, bool *error, const char *sectionName, const char *keyName) {

	char *value = inihelper_getstring(inifile, error, sectionName, keyName);

	if(!value) {

		return 0;

	} else {

		errno = 0;

		long val = strtol(value, NULL, 10);

		if(errno!=0) {

			*error = true;

			return 0;

		}

		if(val<INT_MIN || val>INT_MAX) {

			*error = true;

			return 0;

		}

		return (int)val;

	}

}

bool inihelper_getbool(dictionary *inifile, bool *error, const char *sectionName, const char *keyName) {

	char *value = inihelper_getstring(inifile, error, sectionName, keyName);

	if(!value) {

		return false;

	} else {

		bool trueval = strchr("1tTyY", value[0]);

		bool falseval = strchr("0fFnN", value[0]);

		if(!trueval && !falseval) {

			*error = true;

			return false;

		} else {

			return trueval;

		}

	}

}

int inihelper_setstr(dictionary *dict, const char *sectionName, const char *keyName, const char *value) {

	int result = -1;

	if(!dict || !sectionName || !keyName || !value) {

		flib_log_e("inihelper_setstr called with bad parameters");

	} else {

		char *extendedkey = inihelper_createDictKey(sectionName, keyName);

		if(extendedkey) {

			result = iniparser_set(dict, extendedkey, value);

			free(extendedkey);

		}

	}

	return result;

}

int inihelper_setint(dictionary *dict, const char *sectionName, const char *keyName, int value) {

	int result = -1;

	if(!dict || !sectionName || !keyName) {

		flib_log_e("inihelper_setint called with bad parameters");

	} else {

		char *strvalue = flib_asprintf("%i", value);

		if(strvalue) {

			result = inihelper_setstr(dict, sectionName, keyName, strvalue);

			free(strvalue);

		}

	}

	return result;

}

int inihelper_setbool(dictionary *dict, const char *sectionName, const char *keyName, bool value) {

	int result = -1;

	if(!dict || !sectionName || !keyName) {

		flib_log_e("inihelper_setint called with bad parameters");

	} else {

		result = inihelper_setstr(dict, sectionName, keyName, value ? "true" : "false");

	}

	return result;

}