AttributeTableModel.java

/******************************************************************************
 * Copyright: GPL v3                                                          *
 *                                                                            *
 * 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 dba.gui.newRelation.auxClasses;

import dba.gui.FkWizard.FkWizard;
import dba.utils.Localization;
import dbaCore.data.Attribute;
import dbaCore.data.Database;
import dbaCore.data.RelationSchema;

import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;

/**
 * Custom Tablemodel which can handle our ArrayList with Attributes
 * and contains all Actions to handle it
 *
 * @author Andreas Freitag
 */
public class AttributeTableModel extends AbstractTableModel {
  /**
   *
   */
  private static final long serialVersionUID = 7202056709045229442L;
  private String[] columnNames;
  private ArrayList<Attribute> data;
  private Database database;
  private RelationSchema relation;

  private AttributeTableModel() {
    super();
    Localization locale = Localization.getInstance();
    columnNames = new String[]{locale.getString("AttrName"), locale.getString("AttrConst"), "PK", "FK"};
    data = new ArrayList<>();
  }

  /**
   * Constructor
   *
   * @param data ArrayList with Attributes
   */
  private AttributeTableModel(ArrayList<Attribute> data) {
    this();
    this.data = data;
  }

  public AttributeTableModel(ArrayList<Attribute> data, Database db, RelationSchema rel) {
    this(data);
    relation = rel;
    database = db;
  }

  @Override
  public int getColumnCount() {
    return columnNames.length;
  }

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

  @Override
  public String getColumnName(int col) {
    return columnNames[col];
  }

  @Override
  public Object getValueAt(int row, int col) {
    if (row < 0 || col < 0) {
      return null;
    }
    return getAttributeValueAt(data.get(row), col);
  }

  private Object getAttributeValueAt(Attribute attr, int column) {
    switch (column) {
      case 0:
        return attr.getName();
      case 1:
        return attr.getConstraints().toUpperCase();
      case 2:
        return attr.getIsPrimaryKey();
      case 3:
        return attr.getIsForeignKey();
      default:
        throw new IllegalArgumentException("ColumnIndex out of bounds");
    }
  }

  @Override
  public Class<?> getColumnClass(int c) {
    return getValueAt(0, c).getClass();
  }

  @Override
  public boolean isCellEditable(int row, int col) {
    return true;
  }

  @Override
  public void setValueAt(Object value, int row, int col) {
    setAttributeValueAt(data.get(row), col, value);
    fireTableCellUpdated(row, col);
  }

  private void setAttributeValueAt(Attribute attr, int column, Object value) {
    switch (column) {
      case 0:
        if (value instanceof String) {
          database.updateFkAttributeNames(relation.getName(), attr.getName(), (String) value);
          attr.setName((String) value);
        }
        return;
      case 1:
        if (value instanceof String) {
          attr.setConstraints((String) value);
        }
        return;
      case 2:
        if (value instanceof Boolean) {
          attr.setIsPrimaryKey((Boolean) value);
        }
        return;
      case 3:
        if (value instanceof Boolean) {
          if ((Boolean) value) {
            FkWizard wizard = new FkWizard(database, relation, attr);
            wizard.setVisible(true);
          } else {
            database.removeForeignKey(relation.getName(), attr.getName());
          }
          attr.setIsForeignKey((Boolean) value);
          return;
        }
      default:
        throw new IllegalArgumentException();
    }
  }

  /**
   * Delete row from our Table
   *
   * @param row Row to delete ( 0 until n )
   */
  public void removeRow(int row) {
    relation.removeAttribute(data.get(row));
    fireTableDataChanged();
  }

}