SqlTransformationMappingRootAspect.java

/*
 * JBoss, Home of Professional Open Source.
 *
 * See the LEGAL.txt file distributed with this work for information regarding copyright ownership and licensing.
 *
 * See the AUTHORS.txt file distributed with this work for a full listing of individual contributors.
 */
package org.teiid.designer.transformation.aspects.validation;

import org.eclipse.emf.ecore.EObject;
import org.teiid.core.designer.util.CoreArgCheck;
import org.teiid.designer.core.metamodel.aspect.AspectManager;
import org.teiid.designer.core.metamodel.aspect.MetamodelEntity;
import org.teiid.designer.core.metamodel.aspect.ValidationAspect;
import org.teiid.designer.core.metamodel.aspect.sql.SqlTransformationAspect;
import org.teiid.designer.core.validation.ValidationContext;
import org.teiid.designer.core.validation.ValidationRuleSet;
import org.teiid.designer.metamodels.transformation.SqlTransformationMappingRoot;


/**
 * SqlTransformationMappingRootAspect
 *
 * @since 8.0
 */
public class SqlTransformationMappingRootAspect extends TransformationAspect {

    public SqlTransformationMappingRootAspect(MetamodelEntity entity) {
        super(entity);
    }

	/**
	 * Get validation rules for SqlTransformationMappingRoot
	 */
	@Override
    public ValidationRuleSet getValidationRules() {
		addRule(MAPPINGROOT_RULE);
		return super.getValidationRules();
	}

	/* (non-Javadoc)
	 * @See org.teiid.designer.core.metamodel.aspect.ValidationAspect#updateContext(org.teiid.designer.core.validation.ValidationContext)
	 */
	@Override
    public void updateContext(final EObject eObject, final ValidationContext context) {
		SqlTransformationAspect transformationAspect = (SqlTransformationAspect) AspectManager.getSqlAspect(eObject);
		EObject transformedObject = (EObject) transformationAspect.getTransformedObject(eObject);
		// update the map		
		context.addTargetTransform(transformedObject, eObject);
	}

    /** 
     * @see org.teiid.designer.core.metamodel.aspect.ValidationAspect#shouldValidate(org.eclipse.emf.ecore.EObject)
     * @since 4.2
     */
    @Override
    public boolean shouldValidate(final EObject eObject, final ValidationContext context) {
        CoreArgCheck.isInstanceOf(SqlTransformationMappingRoot.class, eObject);
        
        if(!context.shouldIgnore(eObject)) {
            SqlTransformationMappingRoot mappingRoot = (SqlTransformationMappingRoot) eObject;
            EObject targetObj = mappingRoot.getTarget();
            if(targetObj != null) {
	            ValidationAspect validAspect = AspectManager.getValidationAspect(targetObj);
	            if(validAspect != null) {
	                boolean shouldValidate = validAspect.shouldValidate(targetObj, context);
	                if(!shouldValidate) {
	    	            context.addObjectToIgnore(eObject, true);
	                }
	                return shouldValidate;
	            }
	        }
	        return true;
        }
        return false;        
    }
}