/*
*
* Copyright 2011 Luca Molino (luca.molino--AT--assetdata.it)
*
* 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.FileOutputStream;
import java.io.IOException;
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<StringWriter> {
protected String path;
public OHttpMultipartFileToDiskContentParser(String iPath) {
path = iPath;
}
@Override
public StringWriter 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 = new String(fileName.substring(1));
}
if (fileName.charAt(fileName.length() - 1) == '"') {
fileName = new String(fileName.substring(0, fileName.length() - 1));
}
final OutputStream out = new BufferedOutputStream(new FileOutputStream(path + fileName.toString()));
try {
while (in.available() > 0) {
int value = in.read();
out.write(value);
fileSize++;
}
} finally {
out.flush();
out.close();
}
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 buffer;
}
}