98 lines
3.5 KiB
Java
98 lines
3.5 KiB
Java
/*
|
|
* This file is part of Player Analytics (Plan).
|
|
*
|
|
* Plan is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License v3 as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Plan is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with Plan. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
package com.djrapitops.plan.delivery.webserver.pages;
|
|
|
|
import com.djrapitops.plan.delivery.webserver.Request;
|
|
import com.djrapitops.plan.delivery.webserver.RequestTarget;
|
|
import com.djrapitops.plan.delivery.webserver.auth.Authentication;
|
|
import com.djrapitops.plan.delivery.webserver.response.Response;
|
|
import com.djrapitops.plan.delivery.webserver.response.ResponseFactory;
|
|
import com.djrapitops.plan.exceptions.WebUserAuthException;
|
|
import com.djrapitops.plan.exceptions.connection.WebException;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Abstract PageHandler that allows Tree-like target deduction.
|
|
*
|
|
* @author Rsl1122
|
|
*/
|
|
public abstract class TreePageHandler implements PageHandler {
|
|
|
|
protected final ResponseFactory responseFactory;
|
|
|
|
private final Map<String, PageHandler> pages;
|
|
|
|
public TreePageHandler(ResponseFactory responseFactory) {
|
|
this.responseFactory = responseFactory;
|
|
pages = new HashMap<>();
|
|
}
|
|
|
|
public void registerPage(String targetPage, PageHandler handler) {
|
|
pages.put(targetPage, handler);
|
|
}
|
|
|
|
public void registerPage(String targetPage, PageHandler handler, int requiredPerm) {
|
|
pages.put(targetPage, new PageHandler() {
|
|
@Override
|
|
public Response getResponse(Request request, RequestTarget target) throws WebException {
|
|
return handler.getResponse(request, target);
|
|
}
|
|
|
|
@Override
|
|
public boolean isAuthorized(Authentication auth, RequestTarget target) throws WebUserAuthException {
|
|
return auth.getWebUser().getPermLevel() <= requiredPerm;
|
|
}
|
|
});
|
|
}
|
|
public void registerPage(String targetPage, Response response, int requiredPerm) {
|
|
pages.put(targetPage, new PageHandler() {
|
|
@Override
|
|
public Response getResponse(Request request, RequestTarget target) {
|
|
return response;
|
|
}
|
|
|
|
@Override
|
|
public boolean isAuthorized(Authentication auth, RequestTarget target) throws WebUserAuthException {
|
|
return auth.getWebUser().getPermLevel() <= requiredPerm;
|
|
}
|
|
});
|
|
}
|
|
|
|
@Override
|
|
public Response getResponse(Request request, RequestTarget target) throws WebException {
|
|
PageHandler pageHandler = getPageHandler(target);
|
|
return pageHandler != null
|
|
? pageHandler.getResponse(request, target)
|
|
: responseFactory.pageNotFound404();
|
|
}
|
|
|
|
public PageHandler getPageHandler(RequestTarget target) {
|
|
if (target.isEmpty()) {
|
|
return pages.get("");
|
|
}
|
|
String targetPage = target.get(0);
|
|
target.removeFirst();
|
|
return pages.get(targetPage);
|
|
}
|
|
|
|
public PageHandler getPageHandler(String targetPage) {
|
|
return pages.get(targetPage);
|
|
}
|
|
}
|