001package org.hl7.fhir.dstu2.terminologies;
002
003/*-
004 * #%L
005 * org.hl7.fhir.dstu2
006 * %%
007 * Copyright (C) 2014 - 2019 Health Level 7
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.FileOutputStream;
056import java.io.IOException;
057import java.util.HashMap;
058import java.util.Map;
059
060import org.apache.commons.io.IOUtils;
061import org.hl7.fhir.dstu2.formats.IParser.OutputStyle;
062import org.hl7.fhir.dstu2.model.OperationOutcome;
063import org.hl7.fhir.dstu2.model.Resource;
064import org.hl7.fhir.dstu2.model.ValueSet;
065import org.hl7.fhir.dstu2.terminologies.ValueSetExpander.ValueSetExpansionOutcome;
066import org.hl7.fhir.dstu2.utils.IWorkerContext;
067import org.hl7.fhir.dstu2.utils.ToolingExtensions;
068import org.hl7.fhir.exceptions.FHIRFormatError;
069import org.hl7.fhir.utilities.Utilities;
070import org.hl7.fhir.utilities.xhtml.XhtmlComposer;
071
072public class ValueSetExpansionCache implements ValueSetExpanderFactory {
073
074  public class CacheAwareExpander implements ValueSetExpander {
075
076          @Override
077          public ValueSetExpansionOutcome expand(ValueSet source) throws ETooCostly, IOException {
078                if (expansions.containsKey(source.getUrl()))
079                        return expansions.get(source.getUrl());
080                ValueSetExpander vse = new ValueSetExpanderSimple(context, ValueSetExpansionCache.this);
081                ValueSetExpansionOutcome vso = vse.expand(source);
082                if (vso.getError() != null) {
083                  // well, we'll see if the designated server can expand it, and if it can, we'll cache it locally
084                        vso = context.expandVS(source, false);
085                        FileOutputStream s = new FileOutputStream(Utilities.path(cacheFolder, makeFile(source.getUrl())));
086                        context.newXmlParser().setOutputStyle(OutputStyle.PRETTY).compose(s, vso.getValueset());
087                        s.close();
088                }
089                expansions.put(source.getUrl(), vso);
090                return vso;
091          }
092
093    private String makeFile(String url) {
094      return url.replace("$", "").replace(":", "").replace("//", "/").replace("/", "_")+".xml";
095    }
096  }
097
098  private static final String VS_ID_EXT = "http://tools/cache";
099
100  private final Map<String, ValueSetExpansionOutcome> expansions = new HashMap<String, ValueSetExpansionOutcome>();
101  private final IWorkerContext context;
102  private final String cacheFolder;
103        
104        public ValueSetExpansionCache(IWorkerContext context) {
105    super();
106    cacheFolder = null;
107    this.context = context;
108  }
109  
110        public ValueSetExpansionCache(IWorkerContext context, String cacheFolder) throws FHIRFormatError, IOException {
111    super();
112    this.context = context;
113    this.cacheFolder = cacheFolder;
114    if (this.cacheFolder != null)
115      loadCache();
116  }
117  
118        private void loadCache() throws FHIRFormatError, IOException {
119          File[] files = new File(cacheFolder).listFiles();
120    for (File f : files) {
121      if (f.getName().endsWith(".xml")) {
122        final FileInputStream is = new FileInputStream(f);
123        try {      
124        Resource r = context.newXmlParser().setOutputStyle(OutputStyle.PRETTY).parse(is);
125        if (r instanceof OperationOutcome) {
126          OperationOutcome oo = (OperationOutcome) r;
127          expansions.put(ToolingExtensions.getExtension(oo,VS_ID_EXT).getValue().toString(),
128            new ValueSetExpansionOutcome(new XhtmlComposer(true, false).composePlainText(oo.getText().getDiv())));
129        } else {
130          ValueSet vs = (ValueSet) r; 
131          expansions.put(vs.getUrl(), new ValueSetExpansionOutcome(vs, null));
132        }
133        } finally {
134          IOUtils.closeQuietly(is);
135        }
136      }
137    }
138  }
139
140  @Override
141        public ValueSetExpander getExpander() {
142                return new CacheAwareExpander();
143                // return new ValueSetExpander(valuesets, codesystems);
144        }
145
146}