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.model.control.training.server.transfer;
018
019import com.echothree.model.control.core.server.control.CoreControl;
020import com.echothree.model.control.party.common.transfer.PartyTransfer;
021import com.echothree.model.control.party.server.control.PartyControl;
022import com.echothree.model.control.training.common.TrainingOptions;
023import com.echothree.model.control.training.common.transfer.PartyTrainingClassTransfer;
024import com.echothree.model.control.training.common.transfer.TrainingClassTransfer;
025import com.echothree.model.control.training.server.control.TrainingControl;
026import com.echothree.model.control.training.common.training.PartyTrainingClassStatusConstants;
027import com.echothree.model.control.workflow.common.transfer.WorkflowEntityStatusTransfer;
028import com.echothree.model.control.workflow.server.control.WorkflowControl;
029import com.echothree.model.data.core.server.entity.EntityInstance;
030import com.echothree.model.data.training.server.entity.PartyTrainingClass;
031import com.echothree.model.data.training.server.entity.PartyTrainingClassDetail;
032import com.echothree.model.data.user.server.entity.UserVisit;
033import com.echothree.util.common.transfer.ListWrapper;
034import com.echothree.util.server.persistence.Session;
035import java.util.Set;
036
037public class PartyTrainingClassTransferCache
038        extends BaseTrainingTransferCache<PartyTrainingClass, PartyTrainingClassTransfer> {
039    
040    CoreControl coreControl = Session.getModelController(CoreControl.class);
041    PartyControl partyControl = Session.getModelController(PartyControl.class);
042    WorkflowControl workflowControl = Session.getModelController(WorkflowControl.class);
043    boolean includePartyTrainingClassSessions;
044    
045    /** Creates a new instance of PartyTrainingClassTransferCache */
046    public PartyTrainingClassTransferCache(UserVisit userVisit, TrainingControl trainingControl) {
047        super(userVisit, trainingControl);
048        
049        var options = session.getOptions();
050        if(options != null) {
051            includePartyTrainingClassSessions = options.contains(TrainingOptions.PartyTrainingClassIncludePartyTrainingClassSessions);
052        }
053        
054        setIncludeEntityInstance(true);
055    }
056    
057    public PartyTrainingClassTransfer getPartyTrainingClassTransfer(PartyTrainingClass partyTrainingClass) {
058        PartyTrainingClassTransfer partyTrainingClassTransfer = get(partyTrainingClass);
059
060        if(partyTrainingClassTransfer == null) {
061            PartyTrainingClassDetail partyTrainingClassDetail = partyTrainingClass.getLastDetail();
062            String partyTrainingClassName = partyTrainingClassDetail.getPartyTrainingClassName();
063            PartyTransfer partyTransfer = partyControl.getPartyTransfer(userVisit, partyTrainingClassDetail.getParty());
064            TrainingClassTransfer trainingClassTransfer = trainingControl.getTrainingClassTransfer(userVisit, partyTrainingClassDetail.getTrainingClass());
065            Long unformattedCompletedTime = partyTrainingClassDetail.getCompletedTime();
066            String completedTime = unformattedCompletedTime == null ? null : formatTypicalDateTime(unformattedCompletedTime);
067            Long unformattedValidUntilTime = partyTrainingClassDetail.getValidUntilTime();
068            String validUntilTime = unformattedValidUntilTime == null ? null : formatTypicalDateTime(unformattedValidUntilTime);
069
070            EntityInstance entityInstance = coreControl.getEntityInstanceByBasePK(partyTrainingClass.getPrimaryKey());
071            WorkflowEntityStatusTransfer partyTrainingClassStatus = workflowControl.getWorkflowEntityStatusTransferByEntityInstanceUsingNames(userVisit,
072                    PartyTrainingClassStatusConstants.Workflow_PARTY_TRAINING_CLASS_STATUS, entityInstance);
073
074            partyTrainingClassTransfer = new PartyTrainingClassTransfer(partyTrainingClassName, partyTransfer, trainingClassTransfer,
075                    unformattedCompletedTime, completedTime, unformattedValidUntilTime, validUntilTime, partyTrainingClassStatus);
076            put(partyTrainingClass, partyTrainingClassTransfer);
077            setupOwnedWorkEfforts(null, entityInstance, partyTrainingClassTransfer);
078
079            if(includePartyTrainingClassSessions) {
080                partyTrainingClassTransfer.setPartyTrainingClassSessions(new ListWrapper<>(trainingControl.getPartyTrainingClassSessionTransfersByPartyTrainingClass(userVisit, partyTrainingClass)));
081            }
082        }
083        
084        return partyTrainingClassTransfer;
085    }
086    
087}