package act.controller;

/*-

* #%L

* ACT Framework

* %%

* Copyright (C) 2014 - 2018 ActFramework

* %%

* Licensed 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.

* #L%

*/

import act.ActResponse;

import org.osgl.util.Charsets;

import java.io.IOException;

import java.io.StringWriter;

import java.io.Writer;

import java.nio.ByteBuffer;

public class WriterCache extends Writer {

private StringWriter tee = new StringWriter();

private Writer out;

private ByteBuffer buffer;

public WriterCache(Writer out) {

this.out = out;

}

@Override

public void write(char[] cbuf, int off, int len) throws IOException {

tee.write(cbuf, off, len);

out.write(cbuf, off, len);

}

@Override

public void flush() throws IOException {

out.flush();

}

@Override

public void close() throws IOException {

byte[] ba = tee.toString().getBytes(Charsets.UTF_8);

ByteBuffer buffer = ByteBuffer.allocateDirect(ba.length);

buffer.put(ba);

buffer.flip();

this.buffer = buffer;

out.close();

}

void apply(ActResponse resp) {

resp.writeContent(buffer.duplicate());

}

}