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.training.server.command;
018
019import com.echothree.control.user.training.common.form.GetPartyTrainingClassesForm;
020import com.echothree.control.user.training.common.result.GetPartyTrainingClassesResult;
021import com.echothree.control.user.training.common.result.TrainingResultFactory;
022import com.echothree.model.control.party.common.PartyTypes;
023import com.echothree.model.control.party.server.control.PartyControl;
024import com.echothree.model.control.security.common.SecurityRoleGroups;
025import com.echothree.model.control.security.common.SecurityRoles;
026import com.echothree.model.control.training.server.control.TrainingControl;
027import com.echothree.model.data.party.server.entity.Party;
028import com.echothree.model.data.training.server.entity.TrainingClass;
029import com.echothree.model.data.user.common.pk.UserVisitPK;
030import com.echothree.util.common.message.ExecutionErrors;
031import com.echothree.util.common.validation.FieldDefinition;
032import com.echothree.util.common.validation.FieldType;
033import com.echothree.util.common.command.BaseResult;
034import com.echothree.util.server.control.BaseSimpleCommand;
035import com.echothree.util.server.control.CommandSecurityDefinition;
036import com.echothree.util.server.control.PartyTypeDefinition;
037import com.echothree.util.server.control.SecurityRoleDefinition;
038import com.echothree.util.server.persistence.Session;
039import java.util.Arrays;
040import java.util.Collections;
041import java.util.List;
042
043public class GetPartyTrainingClassesCommand
044        extends BaseSimpleCommand<GetPartyTrainingClassesForm> {
045    
046    private final static CommandSecurityDefinition COMMAND_SECURITY_DEFINITION;
047    private final static List<FieldDefinition> FORM_FIELD_DEFINITIONS;
048    
049    static {
050        COMMAND_SECURITY_DEFINITION = new CommandSecurityDefinition(Collections.unmodifiableList(Arrays.asList(
051                new PartyTypeDefinition(PartyTypes.UTILITY.name(), null),
052                new PartyTypeDefinition(PartyTypes.EMPLOYEE.name(), Collections.unmodifiableList(Arrays.asList(
053                        new SecurityRoleDefinition(SecurityRoleGroups.PartyTrainingClass.name(), SecurityRoles.List.name())
054                        )))
055                )));
056
057        FORM_FIELD_DEFINITIONS = Collections.unmodifiableList(Arrays.asList(
058                new FieldDefinition("PartyName", FieldType.ENTITY_NAME, false, null, null),
059                new FieldDefinition("TrainingClassName", FieldType.ENTITY_NAME, false, null, null)
060                ));
061    }
062    
063    /** Creates a new instance of GetPartyTrainingClassesCommand */
064    public GetPartyTrainingClassesCommand(UserVisitPK userVisitPK, GetPartyTrainingClassesForm form) {
065        super(userVisitPK, form, COMMAND_SECURITY_DEFINITION, FORM_FIELD_DEFINITIONS, true);
066    }
067    
068    @Override
069    protected BaseResult execute() {
070        GetPartyTrainingClassesResult result = TrainingResultFactory.getGetPartyTrainingClassesResult();
071        String partyName = form.getPartyName();
072        String trainingClassName = form.getTrainingClassName();
073        var parameterCount = (partyName == null ? 0 : 1) + (trainingClassName == null ? 0 : 1);
074        
075        if(parameterCount == 1) {
076            var trainingControl = Session.getModelController(TrainingControl.class);
077            
078            if(partyName != null) {
079                var partyControl = Session.getModelController(PartyControl.class);
080                Party party = partyControl.getPartyByName(partyName);
081                
082                if(party != null) {
083                    result.setParty(partyControl.getPartyTransfer(getUserVisit(), party));
084                    result.setPartyTrainingClasses(trainingControl.getPartyTrainingClassTransfersByParty(getUserVisit(), party));
085                } else {
086                    addExecutionError(ExecutionErrors.UnknownPartyName.name(), partyName);
087                }
088            } else if(trainingClassName != null) {
089                TrainingClass trainingClass = trainingControl.getTrainingClassByName(trainingClassName);
090                
091                if(trainingClass != null) {
092                    result.setTrainingClass(trainingControl.getTrainingClassTransfer(getUserVisit(), trainingClass));
093                    result.setPartyTrainingClasses(trainingControl.getPartyTrainingClassTransfersByTrainingClass(getUserVisit(), trainingClass));
094                } else {
095                    addExecutionError(ExecutionErrors.UnknownTrainingClassName.name(), trainingClassName);
096                }
097            }
098        } else {
099            addExecutionError(ExecutionErrors.InvalidParameterCount.name());
100        }
101        
102        return result;
103    }
104    
105}