public with sharing class AddWellnessOppProduct {
                    
 public Opportunity Opp {get; set; }
 public Wellness_Service__c oldWS { get; set; }
 public Opportunity_Products__c oldOppProd {get; set;}
 public Account a { get; set; }
 public List<ExtProduct> products { get; set; }
 public Boolean isWizard { get; set; }
 public Boolean canCreate { get; set; }
 public PageReference returnUrl { get; set; }
    
    public AddWellnessOppProduct () {
        if (apexPages.currentPage().getParameters().get('oid') != null) {
            Opp = [
                select id, Name, RecordType.Name, Account.Id, Account.Name, StageName
                from Opportunity
                where id = : apexPages.currentPage().getParameters().get('oid')];
                
            A = [select name, Health_Group_Number__c, Account_Manager_Servicing_health__c, Account_Manager_Servicing_health__r.id,
            Account_Manager_Servicing_health__r.name from Account where id = : Opp.Account.Id];  
        }
        else if (apexPages.currentPage().getParameters().get('pid') != null) {
            oldOppProd = [
                select Id, Name, 
                    opportunity__r.Account.Id, opportunity__r.Account.Name,
                    Horizon_Product__c, Horizon_Product__r.Name
                from Opportunity_Products__c
                where id = : apexPages.currentPage().getParameters().get('pid')];
                
            Opp = [
                select id, Name, RecordType.Name, Account.Id, Account.Name 
                from Opportunity
                where id = : oldOppProd.Opportunity__c];
                
            A = [select name, Health_group_Number__c, Account_Manager_Servicing_health__c, Account_Manager_Servicing_health__r.id,
            Account_Manager_Servicing_health__r.name from Account where id = : Opp.Account.Id]; 
        }
    
        returnUrl = new PageReference('/apex/WellnessOppWizard?oid=' + Opp.Id);
                                       
         List<RecordType> HPrtList = [
            select Id, Name
            from RecordType
            where sObjectType = 'Horizon_Products__c'
                and Name = 'Wellness'];            
                            
        List<RecordType> OppProdRTList = [
            select Id, Name
            from RecordType
            where sObjectType = 'Opportunity_Products__c' and Name = 'Wellness'];
            
        Map<String, String> wsrtMap = new Map<String, String>();
        
        for (RecordType rt : OppProdRTList) {
            wsrtMap.put(rt.Name, String.ValueOf(rt.Id));
        }
                
        if (HPrtList.size() == 1) {
            String newRtId = '';
            List<Horizon_Products__c> productList = new List<Horizon_Products__c>();
            if (HPrtList[0].Name == 'Wellness') {
                newRtId = wsrtMap.get('Wellness');
                   productList = [
                    select id, name, Plan_Design_Category__c, New_Product_Link_Wellness__c,
                        Change_Product_Link__c, product_category__c,
                        Standard__c
                    from Horizon_Products__c h 
                    WHERE RecordTypeId = : HPrtList[0].Id 
                        and Product_Status__c = 'Active' 
                    order by plan_design_category__c];
            } 
                    
            else {
                canCreate = false;
            }
            products = new List<ExtProduct>();
            for (Horizon_Products__c p : productList) {
                products.add(new ExtProduct(p, Opp, returnUrl.getUrl(), newRtId, HPrtList[0].Name, oldOppProd, a));
            }              
        }
        else {
            canCreate = false;
        }
             
    }
    
    public pageReference Cancel() {
        return returnUrl;
    }
    
    public class ExtProduct {
        public Horizon_Products__c p { get; set; }
        public String link { get; set; }

        public ExtProduct(Horizon_Products__c p, Opportunity Opp, String retUrl, String rtId, String rtName, Opportunity_Products__c oldOppProd, Account a) {
            this.p = p;
            if (oldOppProd == null) {
                this.link = p.New_Product_Link_Wellness__c;
                this.link = this.link.replace('<OppId>', string.valueOf(Opp.Id));
                this.link = this.link.replace('<OppName>', Opp.Name.replace(' ', '+'));
                this.link = this.link.replace('<HPId>', string.valueOf(p.Id));
                this.link = this.link.replace('<HPName>', p.Name.replace(' ', '+'));
                this.link = this.link.replace('<AId>', string.valueOf(a.Id));
                this.link = this.link.replace('<AName>', A.Name.replace(' ', '+'));
                this.link = this.link.replace('<HPPlanCategory>', p.plan_design_category__c.replace(' ', '+'));
                this.link = this.link.replace('<HPProdCategory>', p.product_category__c.replace(' ', '+'));
                this.link = this.link.replace('<rtId>', rtId);
                this.link = this.link.replace('<returnUrl>', EncodingUtil.urlEncode(retUrl, 'UTF-8'));
                
            }
            else {
                this.link = p.Change_Product_Link__c;
                this.link = this.link.replace('<oppProductId>', string.valueOf(oldOppProd.Id));
                this.link = this.link.replace('<returnUrl>', EncodingUtil.urlEncode(retUrl, 'UTF-8'));                                
            }
            
          
        }
    }
}