1 files changed, 110 insertions, 0 deletions
diff --git a/plugins/infra/textedit/org.eclipse.papyrus.infra.textedit.ui/src/org/eclipse/papyrus/infra/textedit/ui/internal/menu/TextDocumentDynamicContribution.java b/plugins/infra/textedit/org.eclipse.papyrus.infra.textedit.ui/src/org/eclipse/papyrus/infra/textedit/ui/internal/menu/TextDocumentDynamicContribution.java
new file mode 100755
index 00000000000..c2cf9ec1128
--- /dev/null
+++ b/plugins/infra/textedit/org.eclipse.papyrus.infra.textedit.ui/src/org/eclipse/papyrus/infra/textedit/ui/internal/menu/TextDocumentDynamicContribution.java
@@ -0,0 +1,110 @@
+/*****************************************************************************
+ * Copyright (c) 2021 CEA LIST 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:
+ *  Vincent Lorenzo (CEA LIST) <vincent.lorenzo@cea.fr> - Initial API and implementation
+ *
+ *****************************************************************************/
+
+package org.eclipse.papyrus.infra.textedit.ui.internal.menu;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.jface.action.Action;
+import org.eclipse.jface.action.ActionContributionItem;
+import org.eclipse.jface.action.IContributionItem;
+import org.eclipse.papyrus.infra.textedit.representation.TextDocumentRepresentation;
+import org.eclipse.papyrus.infra.viewpoints.policy.DynamicContribution;
+import org.eclipse.papyrus.infra.viewpoints.policy.PolicyChecker;
+import org.eclipse.papyrus.infra.viewpoints.policy.ViewPrototype;
+
+/**
+ * Represent the dynamic contribution of the Papyrus TextDocument menu.
+ */
+public class TextDocumentDynamicContribution extends DynamicContribution {
+
+	/**
+	 * Constructor.
+	 */
+	public TextDocumentDynamicContribution() {
+	}
+
+	/**
+	 * Constructor.
+	 *
+	 * @param id
+	 */
+	public TextDocumentDynamicContribution(String id) {
+		super(id);
+	}
+
+	/**
+	 * @see org.eclipse.ui.actions.CompoundContributionItem#getContributionItems()
+	 *
+	 * @return
+	 */
+	@Override
+	protected IContributionItem[] getContributionItems() {
+		final EObject selection = getSelection();
+		if (selection == null) {
+			return new IContributionItem[0];
+		}
+
+		// build a list of all the available prototypes
+		List<ViewPrototype> data = new ArrayList<>();
+		for (final ViewPrototype proto : PolicyChecker.getFor(selection).getPrototypesFor(selection)) {
+			if (!(proto.getRepresentationKind() instanceof TextDocumentRepresentation)) {
+				continue;
+			}
+			data.add(proto);
+		}
+
+		// sort them
+		Collections.sort(data, new ViewPrototype.Comp());
+
+		// build the full labels
+		List<String> labels = new ArrayList<>(data.size());
+		String last = null;
+		boolean first = true;
+		for (ViewPrototype item : data) {
+			String label = item.getLabel();
+			if (last != null && last.equals(label)) {
+				// name collision
+				if (first) {
+					labels.set(labels.size() - 1, data.get(labels.size() - 1).getFullLabel());
+					first = false;
+				}
+				labels.add(item.getFullLabel());
+			} else {
+				labels.add(label);
+				last = label;
+				first = true;
+			}
+		}
+
+		// build the menu
+		List<IContributionItem> items = new ArrayList<>(data.size());
+		for (int i = 0; i != data.size(); i++) {
+			final ViewPrototype proto = data.get(i);
+			String label = labels.get(i);
+			items.add(new ActionContributionItem(new Action(label, proto.getIconDescriptor()) {
+				@Override
+				public void run() {
+					proto.instantiateOn(selection);
+				}
+			}));
+		}
+		return items.toArray(new IContributionItem[items.size()]);
+	}
+
+}