package goodjava.webserver;

import java.io.ByteArrayOutputStream;
import java.io.ByteArrayInputStream;
import java.io.IOException;


// plenty of room for improvement
public class ResponseOutputStream extends ByteArrayOutputStream {
	private final Response response;

	public ResponseOutputStream(Response response) {
		if(response==null) throw new NullPointerException();
		this.response = response;
	}

	@Override public void close() throws IOException {
		super.close();
		int size = size();
		response.body = new Response.Body( size, new ByteArrayInputStream(buf,0,size) );
	}
}