| /******************************************************************************* |
| * Copyright (c) 2016 Chalmers | University of Gothenburg, rt-labs and others. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * Chalmers | University of Gothenburg and rt-labs - initial API and implementation and/or initial documentation |
| *******************************************************************************/ |
| |
| package org.eclipse.capra.ui.notification; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.eclipse.capra.core.adapters.ArtifactMetaModelAdapter; |
| import org.eclipse.capra.core.adapters.Connection; |
| import org.eclipse.capra.core.adapters.TraceMetaModelAdapter; |
| import org.eclipse.capra.core.adapters.TracePersistenceAdapter; |
| import org.eclipse.capra.core.helpers.ExtensionPointHelper; |
| import org.eclipse.capra.core.helpers.TraceHelper; |
| import org.eclipse.core.resources.IMarker; |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.emf.common.util.URI; |
| import org.eclipse.emf.ecore.EObject; |
| import org.eclipse.emf.ecore.resource.Resource; |
| import org.eclipse.emf.ecore.resource.ResourceSet; |
| import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl; |
| import org.eclipse.emf.ecore.util.EcoreUtil; |
| import org.eclipse.jface.dialogs.MessageDialog; |
| import org.eclipse.ui.IMarkerResolution; |
| |
| /** |
| * A quick fix to delete a trace link if one of the linked objects is no longer |
| * available. |
| * |
| * @author Michael Warne |
| */ |
| public class DeleteQuickFix implements IMarkerResolution { |
| ArtifactMetaModelAdapter artifactAdapter = ExtensionPointHelper.getArtifactWrapperMetaModelAdapter().get(); |
| |
| private String label; |
| |
| DeleteQuickFix(String label) { |
| this.label = label; |
| } |
| |
| @Override |
| public String getLabel() { |
| return label; |
| } |
| |
| @Override |
| public void run(IMarker marker) { |
| |
| ResourceSet resourceSet = new ResourceSetImpl(); |
| List<Connection> toDelete = new ArrayList<>(); |
| TracePersistenceAdapter tracePersistenceAdapter = ExtensionPointHelper.getTracePersistenceAdapter().get(); |
| EObject traceModel = tracePersistenceAdapter.getTraceModel(resourceSet); |
| TraceHelper traceHelper = new TraceHelper(traceModel); |
| TraceMetaModelAdapter traceMetamodelAdapter = ExtensionPointHelper.getTraceMetamodelAdapter().get(); |
| EObject artifactModel = tracePersistenceAdapter.getArtifactWrappers(new ResourceSetImpl()); |
| |
| // get all artifacts |
| List<EObject> artifacts = artifactAdapter.getAllArtifacts(artifactModel); |
| |
| String artifactContainerFileName = artifactModel.eResource().getURI().lastSegment(); |
| String markerContainerFileName = new File(marker.getResource().toString()).getName(); |
| String markerUri = marker.getAttribute(CapraNotificationHelper.OLD_URI, null); |
| |
| if (markerContainerFileName.equals(artifactContainerFileName)) { |
| // The element that the marker points to is a Capra artifact. |
| int index = -1; |
| for (EObject aw : artifacts) { |
| if (artifactAdapter.getArtifactUri(aw).equals(markerUri)) { |
| toDelete = traceMetamodelAdapter.getConnectedElements(aw, traceModel); |
| } |
| index++; |
| } |
| // delete trace links associated with deleted artifact |
| traceMetamodelAdapter.deleteTrace(toDelete, traceModel); |
| EObject toRemove = artifacts.get(index); |
| // Delete selected artifacts. |
| EcoreUtil.delete(toRemove); |
| URI artifactModelURI = EcoreUtil.getURI(artifactModel); |
| Resource resourceForArtifacts = resourceSet.createResource(artifactModelURI); |
| resourceForArtifacts.getContents().add(artifactModel); |
| |
| try { |
| resourceForArtifacts.save(null); |
| } catch (IOException e) { |
| e.printStackTrace(); |
| } |
| // delete Marker |
| try { |
| marker.delete(); |
| MessageDialog.openWarning(null, "TraceLinkDeleted", "Links successfully deleted"); |
| } catch (CoreException e) { |
| e.printStackTrace(); |
| } |
| } else { |
| // The element that the marker points to is an EObject and is not |
| // contained in the Capra artifact model. |
| URI deletedEObjectUri = URI.createURI(markerUri); |
| for (Connection c : traceMetamodelAdapter.getAllTraceLinks(traceModel)) { |
| for (EObject item : traceHelper.getTracedElements(c)) { |
| URI itemUri = CapraNotificationHelper.getFileUri(item); |
| if (deletedEObjectUri.equals(itemUri)) { |
| toDelete.add(c); |
| break; |
| } |
| } |
| } |
| |
| // Delete selected traces. |
| traceMetamodelAdapter.deleteTrace(toDelete, traceModel); |
| try { |
| marker.delete(); |
| } catch (CoreException e) { |
| e.printStackTrace(); |
| } |
| } |
| } |
| } |