import javax.swing.table.*;

public class CheckData extends AbstractTableModel
{
    private Checkbook book = new Checkbook();

    public CheckData()
    {
    }
    public Object getValueAt(int row, int col)
    {
       Transaction t = book.getTransaction(row);
       if (col == 0)
          return t.getName();
       else if (col == 1)
          return String.valueOf(t.getAmount());
       else if (col == 2)
          return t.getDate();
       else 
          return String.valueOf(t.getStatus());
    }
    public String getColumnName(int col)
    {   if (col == 0)
           return "Name";
        else if (col == 1)
           return "Amount";
        else if (col == 2)
           return "Date";
        else
           return "Status";
    }
    public int getColumnCount()
    {
       return 4;
    }
    public int getRowCount()
    { 
        return book.getSize();
    }

    public void addTransaction(Transaction t)
    {
       book.addTransaction(t);
       fireTableRowsInserted(book.getSize()-1, book.getSize()-1);
    }

    public double getBalance()
    {  return book.getBalance();
    }
}