/*
 * Copyright (c) 2011 Jiri Svoboda
 * 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.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 THE AUTHOR 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.
 */

/** @addtogroup loc
 * @{
 */
/** @file Categories for location service.
 */

#include <adt/list.h>
#include <errno.h>
#include <fibril_synch.h>
#include <stdlib.h>
#include <str.h>

#include "category.h"
#include "locsrv.h"

/** Initialize category directory. */
void categ_dir_init(categ_dir_t *cdir)
{
	fibril_mutex_initialize(&cdir->mutex);
	list_initialize(&cdir->categories);
}

/** Add new category to directory. */
void categ_dir_add_cat(categ_dir_t *cdir, category_t *cat)
{
	list_append(&cat->cat_list, &cdir->categories);
}

/** Get list of categories. */
int categ_dir_get_categories(categ_dir_t *cdir, category_id_t *id_buf,
    size_t buf_size, size_t *act_size)
{
	size_t act_cnt;
	size_t buf_cnt;

	assert(fibril_mutex_is_locked(&cdir->mutex));

	buf_cnt = buf_size / sizeof(category_id_t);

	act_cnt = list_count(&cdir->categories);
	*act_size = act_cnt * sizeof(category_id_t);

	if (buf_size % sizeof(category_id_t) != 0)
		return EINVAL;

	size_t pos = 0;
	list_foreach(cdir->categories, item) {
		category_t *cat =
		    list_get_instance(item, category_t, cat_list);

		if (pos < buf_cnt)
			id_buf[pos] = cat->id;
		pos++;
	}

	return EOK;
}


/** Initialize category structure. */
static void category_init(category_t *cat, const char *name)
{
	fibril_mutex_initialize(&cat->mutex);
	cat->name = str_dup(name);
	cat->id = loc_create_id();
	link_initialize(&cat->cat_list);
	list_initialize(&cat->svc_memb);
}

/** Allocate new category. */
category_t *category_new(const char *name)
{
	category_t *cat;

	cat = malloc(sizeof(category_t));
	if (cat == NULL)
		return NULL;

	category_init(cat, name);
	return cat;
}

/** Add service to category. */
int category_add_service(category_t *cat, loc_service_t *svc)
{
	assert(fibril_mutex_is_locked(&cat->mutex));
	assert(fibril_mutex_is_locked(&services_list_mutex));

	/* Verify that category does not contain this service yet. */
	list_foreach(cat->svc_memb, item) {
		svc_categ_t *memb = list_get_instance(item, svc_categ_t,
		    cat_link);
		if (memb->svc == svc) {
			return EEXIST;
		}
	}

	svc_categ_t *nmemb = malloc(sizeof(svc_categ_t));
	if (nmemb == NULL)
		return ENOMEM;

	nmemb->svc = svc;
	nmemb->cat = cat;

	list_append(&nmemb->cat_link, &cat->svc_memb);
	list_append(&nmemb->svc_link, &svc->cat_memb);

	return EOK;
}

/** Remove service from category. */
void category_remove_service(svc_categ_t *memb)
{
	assert(fibril_mutex_is_locked(&memb->cat->mutex));
	assert(fibril_mutex_is_locked(&services_list_mutex));

	list_remove(&memb->cat_link);
	list_remove(&memb->svc_link);

	free(memb);
}

/** Get category by ID. */
category_t *category_get(categ_dir_t *cdir, catid_t catid)
{
	assert(fibril_mutex_is_locked(&cdir->mutex));

	list_foreach(cdir->categories, item) {
		category_t *cat = list_get_instance(item, category_t,
		    cat_list);
		if (cat->id == catid)
			return cat;
	}

	return NULL;
}

/** Find category by name. */
category_t *category_find_by_name(categ_dir_t *cdir, const char *name)
{
	assert(fibril_mutex_is_locked(&cdir->mutex));

	list_foreach(cdir->categories, item) {
		category_t *cat = list_get_instance(item, category_t,
		    cat_list);
		if (str_cmp(cat->name, name) == 0)
			return cat;
	}

	return NULL;
}

/** Get list of services in category. */
int category_get_services(category_t *cat, service_id_t *id_buf,
    size_t buf_size, size_t *act_size)
{
	size_t act_cnt;
	size_t buf_cnt;

	assert(fibril_mutex_is_locked(&cat->mutex));

	buf_cnt = buf_size / sizeof(service_id_t);

	act_cnt = list_count(&cat->svc_memb);
	*act_size = act_cnt * sizeof(service_id_t);

	if (buf_size % sizeof(service_id_t) != 0)
		return EINVAL;

	size_t pos = 0;
	list_foreach(cat->svc_memb, item) {
		svc_categ_t *memb =
		    list_get_instance(item, svc_categ_t, cat_link);

		if (pos < buf_cnt)
			id_buf[pos] = memb->svc->id;
		pos++;
	}

	return EOK;
}

/**
 * @}
 */