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.AddCityToCountyForm;
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 AddCityToCountyCommand
043        extends BaseSimpleCommand<AddCityToCountyForm> {
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.County.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("CountyGeoCodeName", FieldType.ENTITY_NAME, false, null, null),
059                new FieldDefinition("CountyName", FieldType.ENTITY_NAME, false, null, null)
060                );
061    }
062    
063    /** Creates a new instance of AddCityToCountyCommand */
064    public AddCityToCountyCommand() {
065        super(COMMAND_SECURITY_DEFINITION, FORM_FIELD_DEFINITIONS, false);
066    }
067    
068    @Override
069    protected BaseResult execute() {
070        var countyGeoCodeName = form.getCountyGeoCodeName();
071        var countyName = form.getCountyName();
072        var parameterCount = (countyGeoCodeName == null ? 0 : 1) + (countyName == null ? 0 : 1);
073        
074        if(parameterCount == 1) {
075            var geoControl = Session.getModelController(GeoControl.class);
076            var geoCodeType = geoControl.getGeoCodeTypeByName(GeoCodeTypes.COUNTY.name());
077            var cityGeoCodeName = form.getCityGeoCodeName();
078            var cityGeoCode = geoControl.getGeoCodeByName(cityGeoCodeName);
079            GeoCode countyGeoCode = null;
080            
081            if(countyGeoCodeName != null) {
082                countyGeoCode = geoControl.getGeoCodeByName(countyGeoCodeName);
083                
084                if(!countyGeoCode.getLastDetail().getGeoCodeType().equals(geoCodeType)) {
085                    countyGeoCode = null;
086                }
087                
088                if(countyGeoCode == null) {
089                    addExecutionError(ExecutionErrors.UnknownCountyGeoCodeName.name(), countyGeoCodeName);
090                }
091            } else if(countyName != 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 stateGeoCodeAliasType = geoControl.getGeoCodeAliasTypeByName(stateGeoCode.getLastDetail().getGeoCodeType(), GeoCodeAliasTypes.POSTAL_2_LETTER.name());
105                var stateGeoCodeAlias = geoControl.getGeoCodeAlias(stateGeoCode, stateGeoCodeAliasType);
106                var statePostal2Letter = stateGeoCodeAlias.getAlias();
107
108                var countryGeoCodeType = geoControl.getGeoCodeTypeByName(GeoCodeTypes.COUNTRY.name());
109                GeoCode countryGeoCode = null;
110                Collection stateRelationships = geoControl.getGeoCodeRelationshipsByFromGeoCode(stateGeoCode);
111                for(var iter = stateRelationships.iterator(); iter.hasNext();) {
112                    var geoCodeRelationship = (GeoCodeRelationship)iter.next();
113                    var toGeoCode = geoCodeRelationship.getToGeoCode();
114                    if(toGeoCode.getLastDetail().getGeoCodeType().equals(countryGeoCodeType)) {
115                        countryGeoCode = toGeoCode;
116                        break;
117                    }
118                }
119
120                var countryGeoCodeAliasType = geoControl.getGeoCodeAliasTypeByName(countryGeoCode.getLastDetail().getGeoCodeType(), GeoCodeAliasTypes.ISO_2_LETTER.name());
121                var countryGeoCodeAlias = geoControl.getGeoCodeAlias(countryGeoCode, countryGeoCodeAliasType);
122                var countryIso2Letter = countryGeoCodeAlias.getAlias();
123
124                var geoCodeScopeName = countryIso2Letter + "_" + statePostal2Letter + "_COUNTIES";
125                var geoCodeScope = geoControl.getGeoCodeScopeByName(geoCodeScopeName);
126
127                var geoCodeAliasType = geoControl.getGeoCodeAliasTypeByName(geoCodeType, GeoCodeAliasTypes.COUNTY_NAME.name());
128                var geoCodeAlias = geoControl.getGeoCodeAliasByAliasWithinScope(geoCodeScope, geoCodeAliasType, countyName);
129                countyGeoCode = geoCodeAlias.getGeoCode();
130                
131                if(countyGeoCode == null) {
132                    addExecutionError(ExecutionErrors.UnknownCountyName.name(), countyName);
133                }
134            }
135            
136            if(countyGeoCode != null) {
137                var geoCodeRelationship = geoControl.getGeoCodeRelationship(cityGeoCode, countyGeoCode);
138                
139                if(geoCodeRelationship == null) {
140                    geoControl.createGeoCodeRelationship(cityGeoCode, countyGeoCode, getPartyPK());
141                } else {
142                    addExecutionError(ExecutionErrors.DuplicateGeoCodeRelationship.name());
143                }
144            }
145        } else {
146            addExecutionError(ExecutionErrors.InvalidParameterCount.name());
147        }
148        
149        return null;
150    }
151    
152}