org.actioncontrollerdemo.jdkhttp.StaticContent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of action-controller-demo-app Show documentation
There is a newer version: 0.0.41
Show newest version
package org.actioncontrollerdemo.jdkhttp;

import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;
import java.net.URLConnection;
import java.util.Objects;
import java.util.Properties;

public class StaticContent implements HttpHandler {
    private final URL baseResource;
    private final String prefix;

    private static Properties mimeTypes = new Properties();
    static {
        try {
            mimeTypes.load(Objects.requireNonNull(StaticContent.class.getClassLoader().getResourceAsStream("mime-types.properties")));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public StaticContent(URL baseResource, String prefix) throws MalformedURLException {
        if (baseResource.getProtocol().equals("file")) {
            File resourceSrc = new File(baseResource.getPath().replace("/target/classes/", "/src/main/resources/"));
            if (resourceSrc.exists()) {
                this.baseResource = resourceSrc.toURI().toURL();
            } else {
                this.baseResource = baseResource;
            }
        } else {
            this.baseResource = baseResource;
        }
        this.prefix = prefix;
    }

    public void handle(HttpExchange exchange) throws IOException {
        URI uri = exchange.getRequestURI();
        URL url = new URL(baseResource + uri.getPath().substring(prefix.length()));
        sendContent(exchange, url);
    }

    private void sendContent(HttpExchange exchange, URL url) throws IOException {
        if (url.getPath().endsWith("/")) {
            url = new URL(url, "index.html");
        }
        try (InputStream inputStream = url.openStream()) {
            String contentType =  URLConnection.getFileNameMap().getContentTypeFor(url.getPath());
            if (contentType != null) {
                exchange.getResponseHeaders().set("Content-type", contentType);
            } else {
                int lastPeriodPos = url.getPath().lastIndexOf('.');
                int lastSlashPos = url.getPath().lastIndexOf('/');
                if (lastPeriodPos > 0 && lastPeriodPos > lastSlashPos) {
                    String extension = url.getPath().substring(lastPeriodPos + 1);
                    contentType = mimeTypes.getProperty(extension);
                    if (contentType != null) {
                        exchange.getResponseHeaders().set("Content-type", contentType);
                    }
                }
            }

            exchange.sendResponseHeaders(200, 0);
            inputStream.transferTo(exchange.getResponseBody());
        } catch (FileNotFoundException e) {
            exchange.sendResponseHeaders(404, 0);
        } catch (Exception e) {
            exchange.sendResponseHeaders(500, 0);
            exchange.getResponseBody().write(e.toString().getBytes());
        } finally {
            exchange.close();
        }
    }
}