/** * Copyright 2010 JBoss Inc * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.drools.verifier.visitor; import java.util.List; import org.drools.lang.descr.TypeDeclarationDescr; import org.drools.verifier.components.Field; import org.drools.verifier.components.Import; import org.drools.verifier.components.ObjectType; import org.drools.verifier.data.VerifierData; public class TypeDeclarationDescrVisitor { private final VerifierData data; public TypeDeclarationDescrVisitor(VerifierData data) { this.data = data; } public void visit(List<TypeDeclarationDescr> typeDeclarationDescrs) { for ( TypeDeclarationDescr typeDeclaration : typeDeclarationDescrs ) { Import objectImport = data.getImportByName( typeDeclaration.getTypeName() ); String objectTypeName; if ( objectImport == null ) { objectTypeName = typeDeclaration.getTypeName(); } else { objectTypeName = objectImport.getName(); } ObjectType objectType = this.data.getObjectTypeByFullName( objectTypeName ); if ( objectType == null ) { objectType = new ObjectType(); objectType.setName( typeDeclaration.getTypeName() ); objectType.setFullName( typeDeclaration.getTypeName() ); data.add( objectType ); } for ( String fieldName : typeDeclaration.getFields().keySet() ) { Field field = data.getFieldByObjectTypeAndFieldName( objectType.getFullName(), fieldName ); if ( field == null ) { field = ObjectTypeFactory.createField( fieldName, objectType ); field.setFieldType( typeDeclaration.getFields().get( fieldName ).getPattern().getObjectType() ); data.add( field ); } } for ( String metadata : typeDeclaration.getMetaAttributes().keySet() ) { objectType.getMetadata().put( metadata, typeDeclaration.getMetaAttribute( metadata ) ); } } } }