ResetBudgetItemsCommand.java

/**
 * Copyright (c) 2012, Lindsay Bradford and other Contributors.
 * All rights reserved.
 * 
 * This program and the accompanying materials  are made available 
 * under the terms of the BSD 3-Clause licence  which accompanies 
 * this distribution, and is available at
 * http://opensource.org/licenses/BSD-3-Clause
 */

package blacksmyth.personalfinancier.control.budget.command;

import java.util.ArrayList;

import javax.swing.undo.CannotRedoException;
import javax.swing.undo.CannotUndoException;

import blacksmyth.personalfinancier.model.budget.BudgetItem;
import blacksmyth.personalfinancier.model.budget.BudgetModel;

public class ResetBudgetItemsCommand extends AbstractBudgetCommand {
  
  private BudgetModel model;
  private ArrayList<BudgetItem> preCommandIncomeItems;
  private ArrayList<BudgetItem> preCommandExpenseItems;
  
  public static ResetBudgetItemsCommand doCmd(BudgetModel model) {
    ResetBudgetItemsCommand command = new ResetBudgetItemsCommand(model);
    command.redo();
    return command;    
  }
  
  protected ResetBudgetItemsCommand(BudgetModel model) {
    this.model = model;
    this.preCommandIncomeItems = model.getIncomeItems();
    this.preCommandExpenseItems = model.getExpenseItems();
  }

  @Override
  public void redo() throws CannotRedoException {
    model.setExpenseItems(new ArrayList<BudgetItem>());
    model.setIncomeItems(new ArrayList<BudgetItem>());
  }

  @Override
  public void undo() throws CannotUndoException {
    model.setExpenseItems(this.preCommandExpenseItems);
    model.setIncomeItems(this.preCommandIncomeItems);
  }
}