package edu.sampleu.bookstore.maintenance;
   
   import java.util.ArrayList;
   import java.util.List;
   
   import org.kuali.rice.kns.maintenance.KualiMaintainableImpl;
   import org.kuali.rice.krad.service.KRADServiceLocator;
   import edu.sampleu.bookstore.bo.Account;
   import edu.sampleu.bookstore.bo.Address;
  import edu.sampleu.bookstore.bo.Author;
  
  /**
   * maintainableClass for Author document.
   * Action to be taken before saving the BO
   */
  
  public class AuthorMaintainable extends KualiMaintainableImpl {
  	
  	
  	private static final long serialVersionUID = 1L;
  
  	@Override
  	public void saveBusinessObject() {
  		// TODO Auto-generated method stub
  		Author author = (Author) this.getBusinessObject();
  		 
  		Account account = (Account)author.getExtension();
  		if(account != null && account.getAuthorId() == null) {			
  			author.setExtension(null);
  		}
  		
  		List<Address> addresses = new ArrayList<Address>();
  		addresses = author.getAddresses();
  		for(Address address : addresses){
  			if(address != null && address.getAuthorId() == null) {			
  				address.setAuthorId(null);
  			}
  		}
  		
  		KRADServiceLocator.getBusinessObjectService().save(author);
  		
  		if(account != null && account.getAuthorId() == null) {			
  			account.setAuthorId(author.getAuthorId());			
  			KRADServiceLocator.getBusinessObjectService().save(account);
  		}
  		for(Address address : addresses){
  			if(address != null && address.getAuthorId() == null) {			
  				address.setAuthorId(author.getAuthorId());			
  				KRADServiceLocator.getBusinessObjectService().save(address);
  			}
  		}
  	}
  
  }