/******************************************************************************
    * PosterController.java - The Spring controller to fetch posters
    * 
    * BuckoVidLib - The BuckoSoft Video Library
    * Copyright(c) 2014 - Dick Balaska
    * 
    */
   package com.buckosoft.BuckoVidLib.web;
   
  import java.awt.Graphics2D;
  import java.awt.Image;
  import java.awt.image.BufferedImage;
  import java.io.IOException;
  import java.io.OutputStream;
  import java.net.URL;
  import java.util.Iterator;
  
  import javax.imageio.ImageIO;
  import javax.imageio.ImageWriter;
  import javax.imageio.stream.ImageOutputStream;
  import javax.servlet.http.HttpServletRequest;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.stereotype.Controller;
  import org.springframework.web.bind.annotation.PathVariable;
  import org.springframework.web.bind.annotation.RequestMapping;
  
  import com.buckosoft.BuckoVidLib.business.BuckoVidLib;
  import com.buckosoft.BuckoVidLib.domain.TVSeason;
  import com.buckosoft.BuckoVidLib.domain.VideoBase;
  import com.buckosoft.BuckoVidLib.util.DomainConverter;
  
  /** The Spring controller to fetch posters
   * @author dick
   * @since 2014-10-07
   */
  @Controller(value="poster")
  @RequestMapping("/poster")
  public class PosterController {
  	private final Log log = LogFactory.getLog(getClass());
  
  	@Autowired
  	private BuckoVidLib		bvl;
  
  	private	int	defaultWidth = 150;
  	private	int	defaultHeight = 225;
  	
  	@RequestMapping("/{key}")
  	public void handleRequest(@PathVariable("key") String key,
  			HttpServletRequest request,
  			OutputStream outputStream) throws Exception {
  		handleRequest(key, defaultWidth, defaultHeight, request, outputStream);
  	}
  	
  	@RequestMapping("/{key}/{width}/{height}")
  	public void handleRequest(@PathVariable("key") String key, 
  			@PathVariable("width") Integer w, @PathVariable("height") Integer h,
  			HttpServletRequest request,
  			OutputStream outputStream) {
  
  		VideoBase v;
  		TVSeason tvs;
  		String printedTitle = null;
  		String vurl;
  		String s;
  		boolean b = false;
  		if (request == null)
  			b = true;
  		else
  			b = bvl.isAllowRestricted(request);
  		v = bvl.getVideoBaseFromKey(DomainConverter.keyToInt(key), b);
  		if (v == null) {
  			tvs = bvl.getTVSeasonFromHashKey(DomainConverter.keyToInt(key));
  			if (tvs == null) {
  				s = "Video not found for key: " + key;
  				log.warn(s);
  				if (request == null)
  					throw new RuntimeException(s);
  				return;
  			} else {
  				printedTitle = tvs.getTitle();
  				vurl = bvl.getPlexUrl() + getPosterUrl(tvs.getPlexKey());
  			}
  		} else {
  			printedTitle = v.getTitle() + " (" + v.getYear() + ")";
  			vurl = bvl.getPlexUrl() + getPosterUrl(v.getPlexKey());
  		}
  		BufferedImage img;
  		try {
  			URL url = new URL(vurl);
  			img = ImageIO.read(url);
  		} catch (IOException e1) {
  			s = "Failed to read image for '" + printedTitle + "'";
  			log.warn(s);
  			log.warn("url was '" + vurl + "'");
  			if (request == null)
  				throw new RuntimeException(s, e1);
 			return;
 		}
 		log.info("Fetch poster for: " + printedTitle);
 		log.debug("original w/h: " + img.getWidth() + "/" + img.getHeight());
 
 		if (w > 1 && h > 1) {
 			// scale the image
 			int	targetw;
 			int	targeth;
 			double wp = (double)img.getWidth() / (double)w;
 			double hp = (double)img.getHeight() / (double)h;
 			log.debug("wp/hp: " + wp + " / " + hp);
 			if (hp < wp) {
 				targetw = (int)((double)img.getWidth() / hp);
 				if (targetw < w)
 					targetw = w;
 				targeth = h;
 			} else {
 				targeth = (int)((double)img.getHeight() / wp);
 				if (targeth < h)
 					targeth = h;
 				targetw = w;
 			}
 			log.debug("target w/h: " + targetw + "/" + targeth);
 			BufferedImage scaled = new BufferedImage(targetw, targeth, BufferedImage.TYPE_INT_BGR);
 			Graphics2D g;
 			g = scaled.createGraphics();
 			g.drawImage(img.getScaledInstance(targetw, targeth, Image.SCALE_SMOOTH), null, null);
 			// replace outgoing image with scaled version
 			img = scaled;
 			g.dispose();
 			// pick an exact sized slice out of the image
 			int overw = img.getWidth() - w;
 			int	overh = img.getHeight() - h;
 			int shiftw = overw/2;
 			int shifth = overh/2;
 			log.debug("final: over: " + overw + "/" + overh + " shift: " + shiftw + "/" + shifth);
 			img = img.getSubimage(shiftw, shifth, w-0, h-0);
 		}
 	
 		// write the image to the stream
 		
 		ImageOutputStream ios;
 		try {
 			ios = ImageIO.createImageOutputStream(outputStream);
 		} catch (IOException e) {
 			s = "Failed to create image for writing from " + vurl + " : " + e.getMessage();
 			log.warn(s);
 			if (request == null)
 				throw new RuntimeException(s);
 			return;
 		}
 		Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("jpg");
 		ImageWriter writer = (ImageWriter)writers.next();
 		writer.setOutput(ios);
 		try {
 			writer.write(img);
 		} catch (IOException e) {
 			s = "Failed to write image from " + vurl + " : " + e.getMessage();
 			log.warn(s);
 			log.warn("Try alternate...");
 			BufferedImage bi = new BufferedImage(img.getWidth(), img.getHeight(), BufferedImage.TYPE_INT_BGR);
 			for (int y=0; y<img.getHeight(); y++) {
 				for (int x=0; x<img.getWidth(); x++) {
 					bi.setRGB(x, y, img.getRGB(x, y));
 				}
 			}
 			try {
 				writer.write(bi);
 			} catch (IOException e1) {
 				log.warn("That failed too", e1);
 				writers = ImageIO.getImageWritersByFormatName("png");
 				writer = (ImageWriter)writers.next();
 				writer.setOutput(ios);
 				try {
 					log.warn("Try png");
 					writer.write(img);
 					ios.close();
 					ios = null;
 					// png was successful
 					log.warn("png writing successful");
 					return;
 				} catch (IOException e2) {
 					log.warn("Writing as png failed too", e2);
 					if (request == null)
 						throw new RuntimeException(s);
 					return;
 				}
 			}
 			log.warn("Alternate jpg passed");
 			return;			
 		} finally {
 			try {
 				if (ios != null)
 					ios.close();
 			} catch (IOException e) {
 				log.warn("finally tipped", e);
 			}
 		}
 	}
 	
 	private	String getPosterUrl(int plexKey) {
 		return(String.format("library/metadata/%d/thumb", plexKey));
 	}
 
 
 }