001package org.hl7.fhir.dstu2.terminologies;
002
003/*
004  Copyright (c) 2011+, HL7, Inc.
005  All rights reserved.
006  
007  Redistribution and use in source and binary forms, with or without modification, 
008  are 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  
019  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
020  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
021  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 
022  IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
023  INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 
024  NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
025  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
026  WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
027  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
028  POSSIBILITY OF SUCH DAMAGE.
029  
030 */
031
032
033
034/*
035Copyright (c) 2011+, HL7, Inc
036All rights reserved.
037
038Redistribution and use in source and binary forms, with or without modification, 
039are permitted provided that the following conditions are met:
040
041 * Redistributions of source code must retain the above copyright notice, this 
042   list of conditions and the following disclaimer.
043 * Redistributions in binary form must reproduce the above copyright notice, 
044   this list of conditions and the following disclaimer in the documentation 
045   and/or other materials provided with the distribution.
046 * Neither the name of HL7 nor the names of its contributors may be used to 
047   endorse or promote products derived from this software without specific 
048   prior written permission.
049
050THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
051ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
052WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 
053IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
054INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 
055NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
056PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
057WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
058ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
059POSSIBILITY OF SUCH DAMAGE.
060
061*/
062
063import java.io.File;
064import java.io.FileInputStream;
065import java.io.FileOutputStream;
066import java.io.IOException;
067import java.util.HashMap;
068import java.util.Map;
069
070import org.apache.commons.io.IOUtils;
071import org.hl7.fhir.dstu2.formats.IParser.OutputStyle;
072import org.hl7.fhir.dstu2.model.OperationOutcome;
073import org.hl7.fhir.dstu2.model.Resource;
074import org.hl7.fhir.dstu2.model.ValueSet;
075import org.hl7.fhir.dstu2.terminologies.ValueSetExpander.ValueSetExpansionOutcome;
076import org.hl7.fhir.dstu2.utils.IWorkerContext;
077import org.hl7.fhir.dstu2.utils.ToolingExtensions;
078import org.hl7.fhir.exceptions.FHIRFormatError;
079import org.hl7.fhir.utilities.Utilities;
080import org.hl7.fhir.utilities.xhtml.XhtmlComposer;
081
082public class ValueSetExpansionCache implements ValueSetExpanderFactory {
083
084  public class CacheAwareExpander implements ValueSetExpander {
085
086          @Override
087          public ValueSetExpansionOutcome expand(ValueSet source) throws ETooCostly, IOException {
088                if (expansions.containsKey(source.getUrl()))
089                        return expansions.get(source.getUrl());
090                ValueSetExpander vse = new ValueSetExpanderSimple(context, ValueSetExpansionCache.this);
091                ValueSetExpansionOutcome vso = vse.expand(source);
092                if (vso.getError() != null) {
093                  // well, we'll see if the designated server can expand it, and if it can, we'll cache it locally
094                        vso = context.expandVS(source, false);
095                        FileOutputStream s = new FileOutputStream(Utilities.path(cacheFolder, makeFile(source.getUrl())));
096                        context.newXmlParser().setOutputStyle(OutputStyle.PRETTY).compose(s, vso.getValueset());
097                        s.close();
098                }
099                expansions.put(source.getUrl(), vso);
100                return vso;
101          }
102
103    private String makeFile(String url) {
104      return url.replace("$", "").replace(":", "").replace("//", "/").replace("/", "_")+".xml";
105    }
106  }
107
108  private static final String VS_ID_EXT = "http://tools/cache";
109
110  private final Map<String, ValueSetExpansionOutcome> expansions = new HashMap<String, ValueSetExpansionOutcome>();
111  private final IWorkerContext context;
112  private final String cacheFolder;
113        
114        public ValueSetExpansionCache(IWorkerContext context) {
115    super();
116    cacheFolder = null;
117    this.context = context;
118  }
119  
120        public ValueSetExpansionCache(IWorkerContext context, String cacheFolder) throws FHIRFormatError, IOException {
121    super();
122    this.context = context;
123    this.cacheFolder = cacheFolder;
124    if (this.cacheFolder != null)
125      loadCache();
126  }
127  
128        private void loadCache() throws FHIRFormatError, IOException {
129          File[] files = new File(cacheFolder).listFiles();
130    for (File f : files) {
131      if (f.getName().endsWith(".xml")) {
132        final FileInputStream is = new FileInputStream(f);
133        try {      
134        Resource r = context.newXmlParser().setOutputStyle(OutputStyle.PRETTY).parse(is);
135        if (r instanceof OperationOutcome) {
136          OperationOutcome oo = (OperationOutcome) r;
137          expansions.put(ToolingExtensions.getExtension(oo,VS_ID_EXT).getValue().toString(),
138            new ValueSetExpansionOutcome(new XhtmlComposer(true, false).composePlainText(oo.getText().getDiv())));
139        } else {
140          ValueSet vs = (ValueSet) r; 
141          expansions.put(vs.getUrl(), new ValueSetExpansionOutcome(vs, null));
142        }
143        } finally {
144          IOUtils.closeQuietly(is);
145        }
146      }
147    }
148  }
149
150  @Override
151        public ValueSetExpander getExpander() {
152                return new CacheAwareExpander();
153                // return new ValueSetExpander(valuesets, codesystems);
154        }
155
156}