/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.cxf.transport.http.netty.server.servlet;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.Locale;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import io.netty.handler.codec.http.ClientCookieEncoder;
import io.netty.handler.codec.http.Cookie;
import io.netty.handler.codec.http.HttpContent;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpResponse;
import io.netty.handler.codec.http.HttpResponseStatus;
import static io.netty.handler.codec.http.HttpHeaders.Names.LOCATION;
import static io.netty.handler.codec.http.HttpHeaders.Names.SET_COOKIE;
public class NettyServletResponse implements HttpServletResponse {
private HttpResponse originalResponse;
private NettyServletOutputStream outputStream;
private PrintWriter writer;
private boolean responseCommited;
public NettyServletResponse(HttpResponse response) {
this.originalResponse = response;
this.outputStream = new NettyServletOutputStream((HttpContent)response);
this.writer = new PrintWriter(this.outputStream);
}
public HttpResponse getOriginalResponse() {
return originalResponse;
}
public void addCookie(Cookie cookie) {
HttpHeaders.addHeader(this.originalResponse,
SET_COOKIE,
ClientCookieEncoder.encode(cookie.getName(), cookie.getValue()));
}
public void addDateHeader(String name, long date) {
HttpHeaders.addHeader(this.originalResponse, name, date);
}
public void addHeader(String name, String value) {
HttpHeaders.addHeader(this.originalResponse, name, value);
}
public void addIntHeader(String name, int value) {
HttpHeaders.addIntHeader(this.originalResponse, name, value);
}
@Override
public void addCookie(javax.servlet.http.Cookie cookie) {
//TODO Do we need to implement it ?
}
public boolean containsHeader(String name) {
return this.originalResponse.headers().contains(name);
}
public void sendError(int sc) throws IOException {
this.originalResponse.setStatus(HttpResponseStatus.valueOf(sc));
}
public void sendError(int sc, String msg) throws IOException {
this.originalResponse.setStatus(new HttpResponseStatus(sc, msg));
}
public void sendRedirect(String location) throws IOException {
setStatus(302);
setHeader(LOCATION, location);
}
public void setDateHeader(String name, long date) {
HttpHeaders.setHeader(this.originalResponse, name, date);
}
public void setHeader(String name, String value) {
HttpHeaders.setHeader(this.originalResponse, name, value);
}
public void setIntHeader(String name, int value) {
HttpHeaders.setIntHeader(this.originalResponse, name, value);
}
@Override
public ServletOutputStream getOutputStream() throws IOException {
return this.outputStream;
}
@Override
public PrintWriter getWriter() throws IOException {
return this.writer;
}
public void setStatus(int sc) {
this.originalResponse.setStatus(HttpResponseStatus.valueOf(sc));
}
public void setStatus(int sc, String sm) {
this.originalResponse.setStatus(new HttpResponseStatus(sc, sm));
}
@Override
public void setContentType(String type) {
HttpHeaders.setHeader(this.originalResponse,
HttpHeaders.Names.CONTENT_TYPE, type);
}
@Override
public void setContentLength(int len) {
HttpHeaders.setContentLength(this.originalResponse, len);
}
@Override
public boolean isCommitted() {
return this.responseCommited;
}
@Override
public void reset() {
if (isCommitted()) {
throw new IllegalStateException("Response already commited!");
}
this.originalResponse.headers().clear();
this.resetBuffer();
}
@Override
public void resetBuffer() {
if (isCommitted()) {
throw new IllegalStateException("Response already commited!");
}
this.outputStream.resetBuffer();
}
@Override
public void flushBuffer() throws IOException {
this.getWriter().flush();
this.responseCommited = true;
}
@Override
public int getBufferSize() {
return this.outputStream.getBufferSize();
}
@Override
public void setBufferSize(int size) {
// we using always dynamic buffer for now
}
public String encodeRedirectURL(String url) {
return this.encodeURL(url);
}
public String encodeRedirectUrl(String url) {
return this.encodeURL(url);
}
public String encodeURL(String url) {
try {
return URLEncoder.encode(url, getCharacterEncoding());
} catch (UnsupportedEncodingException e) {
throw new RuntimeException("Error encoding url!", e);
}
}
public String encodeUrl(String url) {
return this.encodeRedirectURL(url);
}
@Override
public String getCharacterEncoding() {
throw new IllegalStateException(
"Method 'getCharacterEncoding' not yet implemented!");
}
@Override
public String getContentType() {
throw new IllegalStateException(
"Method 'getContentType' not yet implemented!");
}
@Override
public Locale getLocale() {
throw new IllegalStateException(
"Method 'getLocale' not yet implemented!");
}
@Override
public void setCharacterEncoding(String charset) {
throw new IllegalStateException(
"Method 'setCharacterEncoding' not yet implemented!");
}
@Override
public void setLocale(Locale loc) {
throw new IllegalStateException(
"Method 'setLocale' not yet implemented!");
}
}