package net.oni2.aeinstaller.gui.modtable;

import java.util.ResourceBundle;
import java.util.Vector;

import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableColumn;

import net.oni2.aeinstaller.backend.depot.DepotConfig;
import net.oni2.aeinstaller.backend.depot.DepotManager;
import net.oni2.aeinstaller.backend.depot.model.NodeMod;

/**
 * @author Christian Illy
 */
public class ModTableModel extends AbstractTableModel {

	private static final long serialVersionUID = -8278155705802697354L;

	private ResourceBundle bundle = ResourceBundle.getBundle(getClass()
			.getName());

	private Vector<NodeMod> items;
	private int vocabModTypeID = -1;
	private int vocabPlatformID = -1;

	/**
	 * Create a new model
	 */
	public ModTableModel() {
		reloadData();
	}

	@Override
	public Object getValueAt(int row, int col) {
		NodeMod node = items.get(row);
		switch (col) {
			case -1:
				return node;
			case 0:
				return node.getTitle();
			case 1:
				return node.getFields().get("package_number");
			case 2:
				String type = "";
				if (vocabModTypeID < 0) {
					vocabModTypeID = DepotManager.getInstance()
							.getVocabulary(DepotConfig.getVocabularyName_ModType()).getVid();
				}
				for (int tid : node.getTaxonomyTerms().get(vocabModTypeID)) {
					if (type.length() > 0)
						type += ", ";
					type += DepotManager.getInstance().getTaxonomyTerm(tid)
							.getName();
				}
				return type;
			case 3:
				if (vocabPlatformID < 0) {
					vocabPlatformID = DepotManager.getInstance()
							.getVocabulary(DepotConfig.getVocabularyName_Platform()).getVid();
				}
				int tid = node.getTaxonomyTerms().get(vocabPlatformID)
						.iterator().next();
				return DepotManager.getInstance().getTaxonomyTerm(tid)
						.getName();
		}
		return null;
	}

	@Override
	public String getColumnName(int col) {
		switch (col) {
			case 0:
				return bundle.getString("mod.name");
			case 1:
				return bundle.getString("mod.package_number");
			case 2:
				return bundle.getString("mod.type");
			case 3:
				return bundle.getString("mod.platform");
		}
		return null;
	}

	@Override
	public int getRowCount() {
		return items.size();
	}

	@Override
	public int getColumnCount() {
		return 4;
	}

	@Override
	public Class<?> getColumnClass(int col) {
		switch (col) {
			case 0:
				return String.class;
			case 1:
				return Integer.class;
			case 2:
				return String.class;
			case 3:
				return String.class;
		}
		return null;
	}

	/**
	 * Set the constraints on the columns size for the given column
	 * 
	 * @param colNum
	 *            Column number
	 * @param col
	 *            Column object
	 */
	public void setColumnConstraints(int colNum, TableColumn col) {
		int w;
		switch (colNum) {
			case 0:
				col.setPreferredWidth(150);
				break;
			case 1:
				w = 55;
				col.setPreferredWidth(w);
				col.setMinWidth(w);
				col.setMaxWidth(w);
				break;
			case 2:
				col.setPreferredWidth(100);
				break;
			case 3:
				w = 70;
				col.setPreferredWidth(w);
				col.setMinWidth(w);
				col.setMaxWidth(w);
				break;
		}
	}

	/**
	 * Reload the nodes data after an update to the cache
	 */
	public void reloadData() {
		items = DepotManager.getInstance().getModPackageNodes();
	}

	/**
	 * Get the items vector
	 * 
	 * @return Items
	 */
	public Vector<NodeMod> getItems() {
		return items;
	}

}