/*
 * Copyright (c) 2009, SQL Power Group Inc.
 *
 * This file is part of SQL Power Library.
 *
 * SQL Power Library 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.
 *
 * SQL Power 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 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 ca.sqlpower.swingui.table;

import java.io.IOException;
import java.io.PrintWriter;
import java.text.Format;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.swing.table.TableModel;

import au.com.bytecode.opencsv.CSVWriter;

/**
 * This class exports TableModels to a CSV file.
 */
public class TableModelCSVFormatter implements ExportFormatter {

    /**
     * Formatters are given an object from the table model and must output a
     * string representation of that object. This allows classes using the
     * formatter to define different strings to represent an object as other
     * than just the toString version of the object.
     */
    private final Map<Integer, Format> columnFormatters = new HashMap<Integer, Format>();
	
	public TableModelCSVFormatter(){
		
	}

    /**
     * Sets a formatter for the given column of a table model. If the column does
     * not exist because the table is too small the formatter will not be used.
     */
	public void setFormatter(int column, Format formatter) {
	    columnFormatters.put(column, formatter);
	}
	
	/**
	 * Converts the given model into a CSV file and writes it to the output
	 * stream.
	 */	
	public void formatToStream(TableModel model, PrintWriter writer) {
		int[] allRows = new int[model.getRowCount()];
		for (int i = 0; i < model.getRowCount(); i++) {
			allRows[i] = i;
		}
		formatToStream(model, writer, allRows);

	}
	/**
	 * Outputs a table in HTML that represents the TableModel. Only the rows
	 * given will be written to the file.
	 */
	public void formatToStream(TableModel model, PrintWriter writer, int[] selectedRows) {
		try {
			CSVWriter csvWriter = new CSVWriter(writer);
			List<String[]> tableExport = new ArrayList<String[]>();
			int columnCount = model.getColumnCount();
			String[] rowArray = new String[columnCount];
			for (int i = 0; i < columnCount; i++) {
				rowArray[i] = model.getColumnName(i);
			}
			tableExport.add(rowArray);
			for (int row = 0; row < selectedRows.length; row++) {
				rowArray = new String[columnCount];
				for (int col = 0; col < columnCount; col++) {
					Object value = model.getValueAt(selectedRows[row], col);
					if (value != null) {
					    if (columnFormatters.get(col) != null) {
					        rowArray[col] = columnFormatters.get(col).format(value);
					    } else {
					        rowArray[col] = value.toString();
					    }
					} else {
						rowArray[col] = "";
					}
				}
				tableExport.add(rowArray);
			}
			csvWriter.writeAll(tableExport);
			csvWriter.close();

		} catch (IOException ex) {
			throw new RuntimeException("Could not close the CSV Writer", ex);
		}		
	}	  
}