devsnaith/lumiere-server
1
0
Fork 0
mirror of https://github.com/devsnaith/lumiere-server.git synced 2024-11-23 17:13:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add Error Handlergit status!

Browse Source
This commit is contained in:
DevSnaith 2023-04-19 10:53:14 +03:00
parent e99580ef17
commit caa8c59b26
2 changed files with 82 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
src/org/eu/lumiere/error/ErrorHandler.java Normal file
View File

@ -0,0 +1,75 @@
package org.eu.lumiere.error;
import java.io.ByteArrayOutputStream;
import java.io.OutputStream;
import java.util.HashMap;
import org.eu.lumiere.loggers.GlobalLogger;
import org.eu.lumiere.loggers.GlobalLogger.LogLevel;
import org.eu.lumiere.net.http.HTTPResponse;
public class ErrorHandler extends HTTPException {
public boolean debug = true;
private static GlobalLogger l = GlobalLogger.getLogger();
private final HashMap<Integer, String> errors = new HashMap<>();
private static ErrorHandler errorHandler = new ErrorHandler();
private int defaultErrorCode = 500;
public void setError(int errorCode, String statusLine) {
errors.put(errorCode, (String) (debug ? l.printf(LogLevel.INFO, "[%d:%s] added to ErrorHandler", statusLine, errorCode, statusLine) : statusLine));
}
public void initialize() {
setError(500, "Internal Server Error");
setError(501, "Not Implemented");
setError(503, "Service Unavailable");
setError(400, "Bad Request");
setError(401, "Unauthorized");
setError(403, "Forbidden");
setError(404, "Not Found");
setError(411, "Length Required");
setError(414, "URI Too Long");
setError(423, "Locked");
setError(429, "Too Many Requests");
}
@Override
public void onException(HTTPResponse res, int httpCode, String msg) {
String[] sLine = getStatusLine(httpCode).split(" ", 3);
StringBuilder htmlBuilder = new StringBuilder("<!DOCTYPE html><html lang=\"en\">\n<head><meta charset=\"UTF-8\">\n");
htmlBuilder.append("<meta http-equiv=\"X-UA-Compatible\" content=\"IE=edge\">").append("\n");
htmlBuilder.append("<meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">").append("\n");
htmlBuilder.append("<title>").append(String.format("%s %s", sLine[1], sLine[2])).append("</title>").append("\n<head>");
htmlBuilder.append("<body style=\"color: white; background-color: #000000;\">").append("\n");
htmlBuilder.append("<b><code>").append(String.format("%s %s", sLine[1], sLine[2])).append("</code></b><br><code>").append(msg);
res.setStatus(String.format("%s %s %s", sLine[0], sLine[1], sLine[2]));
res.setContentType("text/html");
res.push(htmlBuilder.append("</code></body></html>").toString());
}
public void generateHTML(OutputStream os, int httpCode, String msg) {
HTTPResponse r = new HTTPResponse(os, null, null);
onException(r, httpCode, msg);
}
public byte[] generateHTMLAsBytes(int httpCode, String msg) {
ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
generateHTML(byteArray, httpCode, msg);
return byteArray.toByteArray();
}
public String generateHTMLAsString(int httpCode, String msg) {
return new String(generateHTMLAsBytes(httpCode, msg));
}
public synchronized String getStatusLine(int httpCode) {
if(!errors.containsKey(httpCode))
return errors.containsKey(defaultErrorCode) ? getStatusLine(defaultErrorCode) : "HTTP/1.0 200 OK";
return String.format("HTTP/1.0 %d %s", httpCode, errors.get(httpCode));
}
public static ErrorHandler getHandler() {
return errorHandler;
}
}

7
src/org/eu/lumiere/error/HTTPException.java Normal file
View File

@ -0,0 +1,7 @@
package org.eu.lumiere.error;
import org.eu.lumiere.net.http.HTTPResponse;
public abstract class HTTPException {
public abstract void onException(HTTPResponse os, int httpCode, String msg);
}