MarkerHandler.java

package net.cubespace.dynmap.multiserver.HTTP.Handler;

import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.*;
import io.netty.handler.stream.ChunkedStream;
import net.cubespace.dynmap.multiserver.DynmapServer;
import net.cubespace.dynmap.multiserver.GSON.DynmapWorld;
import net.cubespace.dynmap.multiserver.HTTP.HandlerUtil;
import net.cubespace.dynmap.multiserver.Main;
import net.cubespace.dynmap.multiserver.util.AbstractFile;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;

import static io.netty.handler.codec.http.HttpHeaders.Names.*;
import static io.netty.handler.codec.http.HttpResponseStatus.*;
import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1;

/**
 * @author geNAZt (fabian.fassbender42@googlemail.com)
 */
public class MarkerHandler implements IHandler {
    @Override
    public void handle(ChannelHandlerContext ctx, FullHttpRequest request) throws Exception {
        //Get the correct DynmapServer
        AbstractFile path = null;
        String world = request.getUri().split("/")[3].replace("marker_", "").replace(".json", "");
        for (DynmapServer dynmapServer : Main.getDynmapServers()) {
            for (DynmapWorld dynmapWorld : dynmapServer.getWorlds()) {
                if (dynmapWorld.getName().equals(world)) {
                    path = dynmapServer.getFile(request.getUri());
                }
            }
        }

        AbstractFile file = path;
        if (file == null || file.isHidden() || !file.exists()) {
            HandlerUtil.sendError(ctx, NOT_FOUND);
            return;
        }

        if (!file.isFile()) {
            HandlerUtil.sendError(ctx, FORBIDDEN);

            return;
        }

        // Cache Validation
        String ifModifiedSince = request.headers().get(IF_MODIFIED_SINCE);
        if (ifModifiedSince != null && !ifModifiedSince.isEmpty()) {
            SimpleDateFormat dateFormatter = new SimpleDateFormat(HandlerUtil.HTTP_DATE_FORMAT, Locale.US);
            Date ifModifiedSinceDate = dateFormatter.parse(ifModifiedSince);

            // Only compare up to the second because the datetime format we send to the client
            // does not have milliseconds
            long ifModifiedSinceDateSeconds = ifModifiedSinceDate.getTime() / 1000;
            long fileLastModifiedSeconds = file.lastModified() / 1000;
            if (ifModifiedSinceDateSeconds == fileLastModifiedSeconds) {
                HandlerUtil.sendNotModified(ctx);
                return;
            }
        }

        HttpResponse response = new DefaultHttpResponse(HTTP_1_1, OK);

        HandlerUtil.setContentTypeHeader(response, file.getName());
        HandlerUtil.setDateAndCacheHeaders(response, file.lastModified());

        response.headers().set(CONTENT_LENGTH, file.length());
        response.headers().set(CONNECTION, HttpHeaders.Values.CLOSE);
        response.headers().set(VARY, ACCEPT_ENCODING);

        // Write the initial line and the header.
        ctx.write(response);

        // Write the content.
        ctx.write(new ChunkedStream(file.getInputStream()));
        ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT);
    }
}