/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is part of dcm4che, an implementation of DICOM(TM) in
* Java(TM), hosted at https://github.com/gunterze/dcm4che.
*
* The Initial Developer of the Original Code is
* Agfa Healthcare.
* Portions created by the Initial Developer are Copyright (C) 2011
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* See @authors listed below
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
package org.dcm4chee.archive.qido.client;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import javax.json.Json;
import javax.ws.rs.core.MediaType;
import org.dcm4che3.data.Attributes;
import org.dcm4che3.data.Tag;
import org.dcm4che3.io.SAXReader;
import org.dcm4che3.json.JSONReader;
import org.dcm4che3.json.JSONReader.Callback;
import org.dcm4che3.util.SafeClose;
import org.dcm4chee.archive.conf.ArchiveAEExtension;
import org.dcm4chee.storage.conf.Availability;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* @author Hesham Elbadawi <bsdreko@gmail.com>
*
*/
public class QidoClient {
private static final Logger LOG = LoggerFactory.getLogger(QidoClient.class);
private QidoContext context;
public QidoClient(QidoContext context) {
super();
this.context = context;
}
public QidoResponse verifyStorage(Collection<String> sopInstanceUIDs) {
HashMap<String, Availability> result = new HashMap<String, Availability>();
ArchiveAEExtension aeExt = context.getArchiveAEExtension();
String aeTitle = context.getRemoteAE().getAETitle();
String url = adjustToQidoURL(aeTitle, context.getRemoteBaseURL());
if(aeExt == null) {
throw new IllegalArgumentException("ArchiveAEExtension "
+ "not initialized in context");
}
MediaType type = MediaType.valueOf(aeExt.getQidoClientAcceptType()
!=null ? aeExt.getQidoClientAcceptType(): "application/json");
for (String sopiuid : sopInstanceUIDs) {
Availability externalAvailability = queryOverWebService(aeTitle
, url, sopiuid , context.isFuzzyMatching(),context
.isTimeZoneAdjustment(), type);
result.put(sopiuid, externalAvailability);
}
return new QidoResponse(this.context.getTransactionID()
, result);
}
private Availability queryOverWebService(String aeTitle, String url
, String sopUID, boolean fuzzyMatching, boolean timeZoneAdjustment
, MediaType type) {
HttpURLConnection connection = null;
Availability externalAvaility = Availability.UNAVAILABLE;
try {
url+="?SOPInstanceUID=" + sopUID;
if(timeZoneAdjustment)
url+="&timezoneadjustment=true";
if(fuzzyMatching)
url+="&fuzzymatching=true";
URL qidoURL = new URL(url);
connection = setUpQidoConnection(qidoURL, type);
InputStream in = connection.getInputStream();
if(type.isCompatible(MediaType.APPLICATION_JSON_TYPE))
externalAvaility = readJSON(in);
else
externalAvaility = readXML(in);
} catch (IOException e) {
LOG.error("Error writing to http data output stream"
+ e.getStackTrace().toString());
return Availability.UNAVAILABLE;
}
finally {
connection.disconnect();
}
return externalAvaility;
}
private Availability readXML(InputStream in) {
String full="";
String str;
BufferedReader reader = new BufferedReader(new InputStreamReader(in));
String boundary = null;
try {
boundary = reader.readLine();
while((str = reader.readLine())!=null) {
full+=str;
}
} catch (IOException e) {
LOG.error("Failed to read Input Stream {}", e);
}
finally{
try {
reader.close();
} catch (IOException e) {
LOG.error("Failed to close Stream {}", e);
}
}
String[] parts = full.split(boundary);
Attributes attrs = new Attributes();
for(int i=0;i<parts.length-1;i++) {
try {
attrs = SAXReader.parse(new ByteArrayInputStream(removeXMLHeader(parts[i]).getBytes()));
} catch (Exception e) {
LOG.error("Error while parsing XML stream", e);
}
if(attrs.getString(Tag.InstanceAvailability) != null)
return Availability.valueOf(attrs.getString(
Tag.InstanceAvailability));
}
return Availability.UNAVAILABLE;
}
private Availability readJSON(InputStream in) {
try {
JSONReader reader = null;
try {
reader = new JSONReader(
Json.createParser(new InputStreamReader(in, "UTF-8")));
} catch (UnsupportedEncodingException e) {
LOG.error("Unsupported encoding exception"
+ " while parsing json stream", e);
}
final ArrayList<Attributes> attrs = new ArrayList<Attributes>();
reader.readDatasets(new Callback() {
@Override
public void onDataset(Attributes fmi, Attributes dataset) {
attrs.add(dataset);
}
});
if(attrs.get(0).getString(Tag.InstanceAvailability) != null)
return Availability.valueOf(attrs.get(0).getString(
Tag.InstanceAvailability));
} finally {
SafeClose.close(in);
}
return Availability.UNAVAILABLE;
}
private HttpURLConnection setUpQidoConnection(URL url , MediaType type)
throws IOException{
HttpURLConnection connection = (HttpURLConnection) url
.openConnection();
connection.setDoOutput(true);
connection.setDoInput(true);
connection.setInstanceFollowRedirects(false);
connection.setRequestMethod("GET");
connection.setRequestProperty("Accept", type.toString());
connection.setRequestProperty("charset", "utf-8");
connection.setUseCaches(false);
return connection;
}
private String adjustToQidoURL(String aeTitle, String remoteBaseURL) {
String qidoPath = "qido/" + aeTitle + "/instances";
return remoteBaseURL.endsWith("/") ? remoteBaseURL + qidoPath : remoteBaseURL + "/" + qidoPath;
}
private String removeXMLHeader(String str) {
String buff="";
for(int i=0;i<str.length();i++)
if(str.charAt(i) == '<') {
if(str.charAt(i+1)=='?') {
buff+=str.substring(i,str.length());
break;
}
}
return buff;
}
}