001package org.hl7.fhir.utilities;
002
003/*
004 * #%L
005 * HAPI FHIR - Core Library
006 * %%
007 * Copyright (C) 2014 - 2017 University Health Network
008 * %%
009 * Licensed under the Apache License, Version 2.0 (the "License");
010 * you may not use this file except in compliance with the License.
011 * You may obtain a copy of the License at
012 * 
013 *      http://www.apache.org/licenses/LICENSE-2.0
014 * 
015 * Unless required by applicable law or agreed to in writing, software
016 * distributed under the License is distributed on an "AS IS" BASIS,
017 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
018 * See the License for the specific language governing permissions and
019 * limitations under the License.
020 * #L%
021 */
022
023
024/*
025Copyright (c) 2011+, HL7, Inc
026All rights reserved.
027
028Redistribution and use in source and binary forms, with or without modification, 
029are permitted provided that the following conditions are met:
030
031 * Redistributions of source code must retain the above copyright notice, this 
032   list of conditions and the following disclaimer.
033 * Redistributions in binary form must reproduce the above copyright notice, 
034   this list of conditions and the following disclaimer in the documentation 
035   and/or other materials provided with the distribution.
036 * Neither the name of HL7 nor the names of its contributors may be used to 
037   endorse or promote products derived from this software without specific 
038   prior written permission.
039
040THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
041ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
042WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 
043IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
044INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 
045NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
046PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
047WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
048ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
049POSSIBILITY OF SUCH DAMAGE.
050
051 */
052
053import java.io.File;
054import java.io.FileInputStream;
055import java.io.FileNotFoundException;
056import java.io.IOException;
057
058import javax.xml.transform.Source;
059import javax.xml.transform.TransformerException;
060import javax.xml.transform.TransformerFactory;
061import javax.xml.transform.URIResolver;
062import javax.xml.transform.stream.StreamSource;
063
064public class MyURIResolver implements URIResolver {
065
066        private String path;
067        private URIResolver alt;  
068
069        public MyURIResolver(String path, URIResolver alt) {
070                super();
071                this.path = path;
072                this.alt = alt;
073        }
074
075
076        @Override
077        public Source resolve(String href, String base) throws TransformerException {
078                try {
079                        if (href.startsWith("http://") || href.startsWith("https://")) {
080                                if (alt != null) {
081                                        Source s = alt.resolve(href, base);
082                                        if (s != null)
083                                                return s;
084                                }
085                                return TransformerFactory.newInstance().getURIResolver().resolve(href, base);
086                        } else
087                                //FIXME resource leak
088                                return new StreamSource(new FileInputStream(href.contains(File.separator) ? href : Utilities.path(path, href)));
089                } catch (FileNotFoundException e) {
090                        throw new TransformerException(e);
091                } catch (IOException e) {
092                        throw new TransformerException(e);
093                }
094        }
095
096}