nokogiri-backupify 1.5.0.beta.4

data/ext/java/nokogiri/XmlText.java

@@ -0,0 +1,135 @@
+/**
+ * (The MIT License)
+ *
+ * Copyright (c) 2008 - 2010:
+ *
+ * * {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.isXmlEscaped;
+import static nokogiri.internals.NokogiriHelpers.rubyStringToString;
+import static nokogiri.internals.NokogiriHelpers.stringOrNil;
+import nokogiri.internals.NokogiriHelpers;
+import nokogiri.internals.SaveContext;
+
+import org.jruby.Ruby;
+import org.jruby.RubyClass;
+import org.jruby.anno.JRubyClass;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.javasupport.JavaUtil;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.w3c.dom.Document;
+import org.w3c.dom.Node;
+
+/**
+ * Class for Nokogiri::XML::Text
+ *
+ */
+@JRubyClass(name="Nokogiri::XML::Text", parent="Nokogiri::XML::CharacterData")
+public class XmlText extends XmlNode {
+
+    public XmlText(Ruby runtime, RubyClass rubyClass, Node node) {
+        super(runtime, rubyClass, node);
+    }
+
+    public XmlText(Ruby runtime, RubyClass klass) {
+        super(runtime, klass);
+    }
+
+    @Override
+    protected void init(ThreadContext context, IRubyObject[] args) {
+        if (args.length < 2) {
+            throw getRuntime().newArgumentError(args.length, 2);
+        }
+
+        content = args[0];
+        IRubyObject xNode = args[1];
+
+        XmlNode xmlNode = asXmlNode(context, xNode);
+        XmlDocument xmlDoc = (XmlDocument)xmlNode.document(context);
+        doc = xmlDoc;
+        Document document = xmlDoc.getDocument();
+        // text node content should not be encoded when it is created by Text node.
+        // while content should be encoded when it is created by Element node.
+        Node node = document.createTextNode(rubyStringToString(content));
+        setNode(context, node);
+    }
+    
+    @Override
+    protected IRubyObject getNodeName(ThreadContext context) {
+        if (name == null) name = context.getRuntime().newString("text");
+        return name;
+    }
+    
+    @Override
+    @JRubyMethod(name = {"content", "text", "inner_text"})
+    public IRubyObject content(ThreadContext context) {
+        if (content == null || content.isNil()) {
+            return stringOrNil(context.getRuntime(), node.getTextContent());
+        } else {
+            return content;
+        }
+    }
+
+    @Override
+    public void saveContent(ThreadContext context, SaveContext ctx) {
+        String textContent = node.getTextContent();
+        
+        if (!isXmlEscaped(textContent)) {        
+            textContent = NokogiriHelpers.encodeJavaString(textContent);
+        }
+        if (getEncoding(context, ctx) == null) {
+            textContent = encodeStringToHtmlEntity(textContent);
+        }
+        ctx.append(textContent);
+    }
+    
+    private String getEncoding(ThreadContext context, SaveContext ctx) {
+        String encoding  = ctx.getEncoding();
+        if (encoding != null) return encoding;
+        XmlDocument xmlDocument = (XmlDocument)document(context);
+        IRubyObject ruby_encoding = xmlDocument.encoding(context);
+        if (!ruby_encoding.isNil()) {
+            encoding = rubyStringToString(ruby_encoding);
+        }
+        return encoding;
+    }
+    
+    private String encodeStringToHtmlEntity(String text) {
+        int last = 126; // = U+007E. No need to encode under U+007E.
+        StringBuffer sb = new StringBuffer();
+        for (int i=0; i<text.length(); i++) {
+            int codePoint = text.codePointAt(i);
+            if (codePoint > last) sb.append("&#x" + Integer.toHexString(codePoint) + ";");
+            else sb.append(text.charAt(i));
+        }
+        return new String(sb);
+    }
+}

data/ext/java/nokogiri/XmlXpathContext.java

@@ -0,0 +1,175 @@
+/**
+ * (The MIT License)
+ *
+ * Copyright (c) 2008 - 2010:
+ *
+ * * {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 java.util.Set;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import javax.xml.xpath.XPath;
+import javax.xml.xpath.XPathConstants;
+import javax.xml.xpath.XPathExpression;
+import javax.xml.xpath.XPathExpressionException;
+import javax.xml.xpath.XPathFactory;
+
+import nokogiri.internals.NokogiriNamespaceContext;
+import nokogiri.internals.NokogiriXPathFunctionResolver;
+import nokogiri.internals.NokogiriXPathVariableResolver;
+
+import org.jruby.Ruby;
+import org.jruby.RubyBoolean;
+import org.jruby.RubyClass;
+import org.jruby.RubyException;
+import org.jruby.RubyNumeric;
+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.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.w3c.dom.NodeList;
+
+/**
+ * Class for Nokogiri::XML::XpathContext
+ *
+ */
+@JRubyClass(name="Nokogiri::XML::XPathContext")
+public class XmlXpathContext extends RubyObject {
+    private XmlNode context;
+    private XPath xpath;
+
+    public XmlXpathContext(Ruby ruby, RubyClass rubyClass, XmlNode context) {
+        super(ruby, rubyClass);
+        this.context = context;
+        this.xpath = XPathFactory.newInstance().newXPath();
+        this.xpath.setNamespaceContext(new NokogiriNamespaceContext());
+        this.xpath.setXPathVariableResolver(new NokogiriXPathVariableResolver());
+    }
+
+    @JRubyMethod(name = "new", meta = true)
+    public static IRubyObject rbNew(ThreadContext context, IRubyObject cls, IRubyObject node) {
+        XmlNode xmlNode = (XmlNode)node;
+        return new XmlXpathContext(context.getRuntime(), (RubyClass)cls, xmlNode);
+    }
+
+    @JRubyMethod
+    public IRubyObject evaluate(ThreadContext context, IRubyObject expr, IRubyObject handler) {
+        String src = expr.convertToString().asJavaString();
+        try {
+            if(!handler.isNil()) {
+            	if (!isContainsPrefix(src)) {
+                    Set<String> methodNames = handler.getMetaClass().getMethods().keySet();
+                    for (String name : methodNames) {
+                        src = src.replaceAll(name, NokogiriNamespaceContext.NOKOGIRI_PREFIX+":"+name);
+                    }
+                }
+                xpath.setXPathFunctionResolver(new NokogiriXPathFunctionResolver(handler));
+            }
+            XPathExpression xpathExpression = xpath.compile(src);
+            return node_set(context, xpathExpression);
+            //return new XmlXpath(context.getRuntime(), (RubyClass)context.getRuntime().getClassFromPath("Nokogiri::XML::XPath"), xpathExpression, this.context);
+        } catch (XPathExpressionException xpee) {
+            xpee = new XPathExpressionException(src);
+            RubyException e =
+                XmlSyntaxError.createXPathSyntaxError(getRuntime(), xpee);
+            throw new RaiseException(e);
+        }
+    }
+
+    protected IRubyObject node_set(ThreadContext rbctx, XPathExpression xpathExpression) {
+        XmlNodeSet result = null;
+        try {  
+            result = tryGetNodeSet(xpathExpression);
+//            result.relink_namespace(context);
+            result.setDocument(context.document(rbctx));
+            return result;
+        } catch (XPathExpressionException xpee) {
+            try {
+                return tryGetOpaqueValue(xpathExpression);
+            } catch (XPathExpressionException xpee_opaque) {
+                 RubyException e = XmlSyntaxError.createXPathSyntaxError(getRuntime(), xpee_opaque);
+                 throw new RaiseException(e);
+            }
+        }
+    }
+    
+    private XmlNodeSet tryGetNodeSet(XPathExpression xpathExpression) throws XPathExpressionException {
+        NodeList nodes = (NodeList)xpathExpression.evaluate(context.node, XPathConstants.NODESET);
+        return new XmlNodeSet(getRuntime(), nodes);       
+    }
+    
+    private static Pattern number_pattern = Pattern.compile("\\d.*");
+    private static Pattern boolean_pattern = Pattern.compile("true|false");
+    
+    private IRubyObject tryGetOpaqueValue(XPathExpression xpathExpression) throws XPathExpressionException {
+        String string = (String)xpathExpression.evaluate(context.node, XPathConstants.STRING);
+        if (doesMatch(number_pattern, string)) return RubyNumeric.dbl2num(getRuntime(), Double.parseDouble(string));
+        if (doesMatch(boolean_pattern, string)) return RubyBoolean.newBoolean(getRuntime(), Boolean.parseBoolean(string));
+        return RubyString.newString(getRuntime(), string);
+    }
+    
+    private boolean doesMatch(Pattern pattern, String string) {
+        Matcher m = pattern.matcher(string);
+        return m.matches();
+    }
+
+    private boolean isContainsPrefix(String str) {
+        Set<String> prefixes = ((NokogiriNamespaceContext)xpath.getNamespaceContext()).getAllPrefixes();
+        for (String prefix : prefixes) {
+            if (str.contains(prefix + ":")) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+
+    @JRubyMethod
+    public IRubyObject evaluate(ThreadContext context, IRubyObject expr) {
+        return this.evaluate(context, expr, context.getRuntime().getNil());
+    }
+
+    @JRubyMethod
+    public IRubyObject register_ns(ThreadContext context, IRubyObject prefix, IRubyObject uri) {
+        ((NokogiriNamespaceContext) this.xpath.getNamespaceContext()).registerNamespace(prefix.convertToString().asJavaString(), uri.convertToString().asJavaString());
+        return this;
+    }
+
+    @JRubyMethod
+    public IRubyObject register_variable(ThreadContext context, IRubyObject name, IRubyObject value) {
+        ((NokogiriXPathVariableResolver) this.xpath.getXPathVariableResolver()).
+            registerVariable(name.convertToString().asJavaString(), value.convertToString().asJavaString());
+        return this;
+    }
+}

data/ext/java/nokogiri/XsltStylesheet.java

@@ -0,0 +1,181 @@
+/**
+ * (The MIT License)
+ *
+ * Copyright (c) 2008 - 2010:
+ *
+ * * {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.util.HashMap;
+import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import javax.xml.transform.Templates;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerConfigurationException;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMResult;
+import javax.xml.transform.dom.DOMSource;
+
+import org.jruby.Ruby;
+import org.jruby.RubyArray;
+import org.jruby.RubyClass;
+import org.jruby.RubyObject;
+import org.jruby.anno.JRubyClass;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.javasupport.util.RuntimeHelpers;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.w3c.dom.Document;
+
+/**
+ * Class for Nokogiri::XSLT::Stylesheet
+ *
+ */
+@JRubyClass(name="Nokogiri::XSLT::Stylesheet")
+public class XsltStylesheet extends RubyObject {
+    private static Map<String, Object> registry = new HashMap<String, Object>();
+    private static TransformerFactory factory = null;
+    private Templates sheet;
+
+    public static Map<String, Object> getRegistry() {
+        return registry;
+    }
+
+    public XsltStylesheet(Ruby ruby, RubyClass rubyClass) {
+        super(ruby, rubyClass);
+    }
+
+    private void addParametersToTransformer(ThreadContext context, Transformer transf, IRubyObject parameters) {
+        Ruby ruby = context.getRuntime();
+        RubyArray params = parameters.convertToArray();
+        int limit = params.getLength();
+        if(limit % 2 == 1) limit--;
+
+        for(int i = 0; i < limit; i+=2) {
+            String name = params.aref(ruby.newFixnum(i)).asJavaString();
+            String value = params.aref(ruby.newFixnum(i+1)).asJavaString();
+            transf.setParameter(name, unparseValue(value));
+        }
+    }
+    
+    private Pattern p = Pattern.compile("'.{1,}'");
+
+    private String unparseValue(String orig) {
+        Matcher m = p.matcher(orig);
+        if ((orig.startsWith("\"") && orig.endsWith("\"")) || m.matches()) {
+            orig = orig.substring(1, orig.length()-1);
+        }
+
+        return orig;
+    }
+
+    @JRubyMethod(meta = true)
+    public static IRubyObject parse_stylesheet_doc(ThreadContext context, IRubyObject cls, IRubyObject document) {
+        
+        Ruby ruby = context.getRuntime();
+
+        if(!(document instanceof XmlDocument)) {
+            throw ruby.newArgumentError("doc must be a Nokogiri::XML::Document instance");
+        }
+
+        XmlDocument xmlDoc = (XmlDocument) document;
+
+        RubyArray errors = (RubyArray) xmlDoc.getInstanceVariable("@errors");
+
+        if(!errors.isEmpty()) {
+            throw ruby.newRuntimeError(errors.first().asJavaString());
+        }
+        
+        Document doc = ((XmlDocument) xmlDoc.dup_implementation(context, true)).getDocument();
+
+        XsltStylesheet xslt = new XsltStylesheet(ruby, (RubyClass) cls);
+
+        try {
+            if (factory == null) factory = TransformerFactory.newInstance();
+            xslt.sheet = factory.newTemplates(new DOMSource(doc));
+        } catch (TransformerConfigurationException ex) {
+            ruby.newRuntimeError("could not parse xslt stylesheet");
+        }
+
+        return xslt;
+    }
+
+    @JRubyMethod
+    public IRubyObject serialize(ThreadContext context, IRubyObject doc) {
+        return RuntimeHelpers.invoke(context,
+                RuntimeHelpers.invoke(context, doc, "root"),
+                "to_s");
+    }
+
+    @JRubyMethod(rest = true, required=1, optional=2)
+    public IRubyObject transform(ThreadContext context, IRubyObject[] args) {
+        Ruby ruby = context.getRuntime();
+
+        DOMSource docSource = new DOMSource(((XmlDocument) args[0]).getDocument());
+        DOMResult result = new DOMResult();
+
+        try{
+            Transformer transf = this.sheet.newTransformer();
+            if(args.length > 1) {
+                addParametersToTransformer(context, transf, args[1]);
+            }
+            transf.transform(docSource, result);
+        } catch(TransformerConfigurationException ex) {
+            throw ruby.newRuntimeError("Could not transform the document.");
+        } catch(TransformerException ex) {
+            throw ruby.newRuntimeError("Could not transform the document.");
+        }
+        
+        if ("html".equals(result.getNode().getFirstChild().getNodeName())) {
+            return new HtmlDocument(ruby,
+                    getNokogiriClass(ruby, "Nokogiri::HTML::Document"),
+                    (Document) result.getNode());
+        } else {
+            return new XmlDocument(ruby,
+                    getNokogiriClass(ruby, "Nokogiri::XML::Document"),
+                    (Document) result.getNode());
+        }
+    }
+    
+    @JRubyMethod(name = {"registr", "register"}, meta = true)
+    public static IRubyObject register(ThreadContext context, IRubyObject cls, IRubyObject uri, IRubyObject receiver) {
+        throw context.getRuntime().newNotImplementedError("Nokogiri::XSLT.register method is not implemented");
+        /* When API conflict is solved, this method should be below:
+        // ThreadContext is used while executing xslt extension function
+        registry.put("context", context);
+        registry.put("receiver", receiver);
+        return context.getRuntime().getNil();
+        */
+    }
+}