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);

                         }

                 }

         }

 }