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/*
033Copyright (c) 2011+, HL7, Inc
034All rights reserved.
035
036Redistribution and use in source and binary forms, with or without modification, 
037are permitted provided that the following conditions are met:
038
039 * Redistributions of source code must retain the above copyright notice, this 
040   list of conditions and the following disclaimer.
041 * Redistributions in binary form must reproduce the above copyright notice, 
042   this list of conditions and the following disclaimer in the documentation 
043   and/or other materials provided with the distribution.
044 * Neither the name of HL7 nor the names of its contributors may be used to 
045   endorse or promote products derived from this software without specific 
046   prior written permission.
047
048THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
049ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
050WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 
051IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
052INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 
053NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
054PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
055WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
056ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
057POSSIBILITY OF SUCH DAMAGE.
058
059*/
060
061import java.io.File;
062import java.io.FileInputStream;
063import java.io.FileOutputStream;
064import java.io.IOException;
065import java.util.HashMap;
066import java.util.Map;
067
068import org.apache.commons.io.IOUtils;
069import org.hl7.fhir.dstu2.formats.IParser.OutputStyle;
070import org.hl7.fhir.dstu2.model.OperationOutcome;
071import org.hl7.fhir.dstu2.model.Resource;
072import org.hl7.fhir.dstu2.model.ValueSet;
073import org.hl7.fhir.dstu2.terminologies.ValueSetExpander.ValueSetExpansionOutcome;
074import org.hl7.fhir.dstu2.utils.IWorkerContext;
075import org.hl7.fhir.dstu2.utils.ToolingExtensions;
076import org.hl7.fhir.exceptions.FHIRFormatError;
077import org.hl7.fhir.utilities.Utilities;
078import org.hl7.fhir.utilities.filesystem.ManagedFileAccess;
079import org.hl7.fhir.utilities.xhtml.XhtmlComposer;
080
081public class ValueSetExpansionCache implements ValueSetExpanderFactory {
082
083  public class CacheAwareExpander implements ValueSetExpander {
084
085    @Override
086    public ValueSetExpansionOutcome expand(ValueSet source) throws ETooCostly, IOException {
087      if (expansions.containsKey(source.getUrl()))
088        return expansions.get(source.getUrl());
089      ValueSetExpander vse = new ValueSetExpanderSimple(context, ValueSetExpansionCache.this);
090      ValueSetExpansionOutcome vso = vse.expand(source);
091      if (vso.getError() != null) {
092        // well, we'll see if the designated server can expand it, and if it can, we'll
093        // cache it locally
094        vso = context.expandVS(source, false);
095        FileOutputStream s = ManagedFileAccess.outStream(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 = ManagedFileAccess.file(cacheFolder).listFiles();
130    for (File f : files) {
131      if (f.getName().endsWith(".xml")) {
132        final FileInputStream is = ManagedFileAccess.inStream(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}