OSDN Git Service

MMD Ver7.40 対応
[mikutoga/TogaGem.git] / src / main / java / jp / sfjp / mikutoga / xml / XmlResourceResolver.java
diff --git a/src/main/java/jp/sfjp/mikutoga/xml/XmlResourceResolver.java b/src/main/java/jp/sfjp/mikutoga/xml/XmlResourceResolver.java
new file mode 100644 (file)
index 0000000..e103268
--- /dev/null
@@ -0,0 +1,472 @@
+/*
+ * xml resource resolver
+ *
+ * License : The MIT License
+ * Copyright(c) 2009 olyutorskii
+ */
+
+package jp.sfjp.mikutoga.xml;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Reader;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Map;
+import org.w3c.dom.ls.LSInput;
+import org.w3c.dom.ls.LSResourceResolver;
+import org.xml.sax.EntityResolver;
+import org.xml.sax.InputSource;
+import org.xml.sax.SAXException;
+
+/**
+ * URL変換マップに従い、XML文書からの外部参照をリダイレクトする。
+ * 相対URIはこのクラスをベースに解決される。
+ * 主な用途は外部スキーマのリソース化など。
+ */
+public class XmlResourceResolver
+        implements LSResourceResolver, EntityResolver {
+
+    /** XML Schema. */
+    public static final String SCHEMA_XML =
+            "http://www.w3.org/2001/xml.xsd";
+
+    /** XSD名前空間。 */
+    public static final String NS_XSD =
+            "http://www.w3.org/2001/XMLSchema-instance";
+
+    private static final String LOCAL_SCHEMA_XML =
+            "resources/xmlspace.xsd";
+
+    private static final URI EMPTY_URI = URI.create("");
+
+    private static final Class<?> THISCLASS = XmlResourceResolver.class;
+
+
+    private final Map<URI, URI> uriMap;
+
+
+    /**
+     * コンストラクタ。
+     */
+    public XmlResourceResolver(){
+        super();
+
+        assert this.getClass().equals(THISCLASS);
+
+        Map<URI, URI> map;
+        map = new HashMap<URI, URI>();
+        map = Collections.synchronizedMap(map);
+        this.uriMap = map;
+
+        URL redirectRes = THISCLASS.getResource(LOCAL_SCHEMA_XML);
+        String redirectResName = redirectRes.toString();
+
+        URI originalURI = URI.create(SCHEMA_XML);
+        URI redirectURI = URI.create(redirectResName);
+
+        putRedirectedImpl(originalURI, redirectURI);
+
+        return;
+    }
+
+
+    /**
+     * 絶対URIと相対URIを合成したURIを返す。
+     * 正規化も行われる。
+     * @param base 絶対URIでなければならない。nullでもよい。
+     * @param relative 絶対URIでもよいがその場合baseは無視される。null可。
+     * @return 合成結果のURLオブジェクト。必ず絶対URIになる。
+     * @throws java.net.URISyntaxException URIとして変。
+     * @throws java.lang.IllegalArgumentException 絶対URIが生成できない。
+     */
+    protected static URI buildBaseRelativeURI(String base, String relative)
+            throws URISyntaxException,
+                   IllegalArgumentException {
+        URI baseURI;
+        if(base != null){
+            baseURI = new URI(base);
+            if( ! baseURI.isAbsolute() ){
+                throw new IllegalArgumentException();
+            }
+        }else{
+            baseURI = null;
+        }
+
+        URI relativeURI;
+        if(relative != null){
+            relativeURI = new URI(relative);
+        }else{
+            relativeURI = EMPTY_URI;
+        }
+
+        URI resultURI;
+        if(baseURI == null || relativeURI.isAbsolute()){
+            resultURI = relativeURI;
+        }else{
+            resultURI = baseURI.resolve(relativeURI);
+        }
+
+        if( ! resultURI.isAbsolute() ){
+            throw new IllegalArgumentException();
+        }
+
+        resultURI = resultURI.normalize();
+
+        return resultURI;
+    }
+
+    /**
+     * LSInput実装を生成する。
+     * @return LSInput実装
+     */
+    public static LSInput createLSInput(){
+        LSInput input = new LSInputImpl();
+        return input;
+    }
+
+
+    /**
+     * オリジナルURIとリダイレクト先のURIを登録する。
+     * オリジナルURIへのアクセスはリダイレクトされる。
+     * @param original オリジナルURI
+     * @param redirect リダイレクトURI
+     */
+    private void putRedirectedImpl(URI original, URI redirect){
+        URI oridinalNorm = original.normalize();
+        URI redirectNorm = redirect.normalize();
+
+        this.uriMap.put(oridinalNorm, redirectNorm);
+
+        return;
+    }
+
+    /**
+     * オリジナルURIとリダイレクト先のURIを登録する。
+     * オリジナルURIへのアクセスはリダイレクトされる。
+     * @param original オリジナルURI
+     * @param redirect リダイレクトURI
+     */
+    public void putRedirected(URI original, URI redirect){
+        putRedirectedImpl(original, redirect);
+        return;
+    }
+
+    /**
+     * ローカル版リソース参照解決を登録する。
+     * @param lsc ローカル版リソース参照解決
+     */
+    public void putRedirected(LocalXmlResource lsc){
+        URI original = lsc.getOriginalResource();
+        if(original == null) return;
+
+        URI local = lsc.getLocalResource();
+
+        putRedirected(original, local);
+
+        return;
+    }
+
+    /**
+     * 別リゾルバの登録内容を追加登録する。
+     * @param other 別リゾルバ
+     */
+    public void putRedirected(XmlResourceResolver other){
+        this.uriMap.putAll(other.uriMap);
+        return;
+    }
+
+    /**
+     * 登録済みリダイレクト先URIを返す。
+     * @param original オリジナルURI
+     * @return リダイレクト先URI。未登録の場合はnull
+     */
+    public URI getRedirected(URI original){
+        URI keyURI = original.normalize();
+        URI resourceURI = this.uriMap.get(keyURI);
+        return resourceURI;
+    }
+
+    /**
+     * 登録済みリダイレクト先URIを返す。
+     * @param original オリジナルURI
+     * @return リダイレクト先URI。未登録の場合はオリジナルを返す
+     */
+    public URI resolveRedirected(URI original){
+        URI result = getRedirected(original);
+        if(result == null) result = original;
+        return result;
+    }
+
+    /**
+     * 登録済みリダイレクト先リソースの入力ストリームを得る。
+     * @param originalURI オリジナルURI
+     * @return 入力ストリーム。リダイレクト先が未登録の場合はnull
+     * @throws java.io.IOException 入出力エラー。
+     * もしくはリソースが見つからない。
+     */
+    private InputStream getXMLResourceAsStream(URI originalURI)
+            throws IOException{
+        URI resourceURI = getRedirected(originalURI);
+        if(resourceURI == null) return null;
+
+        URL resourceURL = resourceURI.toURL();
+        InputStream is = resourceURL.openStream();
+
+        return is;
+    }
+
+    /**
+     * {@inheritDoc}
+     * URL変換したあとの入力ソースを返す。
+     * @param type {@inheritDoc}
+     * @param namespaceURI {@inheritDoc}
+     * @param publicId {@inheritDoc}
+     * @param systemId {@inheritDoc}
+     * @param baseURI {@inheritDoc}
+     * @return {@inheritDoc}
+     */
+    @Override
+    public LSInput resolveResource(String type,
+                                     String namespaceURI,
+                                     String publicId,
+                                     String systemId,
+                                     String baseURI ){
+        if(systemId == null) return null;
+
+        URI originalURI;
+        try{
+            originalURI = buildBaseRelativeURI(baseURI, systemId);
+        }catch(URISyntaxException e){
+            return null;
+        }
+
+        InputStream is;
+        try{
+            is = getXMLResourceAsStream(originalURI);
+        }catch(IOException e){
+            return null;
+        }
+        if(is == null) return null;
+
+        LSInput input = createLSInput();
+        input.setBaseURI(baseURI);
+        input.setPublicId(publicId);
+        input.setSystemId(systemId);
+        input.setByteStream(is);
+
+        return input;
+    }
+
+    /**
+     * {@inheritDoc}
+     * URL変換したあとの入力ソースを返す。
+     * @param publicId {@inheritDoc}
+     * @param systemId {@inheritDoc}
+     * @return {@inheritDoc}
+     * @throws org.xml.sax.SAXException {@inheritDoc}
+     * @throws java.io.IOException {@inheritDoc}
+     */
+    @Override
+    public InputSource resolveEntity(String publicId, String systemId)
+            throws SAXException, IOException{
+        if(systemId == null) return null;
+
+        URI originalUri;
+        try{
+            originalUri = new URI(systemId);
+        }catch(URISyntaxException e){
+            return null;
+        }
+
+        InputStream is = getXMLResourceAsStream(originalUri);
+        if(is == null) return null;
+
+        InputSource source = new InputSource(is);
+        source.setPublicId(publicId);
+        source.setSystemId(systemId);
+
+        return source;
+    }
+
+    /**
+     * JRE1.5用LSInput実装。
+     * JRE1.6なら
+     * org.w3c.dom.ls.DOMImplementationLS#createLSInput()
+     * で生成可能かも。
+     */
+    private static final class LSInputImpl implements LSInput {
+
+        private String baseURI = null;
+        private InputStream byteStream = null;
+        private boolean certifiedText = false;
+        private Reader characterStream = null;
+        private String encoding = null;
+        private String publicId = null;
+        private String stringData = null;
+        private String systemId = null;
+
+        /**
+         * コンストラクタ。
+         */
+        LSInputImpl(){
+            super();
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public String getBaseURI(){
+            return this.baseURI;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param baseURI {@inheritDoc}
+         */
+        @Override
+        public void setBaseURI(String baseURI){
+            this.baseURI = baseURI;
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public InputStream getByteStream(){
+            return this.byteStream;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param byteStream {@inheritDoc}
+         */
+        @Override
+        public void setByteStream(InputStream byteStream){
+            this.byteStream = byteStream;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public boolean getCertifiedText(){
+            return this.certifiedText;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param certifiedText {@inheritDoc}
+         */
+        @Override
+        public void setCertifiedText(boolean certifiedText){
+            this.certifiedText = certifiedText;
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public Reader getCharacterStream(){
+            return this.characterStream;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param characterStream {@inheritDoc}
+         */
+        @Override
+        public void setCharacterStream(Reader characterStream){
+            this.characterStream = characterStream;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public String getEncoding(){
+            return this.encoding;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param encoding {@inheritDoc}
+         */
+        @Override
+        public void setEncoding(String encoding){
+            this.encoding = encoding;
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public String getPublicId(){
+            return this.publicId;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param publicId {@inheritDoc}
+         */
+        @Override
+        public void setPublicId(String publicId){
+            this.publicId = publicId;
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public String getStringData(){
+            return this.stringData;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param stringData {@inheritDoc}
+         */
+        @Override
+        public void setStringData(String stringData){
+            this.stringData = stringData;
+            return;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @return {@inheritDoc}
+         */
+        @Override
+        public String getSystemId(){
+            return this.systemId;
+        }
+
+        /**
+         * {@inheritDoc}
+         * @param systemId {@inheritDoc}
+         */
+        @Override
+        public void setSystemId(String systemId){
+            this.systemId = systemId;
+            return;
+        }
+
+    }
+
+}