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);

		}

	}

}