package org.litesoft.ddt;

import org.litesoft.html.*;
import org.litesoft.jetty.Shutdown;

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

public class DeviceDesktopTestServlet extends HttpServlet
{
    private final IndexHtmlGenerator mIndexHtmlGenerator;

    private ServletContext mServletContext;

    public DeviceDesktopTestServlet( IndexHtmlGenerator pIndexHtmlGenerator )
    {
        mIndexHtmlGenerator = pIndexHtmlGenerator;
    }

    @Override
    public void init( ServletConfig config )
            throws ServletException
    {
        super.init( config );
        mServletContext = config.getServletContext();
    }

    @Override
    protected void doGet( HttpServletRequest pRequest, HttpServletResponse pResponse )
            throws ServletException, IOException
    {
        String zPath = pRequest.getPathInfo();
        if ( zPath == null )
        {
            zPath = "/";
        }
        if ( zPath.endsWith( "/" ) )
        {
            zPath += "index.html";
        }
        if ( zPath.endsWith( "/index.html" ) )
        {
            String zAction = pRequest.getParameter( "action" );
            if ( "Restart".equals( zAction ) )
            {
                Shutdown.Instance.get().request( 0 );
            }
            else if ( "ShutDown".equals( zAction ) )
            {
                Shutdown.Instance.get().request( 1 );
            }
            mIndexHtmlGenerator.respond( zPath, pRequest, pResponse );
            return;
        }
        File zFile = new File( "." + zPath );
        if ( !zFile.isFile() )
        {
            System.out.println( "Can't find: " + zFile.getAbsolutePath() );
            pResponse.sendError( HttpServletResponse.SC_NOT_FOUND );
            return;
        }
        if ( !zFile.canRead() )
        {
            System.out.println( "Can't read: " + zFile.getAbsolutePath() );
            pResponse.sendError( HttpServletResponse.SC_FORBIDDEN );
            return;
        }

        CacheHeaders.forever( pResponse );

        // Get the MIME type
        String mimeType = mServletContext.getMimeType( zFile.getName().toLowerCase() );
        if ( null == mimeType )
        {
            System.err.println( "Could not get MIME type of " + zFile.getName() );
            mimeType = "application/octet-stream";
        }
        // Set content type
        pResponse.setContentType( mimeType );

        // Set content size
        pResponse.setContentLength( (int) zFile.length() );

        // Open the file and output streams
        InputStream in = new FileInputStream( zFile );
        try
        {
            OutputStream out = pResponse.getOutputStream();
            try
            {
                copyStream( in, out );

                Closeable zCloseable = out;
                out = null;
                zCloseable.close();
            }
            finally
            {
                closeQuietly( out );
            }
        }
        finally
        {
            closeQuietly( in );
        }
    }

    private static void closeQuietly( Closeable pClosable )
    {
        if ( pClosable != null )
        {
            try
            {
                pClosable.close();
            }
            catch ( IOException e )
            {
                // Whatever
            }
        }
    }

    private static void copyStream( InputStream pIn, OutputStream pOut )
            throws IOException
    {
        // Copy the contents of the file to the output stream
        byte[] buf = new byte[1024];
        for ( int count; (count = pIn.read( buf )) > 0; )
        {
            pOut.write( buf, 0, count );
        }
    }
}