org.tinystruct.http.ResponseBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tinystruct Show documentation

A simple framework for Java development. Simple is hard, Complex is easy. Better thinking, better design.

There is a newer version: 1.4.4

package org.tinystruct.http;

import io.netty.handler.codec.http.FullHttpResponse;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpResponseStatus;

import java.io.IOException;

public class ResponseBuilder extends ResponseWrapper {
    private ResponseStatus status;
    private Version version;
    private final Headers headers = new Headers();

    public ResponseBuilder(FullHttpResponse response) {
        super(response);

        HttpHeaders headers = response.headers();
        Object[] names = headers.names().toArray();
        for (Object o : names) {
            String name = o.toString();
            this.headers.add(Header.value0f(name).set(headers.get(name)));
        }

        this.status = ResponseStatus.valueOf(response.status().code());
    }

    public void setContentType(String contentType) {
        this.response.headers().add(Header.CONTENT_TYPE.name(), contentType);
    }

    @Override
    public void addHeader(String header, Object value) {
        if (!this.response.headers().contains(header)) {
            this.headers.add(new Header(header).set(value));
            if (value instanceof Integer) {
                this.response.headers().addInt(header, (Integer) value);
            } else
                this.response.headers().add(header, value);
        } else {
            if (!this.response.headers().get(header).equalsIgnoreCase(value.toString())) {
                if (value instanceof Integer) {
                    this.response.headers().addInt(header, (Integer) value);
                } else
                    this.response.headers().add(header, value);
            }
        }
    }

    public void addCookie(Cookie cookie) {
        addHeader(Header.SET_COOKIE.name(), cookie.toString());
    }

    @Override
    public Version version() {
        return this.version;
    }

    @Override
    public void setVersion(Version version) {
        this.version = version;
    }

    @Override
    public ResponseStatus status() {
        return this.status;
    }

    @Override
    public Response setStatus(ResponseStatus status) {
        this.status = status;
        this.response.setStatus(HttpResponseStatus.valueOf(status.code()));
        return this;
    }

    @Override
    public Headers headers() {
        return this.headers;
    }

    @Override
    public FullHttpResponse get() {
        return this.response;
    }

    @Override
    public void sendRedirect(String url) throws IOException {
        ResponseHeaders responseHeaders = new ResponseHeaders(this);
        responseHeaders.add(Header.CONTENT_LENGTH.setInt(0));
        responseHeaders.add(Header.LOCATION.set(url));
        this.setStatus(ResponseStatus.TEMPORARY_REDIRECT);
    }

}