bundles/org.simantics.issues.ui/src/org/simantics/issues/ui/handler/MenuActions.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2011 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.issues.ui.handler;
  13
  14 import java.util.ArrayList;
  15 import java.util.Arrays;
  16 import java.util.List;
  17
  18 import org.eclipse.jface.action.ActionContributionItem;
  19 import org.eclipse.jface.action.ContributionItem;
  20 import org.eclipse.jface.action.IAction;
  21 import org.eclipse.jface.action.IContributionItem;
  22 import org.eclipse.jface.resource.ImageDescriptor;
  23 import org.eclipse.jface.viewers.ISelection;
  24 import org.eclipse.swt.SWT;
  25 import org.eclipse.swt.events.SelectionAdapter;
  26 import org.eclipse.swt.events.SelectionEvent;
  27 import org.eclipse.swt.widgets.Menu;
  28 import org.eclipse.swt.widgets.MenuItem;
  29 import org.simantics.Simantics;
  30 import org.simantics.db.ReadGraph;
  31 import org.simantics.db.Resource;
  32 import org.simantics.db.WriteGraph;
  33 import org.simantics.db.common.request.WriteRequest;
  34 import org.simantics.db.exception.DatabaseException;
  35 import org.simantics.db.layer0.SelectionHints;
  36 import org.simantics.issues.Severity;
  37 import org.simantics.issues.common.IssueConstants;
  38 import org.simantics.issues.common.IssueUtils;
  39 import org.simantics.issues.ontology.IssueResource;
  40 import org.simantics.issues.ui.internal.Activator;
  41 import org.simantics.ui.contribution.DynamicMenuContribution;
  42 import org.simantics.utils.datastructures.Callback;
  43 import org.simantics.utils.ui.ErrorLogger;
  44 import org.simantics.utils.ui.ISelectionUtils;
  45
  46 /**
  47  * @author Tuukka Lehtonen
  48  */
  49 public class MenuActions extends DynamicMenuContribution {
  50
  51     @Override
  52     protected Object[] getSelectedObjects() {
  53         ISelection sel = getSelection();
  54         List<Resource> resources = ISelectionUtils.getPossibleKeys(sel, SelectionHints.KEY_MAIN, Resource.class);
  55         return resources.toArray();
  56     }
  57
  58     List<Resource> toResources(Object[] array) {
  59         Resource[] a = new Resource[array.length];
  60         for (int i = 0; i < array.length; ++i)
  61             a[i] = (Resource) array[i];
  62         return Arrays.asList(a);
  63     }
  64
  65     @Override
  66     protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] selection) throws DatabaseException {
  67         if (selection.length == 0)
  68             return new IContributionItem[0];
  69
  70         List<Resource> input = toResources(selection);
  71
  72         @SuppressWarnings("unused")
  73         boolean issues = false;
  74         @SuppressWarnings("unused")
  75         boolean userIssues = false;
  76
  77         IssueResource ISSUE = IssueResource.getInstance(graph);
  78         int issueCount = 0;
  79         int hiddenCount = 0;
  80         int resolvedCount = 0;
  81         int userIssueCount = 0;
  82         for (Resource r : input) {
  83             boolean isIssue = graph.isInstanceOf(r, ISSUE.Issue);
  84             issues |= isIssue;
  85             if (isIssue) {
  86                 ++issueCount;
  87                 if (graph.hasStatement(r, ISSUE.Hidden))
  88                     ++hiddenCount;
  89                 if (graph.hasStatement(r, ISSUE.Resolved))
  90                     ++resolvedCount;
  91             }
  92
  93             boolean userIssue = graph.hasStatement(r, ISSUE.UserIssue);
  94             userIssues |= userIssue;
  95             if (userIssue) {
  96                 ++userIssueCount;
  97             }
  98         }
  99         boolean allHidden = hiddenCount == selection.length;
 100         boolean allResolved = resolvedCount == selection.length;
 101         boolean allUserIssues = userIssueCount == selection.length;
 102         boolean allIssueSelection = issueCount == selection.length;
 103
 104         List<IContributionItem> actions = new ArrayList<IContributionItem>(2);
 105         if (!allHidden && allIssueSelection)
 106             actions.add(new ActionContributionItem(hideAction(input)));
 107         if (hiddenCount > 0 || allHidden && allIssueSelection)
 108             actions.add(new ActionContributionItem(unhideAction(input)));
 109         if (!allResolved && allIssueSelection)
 110             actions.add(new ActionContributionItem(resolveAction(input)));
 111         if (resolvedCount > 0 || allResolved && allIssueSelection)
 112             actions.add(new ActionContributionItem(unresolveAction(input)));
 113         if (allUserIssues)
 114             actions.add(setSeverityAction(input));
 115
 116         return actions.toArray(new IContributionItem[actions.size()]);
 117     }
 118
 119     private IAction unhideAction(List<Resource> input) {
 120         return tagAction("Unhide", Activator.UNHIDE_ICON, IssueResource.URIs.Hidden, false, input);
 121     }
 122
 123     private IAction hideAction(List<Resource> input) {
 124         return tagAction("Hide", Activator.HIDE_ICON, IssueResource.URIs.Hidden, true, input);
 125     }
 126
 127     private IAction resolveAction(List<Resource> input) {
 128         return tagAction("Mark Resolved", Activator.RESOLVE_ICON, IssueResource.URIs.Resolved, true, input);
 129     }
 130
 131     private IAction unresolveAction(List<Resource> input) {
 132         return tagAction("Mark Unresolved", Activator.UNRESOLVE_ICON, IssueResource.URIs.Resolved, false, input);
 133     }
 134
 135     private IAction tagAction(String label, ImageDescriptor image, String tagURI, boolean tag, List<Resource> input) {
 136         return new TagAction(label, image, IssueConstants.ISSUE_VG, tagURI, tag, input);
 137         //return new TagAction(label, image, null, tagURI, tag, input);
 138     }
 139
 140     private IContributionItem setSeverityAction(final List<Resource> userIssues) {
 141         return new ContributionItem() {
 142             @Override
 143             public void fill(Menu menu, int index) {
 144                 MenuItem setSeverityItem = new MenuItem(menu, SWT.CASCADE);
 145                 setSeverityItem.setText("Set Severity");
 146                 Menu setSeverity = new Menu(menu);
 147                 setSeverityItem.setMenu(setSeverity);
 148                 for (final Severity sev : Severity.values()) {
 149                     if (sev.equals(Severity.UNKNOWN))
 150                         continue;
 151
 152                     MenuItem item = new MenuItem(setSeverity, SWT.PUSH);
 153                     item.setText(sev.toString().toLowerCase());
 154                     item.setImage(Activator.getDefault().getImageRegistry().get(sev.toString() + "-full"));
 155                     item.addSelectionListener(new SelectionAdapter() {
 156                         @Override
 157                         public void widgetSelected(SelectionEvent e) {
 158                             setSeverity(userIssues, sev);
 159                         }
 160                     });
 161                 }
 162             }
 163         };
 164     }
 165
 166     private void setSeverity(final List<Resource> userIssues, final Severity severity) {
 167         Simantics.getSession().asyncRequest(new WriteRequest() {
 168             @Override
 169             public void perform(WriteGraph graph) throws DatabaseException {
 170                 IssueResource ISSUE = IssueResource.getInstance(graph);
 171                 Resource sev = IssueUtils.toSeverityResource(ISSUE, severity);
 172                 if (sev != null) {
 173                     for (Resource issue : userIssues) {
 174                         if (!graph.hasStatement(issue, ISSUE.Issue_HasSeverity, sev)) {
 175                             graph.deny(issue, ISSUE.Issue_HasSeverity);
 176                             graph.claim(issue, ISSUE.Issue_HasSeverity, null, sev);
 177                         }
 178                     }
 179                 } else {
 180                     for (Resource issue : userIssues)
 181                         graph.deny(issue, ISSUE.Issue_HasSeverity);
 182                 }
 183             }
 184         }, new Callback<DatabaseException>() {
 185             @Override
 186             public void run(DatabaseException e) {
 187                 if (e != null)
 188                     ErrorLogger.defaultLogError(e);
 189             }
 190         });
 191     }
 192
 193 }