001package common.http.error;
002
003import common.http.interceptor.HttpServletBiConsumer;
004import common.http.interceptor.ServletInterceptor;
005import usecase.auth.AuthenticationRequiredException;
006import usecase.auth.AuthorizationException;
007import usecase.auth.BannedUserException;
008
009import javax.json.Json;
010import javax.json.JsonObject;
011import javax.servlet.ServletException;
012import javax.servlet.http.HttpServletRequest;
013import javax.servlet.http.HttpServletResponse;
014import javax.validation.ConstraintViolation;
015import javax.validation.ConstraintViolationException;
016import java.io.IOException;
017import java.io.PrintWriter;
018import java.time.format.DateTimeFormatter;
019import java.util.Collection;
020import java.util.Collections;
021import java.util.List;
022import java.util.stream.Collectors;
023
024import static javax.servlet.http.HttpServletResponse.*;
025
026/**
027 * @see JSONError
028 */
029public class JSONErrorInterceptor extends ServletInterceptor<JSONError> {
030
031    @Override
032    protected void init(JSONError annotation) {
033
034    }
035
036    @Override
037    public void handle(HttpServletRequest req, HttpServletResponse resp, HttpServletBiConsumer next) throws ServletException, IOException {
038        try{
039            next.handle(req,resp);
040        } catch (IllegalArgumentException e) {
041            sendJSONError(resp, SC_BAD_REQUEST, e.getMessage());
042        } catch (ConstraintViolationException e) {
043            List<String> collect = e.getConstraintViolations().stream()
044                    .map(ConstraintViolation::getMessage)
045                    .collect(Collectors.toList());
046
047            sendJSONError(resp,SC_BAD_REQUEST,collect);
048        } catch (AuthenticationRequiredException e) {
049            sendJSONError(resp,SC_UNAUTHORIZED, e.getMessage());
050        } catch (BannedUserException e){
051            if (e.getDuration() != null){
052                String end  = DateTimeFormatter.ISO_INSTANT.format(e.getDuration());
053                sendJSONError(resp,SC_FORBIDDEN, "Sei bannato fino a " + end);
054            } else {
055                sendJSONError(resp, SC_FORBIDDEN);
056            }
057        } catch (AuthorizationException e) {
058            sendJSONError(resp,SC_FORBIDDEN, e.getMessage());
059        } catch (RuntimeException | ServletException | IOException e){
060            sendJSONError(resp,SC_INTERNAL_SERVER_ERROR, "Internal server error");
061            req.getServletContext().log(e.getMessage(), e);
062        }
063    }
064
065    private void sendJSONError(HttpServletResponse resp, int code, Collection<String> messages) throws IOException {
066        JsonObject errors = Json.createObjectBuilder().add("errors", Json.createArrayBuilder(messages)).build();
067        resp.setStatus(code);
068        resp.setContentType("application/json");
069        PrintWriter writer = resp.getWriter();
070        writer.print(errors.toString());
071        writer.flush();
072
073    }
074
075    private void sendJSONError(HttpServletResponse resp, int code, String message) throws IOException {
076        List<String> singleton = message == null ? Collections.emptyList() : List.of(message);
077        sendJSONError(resp, code, singleton);
078    }
079
080    private void sendJSONError(HttpServletResponse resp, int code) throws IOException{
081        sendJSONError(resp,code,Collections.emptyList());
082    }
083
084    @Override
085    public int priority() {
086        return 2;
087    }
088}