nokogiri 1.1.1-java

data/ext/nokogiri/xml_text.h

@@ -0,0 +1,9 @@
+#ifndef NOKOGIRI_XML_TEXT
+#define NOKOGIRI_XML_TEXT
+
+#include <native.h>
+
+void init_xml_text();
+
+extern VALUE cNokogiriXmlText ;
+#endif

data/ext/nokogiri/xml_xpath.c

@@ -0,0 +1,53 @@
+#include <xml_xpath.h>
+
+static void deallocate(xmlXPathObjectPtr xpath)
+{
+  NOKOGIRI_DEBUG_START(xpath);
+  xmlXPathFreeNodeSetList(xpath); // despite the name, this frees the xpath but not the contained node set
+  NOKOGIRI_DEBUG_END(xpath);
+}
+
+VALUE Nokogiri_wrap_xml_xpath(xmlXPathObjectPtr xpath)
+{
+  return Data_Wrap_Struct(cNokogiriXmlXpath, 0, deallocate, xpath);
+}
+
+/*
+ * call-seq:
+ *  node_set
+ *
+ * Fetch the node set associated with this xpath context.
+ */
+static VALUE node_set(VALUE self)
+{
+  xmlXPathObjectPtr xpath;
+  Data_Get_Struct(self, xmlXPathObject, xpath);
+
+  VALUE node_set = Qnil;
+
+  if (xpath->nodesetval)
+    node_set = Nokogiri_wrap_xml_node_set(xpath->nodesetval);
+
+  if(Qnil == node_set)
+    node_set = Nokogiri_wrap_xml_node_set(xmlXPathNodeSetCreate(NULL));
+
+  rb_funcall(node_set, rb_intern("document="), 1, rb_iv_get(self, "@document"));
+
+  return node_set;
+}
+
+VALUE cNokogiriXmlXpath;
+void init_xml_xpath(void)
+{
+  VALUE module = rb_define_module("Nokogiri");
+  VALUE xml = rb_define_module_under(module, "XML");
+
+  /*
+   * This class wraps an XPath object and should only be instantiated from
+   * XPathContext.
+   */
+  VALUE klass = rb_define_class_under(xml, "XPath", rb_cObject);
+
+  cNokogiriXmlXpath = klass;
+  rb_define_method(klass, "node_set", node_set, 0);
+}

data/ext/nokogiri/xml_xpath.h

@@ -0,0 +1,11 @@
+#ifndef NOKOGIRI_XML_XPATH
+#define NOKOGIRI_XML_XPATH
+
+#include <native.h>
+
+void init_xml_xpath();
+VALUE Nokogiri_wrap_xml_xpath(xmlXPathObjectPtr xpath);
+
+extern VALUE cNokogiriXmlXpath;
+#endif
+

data/ext/nokogiri/xml_xpath_context.c

@@ -0,0 +1,214 @@
+#include <xml_xpath_context.h>
+
+static void deallocate(xmlXPathContextPtr ctx)
+{
+  NOKOGIRI_DEBUG_START(ctx);
+  xmlXPathFreeContext(ctx);
+  NOKOGIRI_DEBUG_END(ctx);
+}
+
+/*
+ * call-seq:
+ *  register_ns(prefix, uri)
+ *
+ * Register the namespace with +prefix+ and +uri+.
+ */
+static VALUE register_ns(VALUE self, VALUE prefix, VALUE uri)
+{
+  xmlXPathContextPtr ctx;
+  Data_Get_Struct(self, xmlXPathContext, ctx);
+
+  xmlXPathRegisterNs( ctx,
+                      (const xmlChar *)StringValuePtr(prefix),
+                      (const xmlChar *)StringValuePtr(uri)
+  );
+  return self;
+}
+
+static void ruby_funcall(xmlXPathParserContextPtr ctx, int nargs)
+{
+  VALUE xpath_handler = Qnil;
+  xmlXPathObjectPtr obj;
+
+  assert(ctx);
+  assert(ctx->context);
+  assert(ctx->context->userData);
+  assert(ctx->context->doc);
+  assert(ctx->context->doc->_private);
+
+  xpath_handler = (VALUE)(ctx->context->userData);
+
+  VALUE * argv = (VALUE *)calloc((unsigned int)nargs, sizeof(VALUE));
+  VALUE doc = (VALUE)ctx->context->doc->_private;
+
+  int i = nargs - 1;
+  do {
+    obj = valuePop(ctx);
+    switch(obj->type) {
+      case XPATH_STRING:
+        argv[i] = rb_str_new2((char *)obj->stringval);
+        break;
+      case XPATH_BOOLEAN:
+        argv[i] = obj->boolval == 1 ? Qtrue : Qfalse;
+        break;
+      case XPATH_NUMBER:
+        argv[i] = rb_float_new(obj->floatval);
+        break;
+      case XPATH_NODESET:
+        argv[i] = Nokogiri_wrap_xml_node_set(obj->nodesetval);
+        break;
+      default:
+        argv[i] = rb_str_new2((char *)xmlXPathCastToString(obj));
+    }
+    xmlXPathFreeNodeSetList(obj);
+  } while(i-- > 0);
+
+  VALUE result = rb_funcall2(
+      xpath_handler,
+      rb_intern((const char *)ctx->context->function),
+      nargs,
+      argv
+  );
+  free(argv);
+
+  VALUE node_set = Qnil;
+  xmlNodeSetPtr xml_node_set = NULL;
+
+  switch(TYPE(result)) {
+    case T_FLOAT:
+    case T_BIGNUM:
+    case T_FIXNUM:
+      xmlXPathReturnNumber(ctx, NUM2DBL(result));
+      break;
+    case T_STRING:
+      xmlXPathReturnString(
+          ctx,
+          (xmlChar *)xmlXPathWrapCString(StringValuePtr(result))
+      );
+      break;
+    case T_TRUE:
+      xmlXPathReturnTrue(ctx);
+      break;
+    case T_FALSE:
+      xmlXPathReturnFalse(ctx);
+      break;
+    case T_NIL:
+      break;
+    case T_ARRAY:
+      node_set = rb_funcall(
+          cNokogiriXmlNodeSet,
+          rb_intern("new"),
+          2,
+          doc,
+          result
+      );
+      Data_Get_Struct(node_set, xmlNodeSet, xml_node_set);
+      xmlXPathReturnNodeSet(ctx, xmlXPathNodeSetMerge(NULL, xml_node_set));
+      break;
+    case T_DATA:
+      if(rb_funcall(result, rb_intern("is_a?"), 1, cNokogiriXmlNodeSet)) {
+        Data_Get_Struct(result, xmlNodeSet, xml_node_set);
+        // Copy the node set, otherwise it will get GC'd.
+        xmlXPathReturnNodeSet(ctx, xmlXPathNodeSetMerge(NULL, xml_node_set));
+        break;
+      }
+    default:
+      rb_raise(rb_eRuntimeError, "Invalid return type");
+  }
+}
+
+static xmlXPathFunction lookup( void *ctx,
+                                const xmlChar * name,
+                                const xmlChar* ns_uri )
+{
+  VALUE xpath_handler = (VALUE)ctx;
+  if(rb_respond_to(xpath_handler, rb_intern((const char *)name)))
+    return ruby_funcall;
+
+  return NULL;
+}
+
+/*
+ * call-seq:
+ *  evaluate(search_path)
+ *
+ * Evaluate the +search_path+ returning an XML::XPath object.
+ */
+static VALUE evaluate(int argc, VALUE *argv, VALUE self)
+{
+  VALUE search_path, xpath_handler;
+  xmlXPathContextPtr ctx;
+  Data_Get_Struct(self, xmlXPathContext, ctx);
+
+  if(rb_scan_args(argc, argv, "11", &search_path, &xpath_handler) == 1)
+    xpath_handler = Qnil;
+
+  xmlChar* query = (xmlChar *)StringValuePtr(search_path);
+
+  if(Qnil != xpath_handler) {
+    // FIXME: not sure if this is the correct place to shove private data.
+    ctx->userData = (void *)xpath_handler;
+    xmlXPathRegisterFuncLookup(ctx, lookup, (void *)xpath_handler);
+  }
+
+  xmlXPathObjectPtr xpath = xmlXPathEvalExpression(query, ctx);
+  if(xpath == NULL) {
+    VALUE xpath = rb_const_get(mNokogiriXml, rb_intern("XPath"));
+    VALUE error = rb_const_get(xpath, rb_intern("SyntaxError"));
+    rb_raise(error, "Couldn't evaluate expression '%s'", query);
+  }
+
+  VALUE xpath_object = Nokogiri_wrap_xml_xpath(xpath);
+
+  assert(ctx->doc);
+  assert(ctx->doc->_private);
+
+  rb_funcall( xpath_object,
+              rb_intern("document="),
+              1,
+              (VALUE)ctx->doc->_private
+    );
+  return xpath_object;
+}
+
+/*
+ * call-seq:
+ *  new(node)
+ *
+ * Create a new XPathContext with +node+ as the reference point.
+ */
+static VALUE new(VALUE klass, VALUE nodeobj)
+{
+  xmlXPathInit();
+
+  xmlNodePtr node ;
+  Data_Get_Struct(nodeobj, xmlNode, node);
+
+  xmlXPathContextPtr ctx = xmlXPathNewContext(node->doc);
+  ctx->node = node;
+  VALUE self = Data_Wrap_Struct(klass, 0, deallocate, ctx);
+  //rb_iv_set(self, "@xpath_handler", Qnil);
+  return self;
+}
+
+VALUE cNokogiriXmlXpathContext;
+void init_xml_xpath_context(void)
+{
+  VALUE module = rb_define_module("Nokogiri");
+
+  /*
+   * Nokogiri::XML
+   */
+  VALUE xml = rb_define_module_under(module, "XML");
+
+  /*
+   * XPathContext is the entry point for searching a Document by using XPath.
+   */
+  VALUE klass = rb_define_class_under(xml, "XPathContext", rb_cObject);
+
+  cNokogiriXmlXpathContext = klass;
+
+  rb_define_singleton_method(klass, "new", new, 1);
+  rb_define_method(klass, "evaluate", evaluate, -1);
+  rb_define_method(klass, "register_ns", register_ns, 2);
+}

data/ext/nokogiri/xml_xpath_context.h

@@ -0,0 +1,9 @@
+#ifndef NOKOGIRI_XML_XPATH_CONTEXT
+#define NOKOGIRI_XML_XPATH_CONTEXT
+
+#include <native.h>
+
+void init_xml_xpath_context();
+
+extern VALUE cNokogiriXmlXpathContext;
+#endif

data/ext/nokogiri/xslt_stylesheet.c

@@ -0,0 +1,123 @@
+#include <xslt_stylesheet.h>
+
+#include <libxslt/xsltInternals.h>
+#include <libxslt/xsltutils.h>
+#include <libxslt/transform.h>
+#include <libexslt/exslt.h>
+
+static void dealloc(xsltStylesheetPtr doc)
+{
+    NOKOGIRI_DEBUG_START(doc);
+    xsltFreeStylesheet(doc); // commented out for now.
+    NOKOGIRI_DEBUG_END(doc);
+}
+
+/*
+ * call-seq:
+ *   parse_stylesheet_doc(document)
+ *
+ * Parse a stylesheet from +document+.
+ */
+static VALUE parse_stylesheet_doc(VALUE klass, VALUE xmldocobj)
+{
+    xmlDocPtr xml ;
+    xsltStylesheetPtr ss ;
+    Data_Get_Struct(xmldocobj, xmlDoc, xml);
+    exsltRegisterAll();
+    ss = xsltParseStylesheetDoc(xmlCopyDoc(xml, 1)); /* 1 => recursive */
+    return Data_Wrap_Struct(klass, NULL, dealloc, ss);
+}
+
+
+/*
+ * call-seq:
+ *   serialize(document)
+ *
+ * Serialize +document+ to an xml string.
+ */
+static VALUE serialize(VALUE self, VALUE xmlobj)
+{
+    xmlDocPtr xml ;
+    xsltStylesheetPtr ss ;
+    xmlChar* doc_ptr ;
+    int doc_len ;
+    VALUE rval ;
+
+    Data_Get_Struct(xmlobj, xmlDoc, xml);
+    Data_Get_Struct(self, xsltStylesheet, ss);
+    xsltSaveResultToString(&doc_ptr, &doc_len, xml, ss);
+    rval = rb_str_new((char*)doc_ptr, doc_len);
+    xmlFree(doc_ptr);
+    return rval ;
+}
+/*
+ *  call-seq:
+ *    transform(document, params)
+ *
+ *  Apply an XSLT stylesheet to an XML::Document.
+ *  +params+ is an array of strings used as XSLT parameters.
+ *  returns Nokogiri::XML::Document
+ */
+static VALUE transform(int argc, VALUE* argv, VALUE self)
+{
+    VALUE xmldoc, paramobj ;
+    xmlDocPtr xml ;
+    xmlDocPtr result ;
+    xsltStylesheetPtr ss ;
+    const char** params ;
+    int param_len, j ;
+
+    rb_scan_args(argc, argv, "11", &xmldoc, &paramobj);
+    if (paramobj == Qnil) { paramobj = rb_ary_new2(0) ; }
+
+    Data_Get_Struct(xmldoc, xmlDoc, xml);
+    Data_Get_Struct(self, xsltStylesheet, ss);
+
+    param_len = NUM2INT(rb_funcall(paramobj, rb_intern("length"), 0));
+    params = calloc((size_t)param_len+1, sizeof(char*));
+    for (j = 0 ; j < param_len ; j++) {
+      VALUE entry = rb_ary_entry(paramobj, j);
+      const char * ptr = StringValuePtr(entry);
+      params[j] = ptr;
+    }
+    params[param_len] = 0 ;
+
+    result = xsltApplyStylesheet(ss, xml, params);
+    free(params);
+
+    return Nokogiri_wrap_xml_document(0, result) ;
+}
+
+/*
+ *  call-seq:
+ *    apply_to(document, params)
+ *
+ *  Apply an XSLT stylesheet to an XML::Document.
+ *  +params+ is an array of strings used as XSLT parameters.
+ *  returns serialized document
+ */
+static VALUE apply_to(int argc, VALUE* argv, VALUE self)
+{
+  return rb_funcall(self, rb_intern("serialize"), 1, 
+                    transform(argc, argv, self));
+}
+
+VALUE cNokogiriXsltStylesheet ;
+void init_xslt_stylesheet()
+{
+  /*
+   * HACK.  This is so that rdoc will work with this C file.
+   */
+  /*
+  VALUE nokogiri = rb_define_module("Nokogiri");
+  VALUE xslt = rb_define_module_under(nokogiri, "XSLT");
+  VALUE klass = rb_define_class_under(xslt, "Stylesheet", rb_cObject);
+  */
+
+  VALUE klass = cNokogiriXsltStylesheet = rb_const_get(mNokogiriXslt, rb_intern("Stylesheet"));
+    
+  rb_define_singleton_method(klass, "parse_stylesheet_doc", parse_stylesheet_doc, 1);
+  rb_define_method(klass, "serialize", serialize, 1);
+  rb_define_method(klass, "apply_to", apply_to, -1);
+  rb_define_method(klass, "transform", transform, -1);
+}

data/ext/nokogiri/xslt_stylesheet.h

@@ -0,0 +1,9 @@
+#ifndef NOKOGIRI_XSLT_STYLESHEET
+#define NOKOGIRI_XSLT_STYLESHEET
+
+#include <native.h>
+
+void init_xslt_stylesheet();
+
+extern VALUE cNokogiriXsltStylesheet ;
+#endif

data/lib/action-nokogiri.rb

@@ -0,0 +1,30 @@
+require 'nokogiri'
+
+#
+#  to use this in your Rails view or controller tests, simply:
+#
+#  require 'action-nokogiri'
+#
+#  class KittehControllerTest < ActionController::TestCase
+#  def test_i_can_does_test_with_nokogiri
+#    get(:index, {:wants => "cheezburgers"})
+#    assert @response.html.at("h2.lolcats")
+#  end
+#
+module ActionController
+  module TestResponseBehavior
+
+    def html(flavor=nil)
+      if flavor == :hpricot
+        @_nokogiri_html_hpricot ||= Nokogiri::Hpricot(body)
+      else
+        @_nokogiri_html_vanilla ||= Nokogiri::HTML(body)
+      end
+    end
+
+    def xml
+      @_nokogiri_xml ||= Nokogiri::XML(body)
+    end
+
+  end
+end