org.bimserver.servlets.DownloadServlet Maven / Gradle / Ivy
package org.bimserver.servlets;
/******************************************************************************
* Copyright (C) 2009-2016 BIMserver.org
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see {@literal roids = new HashSet();
for (Object key : request.getParameterMap().keySet()) {
String keyString = (String) key;
if (keyString.startsWith("download_")) {
if (!request.getParameter(keyString).equals("[off]")) {
roids.add(Long.parseLong(request.getParameter(keyString)));
}
}
}
topicId = serviceMap.getBimsie1ServiceInterface().downloadRevisions(roids, serializer.getOid(), true);
} else if (request.getParameter("compare") != null) {
SCompareType sCompareType = SCompareType.valueOf(request.getParameter("type"));
Long roid1 = Long.parseLong(request.getParameter("roid1"));
Long roid2 = Long.parseLong(request.getParameter("roid2"));
topicId = serviceMap.getServiceInterface().downloadCompareResults(serializer.getOid(), roid1, roid2, Long.valueOf(request.getParameter("mcid")), sCompareType, true);
} else if (request.getParameter("topicId") != null) {
topicId = Long.parseLong(request.getParameter("topicId"));
} else {
long roid = -1;
if (request.getParameter("roid") == null) {
if (request.getParameter("poid") != null) {
long poid = Long.parseLong(request.getParameter("poid"));
SProject projectByPoid = serviceMap.getBimsie1ServiceInterface().getProjectByPoid(poid);
if (projectByPoid == null) {
throw new UserException("Project with oid " + poid + " not found");
}
roid = projectByPoid.getLastRevisionId();
if (roid == -1) {
throw new UserException("No revisions");
}
} else {
throw new UserException("A poid or roid is required for downloading");
}
} else {
roid = Long.parseLong(request.getParameter("roid"));
}
if (request.getParameter("checkout") != null) {
topicId = serviceMap.getBimsie1ServiceInterface().checkout(roid, serializer.getOid(), true);
} else {
if (request.getParameter("classses") != null) {
Set classes = new HashSet();
for (String className : request.getParameter("classses").split(";")) {
classes.add(className);
}
Set roids = new HashSet();
roids.add(roid);
topicId = serviceMap.getBimsie1ServiceInterface().downloadByTypes(roids, "ifc2x3tc1", classes, serializer.getOid(), false, true, true, true);
} else if (request.getParameter("oids") != null) {
Set oids = new HashSet();
for (String oidString : request.getParameter("oids").split(";")) {
oids.add(Long.parseLong(oidString));
}
Set roids = new HashSet();
roids.add(roid);
topicId = serviceMap.getBimsie1ServiceInterface().downloadByOids(roids, oids, serializer.getOid(), true, true);
} else if (request.getParameter("guids") != null) {
Set guids = new HashSet();
for (String guid : request.getParameter("guids").split(";")) {
guids.add(guid);
}
Set roids = new HashSet();
roids.add(roid);
topicId = serviceMap.getBimsie1ServiceInterface().downloadByGuids(roids, guids, serializer.getOid(), false, true);
} else {
topicId = serviceMap.getBimsie1ServiceInterface().download(roid, serializer.getOid(), true, true);
}
}
}
if (topicId == -1) {
response.getWriter().println("No valid topicId");
return;
}
SDownloadResult checkoutResult = serviceMap.getBimsie1ServiceInterface().getDownloadData(topicId);
if (checkoutResult == null) {
LOGGER.error("Invalid topicId: " + topicId);
} else {
DataSource dataSource = checkoutResult.getFile().getDataSource();
PluginConfiguration pluginConfiguration = new PluginConfiguration(serviceMap.getPluginInterface().getPluginSettings(serializer.getOid()));
final ProgressTopic progressTopic = getBimServer().getNotificationsManager().getProgressTopic(topicId);
ProgressReporter progressReporter = new ProgressReporter() {
private long lastMax;
private long lastProgress;
private int stage = 3;
private Date start = new Date();
private String title = "Downloading...";
@Override
public void update(long progress, long max) {
if (progressTopic != null) {
LongActionState ds = StoreFactory.eINSTANCE.createLongActionState();
ds.setStart(start);
ds.setState(progress == max ? ActionState.FINISHED : ActionState.STARTED);
ds.setTitle(title);
ds.setStage(stage);
ds.setProgress((int) Math.round(100.0 * progress / max));
progressTopic.stageProgressUpdate(ds);
this.lastMax = max;
this.lastProgress = progress;
}
}
@Override
public void setTitle(String title) {
if (progressTopic != null) {
stage++;
this.title = title;
LongActionState ds = StoreFactory.eINSTANCE.createLongActionState();
ds.setStart(new Date());
ds.setState(lastProgress == lastMax ? ActionState.FINISHED : ActionState.STARTED);
ds.setTitle(title);
ds.setStage(stage);
ds.setProgress((int) Math.round(100.0 * lastProgress / lastMax));
progressTopic.stageProgressUpdate(ds);
}
}
};
try {
if (zip) {
if (pluginConfiguration.getString("ZipExtension") != null) {
response.setHeader("Content-Disposition", "inline; filename=\"" + dataSource.getName() + "." + pluginConfiguration.getString(SerializerPlugin.ZIP_EXTENSION) + "\"");
} else {
response.setHeader("Content-Disposition", "inline; filename=\"" + dataSource.getName() + ".zip" + "\"");
}
response.setContentType("application/zip");
String nameInZip = dataSource.getName() + "." + pluginConfiguration.getString(SerializerPlugin.EXTENSION);
ZipOutputStream zipOutputStream = new ZipOutputStream(outputStream);
zipOutputStream.putNextEntry(new ZipEntry(nameInZip));
processDataSource(zipOutputStream, dataSource, progressReporter);
try {
zipOutputStream.finish();
} catch (IOException e) {
// Sometimes it's already closed, that's no problem
}
} else {
if (request.getParameter("mime") == null) {
response.setContentType(pluginConfiguration.getString(SerializerPlugin.CONTENT_TYPE));
response.setHeader("Content-Disposition", "inline; filename=\"" + dataSource.getName() + "." + pluginConfiguration.getString(SerializerPlugin.EXTENSION) + "\"");
} else {
response.setContentType(request.getParameter("mime"));
}
processDataSource(outputStream, dataSource, progressReporter);
}
} catch (SerializerException s) {
LOGGER.error("", s);
LongActionState ds = StoreFactory.eINSTANCE.createLongActionState();
ds.setStart(new Date());
ds.setState(ActionState.AS_ERROR);
ds.setTitle("Serialization Error");
ds.setProgress(-1);
ds.setStage(3);
ds.getErrors().add(s.getMessage());
progressTopic.stageProgressUpdate(ds);
}
}
}
if (outputStream instanceof GZIPOutputStream) {
((GZIPOutputStream) outputStream).finish();
}
outputStream.flush();
} catch (NumberFormatException e) {
LOGGER.error("", e);
response.getWriter().println("Some number was incorrectly formatted");
} catch (ServiceException e) {
LOGGER.error("", e);
response.getWriter().println(e.getUserMessage());
} catch (EOFException e) {
} catch (Exception e) {
LOGGER.error("", e);
}
}
private void processDataSource(OutputStream outputStream, DataSource dataSource, ProgressReporter progressReporter) throws IOException, SerializerException, BimserverDatabaseException {
if (dataSource instanceof ExtendedDataSource) {
((ExtendedDataSource) dataSource).writeToOutputStream(outputStream, progressReporter);
} else {
throw new SerializerException("Unsupported datasource type: " + dataSource);
}
}
} © 2015 - 2024 Weber Informatics LLC | Privacy Policy