package net.sourceforge.finmodel.plan;
   
   import java.math.BigDecimal;
   import java.util.ArrayList;
   import java.util.Calendar;
   import java.util.Collections;
   import java.util.HashSet;
   import java.util.List;
   import java.util.Set;
  
  import net.sourceforge.finmodel.account.Account;
  import net.sourceforge.finmodel.account.AccountImpl;
  import net.sourceforge.finmodel.account.CashAccount;
  import net.sourceforge.finmodel.rules.OneTimeRule;
  import net.sourceforge.finmodel.rules.Rule;
  import net.sourceforge.finmodel.rules.RuleImpl;
  
  import org.hibernate.Session;
  import org.hibernate.SessionFactory;
  import org.hibernate.Transaction;
  import org.hibernate.cfg.Configuration;
  import org.jfin.date.ScheduleException;
  
  public class FinancialPlan {
  	private long id;
  	private String name;
  	private Set<RuleImpl> rules = new HashSet<RuleImpl>();
  	private AccountImpl openingBalance;
  	private Calendar initiationDate;
  	
  //	private List<Account> accounts = new ArrayList<Account>();
  	
  	public FinancialPlan() {
  		this.name = "New Plan";
  		this.openingBalance = new CashAccount("Opening Balance");
  	}
  	
  	private static Session getSession() {
  		Session session = null;
  		
  		SessionFactory sessionFactory = new Configuration().configure().buildSessionFactory();
  		session = sessionFactory.openSession();
  		
  		return session;
  	}
  	
  	public void save() {
  		Session session = getSession();
  		Transaction tx = session.beginTransaction();		
  		session.save(this);
  		session.flush();
  		tx.commit();
  	}
  	
  	public static FinancialPlan load( long id )
  	{
  		Session session = getSession();		
  		return (FinancialPlan) session.load(FinancialPlan.class, id);		
  	}
  	
  	public static List<FinancialPlan> allPlans() 
  	{
  		Session session = getSession();
  		List<FinancialPlan> rc = session.createQuery("from net.sourceforge.finmodel.plan.FinancialPlan").list();
  		
  		return rc;
  	}
  	
  	public void addAccount( Account acct, BigDecimal openingBal) {
  		OneTimeRule openRule = new OneTimeRule();
  		openRule.setSource( openingBalance );
  		openRule.setDestination(acct);
  		openRule.setAmount(openingBal);
  		openRule.setDate(initiationDate);
  		
  		addRule(openRule);
  	}
  	
  	public Set<Account> getAccounts() {
  		Set<Account> rc = new HashSet<Account>();
  		
  		for (RuleImpl rule: rules) {
  			rc.add(rule.getSource());
  			rc.add(rule.getDestination());
  		}
  		
  		return rc;
  	}
  	
  	public void setRules( Set<RuleImpl> rules ) {
  		this.rules = rules;
  	}
  	
  	public Set<RuleImpl> getRules() {
  		return this.rules;
  	}
  	
  	public void addRule( RuleImpl rule ) {
  		rules.add( rule );
 		rule.setPlan(this);
 	}
 	
 	public void removeRule( RuleImpl rule ) {
 		rules.remove( rule );
 	}
 	
 	public List<net.sourceforge.finmodel.plan.Transaction> getScheduledTransactions() 
 		throws ScheduleException
 	{
 		List<net.sourceforge.finmodel.plan.Transaction> rc =
 			new ArrayList<net.sourceforge.finmodel.plan.Transaction>();
 		for (Rule rule: rules) {
 			for (Calendar time: rule.getSchedule()) {
 				rc.add( rule.createTransaction( time ));
 			}
 		}
 		Collections.sort(rc, new TransactionComparator());
 		return rc;
 	}
 
 	/**
 	 * @param name the name to set
 	 */
 	public void setName(String name) {
 		this.name = name;
 	}
 
 	/**
 	 * @return the name
 	 */
 	public String getName() {
 		return name;
 	}
 
 	/**
 	 * @param id the id to set
 	 */
 	public void setId(long id) {
 		this.id = id;
 	}
 
 	/**
 	 * @return the id
 	 */
 	public long getId() {
 		return id;
 	}
 }