org.uberfire.server.FileUploadServlet Maven / Gradle / Ivy
/*
* Copyright 2015 Red Hat, Inc. and/or its affiliates.
*
* 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 org.uberfire.server;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import javax.inject.Inject;
import javax.inject.Named;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.uberfire.io.IOService;
import org.uberfire.java.nio.file.Path;
import static org.uberfire.server.UploadUriProvider.getTargetLocation;
public class FileUploadServlet
extends BaseUploadServlet {
private static final String RESPONSE_OK = "OK";
private static final String RESPONSE_FAIL = "FAIL";
private static final String RESPONSE_CONFLICT = "CONFLICT";
@Inject
@Named("ioStrategy")
private IOService ioService;
@Override
protected void doPost(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
try {
URI targetLocation = getTargetLocation(request);
finalizeResponse(response,
getFileItem(request),
targetLocation);
} catch (FileUploadException e) {
logError(e);
writeResponse(response,
RESPONSE_FAIL);
} catch (URISyntaxException e) {
logError(e);
writeResponse(response,
RESPONSE_FAIL);
}
}
private void finalizeResponse(HttpServletResponse response,
FileItem fileItem,
URI uri) throws IOException {
if (!validateAccess(uri,
response)) {
return;
}
final Path path = ioService.get(uri);
try {
ioService.startBatch(path.getFileSystem());
if (ioService.exists(path)) {
writeResponse(response,
RESPONSE_CONFLICT);
response.sendError(HttpServletResponse.SC_CONFLICT);
return;
}
writeFile(ioService,
path,
fileItem);
} finally {
ioService.endBatch();
}
writeResponse(response,
RESPONSE_OK);
}
}