com.devsmart.miniweb.handlers.FileSystemRequestHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of miniweb-core Show documentation
Miniweb is an embedded webserver for Java.
There is a newer version: 0.1.5
package com.devsmart.miniweb.handlers;

import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.entity.FileEntity;
import org.apache.http.protocol.HttpContext;
import org.apache.http.protocol.HttpRequestHandler;
import com.devsmart.miniweb.utils.RequestMethod;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.net.URI;


public class FileSystemRequestHandler implements HttpRequestHandler {

    public final Logger logger = LoggerFactory.getLogger(FileSystemRequestHandler.class);

    private final File mRoot;
    private final String mPrefix;

    public FileSystemRequestHandler(File root, String prefix) {
        this.mRoot = root;
        this.mPrefix = prefix;
    }

    @Override
    public void handle(HttpRequest request, HttpResponse response, HttpContext context) throws HttpException, IOException {

        try {
            if (RequestMethod.GET.name().equals(request.getRequestLine().getMethod())) {

                URI url = new URI(request.getRequestLine().getUri());
                String path = url.getPath();
                if("/".equals(path)){
                    handleRoot(request, response, context);
                    return;
                }
                if(mPrefix != null && mPrefix.trim().length() > 0){
                    path = path.substring(0, mPrefix.length());
                }
                String[] pathsegments = path.split("/");

                File file = new File(mRoot, "");
                for (String part : pathsegments) {
                    file = new File(file, part);
                }

                if (file.exists() && file.isFile()) {
                    logger.debug("sent {}", file.getPath());
                    FileEntity body = new FileEntity(file, "");
                    response.setEntity(body);
                } else {
                    response.setStatusCode(HttpStatus.SC_NOT_FOUND);
                }

            } else {
                //method not allowed
                response.setStatusCode(HttpStatus.SC_METHOD_NOT_ALLOWED);
            }
        } catch (Exception e) {
            logger.error("", e);
            response.setStatusCode(500);
        }
    }

    final String[] sIndexFiles = new String[]{
            "index.html",
            "index.htm"
    };

    private void handleRoot(HttpRequest request, HttpResponse response, HttpContext context) {
        for(String filename : sIndexFiles){
            File file = new File(mRoot, filename);
            if(file.exists() && file.isFile()){
                logger.debug("sent {}", file.getPath());
                FileEntity body = new FileEntity(file, "");
                response.setEntity(body);
                return;
            }
        }

        response.setStatusCode(HttpStatus.SC_NOT_FOUND);
    }
}