#include <socks/object.h>
#include <socks/queue.h>
#include <socks/locks.h>

#define HAS_OP(obj, opname) ((obj)->ob_type->ob_ops.opname)

static queue_t object_types;
static spin_lock_t object_types_lock = SPIN_LOCK_INIT;

kern_status_t object_bootstrap(void)
{
	init_set_objects();
	init_global_namespace();
	return KERN_OK;
}

kern_status_t object_type_register(object_type_t *p)
{
	unsigned long flags;
	spin_lock_irqsave(&object_types_lock, &flags);
	queue_push_back(&object_types, &p->ob_list);
	spin_unlock_irqrestore(&object_types_lock, flags);

	p->ob_cache.c_name = p->ob_name;
	p->ob_cache.c_obj_size = sizeof(object_t) + p->ob_size;
	p->ob_cache.c_page_order = VM_PAGE_16K;

	vm_cache_init(&p->ob_cache);
	p->ob_flags |= OBJTYPE_INIT;

	return KERN_OK;
}

kern_status_t object_type_unregister(object_type_t *p)
{
	unsigned long flags;
	spin_lock_irqsave(&object_types_lock, &flags);
	queue_delete(&object_types, &p->ob_list);
	spin_unlock_irqrestore(&object_types_lock, flags);

	return KERN_OK;
}

object_t *object_create(object_type_t *type)
{
	if (!(type->ob_flags & OBJTYPE_INIT)) {
		return NULL;
	}

	vm_cache_t *cache = &type->ob_cache;
	object_t *obj = vm_cache_alloc(cache, 0);
	if (!obj) {
		return NULL;
	}

	obj->ob_type = type;
	obj->ob_lock = SPIN_LOCK_INIT;
	obj->ob_magic = OBJECT_MAGIC;
	obj->ob_refcount = 1;
	obj->ob_handles = 0;

	return obj;
}

object_t *object_ref(object_t *obj)
{
	obj->ob_refcount++;
	return obj;
}

void object_deref(object_t *obj)
{
	unsigned long flags;
	spin_lock_irqsave(&obj->ob_lock, &flags);

	if (obj->ob_refcount == 0) {
		spin_unlock_irqrestore(&obj->ob_lock, flags);
		return;
	}

	obj->ob_refcount--;

	if (obj->ob_refcount > 0) {
		spin_unlock_irqrestore(&obj->ob_lock, flags);
		return;
	}

	if (HAS_OP(obj, destroy)) {
		obj->ob_type->ob_ops.destroy(obj);
	}

	vm_cache_free(&obj->ob_type->ob_cache, obj);
}

void object_lock(object_t *obj, unsigned long *flags)
{
	spin_lock_irqsave(&obj->ob_lock, flags);
}

void object_unlock(object_t *obj, unsigned long flags)
{
	spin_unlock_irqrestore(&obj->ob_lock, flags);
}

void *object_data(object_t *obj)
{
	return (char *)obj + sizeof *obj;
}

object_t *object_header(void *p)
{
	object_t *obj = (object_t *)((char *)p - sizeof *obj);
	if (obj->ob_magic != OBJECT_MAGIC) {
		return NULL;
	}

	return obj;
}

kern_status_t object_get_child_named(object_t *obj, const char *name, object_t **out)
{
	kern_status_t status = KERN_UNSUPPORTED;

	if (HAS_OP(obj, get_named)) {
		status = obj->ob_type->ob_ops.get_named(obj, name, out);
	}

	return status;
}

kern_status_t object_get_child_at(object_t *obj, size_t at, object_t **out)
{
	kern_status_t status = KERN_UNSUPPORTED;

	if (HAS_OP(obj, get_at)) {
		status = obj->ob_type->ob_ops.get_at(obj, at, out);
	}

	return status;
}

kern_status_t object_query_name(object_t *obj, char name[OBJECT_NAME_MAX])
{
	if (HAS_OP(obj, query_name)) {
		return obj->ob_type->ob_ops.query_name(obj, name);
	}

	return KERN_UNSUPPORTED;
}