package ch.braincell.stickbackup.gui.table;
   
   import java.awt.Color;
   import java.io.File;
   import java.util.ArrayList;
   import java.util.HashMap;
   import java.util.List;
   import java.util.Map;
   
  import javax.swing.table.AbstractTableModel;
  
  import ch.braincell.stickbackup.config.Config;
  
  public class FileTableModel extends AbstractTableModel {
  
  	/***
  	 * generated serialVersionUID.
  	 */
  	private static final long serialVersionUID = 1L;
  
  	private Config config = Config.getInstance();
  
  	private String[] columns = new String[2];
  
  	private List<FileData[]> rows = new ArrayList<FileData[]>();
  
  	private Map<File, FileRow> searchRow = new HashMap<File, FileRow>();
  
  	public int getColumnCount() {
  		return columns.length;
  	}
  
  	@Override
  	public String getColumnName(int column) {
  		return columns[column];
  	}
  
  	public int getRowCount() {
  		return rows.size();
  	}
  
  	public Object getValueAt(int rowIndex, int columnIndex) {
  		return rows.get(rowIndex)[columnIndex];
  	}
  
  	// TODO: die Farbe muss in's Modell --> FileData
  	public void addFiles(File source, File backup) {
  		Color sourcecolor = null == backup ? config.getSourceColor() : config
  				.getChangedColor();
  		Color backupcolor = null == source ? config.getDestinationColor()
  				: config.getChangedColor();
  		FileData sourcedata = new FileData(source, sourcecolor);
  		FileData backupData = new FileData(backup, backupcolor);
  		FileData[] files = { sourcedata, backupData };
  		rows.add(files);
  		if (null != source)
  			searchRow.put(source, new FileRow(rows.size() - 1, 0));
  		if (null != backup)
  			searchRow.put(backup, new FileRow(rows.size() - 1, 1));
  		this.fireTableRowsInserted(rows.size(), rows.size());
  	}
  
  	public void setWorkingColor(File file) {
  		FileRow row = searchRow.get(file);
  		if (null != row) {
  			FileData data = rows.get(row.getRow())[row.getColumn()];
  			data.setColor(config.getWorkingColor());
  			fireTableCellUpdated(row.getRow(), row.getColumn());
  		}
  		else {
  			// TODO: Logging is here...
  			System.err.println("Row not found for file " + file);
  		}
  	}
  
  	public void setFinishedColor(File file) {
  		FileRow row = searchRow.get(file);
  		if (null != row) {
  			FileData data = rows.get(row.getRow())[row.getColumn()];
  			data.setColor(config.getFinishedColor());
  			fireTableCellUpdated(row.getRow(), row.getColumn());
  		}
  		else {
  			// TODO: Logging is here...
  			System.err.println("Row not found for file " + file);
  		}
  	}
  
  	public void clear(String sourceColumnName, String backupColumnName) {
  		rows.clear();
  		searchRow.clear();
  		columns[0] = sourceColumnName;
  		columns[1] = backupColumnName;
  		this.fireTableStructureChanged();
  	}
  
  	public class FileRow {
  		private final int row;
  
 		private final int column;
 
 		public FileRow(int row, int column) {
 			this.row = row;
 			this.column = column;
 		}
 
 		public int getColumn() {
 			return column;
 		}
 
 		public int getRow() {
 			return row;
 		}
 	}
 
 	class FileData {
 		private final File file;
 
 		private Color color;
 
 		public FileData(File file, Color color) {
 			this.file = file;
 			this.color = color;
 		}
 
 		public File getFile() {
 			return file;
 		}
 
 		public Color getColor() {
 			return color;
 		}
 
 		public void setColor(Color color) {
 			this.color = color;
 		}
 	}
 }