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