ImageViewer.java

package com.constellio.app.ui.framework.components.viewers.image;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Arrays;

import javax.imageio.ImageIO;

import org.apache.commons.io.FilenameUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.tika.io.IOUtils;

import com.constellio.app.ui.entities.ContentVersionVO;
import com.constellio.app.ui.entities.RecordVO;
import com.constellio.app.ui.framework.components.resource.ConstellioResourceHandler;
import com.vaadin.annotations.JavaScript;
import com.vaadin.annotations.StyleSheet;
import com.vaadin.server.Page;
import com.vaadin.server.Resource;
import com.vaadin.server.ResourceReference;
import com.vaadin.shared.ui.label.ContentMode;
import com.vaadin.ui.CustomComponent;
import com.vaadin.ui.Label;

@JavaScript({ "theme://jquery/jquery-2.1.4.min.js", "theme://iviewer/jquery-ui.min.js", "theme://iviewer/jquery.mousewheel.min.js", "theme://iviewer/jquery.iviewer.min.js" })
@StyleSheet("theme://iviewer/jquery.iviewer.css")
public class ImageViewer extends CustomComponent {
	
	public static String[] SUPPORTED_EXTENSIONS = { "jpg", "jpeg", "png", "gif" };
	
	private static final int DEFAULT_WIDTH = 800;
	
	private static final int DEFAULT_HEIGHT = 600;
	
	private RecordVO recordVO;
	
	private String metadataCode;
	
	private ContentVersionVO contentVersionVO;
	
	private File file;

	public ImageViewer(RecordVO recordVO, String metadataCode, ContentVersionVO contentVersionVO) {
		this.recordVO = recordVO;
		this.metadataCode = metadataCode;
		this.contentVersionVO = contentVersionVO;
	}
	
	public ImageViewer(File file) {
		this.file = file;
	}

	@Override
	public void attach() {
		super.attach();
		
		try {
			int width = (int) getWidth();
			int height = (int) getHeight();
			
			if (width <= 0 || height <= 0) {
				width = DEFAULT_WIDTH;
				height = DEFAULT_HEIGHT;
			}
			
			int maxWidth = Page.getCurrent().getBrowserWindowWidth();
			if (width > maxWidth) {
				width = maxWidth;
			}
			
			String id = getConnectorId();
			String divId = id + "-viewer";
			
			Resource contentResource;
			InputStream in;
			if (recordVO != null) {
				String version = contentVersionVO.getVersion();
				String filename = contentVersionVO.getFileName();
				contentResource = ConstellioResourceHandler.createResource(recordVO.getId(), metadataCode, version, filename);
				in = contentVersionVO.getInputStreamProvider().getInputStream(getClass().getSimpleName());
			} else if (file != null) {
				contentResource = ConstellioResourceHandler.createResource(file);
				in = new FileInputStream(file);
			} else {
				contentResource = null;
				in = null;
			}
			
			if (contentResource != null) {
				BufferedImage bufferedImage;
				try {
					bufferedImage = ImageIO.read(in);
				} finally {
					IOUtils.closeQuietly(in);
				}
				
				int imageWidth = bufferedImage.getWidth();
				int imageHeight = bufferedImage.getHeight();
				
				float heightWidthRatio = (float) imageHeight / imageWidth;
				float heightF = (float) heightWidthRatio * height;
				height = (int) heightF;
				
				ResourceReference contentResourceReference = ResourceReference.create(contentResource, this, "ImageViewer.file");
				String contentURL = contentResourceReference.getURL();
				
				String divHTML = "<div id=\"" + divId + "\" class=\"viewer\" style=\"position:relative; width:" + width + "px; height:"+ height + "px;\"></div>";
				
				StringBuffer js = new StringBuffer();
				js.append("var $ = jQuery;");
				js.append("\n");
				js.append("$(document).ready(function() {");
				js.append("\n");
				js.append("    var iv1 = $('#" + divId + "').iviewer({");
				js.append("\n");
				js.append("        src: '" + contentURL + "'");
				js.append("\n");
				js.append("    });");
				js.append("\n");
				js.append("});");
				js.append("\n");
				
				setCompositionRoot(new Label(divHTML, ContentMode.HTML));
				
				com.vaadin.ui.JavaScript javascript = com.vaadin.ui.JavaScript.getCurrent();
				javascript.execute("setTimeout(function() {" + js.toString() + "}, 1)");
			}
		} catch (Throwable t) {
			// FIXME
			t.printStackTrace();
			setVisible(false);
		}
	}
	
	public static boolean isSupported(String fileName) {
		String extension = StringUtils.lowerCase(FilenameUtils.getExtension(fileName));
		return Arrays.asList(SUPPORTED_EXTENSIONS).contains(extension);
	}

}