WebsiteHelper.java

package org.oregami.util;

import com.mortennobel.imagescaling.ResampleOp;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.*;

public class WebsiteHelper {

	private String rasterizeLocation = null;//"/Users/sebastian/Downloads/phantomjs/rasterize.js";
	private String phantomjsLocation = null;//"/usr/local/bin/phantomjs";
	
	private static WebsiteHelper instance = null;
	
	public WebsiteHelper(String phantomjsLocation, String rasterizeLocation) {
		this.phantomjsLocation = phantomjsLocation;
		this.rasterizeLocation = rasterizeLocation;
	}
	
	public static void init(String phantomjsLocation, String rasterizeLocation) {
		instance = new WebsiteHelper(phantomjsLocation, rasterizeLocation);
	}
	
	public static WebsiteHelper instance() {
		if (instance==null) {
			throw new RuntimeException("WebsiteHelper must be initialized");
		}
		return instance;
	}
	
	public Map<String, String> createWebsite(String url, String size) throws IOException {
		
		if (!url.startsWith("http://www.oregami.org")
				&& !url.startsWith("http://www.mobygames.com")
				&& !url.startsWith("http://www.kultpower.de")
		) {
			throw new RuntimeException("url_not_allowed");
		}
		List<String> forbiddenCharacters = new ArrayList<String>(Arrays.asList("\"", " "));
		for (String string : forbiddenCharacters) {
			if (url.indexOf(string)>0) {
				throw new RuntimeException("url_security_error_unallowed_characters");
			}
		}
		if (size==null) {
			size = "1200px";
		}
		File temp = File.createTempFile("website-screenshot", ".png"); 
		
		String command = phantomjsLocation + " " + rasterizeLocation + " "
			 + url + " " + temp.getAbsolutePath() + " " + size;
		
		System.out.println("Executing:\n" + command);

		StringBuffer output = new StringBuffer();

		Process p;
		try {
			p = Runtime.getRuntime().exec(command);
			p.waitFor();
			BufferedReader reader = new BufferedReader(new InputStreamReader(
					p.getInputStream()));

			String line = "";
			while ((line = reader.readLine()) != null) {
				output.append(line + "\n");
			}

		} catch (Exception e) {
			e.printStackTrace();
		}

		Map<String, String> ret = new TreeMap<String, String>();
		ret.put("output", output.toString());
		ret.put("command", command);
		ret.put("filename", temp.getAbsolutePath());
		
		return ret;

	}

	public byte[] resize(byte[] imageBytes, Integer targetWidth, Integer targetHeight)
			throws IOException {

		InputStream in = new ByteArrayInputStream(imageBytes);
		BufferedImage bImageFromConvert = ImageIO.read(in);
		
		if (targetHeight==null) {
			targetHeight = targetWidth * bImageFromConvert.getHeight() / bImageFromConvert.getWidth();
		}
		
		ResampleOp  resampleOp = new ResampleOp (targetWidth, targetHeight);
		BufferedImage rescaledImage = resampleOp.filter(bImageFromConvert, null);
		
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		ImageIO.write(rescaledImage, "png", baos );
		baos.flush();
		byte[] rescaledImageInByte = baos.toByteArray();
		baos.close();
		
		return rescaledImageInByte;

	}
	

	public byte[] readFile(String filename) {
		System.out.println("readFile: " + filename);
		File file = new File(filename);
		byte[] bFile = new byte[(int) file.length()];

		try {
			FileInputStream fileInputStream = new FileInputStream(file);
			// convert file into array of bytes
			fileInputStream.read(bFile);
			fileInputStream.close();
			return bFile;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;

	}

}