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