    001// Copyright 2021 The Apache Software Foundation
    002//
    003// Licensed under the Apache License, Version 2.0 (the "License");
    004// you may not use this file except in compliance with the License.
    005// You may obtain a copy of the License at
    006//
    007//     http://www.apache.org/licenses/LICENSE-2.0
    008//
    009// Unless required by applicable law or agreed to in writing, software
    010// distributed under the License is distributed on an "AS IS" BASIS,
    011// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012// See the License for the specific language governing permissions and
    013// limitations under the License.
    014
    015package org.apache.tapestry5.internal.services;
    016
    017import java.io.IOException;
    018import java.util.Map;
    019import java.util.Objects;
    020
    021import org.apache.tapestry5.http.services.Response;
    022import org.apache.tapestry5.services.ComponentEventResultProcessor;
    023import org.apache.tapestry5.services.HttpStatus;
    024
    025/**
    026 * Handles {@link HttpStatus}.values returned by event handler methods.
    027 *
    028 * @since 5.8.0
    029 */
    030public class HttpStatusComponentEventResultProcessor implements ComponentEventResultProcessor<HttpStatus>
    031{
    032    private final Response response;
    033
    034    public HttpStatusComponentEventResultProcessor(Response response)
    035    {
    036        this.response = response;
    037    }
    038
    039    public void processResultValue(HttpStatus value) throws IOException
    040    {
    041        response.setStatus(value.getStatusCode());
    042        final Map<String, String> extraHttpHeaders = value.getExtraHttpHeaders();
    043        for (String header : extraHttpHeaders.keySet())
    044        {
    045            final String headerValue = extraHttpHeaders.get(header);
    046            response.setHeader(header, headerValue);
    047        }
    048        if (value.getResponseBody() != null)
    049        {
    050            Objects.requireNonNull(value.getContentType(), "HttpStatus.mimeType cannot be null");
    051            response.getPrintWriter(value.getContentType()).append(value.getResponseBody()).close();
    052        }
    053    }
    054}