ClasspathUriResolver is now also an LSInputResolver so it can be used for both XSLT...

[utils] / support / general / src / main / java / org / wamblee / xml / ClasspathUriResolver.java

diff --git a/support/general/src/main/java/org/wamblee/xml/ClasspathUriResolver.java b/support/general/src/main/java/org/wamblee/xml/ClasspathUriResolver.java
index ba2d875c3aeb591e2e13829dc47cd88c4bb780cf..6ee7389e4014256d03c67acaa1a7d260e4971f7a 100644 (file)
--- a/support/general/src/main/java/org/wamblee/xml/ClasspathUriResolver.java
+++ b/support/general/src/main/java/org/wamblee/xml/ClasspathUriResolver.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2005 the original author or authors.
+ * Copyright 2005-2010 the original author or authors.
  * 
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -13,24 +13,26 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-
 package org.wamblee.xml;
 
+import org.w3c.dom.ls.DOMImplementationLS;
+import org.w3c.dom.ls.LSInput;
+import org.w3c.dom.ls.LSResourceResolver;
+import org.wamblee.io.ClassPathResource;
+import org.wamblee.io.InputResource;
+
 import java.io.IOException;
+import java.io.InputStream;
 
 import javax.xml.transform.Source;
 import javax.xml.transform.TransformerException;
 import javax.xml.transform.URIResolver;
 import javax.xml.transform.stream.StreamSource;
 
-import org.wamblee.io.ClassPathResource;
-import org.wamblee.io.InputResource;
-
 /**
  * URI resolver that resolves stylesheets through the classpath.
  */
-public class ClasspathUriResolver implements URIResolver {
-
+public class ClasspathUriResolver implements URIResolver, LSResourceResolver {
     /**
      * Constructs the resolver.
      * 
@@ -43,18 +45,38 @@ public class ClasspathUriResolver implements URIResolver {
      * (non-Javadoc)
      * 
      * @see javax.xml.transform.URIResolver#resolve(java.lang.String,
-     *      java.lang.String)
+     * java.lang.String)
      */
     public Source resolve(String aHref, String aBase)
-            throws TransformerException {
+        throws TransformerException {
         InputResource xslt = new ClassPathResource(aHref);
+
         try {
             return new StreamSource(xslt.getInputStream());
         } catch (IOException e) {
             throw new TransformerException(
-                    "Could not get XSLT style sheet in classpath '" + aHref
-                            + "'", e);
+                "Could not get XSLT style sheet in classpath '" + aHref + "'",
+                e);
+        }
+    }
+
+    @Override
+    public LSInput resolveResource(String aType, String aNamespaceURI,
+        String aPublicId, String aSystemId, String aBaseURI) {
+        try {
+            InputStream xslt = new ClassPathResource(aSystemId)
+                .getInputStream();
+            DOMImplementationLS impl = DomUtils.getDomImplementationLS();
+            LSInput input = impl.createLSInput();
+            input.setPublicId(aPublicId);
+            input.setSystemId(aSystemId);
+            input.setByteStream(xslt);
+            return input;
+        } catch (IOException e) {
+            return null;
         }
+       
+       
     }
 
 }