/*
 * The MIT License
 *
 * Copyright (C) 2013-2017 Alexander Saprykin <saprykin.spb@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * 'Software'), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include "plibsys.h"
#include "ptestmacros.h"

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

P_TEST_MODULE_INIT ();

#define PINIFILE_STRESS_LINE	2048
#define PINIFILE_MAX_LINE	1024

extern "C" ppointer pmem_alloc (psize nbytes)
{
	P_UNUSED (nbytes);
	return (ppointer) NULL;
}

extern "C" ppointer pmem_realloc (ppointer block, psize nbytes)
{
	P_UNUSED (block);
	P_UNUSED (nbytes);
	return (ppointer) NULL;
}

extern "C" void pmem_free (ppointer block)
{
	P_UNUSED (block);
}

static bool create_test_ini_file (bool last_empty_section)
{
	FILE *file = fopen ("." P_DIR_SEPARATOR "p_ini_test_file.ini", "w");

	if (file == NULL)
		return false;

	pchar *buf = (pchar *) p_malloc0 (PINIFILE_STRESS_LINE + 1);

	for (int i = 0; i < PINIFILE_STRESS_LINE; ++i)
		buf[i] = (pchar) (97 + i % 20);

	/* Empty section */
	fprintf (file, "[empty_section]\n");

	/* Numeric section */
	fprintf (file, "[numeric_section]\n");
	fprintf (file, "int_parameter_1 = 4\n");
	fprintf (file, "int_parameter_2 = 5 ;This is a comment\n");
	fprintf (file, "int_parameter_3 = 6 #This is another type of a comment\n");
	fprintf (file, "# Whole line is a comment\n");
	fprintf (file, "; Yet another comment line\n");
	fprintf (file, "float_parameter_1 = 3.24\n");
	fprintf (file, "float_parameter_2 = 0.15\n");

	/* String section */
	fprintf (file, "[string_section]\n");
	fprintf (file, "string_parameter_1 = Test string\n");
	fprintf (file, "string_parameter_2 = \"Test string with #'\"\n");
	fprintf (file, "string_parameter_3 = \n");
	fprintf (file, "string_parameter_4 = 12345 ;Comment\n");
	fprintf (file, "string_parameter_4 = 54321\n");
	fprintf (file, "string_parameter_5 = 'Test string'\n");
	fprintf (file, "string_parameter_6 = %s\n", buf);
	fprintf (file, "string_parameter_7 = ''\n");
	fprintf (file, "string_parameter_8 = \"\"\n");
	fprintf (file, "%s = stress line\n", buf);

	/* Boolean section */
	fprintf (file, "[boolean_section]\n");
	fprintf (file, "boolean_parameter_1 = TRUE ;True value\n");
	fprintf (file, "boolean_parameter_2 = 0 ;False value\n");
	fprintf (file, "boolean_parameter_3 = false ;False value\n");
	fprintf (file, "boolean_parameter_4 = 1 ;True value\n");

	/* List section */
	fprintf (file, "[list_section]\n");
	fprintf (file, "list_parameter_1 = {1\t2\t5\t10} ;First list\n");
	fprintf (file, "list_parameter_2 = {2.0 3.0 5.0} #Second list\n");
	fprintf (file, "list_parameter_3 = {true FALSE 1} #Last list\n");

	/* Empty section */
	if (last_empty_section)
		fprintf (file, "[empty_section_2]\n");

	p_free (buf);

	return fclose (file) == 0;
}

P_TEST_CASE_BEGIN (pinifile_nomem_test)
{
	p_libsys_init ();

	P_TEST_REQUIRE (create_test_ini_file (false));

	PIniFile *ini = p_ini_file_new  ("." P_DIR_SEPARATOR "p_ini_test_file.ini");
	P_TEST_CHECK (ini != NULL);

	PMemVTable vtable;

	vtable.free    = pmem_free;
	vtable.malloc  = pmem_alloc;
	vtable.realloc = pmem_realloc;

	P_TEST_CHECK (p_mem_set_vtable (&vtable) == TRUE);

	P_TEST_CHECK (p_ini_file_new ("." P_DIR_SEPARATOR "p_ini_test_file.ini") == NULL);
	P_TEST_CHECK (p_ini_file_parse (ini, NULL) == TRUE);
	P_TEST_CHECK (p_ini_file_sections (ini) == NULL);

	p_mem_restore_vtable ();

	p_ini_file_free (ini);

	ini = p_ini_file_new ("." P_DIR_SEPARATOR "p_ini_test_file.ini");
	P_TEST_CHECK (ini != NULL);

	P_TEST_CHECK (p_ini_file_parse (ini, NULL) == TRUE);
	PList *section_list = p_ini_file_sections (ini);
	P_TEST_CHECK (section_list != NULL);
	P_TEST_CHECK (p_list_length (section_list) == 4);

	p_list_foreach (section_list, (PFunc) p_free, NULL);
	p_list_free (section_list);
	p_ini_file_free (ini);

	P_TEST_CHECK (p_file_remove ("." P_DIR_SEPARATOR "p_ini_test_file.ini", NULL) == TRUE);

	p_libsys_shutdown ();
}
P_TEST_CASE_END ()

P_TEST_CASE_BEGIN (pinifile_bad_input_test)
{
	PIniFile *ini = NULL;

	p_libsys_init ();

	p_ini_file_free (ini);
	P_TEST_CHECK (p_ini_file_new (NULL) == NULL);
	P_TEST_CHECK (p_ini_file_parse (ini, NULL) == FALSE);
	P_TEST_CHECK (p_ini_file_is_parsed (ini) == FALSE);
	P_TEST_CHECK (p_ini_file_is_key_exists (ini, "string_section", "string_paramter_1") == FALSE);
	P_TEST_CHECK (p_ini_file_sections (ini) == NULL);
	P_TEST_CHECK (p_ini_file_keys (ini, "string_section") == NULL);
	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_parameter_1", FALSE) == FALSE);
	P_TEST_CHECK_CLOSE (p_ini_file_parameter_double (ini, "numeric_section", "float_parameter_1", 1.0), 1.0, 0.0001);
	P_TEST_CHECK (p_ini_file_parameter_int (ini, "numeric_section", "int_parameter_1", 0) == 0);
	P_TEST_CHECK (p_ini_file_parameter_list (ini, "list_section", "list_parameter_1") == NULL);
	P_TEST_CHECK (p_ini_file_parameter_string (ini, "string_section", "string_parameter_1", NULL) == NULL);

	ini = p_ini_file_new ("./bad_file_path/fake.ini");
	P_TEST_CHECK (ini != NULL);
	P_TEST_CHECK (p_ini_file_parse (ini, NULL) == FALSE);
	p_ini_file_free (ini);

	P_TEST_REQUIRE (create_test_ini_file (true));

	p_libsys_shutdown ();
}
P_TEST_CASE_END ()

P_TEST_CASE_BEGIN (pinifile_read_test)
{
	p_libsys_init ();

	PIniFile *ini = p_ini_file_new ("." P_DIR_SEPARATOR "p_ini_test_file.ini");
	P_TEST_REQUIRE (ini != NULL);
	P_TEST_CHECK (p_ini_file_is_parsed (ini) == FALSE);

	P_TEST_REQUIRE (p_ini_file_parse (ini, NULL) == TRUE);
	P_TEST_CHECK (p_ini_file_is_parsed (ini) == TRUE);
	P_TEST_REQUIRE (p_ini_file_parse (ini, NULL) == TRUE);
	P_TEST_CHECK (p_ini_file_is_parsed (ini) == TRUE);

	/* Test list of sections */
	PList *list = p_ini_file_sections (ini);
	P_TEST_CHECK (list != NULL);
	P_TEST_CHECK (p_list_length (list) == 4);

	p_list_foreach (list, (PFunc) p_free, NULL);
	p_list_free (list);

	/* Test empty section */
	list = p_ini_file_keys (ini, "empty_section");
	P_TEST_CHECK (list == NULL);

	/* Test numeric section */
	list = p_ini_file_keys (ini, "numeric_section");
	P_TEST_CHECK (p_list_length (list) == 5);
	p_list_foreach (list, (PFunc) p_free, NULL);
	p_list_free (list);

	P_TEST_CHECK (p_ini_file_parameter_list (ini, "numeric_section", "int_parameter_1") == NULL);
	P_TEST_CHECK (p_ini_file_parameter_int (ini, "numeric_section", "int_parameter_1", -1) == 4);
	P_TEST_CHECK (p_ini_file_parameter_int (ini, "numeric_section", "int_parameter_2", -1) == 5);
	P_TEST_CHECK (p_ini_file_parameter_int (ini, "numeric_section", "int_parameter_3", -1) == 6);
	P_TEST_CHECK (p_ini_file_parameter_int (ini, "numeric_section", "int_parameter_def", 10) == 10);
	P_TEST_CHECK_CLOSE (p_ini_file_parameter_double (ini, "numeric_section", "float_parameter_1", -1.0), 3.24, 0.0001);
	P_TEST_CHECK_CLOSE (p_ini_file_parameter_double (ini, "numeric_section", "float_parameter_2", -1.0), 0.15, 0.0001);
	P_TEST_CHECK_CLOSE (p_ini_file_parameter_double (ini, "numeric_section_no", "float_parameter_def", 10.0), 10.0, 0.0001);
	P_TEST_CHECK (p_ini_file_is_key_exists (ini, "numeric_section", "int_parameter_1") == TRUE);
	P_TEST_CHECK (p_ini_file_is_key_exists (ini, "numeric_section", "float_parameter_1") == TRUE);
	P_TEST_CHECK (p_ini_file_is_key_exists (ini, "numeric_section_false", "float_parameter_1") == FALSE);

	/* Test string section */
	list = p_ini_file_keys (ini, "string_section");
	P_TEST_CHECK (p_list_length (list) == 8);
	p_list_foreach (list, (PFunc) p_free, NULL);
	p_list_free (list);

	pchar *str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_1", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "Test string") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_2", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "Test string with #'") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_3", NULL);
	P_TEST_REQUIRE (str == NULL);
	P_TEST_CHECK (p_ini_file_is_key_exists (ini, "string_section", "string_parameter_3") == FALSE);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_4", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "54321") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_5", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "Test string") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_6", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strlen (str) > 0 && strlen (str) < PINIFILE_MAX_LINE);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_7", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_8", NULL);
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "") == 0);
	p_free (str);

	str = p_ini_file_parameter_string (ini, "string_section", "string_parameter_def", "default_value");
	P_TEST_REQUIRE (str != NULL);
	P_TEST_CHECK (strcmp (str, "default_value") == 0);
	p_free (str);

	/* Test boolean section */
	list = p_ini_file_keys (ini, "boolean_section");
	P_TEST_CHECK (p_list_length (list) == 4);
	p_list_foreach (list, (PFunc) p_free, NULL);
	p_list_free (list);

	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_parameter_1", FALSE) == TRUE);
	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_parameter_2", TRUE) == FALSE);
	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_parameter_3", TRUE) == FALSE);
	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_parameter_4", FALSE) == TRUE);
	P_TEST_CHECK (p_ini_file_parameter_boolean (ini, "boolean_section", "boolean_section_def", TRUE) == TRUE);

	/* Test list section */
	list = p_ini_file_keys (ini, "list_section");
	P_TEST_CHECK (p_list_length (list) == 3);
	p_list_foreach (list, (PFunc) p_free, NULL);
	p_list_free (list);

	/* -- First list parameter */
	PList *list_val = p_ini_file_parameter_list (ini, "list_section", "list_parameter_1");
	P_TEST_CHECK (list_val != NULL);
	P_TEST_CHECK (p_list_length (list_val) == 4);

	pint int_sum = 0;
	for (PList *iter = list_val; iter != NULL; iter = iter->next)
		int_sum +=  atoi ((const pchar *) (iter->data));

	P_TEST_CHECK (int_sum == 18);
	p_list_foreach (list_val, (PFunc) p_free, NULL);
	p_list_free (list_val);

	/* -- Second list parameter */
	list_val = p_ini_file_parameter_list (ini, "list_section", "list_parameter_2");
	P_TEST_CHECK (list_val != NULL);
	P_TEST_CHECK (p_list_length (list_val) == 3);

	double flt_sum = 0;
	for (PList *iter = list_val; iter != NULL; iter = iter->next)
		flt_sum +=  atof ((const pchar *) (iter->data));

	P_TEST_CHECK_CLOSE (flt_sum, 10.0, 0.0001);
	p_list_foreach (list_val, (PFunc) p_free, NULL);
	p_list_free (list_val);

	/* -- Third list parameter */
	list_val = p_ini_file_parameter_list (ini, "list_section", "list_parameter_3");
	P_TEST_CHECK (list_val != NULL);
	P_TEST_CHECK (p_list_length (list_val) == 3);

	pboolean bool_sum = TRUE;
	for (PList *iter = list_val; iter != NULL; iter = iter->next)
		bool_sum = bool_sum && atoi ((const pchar *) (iter->data));

	P_TEST_CHECK (bool_sum == FALSE);
	p_list_foreach (list_val, (PFunc) p_free, NULL);
	p_list_free (list_val);

	/* -- False list parameter */
	P_TEST_CHECK (p_ini_file_parameter_list (ini, "list_section_no", "list_parameter_def") == NULL);

	p_ini_file_free (ini);

	P_TEST_CHECK (p_file_remove ("." P_DIR_SEPARATOR "p_ini_test_file.ini", NULL) == TRUE);

	p_libsys_shutdown ();
}
P_TEST_CASE_END ()

P_TEST_SUITE_BEGIN()
{
	P_TEST_SUITE_RUN_CASE (pinifile_nomem_test);
	P_TEST_SUITE_RUN_CASE (pinifile_bad_input_test);
	P_TEST_SUITE_RUN_CASE (pinifile_read_test);
}
P_TEST_SUITE_END()