DebugCFMLWriter.java

/**
 *
 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either 
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 * 
 **/
package lucee.runtime.debug;

import java.io.IOException;
import java.io.OutputStream;
import java.io.Writer;
import java.util.ArrayList;
import java.util.List;

import lucee.commons.io.SystemUtil;
import lucee.commons.io.SystemUtil.TemplateLine;
import lucee.commons.lang.StringUtil;
import lucee.runtime.cache.legacy.CacheItem;
import lucee.runtime.op.Caster;
import lucee.runtime.writer.CFMLWriter;

public class DebugCFMLWriter extends CFMLWriter implements DebugOutputLog {

	private CFMLWriter writer;
	private List<DebugTextFragment> fragments=new ArrayList<DebugTextFragment>(); 

	public DebugCFMLWriter(CFMLWriter writer) {
		super(writer.getBufferSize(), writer.isAutoFlush());
		this.writer=writer;
	}

	@Override
	public int getBufferSize() {
		return writer.getBufferSize();
	}

	@Override
	public boolean isAutoFlush() {
		return writer.isAutoFlush();
	}

	@Override
	public Writer append(CharSequence csq) throws IOException {
		log(csq.toString());
		return writer.append(csq);
	}

	@Override
	public Writer append(char c) throws IOException {
		log(new String(new char[]{c}));
		return writer.append(c);
	}

	@Override
	public Writer append(CharSequence csq, int start, int end) throws IOException {
		log(csq.subSequence(start, end).toString());
		return writer.append(csq, start, end);
	}

	@Override
	public void write(int i) throws IOException {
		print(i);
	}

	@Override
	public void write(char[] cbuf) throws IOException {
		print(cbuf);
	}

	@Override
	public void write(String str) throws IOException {
		print(str);
	}

	@Override
	public void write(String str, int off, int len) throws IOException {
		log(StringUtil.substring(str,off,len));
		writer.write(str, off, len);
	}

	@Override
	public OutputStream getResponseStream() throws IOException {
		return writer.getResponseStream();
	}

	@Override
	public void setClosed(boolean b) {
		writer.setClosed(b);
	}

	@Override
	public void setBufferConfig(int interval, boolean b) throws IOException {
		writer.setBufferConfig(interval, b);
	}

	@Override
	public void appendHTMLBody(String text) throws IOException {
		writer.appendHTMLBody(text);
	}

	@Override
	public void writeHTMLBody(String text) throws IOException {
		writer.writeHTMLBody(text);
	}

	@Override
	public String getHTMLBody() throws IOException {
		return writer.getHTMLBody();
	}

	@Override
	public void flushHTMLBody() throws IOException {
		writer.flushHTMLBody();
	}

	@Override
	public void resetHTMLBody() throws IOException {
		writer.resetHTMLBody();
	}

	@Override
	public void appendHTMLHead(String text) throws IOException {
		writer.appendHTMLHead(text);
	}

	@Override
	public void writeHTMLHead(String text) throws IOException {
		writer.writeHTMLHead(text);
	}

	@Override
	public String getHTMLHead() throws IOException {
		return writer.getHTMLHead();
	}

	@Override
	public void flushHTMLHead() throws IOException {
		writer.flushHTMLHead();
	}

	@Override
	public void resetHTMLHead() throws IOException {
		writer.resetHTMLHead();
	}

	@Override
	public void writeRaw(String str) throws IOException {
		print(str);
	}

	@Override
	public void clear() throws IOException {
		writer.clear();
	}

	@Override
	public void clearBuffer() throws IOException {
		writer.clearBuffer();
	}

	@Override
	public void close() throws IOException {
		writer.close();
	}

	@Override
	public void flush() throws IOException {
		writer.flush();
	}

	@Override
	public int getRemaining() {
		return writer.getRemaining();
	}

	@Override
	public void newLine() throws IOException {
		println();
	}

	@Override
	public void print(boolean b) throws IOException {
		writer.print(b);
		log(b?"true":"false");
	}

	@Override
	public void print(char c) throws IOException {
		log(new String(new char[]{c}));
		writer.write(c);
	}

	@Override
	public void print(int i) throws IOException {
		log(Caster.toString(i));
		writer.write(i);
	}

	@Override
	public void print(long l) throws IOException {
		log(Caster.toString(l));
		writer.print(l);
	}

	@Override
	public void print(float f) throws IOException {
		log(Caster.toString(f));
		writer.print(f);
	}

	@Override
	public void print(double d) throws IOException {
		log(Caster.toString(d));
		writer.print(d);
	}

	@Override
	public void print(char[] carr) throws IOException {
		log(new String(carr));
		writer.write(carr);
	}

	@Override
	public void print(String str) throws IOException {
		log(str);
		writer.write(str);
	}

	@Override
	public void print(Object obj) throws IOException {
		log(String.valueOf(obj));
		writer.print(obj);
	}

	@Override
	public void println() throws IOException {
		print("\n");
	}

	@Override
	public void println(boolean b) throws IOException {
		print(b);
		print("\n");
	}

	@Override
	public void println(char c) throws IOException {
		print(c);
		print("\n");
	}

	@Override
	public void println(int i) throws IOException {
		print(i);
		print("\n");
	}

	@Override
	public void println(long l) throws IOException {
		print(l);
		print("\n");
	}

	@Override
	public void println(float f) throws IOException {
		print(f);
		print("\n");
	}

	@Override
	public void println(double d) throws IOException {
		print(d);
		print("\n");
	}

	@Override
	public void println(char[] carr) throws IOException {
		print(carr);
		print("\n");
	}

	@Override
	public void println(String str) throws IOException {
		print(str);
		print("\n");
	}

	@Override
	public void println(Object obj) throws IOException {
		print(obj);
		print("\n");
	}

	@Override
	public void write(char[] carr, int off, int len) throws IOException {
		log(StringUtil.substring(new String(carr),off,len));
		writer.write(carr,off,len);
	}

	private void log(String str) {
		TemplateLine tl = SystemUtil.getCurrentContext();
		if(tl!=null){
			fragments.add(new DebugTextFragmentImpl(str, tl));
		}
	}

	@Override
	public DebugTextFragment[] getFragments() {
		return fragments.toArray(new DebugTextFragment[fragments.size()]);
	}

	@Override
	public void setAllowCompression(boolean allowCompression) {
		writer.setAllowCompression(allowCompression);
	}

	@Override
	public void doCache(CacheItem ci) {
		writer.doCache(ci);
	}

	@Override
	public CacheItem getCacheItem() {
		return writer.getCacheItem();
	}

}