package org.webslinger.servlet;

 import java.io.IOException;

 import java.io.PrintWriter;

 import java.io.Writer;

 import java.util.Locale;

 import javax.servlet.ServletOutputStream;

 import javax.servlet.ServletResponse;

 public final class FakeServletResponse implements ServletResponse {

     private final PrintWriter writer;

     private final ServletOutputStream outputStream;

     private final ServletResponse response;

     private String encoding;

     private String contentType;

     private Locale locale = Locale.getDefault();

     public FakeServletResponse(ServletOutputStream outputStream) {

         this(outputStream, null, null);

     }

     public FakeServletResponse(ServletResponse response) {

         this(null, null, response);

     }

     public FakeServletResponse(Writer writer) {

         this(null, writer instanceof PrintWriter ? (PrintWriter) writer : new PrintWriter(writer), null);

     }

     private FakeServletResponse(ServletOutputStream outputStream, PrintWriter writer, ServletResponse response) {

         this.outputStream = outputStream;

         this.writer = writer;

         this.response = response;

     }

     public void flushBuffer() throws IOException {

         if (writer != null) getWriter().flush();

         if (response != null) response.flushBuffer();

     }

     public int getBufferSize() {

         return 0;

     }

     public String getCharacterEncoding() {

         return encoding;

     }

     public String getContentType() {

         return contentType;

     }

     public Locale getLocale() {

         return locale;

     }

     public ServletOutputStream getOutputStream() throws IOException {

         if (writer != null) throw new IllegalStateException("getWriter already called");

         if (outputStream == null) {

             if (response == null) throw new UnsupportedOperationException("No output method available");

             return response.getOutputStream();

         }

         return outputStream;

     }

     public PrintWriter getWriter() throws IOException {

         if (outputStream != null) throw new IllegalStateException("getOutputStream already called");

         if (writer == null) {

             if (response == null) throw new UnsupportedOperationException("No output method available");

             return response.getWriter();

         }

         return writer;

     }

     public boolean isCommitted() {

         if (response != null) return response.isCommitted();

         return true;

     }

     public void reset() {

         throw new UnsupportedOperationException();

     }

     public void resetBuffer() {

         throw new UnsupportedOperationException();

     }

     public void setBufferSize(int size) {

     }

     public void setCharacterEncoding(String encoding) {

         this.encoding = encoding;

     }

     public void setContentLength(int contentLength) {

     }

     public void setContentType(String contentType) {

         this.contentType = contentType;

     }

     public void setLocale(Locale locale) {

         this.locale = locale;

     }

 }