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: " + initAdresar);

	    log.log(Level.INFO, "RegExp cesty: " + 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);

	}

    }

}