public class CreateIABookingController{
    
    public ItsApproved_Booking__c ItsApprovedBooking {get; set;}
    public ItsApproved_Product__c ItsApprovedProduct {get; set;}
    public ItsApproved_Design__c ItsApprovedDesign {get; set;}
    public ItsApproved_Delivery__c ItsApprovedDelivery {get; set;}
    List<ItsApproved_Product__c> productList = new List<ItsApproved_Product__c>();
    List<ItsApproved_Design__c> designList = new List<ItsApproved_Design__c>();
    List<ItsApproved_Delivery__c> deliveryList = new List<ItsApproved_Delivery__c>();
    private string OpportunityId;
    public Map<Integer, Row> rows { get; set; }
    public CreateIABookingController(){
        rows = new Map<Integer, Row>();
        addRow();
    }
    public void addRow() {
        Integer rowId = rows.size();
        rows.put(rowId , new Row(rowId, new ItsApproved_Product__c(), new ItsApproved_Design__c(), new ItsApproved_Delivery__c()));
    }
    public void removeRow() {
        rows.remove(Integer.valueOf(ApexPages.currentPage().getParameters().get('rowIdToRemove')));
    }
    public class Row {
        public Integer rowId { get; set; }
        public ItsApproved_Product__c ItsApprovedProduct {get; set;} 
        public ItsApproved_Design__c ItsApprovedDesign {get; set;} 
        public ItsApproved_Delivery__c ItsApprovedDelivery {get; set;}
        public Row(Integer rowId, ItsApproved_Product__c product, ItsApproved_Design__c design, ItsApproved_Delivery__c delivery) {
            this.rowId = rowId;
            ItsApprovedProduct = product;
            ItsApprovedDesign = design;
            ItsApprovedDelivery = delivery;
        }
    }
    {
        opportunityId = ApexPages.currentPage().getParameters().get('oppId');
        ItsApprovedBooking = new ItsApproved_Booking__c();
        ItsApprovedProduct = new ItsApproved_Product__c();
        ItsApprovedDesign = new ItsApproved_Design__c();
        ItsApprovedDelivery = new ItsApproved_Delivery__c();
    }
    
    
    public PageReference Save()
    {
        ItsApprovedBooking.Opportunity__c = opportunityId;
        insert ItsApprovedBooking;
        Map<string,string> product2DesignMap = new Map<string,string>();//map for designs
        Map<string,decimal> product2DeliveryMap = new Map<string,decimal>();//
        
        for (Row r : rows.values())
        {
            
            product2DesignMap.put(r.ItsApprovedProduct.Product_Code__c,r.ItsApprovedDesign.Artwork_Title__c);//maps product and
            //artwork title (both must be unique)
            product2DeliveryMap.put(r.ItsApprovedProduct.Product_Code__c,r.ItsApprovedDelivery.Depot_Code__c);//
            
            r.ItsApprovedProduct.Booking_Reference__c = ItsApprovedBooking.Id;
            r.ItsApprovedProduct.Opp__c = opportunityId;
            productList.add(r.ItsApprovedProduct);
            
            r.ItsApprovedDesign.Opportunity__c = opportunityId;
            r.ItsApprovedDesign.Product_Reference__c = ItsApprovedProduct.Id;
            designList.add(r.ItsApprovedDesign);
            
            r.ItsApprovedDelivery.Product__c = ItsApprovedProduct.Id;
            r.ItsApprovedDelivery.Opportunity__c = opportunityId;
            deliveryList.add(r.ItsApprovedDelivery);
            
        }
        
        if (productList.size() > 0)
        {
            insert productList;
        }
        for(ItsApproved_Product__c product : productList)
        {
            for (ItsApproved_Design__c design : designList)
            {
                if (product2DesignMap.get(product.Product_Code__c) == design.Artwork_Title__c)//adds additional product and design
                    design.Product_Reference__c = product.Id;
                
            }
            for (ItsApproved_Delivery__c delivery : deliveryList)
            {
                delivery.Product__c = product.Id;
            }
        }
        if(designList.size() > 0)
        {
            insert designList;
        }
        
        //Krishna loop
        List<ItsApproved_Design__c> designs = [select Id, Product_Reference__r.Product_Code__c from ItsApproved_Design__c 
                                               where Id IN : designList];
        Map<string, Id> productCode2DesignId = new Map<string, Id>();
        for (ItsApproved_Design__c design : designs)
        {
            productCode2DesignId.put(design.Product_Reference__r.Product_Code__c, design.Id);
        }
        
        //loop through the deliverList and productList and assign the design Id
        for (ItsApproved_Delivery__c delivery : deliveryList)
        {
            for(ItsApproved_Product__c product : productList)
            {
                if (product2DeliveryMap.get(product.Product_Code__c) == delivery.Depot_Code__c)
                {
                    delivery.ItsApproved_Design__c  = productCode2DesignId.get(product.Product_Code__c);
                    break;
                }
            }
        }
        //end Krishna loop
        
        if(deliveryList.size() > 0)
        {
            insert deliveryList;
        }
        
        return new PageReference ('/' + opportunityId);
        
    }
    
}