package org.litesoft.ddt;

import org.litesoft.html.*;
import org.litesoft.template.*;
import org.litesoft.useragent.*;

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

public class IndexHtmlGenerator {
    private static final UserAgentFormFactorDeterminer UAFFD = new UserAgentFormFactorDeterminerImpl();

    private final Template mTemplate;

    public IndexHtmlGenerator( Template pTemplate ) {
        mTemplate = pTemplate;
    }

    public void respond( String pPath, HttpServletRequest pRequest, HttpServletResponse pResponse )
            throws ServletException, IOException {
        CacheHeaders.never( pResponse );

        String zDeviceToken = getCookie( pRequest, "DeviceToken" );
        if ( zDeviceToken != null ) {
            reply( pResponse, "LimitedMode" );
            return;
        }
        String zUserAuthToken = getCookie( pRequest, "UserAuthToken" );
        if ( zUserAuthToken != null ) {
            reply( pResponse, "FullMode" );
            return;
        }
        String zUserAgent = pRequest.getHeader( "user-agent" );
        System.out.println( "User Agent: " + zUserAgent );
        reply( pResponse, "Login" + UAFFD.determine( zUserAgent ) );
    }

    private void reply( HttpServletResponse pResponse, final String pWhat )
            throws IOException {
        pResponse.setContentType( "text/html" );
        mTemplate.dump( new Sink.Simple( pResponse.getWriter() ), new ValueProvider() {
            @Override
            public Object valueFor( String pKey ) {
                return pWhat;
            }
        } );
    }

    //    private static int hasCookie( HttpServletRequest pRequest, String... pCookieNames )
    //    {
    //        if ( pCookieNames != null )
    //        {
    //            for ( int i = 0; i < pCookieNames.length; i++ )
    //            {
    //                if ( null != checkAndClearCookie( pRequest, pCookieNames[i], true ) )
    //                {
    //                    return i;
    //                }
    //            }
    //        }
    //        return -1;
    //    }

    private static String getCookie( HttpServletRequest pRequest, String pName ) {
        return checkAndClearCookie( pRequest, pName, false );
    }

    private static String checkAndClearCookie( HttpServletRequest pRequest, String pName, boolean pClear ) {
        Cookie[] zCookies = pRequest.getCookies();
        if ( zCookies != null ) {
            for ( Cookie zCookie : zCookies ) {
                if ( zCookie.getName().equals( pName ) ) {
                    String zValue = zCookie.getValue();
                    if ( pClear ) {
                        zCookie.setMaxAge( 0 ); // Delete it...
                    }
                    return zValue;
                }
            }
        }
        return null;
    }
}