/**
 *
 * getdns list management functions, note that the internal storage is 
 * accomplished via the libc binary search tree implementation so your
 * pointer foo needs to be keen to digest some of the internal semantics
 * 
 * Interfaces originally taken from the getdns API description pseudo implementation.
 *
 */

/*
 * Copyright (c) 2013, Versign, Inc.
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * * Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * * Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * * Neither the name of the <organization> nor the
 *   names of its contributors may be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL Verisign, Inc. BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <ctype.h>
#include <ldns/buffer.h>
#include "dict.h"

/*---------------------------------------- getdns_dict_find */
/**
 * private function used to locate a key in a dictionary
 * @param dict dicitonary to search
 * @param key key to search for
 * @param addifnotfnd if TRUE then an item will be added if the key is not found
 * @return pointer to dictionary item, caller must not free storage associated with item
 * @return NULL if additnotfnd == FALSE and key is not in dictionary
 */
struct getdns_dict_item *
getdns_dict_find(struct getdns_dict *dict, char *key, bool addifnotfnd)
{
	struct getdns_dict_item *item = NULL;

	if (dict != NULL && key != NULL) {
		item =
		    (struct getdns_dict_item *) ldns_rbtree_search(&(dict->
			root), key);
		if (addifnotfnd == true && item == NULL) {
			/* tsearch will add a node automatically for us */
			item =
			    (struct getdns_dict_item *) malloc(sizeof(struct
				getdns_dict_item));
			item->node.key = strdup(key);
			item->dtype = t_invalid;
			item->data.n = 0;
			ldns_rbtree_insert(&(dict->root),
			    (ldns_rbnode_t *) item);
		}
	}
	return item;
}				/* getdns_dict_find */

/*---------------------------------------- getdns_dict_get_names
*/
getdns_return_t
getdns_dict_get_names(struct getdns_dict * dict, struct getdns_list ** answer)
{
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;
	struct getdns_dict_item *item;
	size_t index;

	if (dict != NULL && answer != NULL) {
		*answer = getdns_list_create();

		LDNS_RBTREE_FOR(item, struct getdns_dict_item *, &(dict->root))
		{
			if (getdns_list_add_item(*answer,
				&index) == GETDNS_RETURN_GOOD) {
				struct getdns_bindata bindata;
				bindata.size = strlen(item->node.key);
				bindata.data = (void *) item->node.key;
				getdns_list_set_bindata(*answer, index,
				    &bindata);
			}
		}
		retval = GETDNS_RETURN_GOOD;
	}
	return retval;
}				/* getdns_dict_get_names */

/*---------------------------------------- getdns_dict_get_data_type */
getdns_return_t
getdns_dict_get_data_type(struct getdns_dict * dict, char *name,
    getdns_data_type * answer)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && answer != NULL) {
		item = getdns_dict_find(dict, name, false);
		if (item != NULL) {
			*answer = item->dtype;
			retval = GETDNS_RETURN_GOOD;
		}
	}

	return retval;
}				/* getdns_dict_get_data_type */

/*---------------------------------------- getdns_dict_get_dict */
getdns_return_t
getdns_dict_get_dict(struct getdns_dict * dict, char *name,
    struct getdns_dict ** answer)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && answer != NULL) {
		item = getdns_dict_find(dict, name, false);
		if (item != NULL) {
			if (item->dtype != t_dict)
				retval = GETDNS_RETURN_WRONG_TYPE_REQUESTED;
			else {
				*answer = item->data.dict;
				retval = GETDNS_RETURN_GOOD;
			}
		}
	}

	return retval;
}				/* getdns_dict_get_dict */

/*---------------------------------------- getdns_dict_get_list */
getdns_return_t
getdns_dict_get_list(struct getdns_dict * dict, char *name,
    struct getdns_list ** answer)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && answer != NULL) {
		item = getdns_dict_find(dict, name, false);
		if (item != NULL) {
			if (item->dtype != t_list)
				retval = GETDNS_RETURN_WRONG_TYPE_REQUESTED;
			else {
				*answer = item->data.list;
				retval = GETDNS_RETURN_GOOD;
			}
		}
	}

	return retval;
}				/* getdns_dict_get_list */

/*---------------------------------------- getdns_dict_get_bindata */
getdns_return_t
getdns_dict_get_bindata(struct getdns_dict * dict, char *name,
    struct getdns_bindata ** answer)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && answer != NULL) {
		item = getdns_dict_find(dict, name, false);
		if (item != NULL) {
			if (item->dtype != t_bindata)
				retval = GETDNS_RETURN_WRONG_TYPE_REQUESTED;
			else {
				*answer = item->data.bindata;
				retval = GETDNS_RETURN_GOOD;
			}
		}
	}

	return retval;
}				/* getdns_dict_get_bindata */

/*---------------------------------------- getdns_dict_get_int */
getdns_return_t
getdns_dict_get_int(struct getdns_dict * dict, char *name, uint32_t * answer)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && answer != NULL) {
		item = getdns_dict_find(dict, name, false);
		if (item != NULL) {
			if (item->dtype != t_int)
				retval = GETDNS_RETURN_WRONG_TYPE_REQUESTED;
			else {
				*answer = item->data.n;
				retval = GETDNS_RETURN_GOOD;
			}
		}
	}

	return retval;
}				/* getdns_dict_get_int */

/*---------------------------------------- getdns_dict_create */
struct getdns_dict *
getdns_dict_create()
{
	struct getdns_dict *dict;

	dict = (struct getdns_dict *) malloc(sizeof(struct getdns_dict));
	ldns_rbtree_init(&(dict->root), (int (*)(const void *,
		    const void *)) strcmp);
	return dict;
}				/* getdns_dict_create */

/*---------------------------------------- getdns_dict_copy */
/**
 * private function used to make a copy of a dict structure,
 * the caller is responsible * for freeing storage allocated to returned value
 * @param srcdict the dictionary structure to copy
 * @param dstdict the copy destination
 * @return the address of the copy of the dictionary structure on success
 * @return NULL on error (out of memory, invalid srcdict)
 */
getdns_return_t
getdns_dict_copy(struct getdns_dict * srcdict, struct getdns_dict ** dstdict)
{
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;
	struct getdns_dict_item *item;
	char *key;

	if (srcdict != NULL && dstdict != NULL) {
		*dstdict = getdns_dict_create();

		LDNS_RBTREE_FOR(item, struct getdns_dict_item *,
		    &(srcdict->root))
		{
			key = (char *) item->node.key;
			switch (item->dtype) {
			case t_bindata:
				getdns_dict_set_bindata(*dstdict, key,
				    item->data.bindata);
				break;

			case t_dict:
				getdns_dict_set_dict(*dstdict, key,
				    item->data.dict);
				break;

			case t_int:
				getdns_dict_set_int(*dstdict, key,
				    item->data.n);
				break;

			case t_list:
				getdns_dict_set_list(*dstdict, key,
				    item->data.list);
				break;

			case t_invalid:
			default:
				// TODO: this is a fault of some kind, for now ignore it
				break;
			}
		}
		retval = GETDNS_RETURN_GOOD;
	}

	return retval;
}				/* getdns_dict_copy */

/*---------------------------------------- getdns_dict_item_free */
/**
 * private function used to release storage associated with a dictionary item
 * @param item all memory in this structure and its children will be freed
 * @return void
 */
void
getdns_dict_item_free(ldns_rbnode_t * node, void *arg)
{
	(void) arg;
	struct getdns_dict_item *item = (struct getdns_dict_item *) node;
	if (item != NULL) {
		if (item->dtype == t_bindata) {
			if (item->data.bindata->size > 0)
				free(item->data.bindata->data);
			free(item->data.bindata);
		} else if (item->dtype == t_dict) {
			getdns_dict_destroy(item->data.dict);
		} else if (item->dtype == t_list) {
			getdns_list_destroy(item->data.list);
		}

		if (item->node.key != NULL)
			free((char *) item->node.key);
		free(item);
	}
}				/* getdns_dict_item_free */

/*---------------------------------------- getdns_dict_destroy */
void
getdns_dict_destroy(struct getdns_dict *dict)
{
	if (dict != NULL) {
		ldns_traverse_postorder(&(dict->root), getdns_dict_item_free,
		    NULL);
		free(dict);
	}

	return;
}				/* getdns_dict_destroy */

/*---------------------------------------- getdns_dict_set_dict */
getdns_return_t
getdns_dict_set_dict(struct getdns_dict * dict, char *name,
    struct getdns_dict * child_dict)
{
	struct getdns_dict_item *item;
	struct getdns_dict *newdict;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL) {
		item = getdns_dict_find(dict, name, true);
		if (item != NULL) {
			retval = getdns_dict_copy(child_dict, &newdict);
			if (retval == GETDNS_RETURN_GOOD) {
				item->dtype = t_dict;
				item->data.dict = newdict;
			} else
				item->dtype = t_invalid;
		}
	}

	return retval;
}				/* getdns_dict_set_dict */

/*---------------------------------------- getdns_dict_set_list */
getdns_return_t
getdns_dict_set_list(struct getdns_dict * dict, char *name,
    struct getdns_list * child_list)
{
	struct getdns_dict_item *item;
	struct getdns_list *newlist;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL) {
		item = getdns_dict_find(dict, name, true);
		if (item != NULL) {
			retval = getdns_list_copy(child_list, &newlist);
			if (retval == GETDNS_RETURN_GOOD) {
				item->dtype = t_list;
				item->data.list = newlist;
			} else
				item->dtype = t_invalid;
		}
	}

	return retval;
}				/* getdns_dict_set_list */

/*---------------------------------------- getdns_dict_set_bindata */
getdns_return_t
getdns_dict_set_bindata(struct getdns_dict * dict, char *name,
    struct getdns_bindata * child_bindata)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL && child_bindata != NULL) {
		item = getdns_dict_find(dict, name, true);
		if (item != NULL) {
			item->dtype = t_bindata;
			item->data.bindata =
			    (struct getdns_bindata *) malloc(sizeof(struct
				getdns_bindata));
			if (item->data.bindata != NULL) {
				item->data.bindata->data =
				    (void *) malloc(child_bindata->size);
				if (item->data.bindata->data != NULL) {
					item->data.bindata->size =
					    child_bindata->size;
					memcpy(item->data.bindata->data,
					    child_bindata->data,
					    child_bindata->size);
					retval = GETDNS_RETURN_GOOD;
				}
			}
		}
	}

	return retval;
}				/* getdns_dict_set_bindata */

/*---------------------------------------- getdns_dict_set_int */
getdns_return_t
getdns_dict_set_int(struct getdns_dict * dict, char *name,
    uint32_t child_uint32)
{
	struct getdns_dict_item *item;
	getdns_return_t retval = GETDNS_RETURN_NO_SUCH_DICT_NAME;

	if (dict != NULL && name != NULL) {
		item = getdns_dict_find(dict, name, true);
		if (item != NULL) {
			item->dtype = t_int;
			item->data.n = child_uint32;
			retval = GETDNS_RETURN_GOOD;
		}
	}

	return retval;
}				/* getdns_dict_set_int */

/*---------------------------------------- getdns_pp_dict */
/**
 * private function to help with indenting. 
 * @param indent number of spaces to return
 * @return       a character string containing min(80, indent) spaces
 */
static const char *
getdns_indent(size_t indent)
{
	static const char *spaces = "                                        "
	    "                                        ";
	return spaces + 80 - (indent < 80 ? indent : 0);
}				/* getdns_indent */

/*---------------------------------------- getdns_pp_bindata */
/**
 * private function to pretty print bindata to a ldns_buffer
 * @param buf     buffer to write to
 * @param indent  number of spaces to append after newline
 * @param bindata the bindata to print
 * @return        on success the number of written characters
 *                if an output error is encountered, a negative value
 */
static int
getdns_pp_bindata(ldns_buffer * buf, size_t indent,
    struct getdns_bindata *bindata)
{
	size_t i, p = ldns_buffer_position(buf);
	uint8_t *dptr;

	if (ldns_buffer_printf(buf, " <bindata ") < 0)
		return -1;

	/* Walk through all printable characters */
	i = 0;
	if (bindata->size && bindata->data[bindata->size - 1] == 0)
		while (i < bindata->size - 1 && isprint(bindata->data[i]))
			i++;

	if (i >= bindata->size - 1) {	/* all chars were printable */
		if (ldns_buffer_printf(buf, "for \"%s\">", bindata->data) < 0)
			return -1;
	} else {
		if (ldns_buffer_printf(buf, "of 0x") < 0)
			return -1;
		for (dptr = bindata->data;
		    dptr < bindata->data + bindata->size; dptr++) {
			if (dptr - bindata->data >= 16) {
				if (ldns_buffer_printf(buf, "...") < 0)
					return -1;
				break;
			}
			if (ldns_buffer_printf(buf, "%.2x", *dptr) < 0)
				return -1;
		}
		if (ldns_buffer_printf(buf, ">") < 0)
			return -1;
	}
	return ldns_buffer_position(buf) - p;
}				/* getdns_pp_bindata */

static int
getdns_pp_dict(ldns_buffer * buf, size_t indent, struct getdns_dict *dict);

/*---------------------------------------- getdns_pp_list */
/**
 * private function to pretty print list to a ldns_buffer
 * @param buf    buffer to write to
 * @param indent number of spaces to append after newline
 * @param list   the to list print
 * @return       on success the number of written characters
 *               if an output error is encountered, a negative value
 */
static int
getdns_pp_list(ldns_buffer * buf, size_t indent, struct getdns_list *list)
{
	size_t i, length, p = ldns_buffer_position(buf);
	getdns_data_type dtype;
	struct getdns_dict *dict_item;
	struct getdns_list *list_item;
	struct getdns_bindata *bindata_item;
	uint32_t int_item;

	if (list == NULL)
		return 0;

	if (ldns_buffer_printf(buf, "[") < 0)
		return -1;

	if (getdns_list_get_length(list, &length) != GETDNS_RETURN_GOOD)
		return -1;

	indent += 2;
	for (i = 0; i < length; i++) {
		if (ldns_buffer_printf(buf, "%s\n%s", (i ? "," : ""),
			getdns_indent(indent)) < 0)
			return -1;

		if (getdns_list_get_data_type(list, i,
			&dtype) != GETDNS_RETURN_GOOD)
			return -1;

		switch (dtype) {
		case t_int:
			if (getdns_list_get_int(list, i, &int_item) !=
			    GETDNS_RETURN_GOOD)
				if (ldns_buffer_printf(buf, " %d",
					(int) int_item) < 0)
					return -1;
			break;

		case t_bindata:
			if (getdns_list_get_bindata(list, i, &bindata_item) !=
			    GETDNS_RETURN_GOOD)
				return -1;
			if (getdns_pp_bindata(buf, indent, bindata_item) < 0)
				return -1;
			break;

		case t_list:
			if (getdns_list_get_list(list, i, &list_item) !=
			    GETDNS_RETURN_GOOD)
				return -1;
			if (getdns_pp_list(buf, indent, list_item) < 0)
				return -1;
			break;

		case t_dict:
			if (getdns_list_get_dict(list, i, &dict_item) !=
			    GETDNS_RETURN_GOOD)
				return -1;
			if (getdns_pp_dict(buf, indent, dict_item) < 0)
				return -1;
			break;

		case t_invalid:
		default:
			if (ldns_buffer_printf(buf, " <invalid>") < 0)
				return -1;
		}
		i++;
	}
	indent -= 2;
	if (ldns_buffer_printf(buf, i ? "\n%s]" : "]",
		getdns_indent(indent)) < 0)
		return -1;

	return ldns_buffer_position(buf) - p;
}				/* getdns_pp_list */

/*---------------------------------------- getdns_pp_dict */
/**
 * private function to pretty print dict to a ldns_buffer
 * @param buf    buffer to write to
 * @param indent number of spaces to append after newline
 * @param dict   the dict to print
 * @return       on success the number of written characters
 *               if an output error is encountered, a negative value
 */
static int
getdns_pp_dict(ldns_buffer * buf, size_t indent, struct getdns_dict *dict)
{
	size_t i, length, p = ldns_buffer_position(buf);
	struct getdns_dict_item *item;

	if (dict == NULL)
		return 0;

	if (ldns_buffer_printf(buf, "{") < 0)
		return -1;

	i = 0;
	indent += 2;
	LDNS_RBTREE_FOR(item, struct getdns_dict_item *, &(dict->root)) {
		if (ldns_buffer_printf(buf, "%s\n%s\"%s\":", (i ? "," : "")
			, getdns_indent(indent)
			, item->node.key) < 0)
			return -1;

		switch (item->dtype) {
		case t_int:
			if (ldns_buffer_printf(buf, " %d", item->data.n) < 0)
				return -1;
			break;

		case t_bindata:
			if (getdns_pp_bindata(buf, indent,
				item->data.bindata) < 0)
				return -1;
			break;

		case t_list:	/* Don't put empty lists on a new line */

			if (getdns_list_get_length(item->data.list,
				&length) != GETDNS_RETURN_GOOD)
				return -1;
			if (length == 0) {
				if (ldns_buffer_printf(buf, " []") < 0)
					return -1;
				break;
			}
			if (ldns_buffer_printf(buf, "\n%s",
				getdns_indent(indent)) < 0)
				return -1;
			if (getdns_pp_list(buf, indent, item->data.list) < 0)
				return -1;
			break;

		case t_dict:
			if (ldns_buffer_printf(buf, "\n%s",
				getdns_indent(indent)) < 0)
				return -1;
			if (getdns_pp_dict(buf, indent, item->data.dict) < 0)
				return -1;
			break;

		case t_invalid:
		default:
			if (ldns_buffer_printf(buf, " <invalid>") < 0)
				return -1;
		}
		i++;
	}
	indent -= 2;
	if (ldns_buffer_printf(buf, i ? "\n%s}" : "}",
		getdns_indent(indent)) < 0)
		return -1;

	return ldns_buffer_position(buf) - p;
}				/* getdns_pp_dict */

/*---------------------------------------- getdns_pretty_print_dict */
/**
 * Return a character string containing a "human readable" representation
 * of dict.
 * @param dict   the dict to pretty print
 * @return       the "human readable" representation of dict
 *               or NULL on error
 */
char *
getdns_pretty_print_dict(struct getdns_dict *dict)
{
	ldns_buffer *buf;
	char *ret;

	if (!dict) {
		return NULL;
	}

	buf = ldns_buffer_new(100);
	if (!buf)
		return NULL;

	if (getdns_pp_dict(buf, 0, dict) < 0) {
		ldns_buffer_free(buf);
		return NULL;
	}
	ret = (char *) ldns_buffer_export(buf);
	ldns_buffer_free(buf);
	return ret;
}				/* getdns_pretty_print_dict */

/* dict.c */