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.geo.server.command;
018
019import com.echothree.control.user.geo.common.form.AddZipCodeToCityForm;
020import com.echothree.model.control.geo.common.GeoCodeAliasTypes;
021import com.echothree.model.control.geo.common.GeoCodeTypes;
022import com.echothree.model.control.geo.server.control.GeoControl;
023import com.echothree.model.control.party.common.PartyTypes;
024import com.echothree.model.control.security.common.SecurityRoleGroups;
025import com.echothree.model.control.security.common.SecurityRoles;
026import com.echothree.model.data.geo.server.entity.GeoCode;
027import com.echothree.model.data.geo.server.entity.GeoCodeRelationship;
028import com.echothree.util.common.command.BaseResult;
029import com.echothree.util.common.message.ExecutionErrors;
030import com.echothree.util.common.validation.FieldDefinition;
031import com.echothree.util.common.validation.FieldType;
032import com.echothree.util.server.control.BaseSimpleCommand;
033import com.echothree.util.server.control.CommandSecurityDefinition;
034import com.echothree.util.server.control.PartyTypeDefinition;
035import com.echothree.util.server.control.SecurityRoleDefinition;
036import com.echothree.util.server.persistence.Session;
037import java.util.Collection;
038import java.util.List;
039import javax.enterprise.context.Dependent;
040
041@Dependent
042public class AddZipCodeToCityCommand
043        extends BaseSimpleCommand<AddZipCodeToCityForm> {
044    
045    private final static CommandSecurityDefinition COMMAND_SECURITY_DEFINITION;
046    private final static List<FieldDefinition> FORM_FIELD_DEFINITIONS;
047    
048    static {
049        COMMAND_SECURITY_DEFINITION = new CommandSecurityDefinition(List.of(
050                new PartyTypeDefinition(PartyTypes.UTILITY.name(), null),
051                new PartyTypeDefinition(PartyTypes.EMPLOYEE.name(), List.of(
052                    new SecurityRoleDefinition(SecurityRoleGroups.City.name(), SecurityRoles.Edit.name())
053                    ))
054                ));
055
056        FORM_FIELD_DEFINITIONS = List.of(
057                new FieldDefinition("CityGeoCodeName", FieldType.ENTITY_NAME, true, null, null),
058                new FieldDefinition("ZipCodeGeoCodeName", FieldType.ENTITY_NAME, false, null, null),
059                new FieldDefinition("ZipCodeName", FieldType.ENTITY_NAME, false, null, null)
060                );
061    }
062    
063    /** Creates a new instance of AddZipCodeToCityCommand */
064    public AddZipCodeToCityCommand() {
065        super(COMMAND_SECURITY_DEFINITION, FORM_FIELD_DEFINITIONS, false);
066    }
067    
068    @Override
069    protected BaseResult execute() {
070        var zipCodeGeoCodeName = form.getZipCodeGeoCodeName();
071        var zipCodeName = form.getZipCodeName();
072        var parameterCount = (zipCodeGeoCodeName == null ? 0 : 1) + (zipCodeName == null ? 0 : 1);
073        
074        if(parameterCount == 1) {
075            var geoControl = Session.getModelController(GeoControl.class);
076            var geoCodeType = geoControl.getGeoCodeTypeByName(GeoCodeTypes.ZIP_CODE.name());
077            var cityGeoCodeName = form.getCityGeoCodeName();
078            var cityGeoCode = geoControl.getGeoCodeByName(cityGeoCodeName);
079            GeoCode zipCodeGeoCode = null;
080            
081            if(zipCodeGeoCodeName != null) {
082                zipCodeGeoCode = geoControl.getGeoCodeByName(zipCodeGeoCodeName);
083                
084                if(!zipCodeGeoCode.getLastDetail().getGeoCodeType().equals(geoCodeType)) {
085                    zipCodeGeoCode = null;
086                }
087                
088                if(zipCodeGeoCode == null) {
089                    addExecutionError(ExecutionErrors.UnknownZipCodeGeoCodeName.name(), zipCodeGeoCodeName);
090                }
091            } else if(zipCodeName != null) {
092                var stateGeoCodeType = geoControl.getGeoCodeTypeByName(GeoCodeTypes.STATE.name());
093                GeoCode stateGeoCode = null;
094                Collection cityRelationships = geoControl.getGeoCodeRelationshipsByFromGeoCode(cityGeoCode);
095                for(var iter = cityRelationships.iterator(); iter.hasNext();) {
096                    var geoCodeRelationship = (GeoCodeRelationship)iter.next();
097                    var toGeoCode = geoCodeRelationship.getToGeoCode();
098                    if(toGeoCode.getLastDetail().getGeoCodeType().equals(stateGeoCodeType)) {
099                        stateGeoCode = toGeoCode;
100                        break;
101                    }
102                }
103
104                var countryGeoCodeType = geoControl.getGeoCodeTypeByName(GeoCodeTypes.COUNTRY.name());
105                GeoCode countryGeoCode = null;
106                Collection stateRelationships = geoControl.getGeoCodeRelationshipsByFromGeoCode(stateGeoCode);
107                for(var iter = stateRelationships.iterator(); iter.hasNext();) {
108                    var geoCodeRelationship = (GeoCodeRelationship)iter.next();
109                    var toGeoCode = geoCodeRelationship.getToGeoCode();
110                    if(toGeoCode.getLastDetail().getGeoCodeType().equals(countryGeoCodeType)) {
111                        countryGeoCode = toGeoCode;
112                        break;
113                    }
114                }
115
116                var countryGeoCodeAliasType = geoControl.getGeoCodeAliasTypeByName(countryGeoCode.getLastDetail().getGeoCodeType(), GeoCodeAliasTypes.ISO_2_LETTER.name());
117                var countryGeoCodeAlias = geoControl.getGeoCodeAlias(countryGeoCode, countryGeoCodeAliasType);
118                var countryIso2Letter = countryGeoCodeAlias.getAlias();
119
120                var geoCodeScopeName = countryIso2Letter + "_ZIP_CODES";
121                var geoCodeScope = geoControl.getGeoCodeScopeByName(geoCodeScopeName);
122
123                var geoCodeAliasType = geoControl.getGeoCodeAliasTypeByName(geoCodeType, GeoCodeAliasTypes.ZIP_CODE.name());
124                var geoCodeAlias = geoControl.getGeoCodeAliasByAliasWithinScope(geoCodeScope, geoCodeAliasType, zipCodeName);
125                zipCodeGeoCode = geoCodeAlias.getGeoCode();
126                
127                if(zipCodeGeoCode == null) {
128                    addExecutionError(ExecutionErrors.UnknownZipCodeName.name(), zipCodeName);
129                }
130            }
131            
132            if(zipCodeGeoCode != null) {
133                var geoCodeRelationship = geoControl.getGeoCodeRelationship(zipCodeGeoCode, cityGeoCode);
134                
135                if(geoCodeRelationship == null) {
136                    geoControl.createGeoCodeRelationship(zipCodeGeoCode, cityGeoCode, getPartyPK());
137                } else {
138                    addExecutionError(ExecutionErrors.DuplicateGeoCodeRelationship.name());
139                }
140            }
141        } else {
142            addExecutionError(ExecutionErrors.InvalidParameterCount.name());
143        }
144        
145        return null;
146    }
147    
148}