1 files changed, 93 insertions, 0 deletions
diff --git a/plugins/toolsmiths/builder/org.eclipse.papyrus.toolsmiths.plugin.builder/src/org/eclipse/papyrus/toolsmiths/plugin/builder/quickfix/NoEcoreGenPackageMarkerResolution.java b/plugins/toolsmiths/builder/org.eclipse.papyrus.toolsmiths.plugin.builder/src/org/eclipse/papyrus/toolsmiths/plugin/builder/quickfix/NoEcoreGenPackageMarkerResolution.java
new file mode 100644
index 00000000000..655ee8e7669
--- /dev/null
+++ b/plugins/toolsmiths/builder/org.eclipse.papyrus.toolsmiths.plugin.builder/src/org/eclipse/papyrus/toolsmiths/plugin/builder/quickfix/NoEcoreGenPackageMarkerResolution.java
@@ -0,0 +1,93 @@
+/*****************************************************************************
+ * Copyright (c) 2020 CEA LIST, EclipseSource and others.
+ *
+ * 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
+ * http://www.eclipse.org/legal/epl-2.0/
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ *
+ * Contributors:
+ *   Alexandra Buzila (EclipseSource) - Initial API and implementation
+ *
+ *****************************************************************************/
+
+package org.eclipse.papyrus.toolsmiths.plugin.builder.quickfix;
+
+import org.eclipse.core.resources.IFile;
+import org.eclipse.core.resources.IMarker;
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.papyrus.infra.emf.utils.ResourceUtils;
+import org.eclipse.papyrus.toolsmiths.plugin.builder.Activator;
+import org.eclipse.papyrus.toolsmiths.plugin.builder.Messages;
+import org.eclipse.papyrus.toolsmiths.validation.profile.constants.ProfilePluginValidationConstants;
+import org.eclipse.pde.core.IBaseModel;
+import org.eclipse.pde.core.plugin.IPluginElement;
+import org.eclipse.pde.core.plugin.IPluginExtension;
+import org.eclipse.pde.core.plugin.IPluginModelBase;
+import org.eclipse.pde.internal.ui.util.ModelModification;
+import org.eclipse.pde.internal.ui.util.PDEModelUtility;
+
+/**
+ * Resolution that adds an entry for the 'org.eclipse.emf.ecore.generated_package' extension and configures it.
+ */
+@SuppressWarnings("restriction")
+public class NoEcoreGenPackageMarkerResolution extends AbstractPapyrusMarkerResolution {
+
+	private IMarker marker;
+
+	@Override
+	public String getDescription() {
+		return Messages.NoEcoreGenPackageMarkerResolution_description;
+	}
+
+	@Override
+	public String getLabel() {
+		return Messages.NoEcoreGenPackageMarkerResolution_label;
+	}
+
+	@Override
+	public void run(IMarker marker) {
+		this.marker = marker;
+		if (!(marker.getResource() instanceof IFile)) {
+			return;
+		}
+		ModelModification modification = new ModelModification((IFile) marker.getResource()) {
+			@Override
+			protected void modifyModel(IBaseModel model, IProgressMonitor monitor) throws CoreException {
+				if (model instanceof IPluginModelBase) {
+					addExtension((IPluginModelBase) model);
+				}
+			}
+		};
+		PDEModelUtility.modifyModel(modification, null);
+	}
+
+	private void addExtension(IPluginModelBase model) {
+		try {
+			IPluginExtension extension = model.getFactory().createExtension();
+			extension.setPoint(ProfilePluginValidationConstants.ECORE_GENERATED_PACKAGE_EXTENSION_POINT);
+			model.getPluginBase().add(extension);
+			IPluginElement packageElement = model.getFactory().createElement(extension);
+			packageElement.setName("package"); //$NON-NLS-1$
+			extension.add(packageElement);
+			IFile genModel = MarkerResolutionUtils.getGenModelFile(marker);
+			if (genModel != null) {
+				packageElement.setAttribute("genModel", ResourceUtils.mapAndEncodePath(genModel)); //$NON-NLS-1$
+			}
+			String uri = MarkerResolutionUtils.getStereotypeUri(marker);
+			if (uri != null) {
+				packageElement.setAttribute("uri", uri); //$NON-NLS-1$
+			}
+
+			// XXX needed?
+			// packageElement.setAttribute("class", "");
+
+		} catch (CoreException e) {
+			Activator.log.error(e);
+		}
+	}
+
+}