/**
 *    TXTWriter.java
 *    Copyright (C) 2010 New Zealand Digital Library, http://expeditee.org
 *
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program 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 General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.expeditee.io;

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

import org.expeditee.gui.Frame;
import org.expeditee.items.Item;
import org.expeditee.items.ItemUtils;
import org.expeditee.items.Text;

public class TXTWriter extends DefaultTreeWriter {

	private boolean _join = false;

	private int _indent = 0;

	@Override
	protected void initialise(Frame start, Writer writer) throws IOException {
		_format = ".txt";
		super.initialise(start, writer);
	}

	@Override
	protected void writeStartFrame(Frame toParse) throws IOException {
		if (ItemUtils.ContainsTag(toParse.getItems(), "@join"))
			_join = !_join;

		if (ItemUtils.ContainsTag(toParse.getItems(), "@indent"))
			_indent++;

		super.writeStartFrame(toParse);
	}

	@Override
	protected void writeTitle(Text title, List<Item> items) throws IOException {
		int indent = 0;

		if (_indent > 0)
			indent = _indent - 1;

		for (String s : title.getTextList()) {

			for (int i = 0; i < indent; i++)
				_writer.write("\t");

			_writer.write(s);
			_writer.write(ItemWriter.NEW_LINE);
		}

		if (!_join)
			_writer.write(ItemWriter.NEW_LINE);
	}

	@Override
	protected void writeEndFrame(Frame toParse) throws IOException {
		if (ItemUtils.ContainsTag(toParse.getItems(), "@indent"))
			if (_indent > 0)
				_indent--;

		if (ItemUtils.ContainsTag(toParse.getItems(), "@join"))
			_join = !_join;

		_writer.write(ItemWriter.NEW_LINE);
	}

	@Override
	protected void resumeFrame(Frame resuming) {
		_join = ItemUtils.ContainsTag(resuming.getItems(), "@join");
	}

	@Override
	protected void writeText(Text text) throws IOException {
		String s = text.getText();

		for (int i = 0; i < _indent; i++)
			_writer.write("\t");

		_writer.write(s);
		_writer.write(ItemWriter.NEW_LINE);

		if (!_join)
			_writer.write(ItemWriter.NEW_LINE);
	}

	@Override
	protected void writeAnnotationText(Text toWrite) throws IOException {
		if (toWrite.startsWith("@BlankLine"))
			_writer.write(ItemWriter.NEW_LINE);
	}
}