/*
*
* Copyright 2011 Luca Molino (molino.luca--AT--gmail.com *
* Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.orientechnologies.orient.server.network.protocol.http.multipart;
import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringWriter;
import java.util.Map;
import com.orientechnologies.orient.core.db.record.ODatabaseRecord;
import com.orientechnologies.orient.core.serialization.serializer.OJSONWriter;
import com.orientechnologies.orient.server.network.protocol.http.OHttpRequest;
import com.orientechnologies.orient.server.network.protocol.http.OHttpUtils;
/**
* @author luca.molino
*
*/
public class OHttpMultipartFileToDiskContentParser implements OHttpMultipartContentParser<InputStream> {
protected boolean overwrite = false;
protected String path;
public OHttpMultipartFileToDiskContentParser(String iPath) {
path = iPath;
if (!path.endsWith("/"))
path += "/";
new File(path).mkdirs();
}
@Override
public InputStream parse(final OHttpRequest iRequest, final Map<String, String> headers,
final OHttpMultipartContentInputStream in, ODatabaseRecord database) throws IOException {
final StringWriter buffer = new StringWriter();
final OJSONWriter json = new OJSONWriter(buffer);
json.beginObject();
String fileName = headers.get(OHttpUtils.MULTIPART_CONTENT_FILENAME);
int fileSize = 0;
if (fileName.charAt(0) == '"')
fileName = fileName.substring(1);
if (fileName.charAt(fileName.length() - 1) == '"')
fileName = fileName.substring(0, fileName.length() - 1);
fileName = path + fileName;
if (!overwrite)
// CHANGE THE FILE NAME TO AVOID OVERWRITING
if (new File(fileName).exists()) {
final String fileExt = fileName.substring(fileName.lastIndexOf("."));
final String fileNoExt = fileName.substring(0, fileName.lastIndexOf("."));
for (int i = 1;; ++i) {
if (!new File(fileNoExt + "_" + i + fileExt).exists()) {
fileName = fileNoExt + "_" + i + fileExt;
break;
}
}
}
// WRITE THE FILE
final OutputStream out = new BufferedOutputStream(new FileOutputStream(fileName.toString()));
try {
int b;
while ((b = in.read()) > -1) {
out.write(b);
fileSize++;
}
} finally {
out.flush();
out.close();
}
// FORMAT THE RETURNING DOCUMENT
json.writeAttribute(1, true, "name", fileName);
json.writeAttribute(1, true, "type", headers.get(OHttpUtils.MULTIPART_CONTENT_TYPE));
json.writeAttribute(1, true, "size", fileSize);
json.endObject();
return new ByteArrayInputStream(buffer.toString().getBytes());
}
public boolean isOverwrite() {
return overwrite;
}
public OHttpMultipartFileToDiskContentParser setOverwrite(boolean overwrite) {
this.overwrite = overwrite;
return this;
}
}