/*******************************************************************************
 * Breakout Cave Survey Visualizer
 *
 * Copyright (C) 2014 James Edwards
 *
 * jedwards8 at fastmail dot fm
 *
 * 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 2 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, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *******************************************************************************/
package org.andork.func;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;

public class BimapperStreamBimapper<T> implements StreamBimapper<T> {
	Bimapper<T, String> format;
	boolean closeStreams;

	public BimapperStreamBimapper(Bimapper<T, String> format) {
		this(format, true);
	}

	public BimapperStreamBimapper(Bimapper<T, String> format, boolean closeStreams) {
		super();
		this.format = format;
		this.closeStreams = closeStreams;
	}

	@Override
	public T read(InputStream in) throws Exception {
		InputStreamReader reader = null;

		try {
			reader = new InputStreamReader(in);
			StringBuffer sb = new StringBuffer();
			int c;

			while ((c = reader.read()) >= 0) {
				sb.append((char) c);
			}
			return format.unmap(sb.toString());
		} finally {
			if (closeStreams && reader != null) {
				try {
					reader.close();
				} catch (Exception ex) {

				}
			}
		}
	}

	@Override
	public void write(T t, OutputStream out) throws Exception {
		OutputStreamWriter writer = null;
		try {
			writer = new OutputStreamWriter(out);
			writer.write(format.map(t));
			writer.flush();
		} finally {
			if (closeStreams && writer != null) {
				try {
					writer.close();
				} catch (Exception ex) {

				}
			}
		}
	}
}