/*
 *
 *  Wireless daemon for Linux
 *
 *  Copyright (C) 2021  Intel Corporation. All rights reserved.
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <assert.h>

#include <ell/ell.h>

#include "src/json.h"
#include "ell/useful.h"

struct json_contents;


/*
 * Object:
 * { "test": "<smile emoji>" }
 */
static void test_json_unicode(const void *data)
{
	struct json_iter iter;
	_auto_(l_free)char *v;
	uint8_t expected[] = { 0xF0, 0x9F, 0x98, 0x80, 0 };

	uint8_t s[] = { '{', '"', 't', 'e', 's', 't', '"', ':',
				'"', 0xf0, 0x9f, 0x98, 0x80, '"', '}', 0 };
	struct json_contents *c = json_contents_new((const char *)s, sizeof(s));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter, JSON_MANDATORY("test", JSON_STRING, &v),
				JSON_UNDEFINED));

	assert(!strcmp(v, (char *)expected));

	json_contents_free(c);
}

static void test_json_escaped_unicode(const void *data)
{
	struct json_iter iter;
	_auto_(l_free)char *v;
	char *s = "{\"test\":\"\\uD8F0\"}";
	char expected[] = "\\uD8F0";
	struct json_contents *c = json_contents_new(s, strlen(s));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter, JSON_MANDATORY("test", JSON_STRING, &v),
				JSON_UNDEFINED));

	assert(!strcmp(v, expected));

	json_contents_free(c);
}

/*
 * Basic test string values and nested objects
 */
static void test_json(const void *data)
{
	char json[] = "{\"wi-fi_tech\":\"infra\","
			"\"discovery\":"
				"{\"ssid\":\"somessid\"},"
			"\"cred\":"
				"{\"akm\":\"psk\","
				"\"pass\":\"somepassphrase\"}}";
	_auto_(l_free)char *tech;
	_auto_(l_free)char *ssid;
	_auto_(l_free)char *akm;
	_auto_(l_free)char *pass;
	_auto_(l_free)char *opt_not_found;
	struct json_iter objnotfound;
	struct json_iter discovery;
	struct json_iter cred;
	struct json_iter iter;
	struct json_contents *c = json_contents_new(json, strlen(json));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
			JSON_MANDATORY("wi-fi_tech", JSON_STRING, &tech),
			JSON_MANDATORY("discovery", JSON_OBJECT, &discovery),
			JSON_MANDATORY("cred", JSON_OBJECT, &cred),
			JSON_OPTIONAL("notfound", JSON_STRING, &opt_not_found),
			JSON_OPTIONAL("objnotfound", JSON_OBJECT, &objnotfound),
			JSON_UNDEFINED));

	assert(!opt_not_found);
	assert(!json_iter_is_valid(&objnotfound));

	assert(!strcmp(tech, "infra"));

	assert(json_iter_parse(&discovery,
			JSON_MANDATORY("ssid", JSON_STRING, &ssid),
			JSON_UNDEFINED));

	assert(!strcmp(ssid, "somessid"));

	assert(json_iter_parse(&cred,
			JSON_MANDATORY("akm", JSON_STRING, &akm),
			JSON_MANDATORY("pass", JSON_STRING, &pass),
			JSON_UNDEFINED));
	assert(!strcmp(akm, "psk"));
	assert(!strcmp(pass, "somepassphrase"));

	json_contents_free(c);
}

/*
 * Tests empty objects parse successfully
 */
static void test_json_empty_objects(const void *data)
{
	char empty[] = "{}";
	char nested_empty[] = "{\"empty\":{}}";
	struct json_iter nested_iter;
	struct json_iter iter;
	struct json_contents *c = json_contents_new(empty, strlen(empty));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter, JSON_UNDEFINED));

	assert(json_iter_parse(&iter,
				JSON_OPTIONAL("optional", JSON_OBJECT, NULL),
				JSON_UNDEFINED));

	assert(!json_iter_parse(&iter,
				JSON_MANDATORY("optional", JSON_OBJECT, NULL),
				JSON_UNDEFINED));
	json_contents_free(c);

	c = json_contents_new(nested_empty, strlen(nested_empty));
	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
			JSON_MANDATORY("empty", JSON_OBJECT, &nested_iter),
			JSON_UNDEFINED));
	assert(json_iter_parse(&nested_iter, JSON_UNDEFINED));
	assert(!json_iter_parse(&nested_iter,
				JSON_MANDATORY("mandatory", JSON_OBJECT, NULL),
				JSON_UNDEFINED));
	json_contents_free(c);
}

/*
 * Tests that expected key/values can be provided in an order different than
 * they appear in the JSON string.
 */
static void test_json_out_of_order(const void *data)
{
	char object[] = "{\"key1\":{},\"key2\":\"val2\",\"key3\":\"val3\"}";
	struct json_iter iter;
	struct json_contents *c = json_contents_new(object, strlen(object));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
				JSON_OPTIONAL("nonexist1", JSON_STRING, NULL),
				JSON_MANDATORY("key3", JSON_STRING, NULL),
				JSON_MANDATORY("key2", JSON_STRING, NULL),
				JSON_OPTIONAL("nonexist2", JSON_OBJECT, NULL),
				JSON_MANDATORY("key1", JSON_OBJECT, NULL),
				JSON_UNDEFINED));
	json_contents_free(c);
}

/*
 * Tests that the token bounds checking works.
 */
static void test_json_larger_object(const void *data)
{
	char json[] = "{\"test1\":\"tester1\","
			"\"test2\":\"tester2\","
			"\"test3\":\"tester3\","
			"\"test4\":\"tester4\","
			"\"test5\":\"tester5\","
			"\"test6\":\"tester6\","
			"\"test7\":\"tester7\","
			"\"test8\":\"tester8\","
			"\"test9\":\"tester9\","
			"\"test10\":\"tester10\","
			"\"test11\":\"tester11\","
			"\"test12\":\"tester12\","
			"\"test13\":\"tester13\"}";

	struct json_iter iter;
	struct json_contents *c = json_contents_new(json, strlen(json));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
				JSON_MANDATORY("test13", JSON_STRING, NULL),
				JSON_UNDEFINED));
	json_contents_free(c);
}

static void check_primitives(struct json_iter *i, struct json_iter *ui,
				struct json_iter *t, struct json_iter *f,
				struct json_iter *null, struct json_iter *obj)
{

	int i_val;
	unsigned int ui_val;
	bool b_val;

	assert(json_iter_is_valid(i));
	assert(!json_iter_get_uint(i, NULL));
	assert(!json_iter_get_boolean(i, NULL));
	assert(!json_iter_get_null(i));
	assert(!json_iter_next(i));
	assert(json_iter_get_int(i, &i_val));
	assert(i_val == -10);

	assert(json_iter_is_valid(ui));
	assert(!json_iter_get_boolean(ui, NULL));
	assert(!json_iter_get_null(ui));
	assert(!json_iter_next(ui));
	assert(json_iter_get_int(ui, &i_val));
	assert(json_iter_get_uint(ui, &ui_val));
	assert(i_val == 10 && ui_val == 10);

	assert(json_iter_is_valid(t));
	assert(!json_iter_get_null(t));
	assert(!json_iter_get_int(t, NULL));
	assert(!json_iter_get_uint(t, NULL));
	assert(!json_iter_next(t));
	assert(json_iter_get_boolean(t, &b_val));
	assert(b_val == true);

	assert(json_iter_is_valid(f));
	assert(!json_iter_get_null(f));
	assert(!json_iter_get_int(f, NULL));
	assert(!json_iter_get_uint(f, NULL));
	assert(!json_iter_next(f));
	assert(json_iter_get_boolean(f, &b_val));
	assert(b_val == false);

	assert(json_iter_is_valid(null));
	assert(!json_iter_get_int(null, NULL));
	assert(!json_iter_get_uint(null, NULL));
	assert(!json_iter_get_boolean(null, NULL));
	assert(!json_iter_next(null));
	assert(json_iter_get_null(null));

	if (obj) {
		assert(json_iter_is_valid(obj));
		assert(!json_iter_next(obj));
		assert(json_iter_parse(obj,
			JSON_MANDATORY("null_val", JSON_PRIMITIVE, null),
			JSON_MANDATORY("false_val", JSON_PRIMITIVE, f),
			JSON_MANDATORY("true_val", JSON_PRIMITIVE, t),
			JSON_MANDATORY("int_val", JSON_PRIMITIVE, i),
			JSON_MANDATORY("uint_val", JSON_PRIMITIVE, ui),
			JSON_UNDEFINED));

		check_primitives(i, ui, t, f, null, NULL);
	}
}

static void test_json_primitives(const void *data)
{
	char json[] = "{\"int_val\": -10,"
			"\"uint_val\": 10,"
			"\"true_val\": true,"
			"\"false_val\": false,"
			"\"null_val\": null,"
			"\"obj_val\":{"
				"\"int_val\": -10,"
				"\"uint_val\": 10,"
				"\"true_val\": true,"
				"\"false_val\": false,"
				"\"null_val\": null}}";
	struct json_contents *c = json_contents_new(json, strlen(json));
	struct json_iter outer, inner, null, f, t, i, ui;
	struct json_iter not_found;

	json_iter_init(&outer, c);
	assert(json_iter_parse(&outer,
			JSON_MANDATORY("obj_val", JSON_OBJECT, &inner),
			JSON_MANDATORY("null_val", JSON_PRIMITIVE, &null),
			JSON_MANDATORY("false_val", JSON_PRIMITIVE, &f),
			JSON_MANDATORY("true_val", JSON_PRIMITIVE, &t),
			JSON_MANDATORY("int_val", JSON_PRIMITIVE, &i),
			JSON_MANDATORY("uint_val", JSON_PRIMITIVE, &ui),
			JSON_OPTIONAL("not_found", JSON_PRIMITIVE, &not_found),
			JSON_UNDEFINED));

	assert(!json_iter_is_valid(&not_found));

	check_primitives(&i, &ui, &t, &f, &null, &inner);

	json_contents_free(c);
}

static void test_json_arrays(const void *data)
{
	unsigned int ui;
	int i;
	bool b;
	int count;
	char json[] = "{\"uint_array\":[1, 2, 3, 4, 5, 6],"
			"\"int_array\":[-1, -2, -3, -4, -5, -6],"
			"\"bool_array\":[true, false, true, false],"
			"\"null_array\":[null, null, null, null],"
			"\"obj_array\":[{}, {\"key\":\"value\", \"key2\":\"value2\"}],"
			"\"mixed_array\":[1, -1, true, false, null, \"string\"]}";

	struct json_iter iter;
	struct json_iter i_array, ui_array, b_array, n_array,
				m_array, obj_array;
	struct json_contents *c = json_contents_new(json, strlen(json));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
			JSON_MANDATORY("mixed_array", JSON_ARRAY, &m_array),
			JSON_MANDATORY("null_array", JSON_ARRAY, &n_array),
			JSON_MANDATORY("bool_array", JSON_ARRAY, &b_array),
			JSON_MANDATORY("int_array", JSON_ARRAY, &i_array),
			JSON_MANDATORY("uint_array", JSON_ARRAY, &ui_array),
			JSON_MANDATORY("obj_array", JSON_ARRAY, &obj_array),
			JSON_UNDEFINED));

	count = 1;

	while (json_iter_next(&ui_array)) {
		assert(json_iter_get_type(&ui_array) == JSON_PRIMITIVE);
		assert(!json_iter_parse(&ui_array, JSON_UNDEFINED));
		assert(json_iter_get_uint(&ui_array, &ui));
		assert(ui == (unsigned int) count);
		count++;
	}

	count = -1;

	while (json_iter_next(&i_array)) {
		assert(json_iter_get_type(&i_array) == JSON_PRIMITIVE);
		assert(!json_iter_parse(&i_array, JSON_UNDEFINED));
		assert(json_iter_get_int(&i_array, &i));
		assert(i == count);
		count--;
	}

	count = 0;

	while (json_iter_next(&b_array)) {
		assert(json_iter_get_type(&b_array) == JSON_PRIMITIVE);
		assert(!json_iter_parse(&b_array, JSON_UNDEFINED));
		assert(json_iter_get_boolean(&b_array, &b));
		assert(b == count % 2 ? false : true);
		count++;
	}

	count = 0;

	while (json_iter_next(&n_array)) {
		assert(json_iter_get_type(&n_array) == JSON_PRIMITIVE);
		assert(!json_iter_parse(&n_array, JSON_UNDEFINED));
		assert(json_iter_get_null(&n_array));
		count++;
	}

	assert(count == 4);

	count = 0;

	while (json_iter_next(&m_array)) {
		_auto_(l_free) char *str = NULL;

		switch (count) {
		case 0:
			assert(json_iter_get_type(&m_array) == JSON_PRIMITIVE);
			assert(json_iter_get_uint(&m_array, &ui));
			assert(ui == 1);
			break;
		case 1:
			assert(json_iter_get_type(&m_array) == JSON_PRIMITIVE);
			assert(json_iter_get_int(&m_array, &i));
			assert(i == -1);
			break;
		case 2:
		case 3:
			assert(json_iter_get_type(&m_array) == JSON_PRIMITIVE);
			assert(json_iter_get_boolean(&m_array, &b));
			assert(b == count % 2 ? false : true);
			break;
		case 4:
			assert(json_iter_get_type(&m_array) == JSON_PRIMITIVE);
			assert(json_iter_get_null(&m_array));
			break;
		case 5:
			assert(json_iter_get_type(&m_array) == JSON_STRING);
			assert(json_iter_get_string(&m_array, &str));
			assert(!strcmp(str, "string"));
			break;
		}

		count++;
	}

	count = 0;

	assert(!json_iter_parse(&obj_array, JSON_UNDEFINED));

	while (json_iter_next(&obj_array)) {
		struct json_iter object;

		assert(json_iter_get_type(&obj_array) == JSON_OBJECT);
		assert(json_iter_get_container(&obj_array, &object));

		switch (count) {
		case 0:
			assert(json_iter_parse(&object, JSON_UNDEFINED));
			break;
		case 1:
			assert(json_iter_parse(&object,
				JSON_MANDATORY("key", JSON_STRING, NULL),
				JSON_UNDEFINED));
			break;
		}

		count++;
	}

	assert(count == 2);

	json_contents_free(c);
}

static void test_json_nested_arrays(const void *data)
{
	char json[] = "{\"array\":[[], {}, [1, 2], {\"key\":\"value\"}, [\"one\",\"two\"]]}";
	int count = 0;
	struct json_iter iter;
	struct json_iter array;
	struct json_iter inner;
	struct json_contents *c = json_contents_new(json, strlen(json));

	json_iter_init(&iter, c);
	assert(json_iter_parse(&iter,
			JSON_MANDATORY("array", JSON_ARRAY, &array),
			JSON_UNDEFINED));

	while (json_iter_next(&array)) {
		int count2 = 0;

		assert(json_iter_get_container(&array, &inner));

		while (json_iter_next(&inner)) {
			_auto_(l_free) char *str = NULL;

			switch (count) {
			case 0:
			case 1:
				assert(false);
				break;
			case 4:
				assert(json_iter_get_type(&inner) ==
							JSON_STRING);
				assert(json_iter_get_string(&inner, &str));

				if (count2 == 0)
					assert(!strcmp("one", str));
				else
					assert(!strcmp("two", str));

				break;
			}

			count2++;
		}

		switch (count) {
		case 0:
			assert(count2 == 0);
			break;
		case 1:
			assert(count2 == 0);
			break;
		case 2:
			assert(count2 == 2);
			break;
		case 3:
			assert(count2 == 0);
			break;
		case 4:
			assert(count2 == 2);
			break;
		default:
			break;
		}

		count++;
	}

	assert(count == 5);

	json_contents_free(c);
}

int main(int argc, char *argv[])
{
	l_test_init(&argc, &argv);

	l_test_add("json unicode", test_json_unicode, NULL);
	l_test_add("json escaped unicode", test_json_escaped_unicode, NULL);
	l_test_add("json nested objects", test_json, NULL);
	l_test_add("json empty objects", test_json_empty_objects, NULL);
	l_test_add("json parse out of order", test_json_out_of_order, NULL);
	l_test_add("json larger object", test_json_larger_object, NULL);
	l_test_add("json test primitives", test_json_primitives, NULL);
	l_test_add("json test arrays", test_json_arrays, NULL);
	l_test_add("json test nested arrays", test_json_nested_arrays, NULL);

	return l_test_run();
}