/**
* Copyright (C) 2010-2017 Structr GmbH
*
* This file is part of Structr <http://structr.org>.
*
* Structr 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.
*
* Structr 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 Structr. If not, see <http://www.gnu.org/licenses/>.
*/
package org.structr.websocket.command;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.structr.common.error.FrameworkException;
import org.structr.core.property.PropertyMap;
import org.structr.dynamic.File;
import org.structr.web.common.FileHelper;
import org.structr.web.entity.FileBase;
//~--- classes ----------------------------------------------------------------
/**
*
*
*/
public class FileUploadHandler {
private static final Logger logger = LoggerFactory.getLogger(FileUploadHandler.class.getName());
//~--- fields ---------------------------------------------------------
private FileBase file = null;
private FileChannel privateFileChannel = null;
private Long size = 0L;
//~--- constructors ---------------------------------------------------
public FileUploadHandler(final FileBase file) {
this.size = file.getProperty(File.size);
this.file = file;
if (this.size == null) {
FileChannel channel;
try {
channel = getChannel(false);
this.size = channel.size();
updateSize(this.size);
} catch (IOException ex) {
logger.error("Could not access file", ex);
}
}
}
//~--- methods --------------------------------------------------------
public void handleChunk(int sequenceNumber, int chunkSize, byte[] data, int chunks) throws IOException {
FileChannel channel = getChannel(sequenceNumber > 0);
if (channel != null) {
channel.position(sequenceNumber * chunkSize);
channel.write(ByteBuffer.wrap(data));
if (this.size == null) {
this.size = channel.size();
}
// finish upload
if (sequenceNumber + 1 == chunks) {
finish();
updateSize(this.size);
}
}
}
private void updateSize(final Long size) {
if (size == null) {
return;
}
try {
file.unlockSystemPropertiesOnce();
file.setProperties(file.getSecurityContext(), new PropertyMap(File.size, size));
} catch (FrameworkException ex) {
logger.warn("Could not set size to " + size, ex);
}
}
/**
* Called when the WebSocket connection is closed
*/
public void finish() {
try {
FileChannel channel = getChannel(false);
if (channel != null && channel.isOpen()) {
channel.force(true);
channel.close();
this.privateFileChannel = null;
//file.increaseVersion();
file.notifyUploadCompletion();
}
} catch (IOException e) {
logger.warn("Unable to finish file upload", e);
}
}
//~--- get methods ----------------------------------------------------
// ----- private methods -----
private FileChannel getChannel(final boolean append) throws IOException {
if (this.privateFileChannel == null) {
String relativeFilePath = file.getRelativeFilePath();
if (relativeFilePath != null) {
if (relativeFilePath.contains("..")) {
throw new IOException("Security violation: File path contains ..");
}
String filePath = FileHelper.getFilePath(relativeFilePath);
java.io.File fileOnDisk = new java.io.File(filePath);
fileOnDisk.getParentFile().mkdirs();
this.privateFileChannel = new FileOutputStream(fileOnDisk, append).getChannel();
}
}
return this.privateFileChannel;
}
}