package cz.frantovo.nekurak.servlet;

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.IOException;

 import java.io.InputStream;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import java.util.regex.Pattern;

 import javax.servlet.ServletException;

 import javax.servlet.ServletOutputStream;

 import javax.servlet.http.HttpServlet;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 /**

  * <p>Servlet pro zpřístupnění fotek, které se nacházejí ve zvláštním adresáři.</p>

  *

  * <p>Má jeden inicializační parametr:</p>

  *  <ul>

  *     <li><code>adresar</code> – cesta k adresáři na disku, kde se nacházejí fotky.</li>

  *     <li>např. <code>/var/www/nekurak.net/fotky</code></li>

  *  </ul>

  *

  * <p>Očekává se struktura adresářů tohoto typu:</p>

  *  <ul>

  *	<li>Plné rozlišení: <code>/var/www/nekurak.net/fotky/original/1.jpg</code></li>

  *      <li>Náhled fotky:   <code>/var/www/nekurak.net/fotky/nahled/1.jpg</code></li>

  *  </ul>

  * <p>Lze ovlivnit konstantami níže.</p>

  *

  * <p>URL respektují fyzické umístění fotek na serveru (není např. ID v GET parametru stylem <code>/fotky?id=123</code>),

  * díky tomu je možné, aby fotky později servírovala přímo reverzní proxy a nemusely procházet přes naši aplikaci

  * (fotky jsou veřejné, není potřeba řešit autorizaci)</p>

  *

  * @author fiki

  */

 public class Fotky extends HttpServlet {

 	/** Název inicializačního parametru */

 	private static final String INIT_ADRESAR = "adresar";

 	/** Název podadresáře obsahujícího fotku v plném rozlišení */

 	public static final String PODADRESAR_ORIGINAL = "original";

 	/** Název podadresáře obsahujícího výchozí náhled fotky */

 	public static final String PODADRESAR_NAHLED = "nahled";

 	public static final String PRIPONA = "jpg";

 	private static final String LOMITKO = File.separator;

 	/** Regulární výraz */

 	private static final String VZOR_CESTY = "^" + LOMITKO + "(" + PODADRESAR_ORIGINAL + "|" + PODADRESAR_NAHLED + ")" + LOMITKO + "\\d+\\." + PRIPONA + "$";

 	private static final String MIME_TYP = "image/jpeg";

 	private File adresar;

 	private static final Logger log = Logger.getLogger(Fotky.class.getSimpleName());

 	@Override

 	public void init() throws ServletException {

 		super.init();

 		String initAdresar = getServletConfig().getInitParameter(INIT_ADRESAR);

 		adresar = new File(initAdresar);

 		if (adresar.isDirectory()) {

 			log.log(Level.INFO, "Servlet „Fotka“ byl úspěšně inicializován.");

 			log.log(Level.INFO, "Adresář s fotkami: {0}", initAdresar);

 			log.log(Level.INFO, "RegExp cesty: {0}", VZOR_CESTY);

 		} else {

 			throw new ServletException("Servlet „Fotka“ se nepodařilo inicializovat. Cesta: " + initAdresar);

 		}

 	}

 	/**

 	 * @param pozadavek pouze GET (není důvod podporovat POST)

 	 * @param odpoved odešleme fotku s MIME typem podle konstanty, délkou a datem podle souboru.

 	 * @throws ServletException pokud je požadovaná cesta chybná (nevyhovuje vzoru)

 	 * @throws IOException

 	 */

 	@Override

 	protected void doGet(HttpServletRequest pozadavek, HttpServletResponse odpoved) throws ServletException, IOException {

 		String cesta = zkontrolujParametr(pozadavek.getPathInfo());

 		File soubor = new File(adresar, cesta);

 		if (soubor.isFile() && soubor.canRead()) {

 			if (soubor.lastModified() > pozadavek.getDateHeader("If-Modified-Since")) {

 				/** Soubor se změnil nebo ho klient ještě nemá načtený. */

 				odpoved.setContentType(MIME_TYP);

 				odpoved.setContentLength((int) soubor.length());

 				odpoved.setDateHeader("Last-Modified", soubor.lastModified());

 				ServletOutputStream vystup = odpoved.getOutputStream();

 				InputStream vstup = new FileInputStream(soubor);

 				try {

 					byte[] zasobnik = new byte[1024];

 					int bajtuNacteno;

 					while ((bajtuNacteno = vstup.read(zasobnik)) != -1) {

 						vystup.write(zasobnik, 0, bajtuNacteno);

 					}

 				} catch (Exception e) {

 					throw new ServletException("Chyba při odesílání obrázku klientovi.", e);

 				} finally {

 					vstup.close();

 					vystup.close();

 				}

 			} else {

 				/** Soubor se od posledního načtení klientem nezměnil → není potřeba ho posílat znova. */

 				odpoved.setStatus(HttpServletResponse.SC_NOT_MODIFIED);

 			}

 		} else {

 			/** Neexistující nebo nečitelný soubor → HTTP 404 chyba */

 			odpoved.sendError(HttpServletResponse.SC_NOT_FOUND);

 		}

 	}

 	/**

 	 * @param cesta cesta požadovaná klientem: <code>request.getPathInfo()</code>

 	 * @throws ServletException pokud cesta nevyhovuje vzoru

 	 */

 	private static String zkontrolujParametr(String cesta) throws ServletException {

 		if (Pattern.matches(VZOR_CESTY, cesta)) {

 			/** cesta je v pořádku → pokračujeme */

 			return cesta;

 		} else {

 			/** Chybná cesta → HTTP 500 chyba */

 			throw new ServletException("Chybná cesta k obrázku: " + cesta);

 		}

 	}

 }