nokogiri 1.4.3.1-java → 1.5.0-java

data/ext/java/nokogiri/XmlReader.java

@@ -0,0 +1,408 @@
+/**
+ * (The MIT License)
+ *
+ * Copyright (c) 2008 - 2011:
+ *
+ * * {Aaron Patterson}[http://tenderlovemaking.com]
+ * * {Mike Dalessio}[http://mike.daless.io]
+ * * {Charles Nutter}[http://blog.headius.com]
+ * * {Sergio Arbeo}[http://www.serabe.com]
+ * * {Patrick Mahoney}[http://polycrystal.org]
+ * * {Yoko Harada}[http://yokolet.blogspot.com]
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * 'Software'), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package nokogiri;
+
+import static nokogiri.internals.NokogiriHelpers.getNokogiriClass;
+import static nokogiri.internals.NokogiriHelpers.stringOrBlank;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.util.ArrayDeque;
+import java.util.Stack;
+
+import nokogiri.internals.ReaderNode;
+import nokogiri.internals.ReaderNode.ElementNode;
+
+import org.jruby.Ruby;
+import org.jruby.RubyArray;
+import org.jruby.RubyBoolean;
+import org.jruby.RubyClass;
+import org.jruby.RubyFixnum;
+import org.jruby.RubyObject;
+import org.jruby.RubyString;
+import org.jruby.anno.JRubyClass;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.exceptions.RaiseException;
+import org.jruby.javasupport.util.RuntimeHelpers;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.jruby.util.ByteList;
+import org.xml.sax.Attributes;
+import org.xml.sax.InputSource;
+import org.xml.sax.SAXException;
+import org.xml.sax.SAXParseException;
+import org.xml.sax.XMLReader;
+import org.xml.sax.ext.DefaultHandler2;
+import org.xml.sax.helpers.XMLReaderFactory;
+
+/**
+ * Class for Nokogiri:XML::Reader
+ *
+ * @author sergio
+ * @author Yoko Harada <yokolet@gmail.com>
+ */
+@JRubyClass(name="Nokogiri::XML::Reader")
+public class XmlReader extends RubyObject {
+
+    private static final int XML_TEXTREADER_MODE_INITIAL = 0;
+    private static final int XML_TEXTREADER_MODE_INTERACTIVE = 1;
+    private static final int XML_TEXTREADER_MODE_ERROR = 2;
+    private static final int XML_TEXTREADER_MODE_EOF = 3;
+    private static final int XML_TEXTREADER_MODE_CLOSED = 4;
+    private static final int XML_TEXTREADER_MODE_READING = 5;
+
+    ArrayDeque<ReaderNode> nodeQueue;
+    private int state;
+    
+    public XmlReader(Ruby runtime, RubyClass klazz) {
+        super(runtime, klazz);
+    }
+    
+    /**
+     * Create and return a copy of this object.
+     *
+     * @return a clone of this object
+     */
+    @Override
+    public Object clone() throws CloneNotSupportedException {
+        return super.clone();
+    }
+    
+    public void init(Ruby runtime) {
+        nodeQueue = new ArrayDeque<ReaderNode>();
+        nodeQueue.add(new ReaderNode.EmptyNode(runtime));
+    }
+
+    private void parseRubyString(ThreadContext context, RubyString content){
+        Ruby ruby = context.getRuntime();
+        try {
+            this.setState(XML_TEXTREADER_MODE_READING);
+            XMLReader reader = this.createReader(ruby);
+            ByteList byteList = content.getByteList();
+            ByteArrayInputStream bais = new ByteArrayInputStream(byteList.unsafeBytes(), byteList.begin(), byteList.length());
+            reader.parse(new InputSource(bais));
+            this.setState(XML_TEXTREADER_MODE_CLOSED);
+        } catch (SAXParseException spe) {
+            this.setState(XML_TEXTREADER_MODE_ERROR);
+            this.nodeQueue.add(new ReaderNode.ExceptionNode(ruby, spe));
+        } catch (IOException ioe) {
+            throw RaiseException.createNativeRaiseException(ruby, ioe);
+        } catch (SAXException saxe) {
+            throw RaiseException.createNativeRaiseException(ruby, saxe);
+        }
+    }
+
+    private void setState(int state) { this.state = state; }
+
+    @JRubyMethod
+    public IRubyObject attribute(ThreadContext context, IRubyObject name) {
+        return nodeQueue.peek().getAttributeByName(name);
+    }
+
+    @JRubyMethod
+    public IRubyObject attribute_at(ThreadContext context, IRubyObject index) {
+        return nodeQueue.peek().getAttributeByIndex(index);
+    }
+
+    @JRubyMethod
+    public IRubyObject attribute_count(ThreadContext context) {
+        return nodeQueue.peek().getAttributeCount();
+    }
+
+    @JRubyMethod
+    public IRubyObject attribute_nodes(ThreadContext context) {
+        return nodeQueue.peek().getAttributesNodes();
+    }
+
+    @JRubyMethod
+    public IRubyObject attr_nodes(ThreadContext context) {
+        return nodeQueue.peek().getAttributesNodes();
+    }
+
+    @JRubyMethod(name = "attributes?")
+    public IRubyObject attributes_p(ThreadContext context) {
+        return nodeQueue.peek().hasAttributes();
+    }
+    
+    @JRubyMethod
+    public IRubyObject base_uri(ThreadContext context) {
+        return nodeQueue.peek().getXmlBase();
+    }
+
+    @JRubyMethod(name="default?")
+    public IRubyObject default_p(ThreadContext context){
+        return nodeQueue.peek().isDefault();
+    }
+
+    @JRubyMethod
+    public IRubyObject depth(ThreadContext context) {
+        return nodeQueue.peek().getDepth();
+    }
+    
+    @JRubyMethod(name = {"empty_element?", "self_closing?"})
+    public IRubyObject empty_element_p(ThreadContext context) {
+        ReaderNode readerNode = nodeQueue.peek();
+        if (readerNode == null) return context.getRuntime().getNil();
+        if (!(readerNode instanceof ElementNode)) context.getRuntime().getFalse();
+        return RubyBoolean.newBoolean(context.getRuntime(), !readerNode.hasChildren);
+    }
+
+    @JRubyMethod(meta = true, rest = true)
+    public static IRubyObject from_io(ThreadContext context, IRubyObject cls, IRubyObject args[]) {
+        // Only to pass the  source test.
+        Ruby runtime = context.getRuntime();
+        // Not nil allowed!
+        if(args[0].isNil()) throw runtime.newArgumentError("io cannot be nil");
+
+        XmlReader reader = (XmlReader) NokogiriService.XML_READER_ALLOCATOR.allocate(runtime, getNokogiriClass(runtime, "Nokogiri::XML::Reader"));
+        reader.init(runtime);
+        reader.setInstanceVariable("@source", args[0]);
+        reader.setInstanceVariable("@errors", runtime.newArray());
+        if (args.length > 2) reader.setInstanceVariable("@encoding", args[2]);
+
+        RubyString content = RuntimeHelpers.invoke(context, args[0], "read").convertToString();
+        reader.parseRubyString(context, content);
+        return reader;
+    }
+
+    @JRubyMethod(meta = true, rest = true)
+    public static IRubyObject from_memory(ThreadContext context, IRubyObject cls, IRubyObject args[]) {
+        // args[0]: string, args[1]: url, args[2]: encoding, args[3]: options 
+        Ruby runtime = context.getRuntime();
+        // Not nil allowed!
+        if(args[0].isNil()) throw runtime.newArgumentError("string cannot be nil");
+
+        XmlReader reader = (XmlReader) NokogiriService.XML_READER_ALLOCATOR.allocate(runtime, getNokogiriClass(runtime, "Nokogiri::XML::Reader"));
+        reader.init(runtime);
+        reader.setInstanceVariable("@source", args[0]);
+        reader.setInstanceVariable("@errors", runtime.newArray());
+        if (args.length > 2) reader.setInstanceVariable("@encoding", args[2]);
+
+        reader.parseRubyString(context, args[0].convertToString());
+        return reader;
+    }
+
+    @JRubyMethod
+    public IRubyObject node_type(ThreadContext context) {
+        IRubyObject node_type = nodeQueue.peek().getNodeType();
+        return node_type == null ? RubyFixnum.zero(context.getRuntime()) : node_type;
+    }
+
+    @JRubyMethod
+    public IRubyObject inner_xml(ThreadContext context) {
+        return stringOrBlank(context.getRuntime(), getInnerXml(nodeQueue, nodeQueue.peek()));
+    }
+    
+    private String getInnerXml(ArrayDeque<ReaderNode> nodeQueue, ReaderNode current) {
+        if (current.depth < 0) return null;
+        if (!current.hasChildren) return null;
+        StringBuffer sb = new StringBuffer();
+        int currentDepth = (Integer)current.depth;
+        for (ReaderNode node : nodeQueue) {
+            if (((Integer)node.depth) > currentDepth) sb.append(node.getString());
+        }
+        return new String(sb);
+    }
+    
+    @JRubyMethod
+    public IRubyObject outer_xml(ThreadContext context) {
+        return stringOrBlank(context.getRuntime(), getOuterXml(nodeQueue, nodeQueue.peek()));
+    }
+    
+    private String getOuterXml(ArrayDeque<ReaderNode> nodeQueue, ReaderNode current) {
+        if (current.depth < 0) return null;
+        StringBuffer sb = new StringBuffer();
+        int initialDepth = (Integer)current.depth - 1;
+        for (ReaderNode node : nodeQueue) {
+            if (((Integer)node.depth) > initialDepth) sb.append(node.getString());
+        }
+        return new String(sb);
+    }
+
+    @JRubyMethod
+    public IRubyObject lang(ThreadContext context) {
+        return nodeQueue.peek().getLang();
+    }
+
+    @JRubyMethod
+    public IRubyObject local_name(ThreadContext context) {
+        return nodeQueue.peek().getLocalName();
+    }
+
+    @JRubyMethod
+    public IRubyObject name(ThreadContext context) {
+        return nodeQueue.peek().getName();
+    }
+
+    @JRubyMethod
+    public IRubyObject namespace_uri(ThreadContext context) {
+        return nodeQueue.peek().getUri();
+    }
+
+    @JRubyMethod
+    public IRubyObject namespaces(ThreadContext context) {
+        return nodeQueue.peek().getNamespaces(context);
+    }
+
+    @JRubyMethod
+    public IRubyObject prefix(ThreadContext context) {
+        return nodeQueue.peek().getPrefix();
+    }
+
+    @JRubyMethod
+    public IRubyObject read(ThreadContext context) {
+        this.nodeQueue.poll();
+        if(nodeQueue.peek() == null) {
+            return context.getRuntime().getNil();
+        } else if(nodeQueue.peek().isError()) {
+            RubyArray errors = (RubyArray) this.getInstanceVariable("@errors");
+            errors.append(nodeQueue.peek().toSyntaxError());
+
+            this.setInstanceVariable("@errors", errors);
+
+            throw new RaiseException((XmlSyntaxError) nodeQueue.peek().toSyntaxError());
+        } else {
+            return this;
+        }
+    }
+
+    @JRubyMethod
+    public IRubyObject state(ThreadContext context) {
+        return context.getRuntime().newFixnum(this.state);
+    }
+
+    @JRubyMethod
+    public IRubyObject value(ThreadContext context) {
+        return nodeQueue.peek().getValue();
+    }
+
+    @JRubyMethod(name = "value?")
+    public IRubyObject value_p(ThreadContext context) {
+        return nodeQueue.peek().hasValue();
+    }
+
+    @JRubyMethod
+    public IRubyObject xml_version(ThreadContext context) {
+        return nodeQueue.peek().getXmlVersion();
+    }
+
+    protected XMLReader createReader(final Ruby ruby) {
+        DefaultHandler2 handler = new DefaultHandler2() {
+
+            Stack<String> langStack;
+            int depth;
+            Stack<String> xmlBaseStack;
+            Stack<ReaderNode.ElementNode> elementStack;
+
+            @Override
+            public void characters(char[] chars, int start, int length) {
+                ReaderNode.TextNode node = ReaderNode.createTextNode(ruby, new String(chars, start, length), depth, langStack, xmlBaseStack);
+                nodeQueue.add(node);
+            }
+            
+            @Override
+            public void endDocument() throws SAXException {
+                langStack = null;
+                xmlBaseStack = null;
+                elementStack = null;
+            }
+
+            @Override
+            public void endElement(String uri, String localName, String qName) {
+                depth--;     
+                ReaderNode previous = nodeQueue.getLast();
+                ElementNode startElementNode = elementStack.pop();
+                if (previous instanceof ReaderNode.ElementNode && qName.equals(previous.name)) {
+                    previous.hasChildren = false;
+                } else {
+                    ReaderNode node = ReaderNode.createClosingNode(ruby, uri, localName, qName, depth, langStack, xmlBaseStack);
+                    if (startElementNode != null) {
+                        node.attributeList = startElementNode.attributeList;
+                        node.namespaces = startElementNode.namespaces;
+                    }
+                    nodeQueue.add(node);
+                }
+                if (!langStack.isEmpty()) langStack.pop();
+                if (!xmlBaseStack.isEmpty()) xmlBaseStack.pop();
+            }
+
+            @Override
+            public void error(SAXParseException ex) throws SAXParseException {
+                nodeQueue.add(new ReaderNode.ExceptionNode(ruby, ex));
+                throw ex;
+            }
+
+            @Override
+            public void fatalError(SAXParseException ex) throws SAXParseException {
+                nodeQueue.add(new ReaderNode.ExceptionNode(ruby, ex));
+                throw ex;
+            }
+
+            @Override
+            public void startDocument() {
+                depth = 0;
+                langStack = new Stack<String>();
+                xmlBaseStack = new Stack<String>();
+                elementStack = new Stack<ReaderNode.ElementNode>();
+            }
+
+            @Override
+            public void startElement(String uri, String localName, String qName, Attributes attrs) {
+                ReaderNode readerNode = ReaderNode.createElementNode(ruby, uri, localName, qName, attrs, depth, langStack, xmlBaseStack);
+                nodeQueue.add(readerNode);
+                depth++;
+                if (readerNode.lang != null) langStack.push(readerNode.lang);
+                if (readerNode.xmlBase != null) xmlBaseStack.push(readerNode.xmlBase);
+                elementStack.push((ReaderNode.ElementNode)readerNode);
+            }
+
+            @Override
+            public void warning(SAXParseException ex) throws SAXParseException {
+                nodeQueue.add(new ReaderNode.ExceptionNode(ruby, ex));
+                throw ex;
+            }
+        };
+        try {
+            XMLReader reader = XMLReaderFactory.createXMLReader();
+            reader.setContentHandler(handler);
+            reader.setDTDHandler(handler);
+            reader.setErrorHandler(handler);
+            reader.setFeature("http://xml.org/sax/features/xmlns-uris", true);
+            reader.setFeature("http://xml.org/sax/features/namespace-prefixes", true);
+            reader.setFeature("http://apache.org/xml/features/nonvalidating/load-external-dtd", false);
+            return reader;
+        } catch (SAXException saxe) {
+            throw RaiseException.createNativeRaiseException(ruby, saxe);
+        }
+    }
+}

data/ext/java/nokogiri/XmlRelaxng.java

@@ -0,0 +1,144 @@
+/**
+ * (The MIT License)
+ *
+ * Copyright (c) 2008 - 2011:
+ *
+ * * {Aaron Patterson}[http://tenderlovemaking.com]
+ * * {Mike Dalessio}[http://mike.daless.io]
+ * * {Charles Nutter}[http://blog.headius.com]
+ * * {Sergio Arbeo}[http://www.serabe.com]
+ * * {Patrick Mahoney}[http://polycrystal.org]
+ * * {Yoko Harada}[http://yokolet.blogspot.com]
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * 'Software'), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package nokogiri;
+
+import static nokogiri.internals.NokogiriHelpers.getNokogiriClass;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.StringWriter;
+import java.io.UnsupportedEncodingException;
+
+import javax.xml.transform.Source;
+import javax.xml.transform.TransformerConfigurationException;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamResult;
+import javax.xml.transform.stream.StreamSource;
+
+import org.iso_relax.verifier.Schema;
+import org.iso_relax.verifier.Verifier;
+import org.iso_relax.verifier.VerifierConfigurationException;
+import org.iso_relax.verifier.VerifierFactory;
+import org.jruby.Ruby;
+import org.jruby.RubyClass;
+import org.jruby.anno.JRubyClass;
+import org.jruby.runtime.ThreadContext;
+import org.w3c.dom.Document;
+import org.xml.sax.ErrorHandler;
+import org.xml.sax.SAXException;
+
+/**
+ * Class for Nokogiri::XML::RelaxNG
+ * 
+ * @author sergio
+ * @author Yoko Harada <yokolet@gmail.com>
+ */
+@JRubyClass(name="Nokogiri::XML::RelaxNG", parent="Nokogiri::XML::Schema")
+public class XmlRelaxng extends XmlSchema {
+    private Verifier verifier;
+
+    public XmlRelaxng(Ruby ruby, RubyClass klazz) {
+        super(ruby, klazz);
+    }
+    
+    private void setVerifier(Verifier verifier) {
+        this.verifier = verifier;
+    }
+    
+    static XmlSchema createSchemaInstance(ThreadContext context, RubyClass klazz, Source source) {
+        Ruby runtime = context.getRuntime();
+        XmlRelaxng xmlRelaxng = (XmlRelaxng) NokogiriService.XML_RELAXNG_ALLOCATOR.allocate(runtime, klazz);
+        xmlRelaxng.setInstanceVariable("@errors", runtime.newEmptyArray());
+        
+        try {
+            Schema schema = xmlRelaxng.getSchema(source, context);
+            xmlRelaxng.setVerifier(schema.newVerifier());
+            return xmlRelaxng;
+        } catch (VerifierConfigurationException ex) {
+            throw context.getRuntime().newRuntimeError("Could not parse document: " + ex.getMessage());
+        }
+    }
+
+    private Schema getSchema(Source source, ThreadContext context) {
+        InputStream is = null;
+        VerifierFactory factory = new com.thaiopensource.relaxng.jarv.VerifierFactoryImpl();
+        if (source instanceof StreamSource) {
+            StreamSource ss = (StreamSource)source;
+            is = ss.getInputStream();
+        } else { //if (this.source instanceof DOMSource)
+            DOMSource ds = (DOMSource)source;
+            StringWriter xmlAsWriter = new StringWriter();
+            StreamResult result = new StreamResult(xmlAsWriter);
+            try {
+                TransformerFactory.newInstance().newTransformer().transform(ds, result);
+            } catch (TransformerConfigurationException ex) {
+                throw context.getRuntime()
+                    .newRuntimeError("Could not parse document: "+ex.getMessage());
+            } catch (TransformerException ex) {
+                throw context.getRuntime()
+                    .newRuntimeError("Could not parse document: "+ex.getMessage());
+            }
+            try {
+                is = new ByteArrayInputStream(xmlAsWriter.toString().getBytes("UTF-8"));
+            } catch (UnsupportedEncodingException ex) {
+                throw context.getRuntime()
+                    .newRuntimeError("Could not parse document: "+ex.getMessage());
+            }
+        }
+
+        try {
+            return factory.compileSchema(is);
+        } catch (VerifierConfigurationException ex) {
+            throw context.getRuntime()
+                .newRuntimeError("Could not parse document: "+ex.getMessage());
+        } catch (SAXException ex) {
+            throw context.getRuntime()
+                .newRuntimeError("Could not parse document: "+ex.getMessage());
+        } catch (IOException ex) {
+            throw context.getRuntime().newIOError(ex.getMessage());
+        }
+    }
+    
+    @Override
+    protected void setErrorHandler(ErrorHandler errorHandler) {
+        verifier.setErrorHandler(errorHandler);
+    }
+    
+    @Override
+    protected void validate(Document document) throws SAXException, IOException {
+        verifier.verify(document);
+    }
+}