aas/plugins/modeling/org.eclipse.papyrus.aas.validation/src/org/eclipse/papyrus/aas/validation/constraints/AssetContainmentConstraint.java

/*****************************************************************************
 * Copyright (c) 2022 CEA LIST.
 *
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *  Saadia Dhouib (CEA LIST) saadia.dhouib@cea.fr - Initial API and implementation
 *
 *****************************************************************************/
package org.eclipse.papyrus.aas.validation.constraints;

import org.eclipse.core.runtime.IStatus;
import org.eclipse.emf.validation.AbstractModelConstraint;
import org.eclipse.emf.validation.IValidationContext;
import org.eclipse.papyrus.aas.Asset;
import org.eclipse.papyrus.aas.AssetAdministrationShell;
import org.eclipse.uml2.uml.Classifier;
import org.eclipse.uml2.uml.NamedElement;
import org.eclipse.uml2.uml.util.UMLUtil;

public class AssetContainmentConstraint extends AbstractModelConstraint{

	@Override
	public IStatus validate(IValidationContext ctx) {
		// TODO Auto-generated method stub
		IStatus status = ctx.createSuccessStatus();
		if(ctx.getTarget() instanceof NamedElement) {
			NamedElement elt = (NamedElement)ctx.getTarget();
			Asset asset = UMLUtil.getStereotypeApplication(elt, Asset.class);
			if (asset != null && elt instanceof org.eclipse.uml2.uml.Class) {
				if(! ((org.eclipse.uml2.uml.Class)elt).getAllAttributes().isEmpty()) {
					status = ctx.createFailureStatus("an asset must not contain attributes");
				}
				if(! ((org.eclipse.uml2.uml.Class)elt).getAllOperations().isEmpty() ) {
					status = ctx.createFailureStatus("an asset must not contain operations");
				}
				if(! ((org.eclipse.uml2.uml.Class)elt).getNestedClassifiers().isEmpty() ) {
					for (Classifier cl: ((org.eclipse.uml2.uml.Class)elt).getNestedClassifiers()) {
						if(UMLUtil.getStereotypeApplication(cl, AssetAdministrationShell.class)==null) {
							status = ctx.createFailureStatus("an asset must not contain sub components");
						}
						
					}
					
				}
				
			}
		}
		return status;
	}

}