org.apache.juneau.examples.rest.TempDirResource.java Source code

Introduction

Here is the source code for org.apache.juneau.examples.rest.TempDirResource.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.juneau.examples.rest;

import static org.apache.juneau.dto.html5.HtmlBuilder.*;

import java.io.*;

import org.apache.commons.fileupload.*;
import org.apache.commons.fileupload.servlet.*;
import org.apache.juneau.dto.html5.*;
import org.apache.juneau.rest.*;
import org.apache.juneau.rest.annotation.*;
import org.apache.juneau.utils.*;

/**
 * Sample resource that extends {@link DirectoryResource} to open up the temp directory as a REST resource.
 */
@RestResource(path = "/tempDir", title = "Temp Directory View Service", description = "View and download files in the '$S{java.io.tmpdir}' directory.", pageLinks = "{up:'request:/..',options:'servlet:/?method=OPTIONS',upload:'servlet:/upload',source:'$C{Source/gitHub}/org/apache/juneau/examples/rest/TempDirResource.java'}", properties = {
        @Property(name = "rootDir", value = "$S{java.io.tmpdir}"), @Property(name = "allowViews", value = "true"),
        @Property(name = "allowDeletes", value = "true"),
        @Property(name = "allowPuts", value = "false") }, stylesheet = "styles/devops.css")
public class TempDirResource extends DirectoryResource {
    private static final long serialVersionUID = 1L;

    /**
     * [GET /upload] - Display the form entry page for uploading a file to the temp directory.
     */
    @RestMethod(name = "GET", path = "/upload")
    public Form getUploadForm(RestRequest req) {
        return form().id("form").action(req.getServletURI() + "/upload").method("POST")
                .enctype("multipart/form-data")
                .children(input().name("contents").type("file"), button("submit", "Submit"));
    }

    /**
     * [POST /upload] - Upload a file as a multipart form post.
     * Shows how to use the Apache Commons ServletFileUpload class for handling multi-part form posts.
     */
    @RestMethod(name = "POST", path = "/upload", matchers = TempDirResource.MultipartFormDataMatcher.class)
    public Redirect uploadFile(RestRequest req) throws Exception {
        ServletFileUpload upload = new ServletFileUpload();
        FileItemIterator iter = upload.getItemIterator(req);
        while (iter.hasNext()) {
            FileItemStream item = iter.next();
            if (item.getFieldName().equals("contents")) { //$NON-NLS-1$
                File f = new File(getRootDir(), item.getName());
                IOPipe.create(item.openStream(), new FileOutputStream(f)).closeOut().run();
            }
        }
        return new Redirect(); // Redirect to the servlet root.
    }

    /** Causes a 404 if POST isn't multipart/form-data */
    public static class MultipartFormDataMatcher extends RestMatcher {
        @Override /* RestMatcher */
        public boolean matches(RestRequest req) {
            String contentType = req.getContentType();
            return contentType != null && contentType.startsWith("multipart/form-data"); //$NON-NLS-1$
        }
    }
}