org.apache.wicket.protocol.http.servlet.ServletPartFileItem.java Source code

Introduction

Here is the source code for org.apache.wicket.protocol.http.servlet.ServletPartFileItem.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.wicket.protocol.http.servlet;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.util.Collection;

import javax.servlet.http.Part;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemHeaders;
import org.apache.commons.fileupload.util.FileItemHeadersImpl;
import org.apache.wicket.WicketRuntimeException;
import org.apache.wicket.request.resource.AbstractResource;
import org.apache.wicket.util.io.IOUtils;
import org.apache.wicket.util.lang.Args;
import org.apache.wicket.util.string.Strings;

/**
 * An adapter of Servlet 3.0 {@link Part} to Apache Commons FileUpload's {@link FileItem}
 */
class ServletPartFileItem implements FileItem {
    /**
     * The adapted part
     */
    private final Part part;
    private final boolean isFormField;

    /**
     * Constructor
     */
    ServletPartFileItem(Part part) {
        Args.notNull(part, "part");
        this.part = part;

        String contentType = part.getContentType();
        this.isFormField = contentType == null;
    }

    @Override
    public InputStream getInputStream() throws IOException {
        return part.getInputStream();
    }

    @Override
    public String getContentType() {
        return part.getContentType();
    }

    @Override
    public String getName() {
        return getFileName(part);
    }

    private String getFileName(Part part) {
        String contentDisposition = part.getHeader(AbstractResource.CONTENT_DISPOSITION_HEADER_NAME);
        for (String cd : Strings.split(contentDisposition, ';')) {
            if (cd.trim().startsWith("filename")) {
                return cd.substring(cd.indexOf('=') + 1).trim().replace("\"", "");
            }
        }
        return null;
    }

    @Override
    public boolean isInMemory() {
        return true;
    }

    @Override
    public long getSize() {
        return part.getSize();
    }

    @Override
    public byte[] get() {
        try {
            return IOUtils.toByteArray(getInputStream());
        } catch (IOException iox) {
            throw new WicketRuntimeException("Could not read upload's part input stream", iox);
        }
    }

    @Override
    public String getString(String encoding) throws UnsupportedEncodingException {
        byte[] bytes = get();
        return new String(bytes, encoding);
    }

    @Override
    public String getString() {
        try {
            return getString("UTF-8");
        } catch (UnsupportedEncodingException uex) {
            throw new WicketRuntimeException("UTF-8 must be supported", uex);
        }
    }

    @Override
    public void write(File file) throws Exception {
        part.write(file.getName());
    }

    @Override
    public void delete() {
        try {
            part.delete();
        } catch (IOException iox) {
            throw new WicketRuntimeException("A problem occurred while deleting an upload part", iox);
        }
    }

    @Override
    public String getFieldName() {
        return part.getName();
    }

    @Override
    public void setFieldName(String name) {
        throw new UnsupportedOperationException("setFieldName");
    }

    @Override
    public boolean isFormField() {
        return isFormField;
    }

    @Override
    public void setFormField(boolean state) {
        throw new UnsupportedOperationException("setFormField");
    }

    @Override
    public OutputStream getOutputStream() throws IOException {
        throw new UnsupportedOperationException("getOutputStream");
    }

    @Override
    public FileItemHeaders getHeaders() {
        FileItemHeadersImpl fileItemHeaders = new FileItemHeadersImpl();
        for (String headerName : part.getHeaderNames()) {
            Collection<String> headerValues = part.getHeaders(headerName);
            for (String headerValue : headerValues) {
                fileItemHeaders.addHeader(headerName, headerValue);
            }
        }
        return fileItemHeaders;
    }

    @Override
    public void setHeaders(FileItemHeaders headers) {
        throw new UnsupportedOperationException("setHeaders");
    }
}