Examples with HttpServerResponse org.wildfly.security.http.HttpServerResponse used on opensource projects

Example 1 with HttpServerResponse

use of org.wildfly.security.http.HttpServerResponse in project keycloak by keycloak.

the class ElytronHttpFacade method getResponse.

@Override
public Response getResponse() {
    return new Response() {

        @Override
        public void setStatus(final int status) {
            if (status < 200 || status > 300) {
                responseConsumer = responseConsumer.andThen(response -> response.setStatusCode(status));
            }
        }

        @Override
        public void addHeader(final String name, final String value) {
            headers.put(name, value);
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse response) {
                    String latestValue = headers.get(name);
                    if (latestValue.equals(value)) {
                        response.addResponseHeader(name, latestValue);
                    }
                }
            });
        }

        @Override
        public void setHeader(String name, String value) {
            addHeader(name, value);
        }

        @Override
        public void resetCookie(final String name, final String path) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, "", path, null, 0, false, false, response));
            HttpScope exchangeScope = getScope(Scope.EXCHANGE);
            ProtectedHttpServerExchange undertowExchange = ProtectedHttpServerExchange.class.cast(exchangeScope.getAttachment(UNDERTOW_EXCHANGE));
            if (undertowExchange != null) {
                CookieImpl cookie = new CookieImpl(name, "");
                cookie.setMaxAge(0);
                cookie.setPath(path);
                undertowExchange.getExchange().setResponseCookie(cookie);
            }
        }

        @Override
        public void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, value, path, domain, maxAge, secure, httpOnly, response));
        }

        private void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly, HttpServerResponse response) {
            response.setResponseCookie(new HttpServerCookie() {

                @Override
                public String getName() {
                    return name;
                }

                @Override
                public String getValue() {
                    return value;
                }

                @Override
                public String getDomain() {
                    return domain;
                }

                @Override
                public int getMaxAge() {
                    return maxAge;
                }

                @Override
                public String getPath() {
                    return path;
                }

                @Override
                public boolean isSecure() {
                    return secure;
                }

                @Override
                public int getVersion() {
                    return 0;
                }

                @Override
                public boolean isHttpOnly() {
                    return httpOnly;
                }
            });
        }

        @Override
        public OutputStream getOutputStream() {
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse httpServerResponse) {
                    try {
                        httpServerResponse.getOutputStream().write(stream.toByteArray());
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to write to response output stream", e);
                    }
                }
            });
            return stream;
        }

        @Override
        public void sendError(int code) {
            setStatus(code);
        }

        @Override
        public void sendError(final int code, final String message) {
            responseConsumer = responseConsumer.andThen(response -> {
                response.setStatusCode(code);
                response.addResponseHeader("Content-Type", "text/html");
                try {
                    response.getOutputStream().write(message.getBytes());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            });
        }

        @Override
        public void end() {
        }
    };
}

Example 2 with HttpServerResponse

use of org.wildfly.security.http.HttpServerResponse in project keycloak by keycloak.

the class ElytronHttpFacade method getResponse.

@Override
public Response getResponse() {
    return new Response() {

        @Override
        public void setStatus(final int status) {
            responseConsumer = responseConsumer.andThen(response -> response.setStatusCode(status));
        }

        @Override
        public void addHeader(final String name, final String value) {
            responseConsumer = responseConsumer.andThen(response -> response.addResponseHeader(name, value));
        }

        @Override
        public void setHeader(String name, String value) {
            addHeader(name, value);
        }

        @Override
        public void resetCookie(final String name, final String path) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, "", path, null, 0, false, false, response));
        }

        @Override
        public void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, value, path, domain, maxAge, secure, httpOnly, response));
        }

        private void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly, HttpServerResponse response) {
            response.setResponseCookie(new HttpServerCookie() {

                @Override
                public String getName() {
                    return name;
                }

                @Override
                public String getValue() {
                    return value;
                }

                @Override
                public String getDomain() {
                    return domain;
                }

                @Override
                public int getMaxAge() {
                    return maxAge;
                }

                @Override
                public String getPath() {
                    return path;
                }

                @Override
                public boolean isSecure() {
                    return secure;
                }

                @Override
                public int getVersion() {
                    return 0;
                }

                @Override
                public boolean isHttpOnly() {
                    return httpOnly;
                }
            });
        }

        @Override
        public OutputStream getOutputStream() {
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse httpServerResponse) {
                    try {
                        httpServerResponse.getOutputStream().write(stream.toByteArray());
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to write to response output stream", e);
                    }
                }
            });
            return stream;
        }

        @Override
        public void sendError(int code) {
            setStatus(code);
        }

        @Override
        public void sendError(final int code, final String message) {
            responseConsumer = responseConsumer.andThen(response -> {
                response.setStatusCode(code);
                response.addResponseHeader("Content-Type", "text/html");
                try {
                    response.getOutputStream().write(message.getBytes());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            });
        }

        @Override
        public void end() {
        }
    };
}