001package org.hl7.fhir.utilities;
002
003/*
004Copyright (c) 2011+, HL7, Inc
005All rights reserved.
006
007Redistribution and use in source and binary forms, with or without modification, 
008are permitted provided that the following conditions are met:
009
010 * Redistributions of source code must retain the above copyright notice, this 
011   list of conditions and the following disclaimer.
012 * Redistributions in binary form must reproduce the above copyright notice, 
013   this list of conditions and the following disclaimer in the documentation 
014   and/or other materials provided with the distribution.
015 * Neither the name of HL7 nor the names of its contributors may be used to 
016   endorse or promote products derived from this software without specific 
017   prior written permission.
018
019THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
020ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
021WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 
022IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
023INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 
024NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
025PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
026WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
027ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
028POSSIBILITY OF SUCH DAMAGE.
029
030*/
031
032import java.io.File;
033import java.io.FileInputStream;
034import java.io.FileNotFoundException;
035import java.io.IOException;
036
037import javax.xml.transform.Source;
038import javax.xml.transform.TransformerException;
039import javax.xml.transform.TransformerFactory;
040import javax.xml.transform.URIResolver;
041import javax.xml.transform.stream.StreamSource;
042
043public class MyURIResolver implements URIResolver {
044
045  private String path;
046  private URIResolver alt;  
047  
048  public MyURIResolver(String path, URIResolver alt) {
049    super();
050    this.path = path;
051    this.alt = alt;
052  }
053
054
055  @Override
056  public Source resolve(String href, String base) throws TransformerException {
057    try {
058      if (href.startsWith("http://") || href.startsWith("https://")) {
059        if (alt != null) {
060          Source s = alt.resolve(href, base);
061          if (s != null)
062            return s;
063        }
064        return TransformerFactory.newInstance().getURIResolver().resolve(href, base);
065      } else
066        return new StreamSource(new FileInputStream(href.contains(File.separator) ? href : Utilities.path(path, href)));
067    } catch (FileNotFoundException e) {
068      throw new TransformerException(e);
069    } catch (IOException e) {
070      throw new TransformerException(e);
071    }
072  }
073
074}