/* vim: tabstop=4 shiftwidth=4 noexpandtab
 * This file is part of ToaruOS and is released under the terms
 * of the NCSA / University of Illinois License - see LICENSE.md
 * Copyright (C) 2015 Kevin Lange
 *
 * Configuration File Reader
 *
 * Reads an implementation of the INI "standard". Note that INI
 * isn't actually a standard. We support the following:
 * - ; comments
 * - foo=bar keyword assignment
 * - [sections]
 */
#include <stdio.h>

#include "confreader.h"

#include "lib/hashmap.h"

static void free_hashmap(void * h) {
	hashmap_free(h);
	free(h);
}

confreader_t * confreader_load(const char * file) {

	confreader_t * out = malloc(sizeof(confreader_t));

	out->sections = hashmap_create(10);

	FILE * f = fopen(file, "r");

	hashmap_t * current_section = hashmap_create(10);
	current_section->hash_val_free = free_hashmap;

	hashmap_set(out->sections, "", current_section);

	if (!f) {
		/* File does not exist, no configuration values, but continue normally. */
		return out;
	}

	char tmp[1024];
	char tmp2[1024];
	char eq[2];

	while (!feof(f)) {
		char c = fgetc(f);
		tmp[0] = '\0';
		eq[0] = '\0';
		tmp2[0] = '\0';
		if (c == ';') {
			fscanf(f, " %[^\n]", tmp);
			while (!feof(f) && fgetc(f) != '\n');
		} else if (c == '\n' || c == EOF) {
			continue;
		} else if (c == '[') {
			fscanf(f, " %[^]] ", tmp);
			while (!feof(f) && fgetc(f) != '\n');
			current_section = hashmap_create(10);
			hashmap_set(out->sections, tmp, current_section);
		} else {
			ungetc(c, f);
			fscanf(f, " %[^=]%1[=]%[^=\n]", tmp, eq, tmp2);
			while (!feof(f) && fgetc(f) != '\n');
			if (strcmp(eq, "=")) {
				continue;
			}

			hashmap_set(current_section, tmp, strdup(tmp2));
		}
	}

	fclose(f);

	return out;
}

void confreader_free(confreader_t * conf) {
	free_hashmap(conf->sections);
	free(conf);
}

char * confreader_get(confreader_t * ctx, char * section, char * value) {
	if (!ctx) return NULL;

	hashmap_t * s = hashmap_get(ctx->sections, section);

	if (!s) return NULL;

	return hashmap_get(s, value);
}

char * confreader_getd(confreader_t * ctx, char * section, char * value, char * def) {
	char * result = confreader_get(ctx, section, value);
	return result ? result : def;
}

int confreader_int(confreader_t * ctx, char * section, char * value) {
	char * result = confreader_get(ctx, section, value);
	if (!result) return 0;
	return atoi(result);
}

int confreader_intd(confreader_t * ctx, char * section, char * value, int def) {
	char * result = confreader_get(ctx, section, value);
	if (!result) return def;
	return atoi(result);
}