tenderlove-nokogiri 0.0.0-x86-mswin32-60

data/ext/nokogiri/xml_node.c

@@ -0,0 +1,709 @@
+#include <xml_node.h>
+
+/*
+ * call-seq:
+ *  encode_special_chars(string)
+ *
+ * Encode any special characters in +string+
+ */
+static VALUE encode_special_chars(VALUE self, VALUE string)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  xmlChar * encoded = xmlEncodeSpecialChars(
+      node->doc,
+      (const xmlChar *)StringValuePtr(string)
+  );
+
+  VALUE encoded_str = rb_str_new2((const char *)encoded);
+  free(encoded);
+
+  return encoded_str;
+}
+
+/*
+ * call-seq:
+ *  internal_subset
+ *
+ * Get the internal subset
+ */
+static VALUE internal_subset(VALUE self)
+{
+  xmlNodePtr node;
+  xmlDocPtr doc;
+  Data_Get_Struct(self, xmlNode, node);
+
+  if(!node->doc) return Qnil;
+
+  doc = node->doc;
+
+  if(!doc->intSubset) return Qnil;
+
+  return Nokogiri_wrap_xml_node((xmlNodePtr)doc->intSubset);
+}
+
+/*
+ * call-seq:
+ *  dup
+ *
+ * Copy this node
+ */
+static VALUE duplicate_node(VALUE self)
+{
+  xmlNodePtr node, dup;
+  Data_Get_Struct(self, xmlNode, node);
+
+  dup = xmlCopyNode(node, 1);
+  if(dup == NULL) return Qnil;
+
+  return Nokogiri_wrap_xml_node(dup);
+}
+
+/*
+ * call-seq:
+ *  unlink
+ *
+ * Unlink this node from its current context.
+ */
+static VALUE unlink_node(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  xmlUnlinkNode(node);
+  Nokogiri_xml_node_owned_set(node);
+  return self;
+}
+
+/*
+ * call-seq:
+ *  blank?
+ *
+ * Is this node blank?
+ */
+static VALUE blank_eh(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  if(1 == xmlIsBlankNode(node))
+    return Qtrue;
+  return Qfalse;
+}
+
+/*
+ * call-seq:
+ *  next_sibling
+ *
+ * Returns the next sibling node
+ */
+static VALUE next_sibling(VALUE self)
+{
+  xmlNodePtr node, sibling;
+  Data_Get_Struct(self, xmlNode, node);
+
+  sibling = node->next;
+  if(!sibling) return Qnil;
+
+  return Nokogiri_wrap_xml_node(sibling) ;
+}
+
+/*
+ * call-seq:
+ *  previous_sibling
+ *
+ * Returns the previous sibling node
+ */
+static VALUE previous_sibling(VALUE self)
+{
+  xmlNodePtr node, sibling;
+  Data_Get_Struct(self, xmlNode, node);
+
+  sibling = node->prev;
+  if(!sibling) return Qnil;
+
+  return Nokogiri_wrap_xml_node(sibling);
+}
+
+/*
+ *  call-seq:
+ *    replace(new_node)
+ *
+ *  replace node with the new node in the document.
+ */
+static VALUE replace(VALUE self, VALUE _new_node)
+{
+  xmlNodePtr node, new_node;
+  Data_Get_Struct(self, xmlNode, node);
+  Data_Get_Struct(_new_node, xmlNode, new_node);
+
+  xmlReplaceNode(node, new_node);
+  Nokogiri_xml_node_owned_set(node);
+  Nokogiri_xml_node_owned_set(new_node);
+  return self ;
+}
+
+
+/*
+ * call-seq:
+ *  child
+ *
+ * Returns the child node
+ */
+static VALUE child(VALUE self)
+{
+  xmlNodePtr node, child;
+  Data_Get_Struct(self, xmlNode, node);
+
+  child = node->children;
+  if(!child) return Qnil;
+
+  return Nokogiri_wrap_xml_node(child);
+}
+
+/*
+ * call-seq:
+ *  key?(attribute)
+ *
+ * Returns true if +attribute+ is set
+ */
+static VALUE key_eh(VALUE self, VALUE attribute)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  if(xmlHasProp(node, (xmlChar *)StringValuePtr(attribute)))
+    return Qtrue;
+  return Qfalse;
+}
+
+/*
+ * call-seq:
+ *  []=(property, value)
+ *
+ * Set the +property+ to +value+
+ */
+static VALUE set(VALUE self, VALUE property, VALUE value)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  xmlSetProp(node, (xmlChar *)StringValuePtr(property),
+      (xmlChar *)StringValuePtr(value));
+
+  return value;
+}
+
+/*
+ *  call-seq:
+ *    remove_attribute(property)
+ *
+ *  remove the property +property+
+ */
+static VALUE remove_prop(VALUE self, VALUE property)
+{
+  xmlNodePtr node;
+  xmlAttrPtr attr ;
+  Data_Get_Struct(self, xmlNode, node);
+  attr = xmlHasProp(node, (xmlChar *)StringValuePtr(property));
+  if (attr) { xmlRemoveProp(attr); }
+  return Qnil;
+}
+
+/*
+ * call-seq:
+ *   get(attribute)
+ *
+ * Get the value for +attribute+
+ */
+static VALUE get(VALUE self, VALUE attribute)
+{
+  xmlNodePtr node;
+  xmlChar* propstr ;
+  VALUE rval ;
+  Data_Get_Struct(self, xmlNode, node);
+  propstr = xmlGetProp(node, (xmlChar *)StringValuePtr(attribute));
+  rval = rb_str_new2((char *)propstr) ;
+  xmlFree(propstr);
+  return rval ;
+}
+
+/*
+ *  call-seq:
+ *    attributes()
+ *
+ *  returns a hash containing the node's attributes.
+ */
+static VALUE attributes(VALUE self)
+{
+    /* this code in the mode of xmlHasProp() */
+    xmlNodePtr node ;
+    VALUE attr ;
+
+    attr = rb_hash_new() ;
+    Data_Get_Struct(self, xmlNode, node);
+
+    Nokogiri_xml_node_properties(node, attr);
+
+    return attr ;
+}
+
+/*
+ *  call-seq:
+ *    namespaces()
+ *
+ *  returns a hash containing the node's namespaces.
+ */
+static VALUE namespaces(VALUE self)
+{
+    /* this code in the mode of xmlHasProp() */
+    xmlNodePtr node ;
+    VALUE attr ;
+
+    attr = rb_hash_new() ;
+    Data_Get_Struct(self, xmlNode, node);
+
+    Nokogiri_xml_node_namespaces(node, attr);
+
+    return attr ;
+}
+
+/*
+ * call-seq:
+ *  type
+ *
+ * Get the type for this node
+ */
+static VALUE type(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  return INT2NUM((int)node->type);
+}
+
+/*
+ * call-seq:
+ *  content=
+ *
+ * Set the content for this Node
+ */
+static VALUE set_content(VALUE self, VALUE content)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  xmlNodeSetContent(node, (xmlChar *)StringValuePtr(content));
+  return content;
+}
+
+/*
+ * call-seq:
+ *  content
+ *
+ * Returns the content for this Node
+ */
+static VALUE get_content(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+
+  xmlChar * content = xmlNodeGetContent(node);
+  if(content) {
+    VALUE rval = rb_str_new2((char *)content);
+    xmlFree(content);
+    return rval;
+  }
+  return Qnil;
+}
+
+/*
+ * call-seq:
+ *  parent=(parent_node)
+ *
+ * Set the parent Node for this Node
+ */
+static VALUE set_parent(VALUE self, VALUE parent_node)
+{
+  xmlNodePtr node, parent;
+  Data_Get_Struct(self, xmlNode, node);
+  Data_Get_Struct(parent_node, xmlNode, parent);
+
+  xmlAddChild(parent, node);
+  Nokogiri_xml_node_owned_set(node);
+  return parent_node;
+}
+
+/*
+ * call-seq:
+ *  parent
+ *
+ * Get the parent Node for this Node
+ */
+static VALUE get_parent(VALUE self)
+{
+  xmlNodePtr node, parent;
+  Data_Get_Struct(self, xmlNode, node);
+
+  parent = node->parent;
+  if(!parent) return Qnil;
+
+  return Nokogiri_wrap_xml_node(parent) ;
+}
+
+/*
+ * call-seq:
+ *  name=(new_name)
+ *
+ * Set the name for this Node
+ */
+static VALUE set_name(VALUE self, VALUE new_name)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  xmlNodeSetName(node, (xmlChar*)StringValuePtr(new_name));
+  return new_name;
+}
+
+/*
+ * call-seq:
+ *  name
+ *
+ * Returns the name for this Node
+ */
+static VALUE get_name(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+  return rb_str_new2((const char *)node->name);
+}
+
+/*
+ * call-seq:
+ *  path
+ *
+ * Returns the path associated with this Node
+ */
+static VALUE path(VALUE self)
+{
+  xmlNodePtr node;
+  xmlChar *path ;
+  VALUE rval ;
+  Data_Get_Struct(self, xmlNode, node);
+  
+  path = xmlGetNodePath(node);
+  rval = rb_str_new2((char *)path);
+  xmlFree(path);
+  return rval ;
+}
+
+/*
+ * call-seq:
+ *  document
+ *
+ * Returns the Nokogiri::XML::Document associated with this Node
+ */
+static VALUE document(VALUE self)
+{
+  xmlNodePtr node;
+  Data_Get_Struct(self, xmlNode, node);
+
+  if(!node->doc) return Qnil;
+  return (VALUE)node->doc->_private;
+}
+
+/*
+ *  call-seq:
+ *    add_next_sibling(node)
+ *
+ *  Insert +node+ after this node (as a sibling).
+ */
+static VALUE add_next_sibling(VALUE self, VALUE rb_node)
+{
+  xmlNodePtr node, new_sibling;
+  Data_Get_Struct(self, xmlNode, node);
+  Data_Get_Struct(rb_node, xmlNode, new_sibling);
+  xmlAddNextSibling(node, new_sibling);
+
+  rb_funcall(rb_node, rb_intern("decorate!"), 0);
+  Nokogiri_xml_node_owned_set(new_sibling);
+
+  return rb_node;
+}
+
+/*
+ * call-seq:
+ *  add_previous_sibling(node)
+ *
+ * Insert +node+ before this node (as a sibling).
+ */
+static VALUE add_previous_sibling(VALUE self, VALUE rb_node)
+{
+  xmlNodePtr node, new_sibling;
+  Data_Get_Struct(self, xmlNode, node);
+  Data_Get_Struct(rb_node, xmlNode, new_sibling);
+  xmlAddPrevSibling(node, new_sibling);
+
+  rb_funcall(rb_node, rb_intern("decorate!"), 0);
+  Nokogiri_xml_node_owned_set(new_sibling);
+
+  return rb_node;
+}
+
+/*
+ * call-seq:
+ *  to_xml
+ *
+ * Returns this node as XML
+ */
+static VALUE to_xml(VALUE self)
+{
+  xmlBufferPtr buf ;
+  xmlNodePtr node ;
+  VALUE xml ;
+
+  Data_Get_Struct(self, xmlNode, node);
+
+  buf = xmlBufferCreate() ;
+  xmlNodeDump(buf, node->doc, node, 2, 1);
+  xml = rb_str_new2((char*)buf->content);
+  xmlBufferFree(buf);
+  return xml ;
+}
+
+
+/*
+ * call-seq:
+ *   new(name)
+ *
+ * Create a new node with +name+
+ */
+static VALUE new(VALUE klass, VALUE name)
+{
+  xmlNodePtr node = xmlNewNode(NULL, (xmlChar *)StringValuePtr(name));
+  VALUE rb_node = Nokogiri_wrap_xml_node(node) ;
+
+  if(rb_block_given_p()) rb_yield(rb_node);
+
+  Nokogiri_xml_node_owned_set(node);
+
+  return rb_node;
+}
+
+
+/*
+ * call-seq:
+ *   new_from_str(string)
+ *
+ * Create a new node by parsing +string+
+ */
+static VALUE new_from_str(VALUE klass, VALUE xml)
+{
+    /*
+     *  I couldn't find a more efficient way to do this. So we create a new
+     *  document and copy (recursively) the root node.
+     */
+    VALUE rb_doc ;
+    xmlDocPtr doc ;
+    xmlNodePtr node ;
+
+    rb_doc = rb_funcall(cNokogiriXmlDocument, rb_intern("read_memory"), 4,
+                        xml, Qnil, Qnil, INT2NUM(0));
+    Data_Get_Struct(rb_doc, xmlDoc, doc);
+    node = xmlCopyNode(xmlDocGetRootElement(doc), 1); /* 1 => recursive */
+    return Nokogiri_wrap_xml_node(node);
+}
+
+static void deallocate(xmlNodePtr node)
+{
+  if (! Nokogiri_xml_node_owned_get(node)) {
+    NOKOGIRI_DEBUG_START_NODE(node);
+    xmlFreeNode(node);
+    NOKOGIRI_DEBUG_END(node);
+  }
+}
+
+static void gc_mark_node(xmlNodePtr node)
+{
+  xmlNodePtr child ;
+  /* mark document */
+  if (node && node->doc && node->doc->_private)
+    rb_gc_mark((VALUE)node->doc->_private);
+  /* mark parent node */
+  if (node && node->parent && node->parent->_private)
+    rb_gc_mark((VALUE)node->parent->_private);
+  /* mark children nodes */
+  for (child = node->children ; child ; child = child->next) {
+    if (child->_private)
+      rb_gc_mark((VALUE)child->_private);
+  }
+  /* mark sibling nodes */
+  if (node->next && node->next->_private)
+    rb_gc_mark((VALUE)node->next->_private) ;
+  if (node->prev && node->prev->_private)
+    rb_gc_mark((VALUE)node->prev->_private) ;
+}
+
+VALUE Nokogiri_wrap_xml_node(xmlNodePtr node)
+{
+  if (node->_private)
+    return (VALUE)node->_private ;
+
+  VALUE rb_node = Qnil;
+  
+  switch(node->type)
+  {
+    VALUE klass;
+
+    case XML_TEXT_NODE:
+      klass = rb_const_get(mNokogiriXml, rb_intern("Text"));
+      rb_node = Data_Wrap_Struct(klass, gc_mark_node, deallocate, node) ;
+      break;
+    case XML_ELEMENT_NODE:
+      klass = rb_const_get(mNokogiriXml, rb_intern("Element"));
+      rb_node = Data_Wrap_Struct(klass, gc_mark_node, deallocate, node) ;
+      break;
+    case XML_ENTITY_DECL:
+      klass = rb_const_get(mNokogiriXml, rb_intern("EntityDeclaration"));
+      rb_node = Data_Wrap_Struct(klass, gc_mark_node, deallocate, node) ;
+      break;
+    case XML_CDATA_SECTION_NODE:
+      klass = rb_const_get(mNokogiriXml, rb_intern("CDATA"));
+      rb_node = Data_Wrap_Struct(klass, gc_mark_node, deallocate, node) ;
+      break;
+    case XML_DTD_NODE:
+      klass = rb_const_get(mNokogiriXml, rb_intern("DTD"));
+      rb_node = Data_Wrap_Struct(klass, gc_mark_node, deallocate, node) ;
+      break;
+    default:
+      rb_node = Data_Wrap_Struct(cNokogiriXmlNode, gc_mark_node, deallocate, node) ;
+  }
+
+  node->_private = (void*)rb_node ;
+  rb_funcall(rb_node, rb_intern("decorate!"), 0);
+  Nokogiri_xml_node_owned_set(node);
+  return rb_node ;
+}
+
+
+void Nokogiri_xml_node_properties(xmlNodePtr node, VALUE attr_hash)
+{
+  xmlAttrPtr prop;
+  xmlChar* propstr ;
+  prop = node->properties ;
+  while (prop != NULL) {
+    propstr = xmlGetProp(node, prop->name) ;
+    rb_hash_aset(attr_hash, rb_str_new2((const char*)prop->name),
+                 rb_str_new2((char*)propstr));
+    xmlFree(propstr);
+    prop = prop->next ;
+  }
+}
+
+
+#define XMLNS_PREFIX "xmlns"
+#define XMLNS_PREFIX_LEN 6 /* including either colon or \0 */
+#define XMLNS_BUFFER_LEN 128
+void Nokogiri_xml_node_namespaces(xmlNodePtr node, VALUE attr_hash)
+{
+  xmlNsPtr ns;
+  static char buffer[XMLNS_BUFFER_LEN] ;
+  char *key ;
+  size_t keylen ;
+
+  if (node->type != XML_ELEMENT_NODE) return ;
+
+  ns = node->nsDef;
+  while (ns != NULL) {
+
+    keylen = XMLNS_PREFIX_LEN + (ns->prefix ? (strlen((const char*)ns->prefix) + 1) : 0) ;
+    if (keylen > XMLNS_BUFFER_LEN) {
+      key = (char*)malloc(keylen) ;
+    } else {
+      key = buffer ;
+    }
+
+    if (ns->prefix) {
+      sprintf(key, "%s:%s", XMLNS_PREFIX, ns->prefix);
+    } else {
+      sprintf(key, "%s", XMLNS_PREFIX);
+    }
+
+    rb_hash_aset(attr_hash, rb_str_new2(key), rb_str_new2((const char*)ns->href)) ;
+    if (key != buffer) {
+      free(key);
+    }
+    ns = ns->next ;
+  }
+}
+
+
+void Nokogiri_xml_node_owned_set(xmlNodePtr node)
+{
+  VALUE hash = rb_cvar_get(cNokogiriXmlNode, rb_intern("@@owned"));
+  rb_hash_aset(hash, INT2NUM((long)node), node->parent ? Qtrue : Qfalse) ;
+}
+
+int Nokogiri_xml_node_owned_get(xmlNodePtr node)
+{
+  VALUE hash = rb_cvar_get(cNokogiriXmlNode, rb_intern("@@owned"));
+  VALUE q = rb_hash_aref(hash, INT2NUM((long)node)) ;
+  return q == Qtrue ? Qtrue : Qfalse ;
+}
+
+/*
+ * call-seq:
+ *   owned?
+ *
+ * Is this node owned by a document?
+ */
+static VALUE owned_eh(VALUE self)
+{
+  xmlNodePtr node ;
+  VALUE hash = rb_cvar_get(cNokogiriXmlNode, rb_intern("@@owned"));
+  Data_Get_Struct(self, xmlNode, node);
+  return rb_hash_aref(hash, INT2NUM((long)node)) == Qtrue ? Qtrue : Qfalse ;
+}
+
+
+VALUE cNokogiriXmlNode ;
+void init_xml_node()
+{
+  /*
+   * HACK.  This is so that rdoc will work with this C file.
+   */
+  /*
+  VALUE nokogiri = rb_define_module("Nokogiri");
+  VALUE xml = rb_define_module_under(nokogiri, "XML");
+  VALUE klass = rb_define_class_under(xml, "Node", rb_cObject);
+  */
+
+  VALUE klass = cNokogiriXmlNode = rb_const_get(mNokogiriXml, rb_intern("Node"));
+
+  rb_define_singleton_method(klass, "new", new, 1);
+  rb_define_singleton_method(klass, "new_from_str", new_from_str, 1);
+
+  rb_define_method(klass, "document", document, 0);
+  rb_define_method(klass, "name", get_name, 0);
+  rb_define_method(klass, "name=", set_name, 1);
+  rb_define_method(klass, "parent=", set_parent, 1);
+  rb_define_method(klass, "parent", get_parent, 0);
+  rb_define_method(klass, "child", child, 0);
+  rb_define_method(klass, "next_sibling", next_sibling, 0);
+  rb_define_method(klass, "previous_sibling", previous_sibling, 0);
+  rb_define_method(klass, "replace", replace, 1);
+  rb_define_method(klass, "type", type, 0);
+  rb_define_method(klass, "content", get_content, 0);
+  rb_define_method(klass, "path", path, 0);
+  rb_define_method(klass, "key?", key_eh, 1);
+  rb_define_method(klass, "blank?", blank_eh, 0);
+  rb_define_method(klass, "[]=", set, 2);
+  rb_define_method(klass, "remove_attribute", remove_prop, 1);
+  rb_define_method(klass, "attributes", attributes, 0);
+  rb_define_method(klass, "namespaces", namespaces, 0);
+  rb_define_method(klass, "add_previous_sibling", add_previous_sibling, 1);
+  rb_define_method(klass, "add_next_sibling", add_next_sibling, 1);
+  rb_define_method(klass, "encode_special_chars", encode_special_chars, 1);
+  rb_define_method(klass, "to_xml", to_xml, 0);
+  rb_define_method(klass, "dup", duplicate_node, 0);
+  rb_define_method(klass, "unlink", unlink_node, 0);
+  rb_define_method(klass, "internal_subset", internal_subset, 0);
+
+  rb_define_private_method(klass, "native_content=", set_content, 1);
+  rb_define_private_method(klass, "get", get, 1);
+  rb_define_private_method(klass, "owned?", owned_eh, 0);
+}

data/ext/nokogiri/xml_node.h

@@ -0,0 +1,15 @@
+#ifndef NOKOGIRI_XML_NODE
+#define NOKOGIRI_XML_NODE
+
+#include <native.h>
+
+void init_xml_node();
+VALUE Nokogiri_wrap_xml_node(xmlNodePtr root);
+
+extern VALUE cNokogiriXmlNode ;
+VALUE Nokogiri_wrap_xml_node(xmlNodePtr node) ;
+void Nokogiri_xml_node_properties(xmlNodePtr node, VALUE attr_hash) ;
+void Nokogiri_xml_node_namespaces(xmlNodePtr node, VALUE attr_hash) ;
+int Nokogiri_xml_node_owned_get(xmlNodePtr node) ;
+void Nokogiri_xml_node_owned_set(xmlNodePtr node) ;
+#endif