001/**
002The contents of this file are subject to the Mozilla Public License Version 1.1 
003(the "License"); you may not use this file except in compliance with the License. 
004You may obtain a copy of the License at http://www.mozilla.org/MPL/ 
005Software distributed under the License is distributed on an "AS IS" basis, 
006WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the 
007specific language governing rights and limitations under the License. 
008
009The Original Code is "OnlyKnownSegmentsRule.java".  Description: 
010"Validation rule for detecting unknown Segments in a message" 
011
012The Initial Developer of the Original Code is University Health Network. Copyright (C) 
0132012.  All Rights Reserved. 
014
015Contributor(s): ______________________________________. 
016
017Alternatively, the contents of this file may be used under the terms of the 
018GNU General Public License (the "GPL"), in which case the provisions of the GPL are 
019applicable instead of those above.  If you wish to allow use of your version of this 
020file only under the terms of the GPL and not to allow others to use your version 
021of this file under the MPL, indicate your decision by deleting  the provisions above 
022and replace  them with the notice and other provisions required by the GPL License.  
023If you do not delete the provisions above, a recipient may use your version of 
024this file under either the MPL or the GPL. 
025 */
026package ca.uhn.hl7v2.validation.builder.support;
027
028import java.util.ArrayList;
029import java.util.List;
030import java.util.Set;
031
032import ca.uhn.hl7v2.HL7Exception;
033import ca.uhn.hl7v2.model.Message;
034import ca.uhn.hl7v2.model.Structure;
035import ca.uhn.hl7v2.model.SuperStructure;
036import ca.uhn.hl7v2.validation.MessageRule;
037import ca.uhn.hl7v2.validation.ValidationException;
038import ca.uhn.hl7v2.validation.impl.AbstractMessageRule;
039
040/**
041 * Validation rule for detecting populated elements in a {@link SuperStructure}
042 * which do not belong to the given message type
043 * 
044 * @author James Agnew
045 */
046@SuppressWarnings("serial")
047public class OnlyAllowableSegmentsInSuperstructureRule extends AbstractMessageRule {
048
049        public static final MessageRule ONLY_ALLOWABLE_SEGMENTS = new OnlyAllowableSegmentsInSuperstructureRule();
050        
051        public ValidationException[] apply(Message msg) {
052                List<ValidationException> exceptions = new ArrayList<ValidationException>();
053                
054                if (msg instanceof SuperStructure) {
055                        checkStructure((SuperStructure)msg, exceptions);
056                }
057                
058                return exceptions.toArray(new ValidationException[exceptions.size()]);
059        }
060
061        private void checkStructure(SuperStructure theMsg, List<ValidationException> theExceptions) {
062                String messageStructure = theMsg.getMessage().getName();
063
064                FORNAME:
065                for (String name : theMsg.getNames()) {
066                        
067                        try {
068                                for (Structure rep : theMsg.getAll(name)) {
069                                        
070                                        if (!rep.isEmpty()) {
071                                                Set<String> structuresWhichChildAppliesTo = theMsg.getStructuresWhichChildAppliesTo(name);
072                                                if (!structuresWhichChildAppliesTo.contains(messageStructure)) {
073                                                        String msgSimpleName = theMsg.getMessage().getClass().getSimpleName();
074                                                        theExceptions.add(new ValidationException("Message (superstructure " + msgSimpleName + ") of type " + messageStructure + " must not have content in " + name));
075                                                }
076                                                continue FORNAME;
077                                        }
078                                        
079                                }
080                        } catch (HL7Exception e) {
081                                // should not happen
082                                throw new Error("Can't get rep of structure " + name + ". This is probably a HAPI bug");
083                        }
084                        
085                }
086        }
087
088}