org.mayocat.rest.resources.AbstractImageResource.java Source code

Introduction

Here is the source code for org.mayocat.rest.resources.AbstractImageResource.java

Source

/*
 * Copyright (c) 2012, Mayocat <hello@mayocat.org>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.mayocat.rest.resources;

import com.google.common.base.Optional;
import java.awt.Dimension;
import java.awt.Rectangle;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import javax.inject.Inject;
import javax.inject.Provider;
import javax.servlet.ServletContext;
import javax.ws.rs.core.Response;
import org.mayocat.attachment.model.Attachment;
import org.mayocat.attachment.model.LoadedAttachment;
import org.mayocat.attachment.store.AttachmentStore;
import org.mayocat.image.ImageService;
import org.mayocat.rest.parameters.ImageOptions;
import org.slf4j.Logger;

/**
 * @version $Id: fdd665b717beb3d50d19336a71ea1216b826c7bd $
 */
public class AbstractImageResource {
    @Inject
    private ImageService imageService;

    @Inject
    private Logger logger;

    @Inject
    protected Provider<AttachmentStore> attachmentStore;

    private static final List<String> IMAGE_EXTENSIONS = new ArrayList<String>();

    static {
        IMAGE_EXTENSIONS.add("jpg");
        IMAGE_EXTENSIONS.add("jpeg");
        IMAGE_EXTENSIONS.add("gif");
        IMAGE_EXTENSIONS.add("png");
    }

    public Response downloadThumbnail(String slug, String extension, Integer x, Integer y, Integer width,
            Integer height, ServletContext servletContext, Optional<ImageOptions> imageOptions) {
        return this.downloadThumbnail(null, slug, extension, x, y, width, height, servletContext, imageOptions);
    }

    public Response downloadThumbnail(String tenantSlug, String slug, String extension, Integer x, Integer y,
            Integer width, Integer height, ServletContext servletContext, Optional<ImageOptions> imageOptions) {
        if (!IMAGE_EXTENSIONS.contains(extension)) {
            // Refuse to treat a request with image options for a non-image attachment
            return Response.status(Response.Status.BAD_REQUEST).entity("Not an image").build();
        }

        String fileName = slug + "." + extension;
        Attachment file;

        if (tenantSlug == null) {
            file = this.attachmentStore.get().findBySlugAndExtension(slug, extension);
        } else {
            file = this.attachmentStore.get().findByTenantAndSlugAndExtension(tenantSlug, slug, extension);
        }

        if (file == null) {
            return Response.status(Response.Status.NOT_FOUND).build();
        }

        try {
            Rectangle boundaries = new Rectangle(x, y, width, height);

            if (imageOptions.isPresent()) {
                Optional<Dimension> newDimension = imageService.newDimension(boundaries,
                        imageOptions.get().getWidth(), imageOptions.get().getHeight());

                Dimension dimensions = newDimension.or(
                        new Dimension(imageOptions.get().getWidth().or(-1), imageOptions.get().getHeight().or(-1)));

                return Response
                        .ok(imageService.getImage(file, dimensions, boundaries),
                                servletContext.getMimeType(fileName))
                        .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
            } else {
                return Response.ok(imageService.getImage(file, boundaries), servletContext.getMimeType(fileName))
                        .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
            }
        } catch (IOException e) {
            this.logger.warn("Failed to scale image for attachment [{slug}]", slug);
            return Response.serverError().entity("Failed to scale image").build();
        }
    }

    public Response downloadImage(String slug, String extension, ServletContext servletContext,
            Optional<ImageOptions> imageOptions) {
        return this.downloadImage(null, slug, extension, servletContext, imageOptions);
    }

    public Response downloadImage(String tenantSlug, String slug, String extension, ServletContext servletContext,
            Optional<ImageOptions> imageOptions) {
        if (!IMAGE_EXTENSIONS.contains(extension)) {
            // Refuse to treat a request with image options for a non-image attachment
            return Response.status(Response.Status.BAD_REQUEST).entity("Not an image").build();
        }

        if (imageOptions.isPresent()) {

            String fileName = slug + "." + extension;
            Attachment file;
            if (tenantSlug == null) {
                file = this.attachmentStore.get().findBySlugAndExtension(slug, extension);
            } else {
                file = this.attachmentStore.get().findByTenantAndSlugAndExtension(tenantSlug, slug, extension);
            }

            if (file == null) {
                return Response.status(Response.Status.NOT_FOUND).build();
            }

            try {
                if (imageOptions.get().getHeight().isPresent() && imageOptions.get().getWidth().isPresent()) {
                    // Both width and height set -> calculate a fitting box

                    Dimension dimension = new Dimension(imageOptions.get().getWidth().get(),
                            imageOptions.get().getHeight().get());
                    Optional<Rectangle> fittingBox = imageService.getFittingRectangle(file, dimension);

                    InputStream image;
                    if (fittingBox.isPresent()) {
                        image = imageService.getImage(file, dimension, fittingBox.get());
                    } else {
                        image = imageService.getImage(file, dimension);
                    }
                    return Response.ok(image, servletContext.getMimeType(fileName))
                            .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
                } else {
                    Optional<Dimension> newDimension = imageService.newDimension(file,
                            imageOptions.get().getWidth(), imageOptions.get().getHeight());

                    if (newDimension.isPresent()) {
                        return Response
                                .ok(imageService.getImage(file, newDimension.get()),
                                        servletContext.getMimeType(fileName))
                                .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
                    }

                    // data stream has been consumed, load it again
                    file = this.attachmentStore.get().findAndLoadBySlugAndExtension(slug, extension);

                    LoadedAttachment loadedAttachment = attachmentStore.get().findAndLoadById(file.getId());
                    return Response.ok(loadedAttachment.getData().getStream(), servletContext.getMimeType(fileName))
                            .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
                }
            } catch (IOException e) {
                this.logger.warn("Failed to scale image for attachment [{slug}]", slug);
                return Response.serverError().entity("Failed to scale image").build();
            }
        }

        return this.downloadFile(tenantSlug, slug, extension, servletContext);
    }

    public Response downloadFile(String slug, String extension, ServletContext servletContext) {
        return this.downloadFile(null, slug, extension, servletContext);
    }

    public Response downloadFile(String tenantSlug, String slug, String extension, ServletContext servletContext) {
        String fileName = slug + "." + extension;
        LoadedAttachment file;
        if (tenantSlug == null) {
            file = this.attachmentStore.get().findAndLoadBySlugAndExtension(slug, extension);
        } else {
            file = this.attachmentStore.get().findAndLoadByTenantAndSlugAndExtension(tenantSlug, slug, extension);
        }
        if (file == null) {
            return Response.status(Response.Status.NOT_FOUND).build();
        }

        return Response.ok(file.getData().getStream(), servletContext.getMimeType(fileName))
                .header("Content-disposition", "inline; filename*=utf-8''" + fileName).build();
    }
}