// This Source Code is in the Public Domain per: http://litesoft.org/License.txt
package org.litesoft.servlets;

import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;

import org.litesoft.commonfoundation.typeutils.*;
import org.litesoft.util.*;
import org.litesoft.util.template.*;

public abstract class TemplateServlet extends HttpServlet
{
    private String mCSS = "";
    private String mJS = "";

    protected TemplateServlet( String pCSShref, String... pJSsrcs )
    {
        addCSS( pCSShref );
        if ( pJSsrcs != null )
        {
            for ( String zJSsrc : pJSsrcs )
            {
                addJS( zJSsrc );
            }
        }
    }

    public void addCSS( String pCSShref )
    {
        if ( null != (pCSShref = Strings.noEmpty( pCSShref )) )
        {
            mCSS += "\n<link rel=\"stylesheet\" type=\"text/css\" href=\"" + pCSShref + "\" />";
        }
    }

    public void addJS( String pJSsrc )
    {
        if ( null != (pJSsrc = Strings.noEmpty( pJSsrc )) )
        {
            mJS += "\n<script type=\"text/javascript\" src=\"" + pJSsrc + "\"></script>";
        }
    }

    @Override
    public final void doGet( HttpServletRequest pRequest, HttpServletResponse pResponse )
            throws ServletException, IOException
    {
        HttpSession zSession = pRequest.getSession( true );

        String query = Utils.fixPercentsInURLs( Strings.noEmpty( pRequest.getQueryString() ) );
//        System.out.println( "service: " + query );
//        System.out.println( "RequestURI: " + pHttpServletRequest.getRequestURI() );
//        System.out.println( "RequestURL: " + pHttpServletRequest.getRequestURL() );
//        System.out.println( "PathInfo: " + pHttpServletRequest.getPathInfo() );
//        System.out.println( "PathTranslated: " + pHttpServletRequest.getPathTranslated() );
        TemplateControl zTC = resolveQuery( pRequest, query, pResponse, zSession );
        if ( zTC == null )
        {
            throw new ServletException( "Unable to process request: " + query );
        }
        try
        {
            if ( zTC.isForwardTo() )
            {
                String newURL = pResponse.encodeRedirectURL( zTC.getForwardTo() );
                pResponse.setStatus( HttpServletResponse.SC_SEE_OTHER );
                pResponse.setHeader( "Location", newURL );
            }
            else
            {
                zTC.add( "CSS", mCSS );
                zTC.add( "JS", mJS );
                pResponse.setCharacterEncoding( FileUtils.UTF_8 );
                pResponse.setContentType( "text/html" );
                ServletUtils.noCache( pResponse );
                Writer zWriter = new OutputStreamWriter( pResponse.getOutputStream(), FileUtils.UTF_8 );
                try
                {
                    zTC.applyTo( zWriter, null );
                }
                finally
                {
                    zWriter.close();
                }
            }
        }
        finally
        {
            zTC.dispose();
        }
    }

    protected String getTemplatePath( String pTemplateName )
    {
        String zTemplateFilePath = Strings.noEmpty( getServletConfig().getInitParameter( pTemplateName ) );
        if ( zTemplateFilePath == null )
        {
            String zTemplateBase = Strings.noEmpty( getServletConfig().getServletContext().getInitParameter( "TemplateFilePathBase" ) );
            zTemplateFilePath = ((zTemplateBase != null) ? zTemplateBase + "/" + pTemplateName : pTemplateName) + ".template.html";
        }
        if ( !zTemplateFilePath.startsWith( "/" ) )
        {
            String zBasePath = Strings.noEmpty( System.getenv( "TemplateDevBasePath" ) );
            if ( zBasePath != null )
            {
                zTemplateFilePath = zBasePath + "/" + zTemplateFilePath;
            }
        }
        return zTemplateFilePath;
    }

    abstract protected TemplateControl resolveQuery( HttpServletRequest pRequest, String pQuery, HttpServletResponse pResponse, HttpSession pSession )
            throws IOException;
}