001// --------------------------------------------------------------------------------
002// Copyright 2002-2024 Echo Three, LLC
003//
004// Licensed under the Apache License, Version 2.0 (the "License");
005// you may not use this file except in compliance with the License.
006// You may obtain a copy of the License at
007//
008//     http://www.apache.org/licenses/LICENSE-2.0
009//
010// Unless required by applicable law or agreed to in writing, software
011// distributed under the License is distributed on an "AS IS" BASIS,
012// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013// See the License for the specific language governing permissions and
014// limitations under the License.
015// --------------------------------------------------------------------------------
016
017package com.echothree.control.user.associate.server.command;
018
019import com.echothree.control.user.associate.common.form.GetAssociateReferralsForm;
020import com.echothree.control.user.associate.common.result.AssociateResultFactory;
021import com.echothree.control.user.associate.common.result.GetAssociateReferralsResult;
022import com.echothree.model.control.associate.server.control.AssociateControl;
023import com.echothree.model.data.associate.server.entity.Associate;
024import com.echothree.model.data.associate.server.entity.AssociateProgram;
025import com.echothree.model.data.user.common.pk.UserVisitPK;
026import com.echothree.util.common.message.ExecutionErrors;
027import com.echothree.util.common.validation.FieldDefinition;
028import com.echothree.util.common.validation.FieldType;
029import com.echothree.util.common.command.BaseResult;
030import com.echothree.util.server.control.BaseSimpleCommand;
031import com.echothree.util.server.persistence.Session;
032import java.util.Arrays;
033import java.util.Collections;
034import java.util.List;
035
036public class GetAssociateReferralsCommand
037        extends BaseSimpleCommand<GetAssociateReferralsForm> {
038    
039    private final static List<FieldDefinition> FORM_FIELD_DEFINITIONS;
040    
041    static {
042        FORM_FIELD_DEFINITIONS = Collections.unmodifiableList(Arrays.asList(
043                new FieldDefinition("AssociateProgramName", FieldType.ENTITY_NAME, true, null, null),
044                new FieldDefinition("AssociateName", FieldType.ENTITY_NAME, true, null, null)
045                ));
046    }
047    
048    /** Creates a new instance of GetAssociateReferralsCommand */
049    public GetAssociateReferralsCommand(UserVisitPK userVisitPK, GetAssociateReferralsForm form) {
050        super(userVisitPK, form, null, FORM_FIELD_DEFINITIONS, true);
051    }
052    
053    @Override
054    protected BaseResult execute() {
055        var associateControl = Session.getModelController(AssociateControl.class);
056        GetAssociateReferralsResult result = AssociateResultFactory.getGetAssociateReferralsResult();
057        String associateProgramName = form.getAssociateProgramName();
058        AssociateProgram associateProgram = associateControl.getAssociateProgramByName(associateProgramName);
059        
060        if(associateProgram != null) {
061            String associateName = form.getAssociateName();
062            Associate associate = associateControl.getAssociateByName(associateProgram, associateName);
063            
064            result.setAssociateProgram(associateControl.getAssociateProgramTransfer(getUserVisit(), associateProgram));
065            
066            if(associate != null) {
067                result.setAssociate(associateControl.getAssociateTransfer(getUserVisit(), associate));
068                result.setAssociateReferrals(associateControl.getAssociateReferralTransfersByAssociate(associate, getUserVisit()));
069            } else {
070                addExecutionError(ExecutionErrors.UnknownAssociateName.name(), associateName);
071            }
072        } else {
073            addExecutionError(ExecutionErrors.UnknownAssociateProgramName.name(), associateProgramName);
074        }
075        
076        return result;
077    }
078    
079}