#include <rblm.h>

#include <rblm-private.h>

VALUE lm_cMessageNode;

LmMessageNode *

rb_lm_message_node_from_ruby_object (VALUE obj)

{

	LmMessageNode *node;

	if (!rb_lm__is_kind_of (obj, lm_cMessageNode)) {

		rb_raise (rb_eTypeError, "not a LmMessageNode");

	}

	Data_Get_Struct (obj, LmMessageNode, node);

	return node;

}

void

msg_node_free (LmMessageNode *node)

{

	lm_message_node_unref (node);

}

VALUE

rb_lm_message_node_to_ruby_object (LmMessageNode *node)

{

	if (node) {

		lm_message_node_ref (node);

		return Data_Wrap_Struct (lm_cMessageNode, NULL, 

					 msg_node_free, node);

	} else {

		return Qnil;

	}

}

VALUE

msg_node_allocate (VALUE klass)

{

	return Data_Wrap_Struct (klass, NULL, msg_node_free, NULL);

}

VALUE

msg_node_get_name (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return rb_str_new2 (node->name);

}

VALUE

msg_node_get_value (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	if (lm_message_node_get_value (node)) {

		return rb_str_new2 (lm_message_node_get_value (node));

	} 

	return Qnil;

}

VALUE

msg_node_set_value (VALUE self, VALUE value)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);	

	char          *value_str = NULL;

	if (!rb_respond_to (value, rb_intern ("to_s"))) {

		rb_raise (rb_eArgError, "value should respond to to_s");

	} else {

		VALUE str_val = rb_funcall (value, rb_intern ("to_s"), 0);

		value_str = StringValuePtr (str_val);

	}

	lm_message_node_set_value (node, value_str);

}

VALUE

msg_node_add_child (int argc, VALUE *argv, VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	LmMessageNode *child;

	VALUE          name, value;

	char          *value_str = NULL;

	rb_scan_args (argc, argv, "11", &name, &value);

	if (!NIL_P (value)) {

		if (!rb_respond_to (value, rb_intern ("to_s"))) {

			rb_raise (rb_eArgError, "value should respond to to_s");

		} else {

			VALUE str_val = rb_funcall (value, rb_intern ("to_s"), 0);

			value_str = StringValuePtr (str_val);

		}

	}

	child = lm_message_node_add_child (node, StringValuePtr (name),

					   value_str);

	return LMMESSAGENODE2RVAL (child);

}

VALUE

msg_node_get_attribute (VALUE self, VALUE attr)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	if (lm_message_node_get_attribute (node, StringValuePtr (attr))) {

	    return rb_str_new2 (lm_message_node_get_attribute (node, 

							       StringValuePtr (attr)));

	}

	return Qnil;

}

VALUE

msg_node_set_attribute (VALUE self, VALUE attr, VALUE value)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	char          *value_str = NULL;

	if (!NIL_P (value)) {

		value_str = StringValuePtr (value);

	}

	lm_message_node_set_attribute (node, StringValuePtr (attr), value_str);

	return Qnil;

}

VALUE 

msg_node_get_child (VALUE self, VALUE name)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	LmMessageNode *child;

	child = lm_message_node_get_child (node, StringValuePtr (name));

	return LMMESSAGENODE2RVAL (child);

}

VALUE

msg_node_find_child (VALUE self, VALUE name)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	LmMessageNode *child;

	child = lm_message_node_find_child (node, StringValuePtr (name));

	return LMMESSAGENODE2RVAL (child);

}

VALUE

msg_node_get_is_raw_mode (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return GBOOL2RVAL (lm_message_node_get_raw_mode (node));

}

VALUE

msg_node_set_is_raw_mode (VALUE self, VALUE raw_mode)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	lm_message_node_set_raw_mode (node, RVAL2GBOOL (raw_mode));

	return Qnil;

}

VALUE

msg_node_to_string (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return rb_str_new2 (lm_message_node_to_string (node));

}

VALUE

msg_node_get_next (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return LMMESSAGENODE2RVAL (node->next);

}

VALUE

msg_node_get_prev (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return LMMESSAGENODE2RVAL (node->prev);

}

VALUE

msg_node_get_parent (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return LMMESSAGENODE2RVAL (node->parent);

}

VALUE 

msg_node_get_children (VALUE self)

{

	LmMessageNode *node = rb_lm_message_node_from_ruby_object (self);

	return LMMESSAGENODE2RVAL (node->children);

}

extern void 

Init_lm_message_node (VALUE lm_mLM)

{

	lm_cMessageNode = rb_define_class_under (lm_mLM, "MessageNode", 

						 rb_cObject);

	rb_define_alloc_func (lm_cMessageNode, msg_node_allocate);

	rb_define_method (lm_cMessageNode, "name", msg_node_get_name, 0);

	rb_define_method (lm_cMessageNode, "value", msg_node_get_value, 0);

	rb_define_method (lm_cMessageNode, "value=", msg_node_set_value, 1);

	rb_define_method (lm_cMessageNode, "add_child", msg_node_add_child, -1);

	rb_define_method (lm_cMessageNode, "get_attribute", msg_node_get_attribute, 1);

	rb_define_method (lm_cMessageNode, "set_attribute", msg_node_set_attribute, 2);

	rb_define_method (lm_cMessageNode, "get_child", msg_node_get_child, 1);

	rb_define_method (lm_cMessageNode, "find_child", msg_node_find_child, 1);

	rb_define_method (lm_cMessageNode, "raw_mode", msg_node_get_is_raw_mode, 0);

	rb_define_method (lm_cMessageNode, "raw_mode=", msg_node_set_is_raw_mode, 1);

	rb_define_method (lm_cMessageNode, "to_s", msg_node_to_string, 0);

	rb_define_method (lm_cMessageNode, "next", msg_node_get_next, 0);

	rb_define_method (lm_cMessageNode, "prev", msg_node_get_prev, 0);

	rb_define_method (lm_cMessageNode, "parent", msg_node_get_parent, 0);

	rb_define_method (lm_cMessageNode, "children", msg_node_get_children, 0);

}