jhi.buntata.server.DatasourceIcon.java Source code

Introduction

Here is the source code for jhi.buntata.server.DatasourceIcon.java

Source

/*
 * Copyright 2016 Information & Computational Sciences, The James Hutton Institute
 *
 * 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 jhi.buntata.server;

import org.apache.commons.fileupload.*;
import org.apache.commons.fileupload.disk.*;
import org.restlet.data.*;
import org.restlet.data.Status;
import org.restlet.ext.fileupload.*;
import org.restlet.representation.*;
import org.restlet.resource.*;

import java.io.*;
import java.nio.file.*;

import jhi.buntata.data.*;
import jhi.buntata.resource.*;

/**
 * {@link ServerResource} handling {@link BuntataDatasource} icon requests.
 *
 * @author Sebastian Raubach
 */
public class DatasourceIcon extends ServerResource {
    private static String dataDir;

    private final DatasourceDAO dao = new DatasourceDAO();
    private Long id = null;

    @Override
    public void doInit() {
        super.doInit();

        try {
            this.id = Long.parseLong(getRequestAttributes().get("id").toString());
        } catch (NullPointerException | NumberFormatException e) {
        }
    }

    @Put
    public void putImage(Representation entity) {
        if (entity != null && MediaType.MULTIPART_FORM_DATA.equals(entity.getMediaType(), true)) {
            try {
                DiskFileItemFactory factory = new DiskFileItemFactory();
                RestletFileUpload upload = new RestletFileUpload(factory);
                FileItemIterator fileIterator = upload.getItemIterator(entity);

                if (fileIterator.hasNext()) {
                    FileItemStream fi = fileIterator.next();

                    String name = fi.getName();

                    File file = new File(dataDir, name);

                    int i = 1;
                    while (file.exists())
                        file = new File(dataDir, (i++) + name);

                    Files.copy(fi.openStream(), file.toPath());

                    BuntataDatasource ds = dao.get(id);
                    ds.setIcon(file.getName());
                    dao.updateIcon(ds);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            throw new ResourceException(Status.CLIENT_ERROR_UNSUPPORTED_MEDIA_TYPE);
        }
    }

    @Get
    public FileRepresentation getImage() {
        BuntataDatasource datasource = dao.get(id);

        FileRepresentation representation = null;

        if (datasource != null) {
            // Get the icon
            String icon = datasource.getIcon();

            if (icon != null) {
                File file = new File(dataDir, icon);

                // Check if the icon exists
                if (file.exists() && file.isFile()) {
                    MediaType mt = MediaType.IMAGE_ALL;
                    representation = new FileRepresentation(file, mt);

                    Disposition disp = new Disposition(Disposition.TYPE_ATTACHMENT);
                    disp.setFilename(file.getName());
                    disp.setSize(file.length());
                    representation.setDisposition(disp);
                }
            }
        }

        // Return the result
        return representation;
    }

    public static void setDataDir(String dataDir) {
        DatasourceIcon.dataDir = dataDir;
    }
}