dynmap-neoforge/src/main/java/org/dynmap/WebServerRequest.java

package org.dynmap;

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.Socket;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Logger;

import org.bukkit.Server;
import org.bukkit.entity.Player;
import org.dynmap.debug.Debugger;

public class WebServerRequest extends Thread {
	protected static final Logger log = Logger.getLogger("Minecraft");

	private Debugger debugger;
	private Socket socket;
	private MapManager mgr;
	private Server server;
	private PlayerList playerList;

	public WebServerRequest(Socket socket, MapManager mgr, Server server, PlayerList playerList, Debugger debugger)
	{
		this.debugger = debugger;
		this.socket = socket;
		this.mgr = mgr;
		this.server = server;
		this.playerList = playerList;
	}
	
	private static void writeHttpHeader(BufferedOutputStream out, int statusCode, String statusText) throws IOException {
		out.write("HTTP/1.0 ".getBytes());
		out.write(Integer.toString(statusCode).getBytes());
		out.write((" " + statusText + "\r\n").getBytes());
	}
	
	private static void writeHeaderField(BufferedOutputStream out, String name, String value) throws IOException {
		out.write(name.getBytes());
		out.write((int)':');
		out.write((int)' ');
		out.write(value.getBytes());
		out.write(13);
		out.write(10);
	}
	
	private static void writeEndOfHeaders(BufferedOutputStream out) throws IOException {
		out.write(13);
		out.write(10);
	}

	public void run()
	{
		InputStream reader = null;
		try {
			socket.setSoTimeout(30000);
			BufferedReader in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
			BufferedOutputStream out = new BufferedOutputStream(socket.getOutputStream());

			String request = in.readLine();
			if (request == null || !request.startsWith("GET ") || !(request.endsWith(" HTTP/1.0") || request.endsWith("HTTP/1.1"))) {
				// Invalid request type (no "GET")
				writeHttpHeader(out, 500, "Invalid Method.");
				writeEndOfHeaders(out);
				return;
			}

			String path = request.substring(4, request.length() - 9);
			debugger.debug("request: " + path);
			if (path.startsWith("/up/")) {
				handleUp(out, path.substring(3));
			} else if (path.startsWith("/tiles/")) {
				handleMapToDirectory(out, path.substring(6), mgr.tileDirectory);
			} else if (path.startsWith("/")) {
				handleMapToDirectory(out, path, mgr.webDirectory);
			}
			out.flush();
			out.close();
		}
		catch (IOException e) {
			if (reader != null) {
				try {
					reader.close();
				}
				catch (Exception anye) {
					// Do nothing.
				}
			}
		}
		catch(Exception ex) {
			debugger.error("Exception on WebRequest-thread: " + ex.toString());
		}
	}
	
	public void handleUp(BufferedOutputStream out, String path) throws IOException {
		int current = (int) (System.currentTimeMillis() / 1000);
		long cutoff = 0;

		if(path.charAt(0) == '/') {
			try {
				cutoff = ((long) Integer.parseInt(path.substring(1))) * 1000;
			} catch(NumberFormatException e) {
			}
		}

		StringBuilder sb = new StringBuilder();
		
		long relativeTime = server.getTime() % 24000;
		sb.append(current + " " + relativeTime + "\n");

		Player[] players = playerList.getVisiblePlayers();
		for(Player player : players) {
			sb.append("player " + player.getName() + " " + player.getLocation().getX() + " " + player.getLocation().getY() + " " + player.getLocation().getZ() + "\n");
		}
		
		TileUpdate[] tileUpdates = mgr.staleQueue.getTileUpdates(cutoff);
		for(TileUpdate tu : tileUpdates) {
			sb.append("tile " + tu.tile.getName() + "\n");
		}
		
		ChatQueue.ChatMessage[] messages = mgr.chatQueue.getChatMessages(cutoff);
		for(ChatQueue.ChatMessage cu : messages) {
			sb.append("chat " + cu.playerName + " " + cu.message + "\n");
		}

		debugger.debug("Sending " + players.length + " players, " + tileUpdates.length + " tile-updates, and " + messages.length + " chats. "+ path + ";" + cutoff);
		
		byte[] bytes = sb.toString().getBytes();
		
		String dateStr = new Date().toString();
		writeHttpHeader(out, 200, "OK");
		writeHeaderField(out, "Date", dateStr);
		writeHeaderField(out, "Content-Type", "text/plain");
		writeHeaderField(out, "Expires", "Thu, 01 Dec 1994 16:00:00 GMT");
		writeHeaderField(out, "Last-modified", dateStr);
		writeHeaderField(out, "Content-Length", Integer.toString(bytes.length));
		writeEndOfHeaders(out);
		out.write(bytes);
	}
	
	private byte[] readBuffer = new byte[40960];
	
	public void writeFile(BufferedOutputStream out, String path, InputStream fileInput) throws IOException {
		int dotindex = path.lastIndexOf('.');
		String extension = null;
		if (dotindex > 0) extension = path.substring(dotindex);
		
		writeHttpHeader(out, 200, "OK");
		writeHeaderField(out, "Content-Type", getMimeTypeFromExtension(extension));
		writeHeaderField(out, "Connection", "close");
		writeEndOfHeaders(out);
		try {
			int readBytes;
			while((readBytes = fileInput.read(readBuffer)) > 0) {
				out.write(readBuffer, 0, readBytes);
			}
		} catch(IOException e) {
			fileInput.close();
			throw e;
		}
		fileInput.close();
	}
	
	public String getFilePath(String path) {
		int qmark = path.indexOf('?');
		if (qmark >= 0) path = path.substring(0, qmark);
		path = path.substring(1);
		
		if (path.startsWith("/") || path.startsWith("."))
			return null;
		if (path.length() == 0) path = "index.html";
		return path;
	}
	
	public void handleMapToJar(BufferedOutputStream out, String path) throws IOException {
		path = getFilePath(path);
		if (path != null) {
			InputStream s = this.getClass().getResourceAsStream("/web/" + path);
			if (s != null) {
				writeFile(out, path, s);
				return;
			}
		}
		writeHttpHeader(out, 404, "Not found");
		writeEndOfHeaders(out);
	}
	
	public void handleMapToDirectory(BufferedOutputStream out, String path, File directory) throws IOException {
		path = getFilePath(path);
		if (path != null) {
			File tileFile = new File(directory, path);
	
			if (tileFile.getAbsolutePath().startsWith(directory.getAbsolutePath()) && tileFile.isFile()) {
				FileInputStream s = new FileInputStream(tileFile);
				writeFile(out, path, s);
				return;
			}
		}
		writeHttpHeader(out, 404, "Not found");
		writeEndOfHeaders(out);
	}
	
	private static Map<String, String> mimes = new HashMap<String, String>();
	static {
		mimes.put(".html", "text/html");
		mimes.put(".htm", "text/html");
		mimes.put(".js", "text/javascript");
		mimes.put(".png", "image/png");
		mimes.put(".css", "text/css");
		mimes.put(".txt", "text/plain");
	}
	public static String getMimeTypeFromExtension(String extension) {
		String m = mimes.get(extension);
		if (m != null) return m;
		return "application/octet-steam";
	}
}