refactor: extracted nested project tree into reusable composite

Signed-off-by: Igor Fedorenko <igor@ifedorenko.com>

3 files changed, 371 insertions, 331 deletions

diff --git a/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/MavenImages.java b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/MavenImages.java
index 43a2a1cc..22a42789 100644
--- a/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/MavenImages.java
+++ b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/MavenImages.java
@@ -94,10 +94,6 @@ public class MavenImages {
 
   public static final String MVN_PROJECT = "project-mlabel.gif"; //$NON-NLS-1$
 
-  static {
-
-  }
-
   private static ImageDescriptor createDescriptor(String key) {
     try {
       ImageRegistry imageRegistry = getImageRegistry();
diff --git a/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/components/NestedProjectsComposite.java b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/components/NestedProjectsComposite.java
new file mode 100644
index 00000000..e140437f
--- /dev/null
+++ b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/components/NestedProjectsComposite.java
@@ -0,0 +1,357 @@
+/*******************************************************************************
+ * Copyright (c) 2010 Sonatype, Inc.
+ * 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:
+ *      Sonatype, Inc. - initial API and implementation
+ *******************************************************************************/
+
+package org.eclipse.m2e.core.ui.internal.components;
+
+import java.io.File;
+import java.net.URI;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import org.eclipse.core.filesystem.EFS;
+import org.eclipse.core.filesystem.IFileStore;
+import org.eclipse.core.resources.IContainer;
+import org.eclipse.core.resources.IProject;
+import org.eclipse.core.resources.IWorkspaceRoot;
+import org.eclipse.core.resources.ResourcesPlugin;
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.jface.action.Action;
+import org.eclipse.jface.action.IMenuListener;
+import org.eclipse.jface.action.IMenuManager;
+import org.eclipse.jface.action.MenuManager;
+import org.eclipse.jface.viewers.CheckboxTreeViewer;
+import org.eclipse.jface.viewers.IDecoration;
+import org.eclipse.jface.viewers.ISelection;
+import org.eclipse.jface.viewers.IStructuredSelection;
+import org.eclipse.jface.viewers.ITreeContentProvider;
+import org.eclipse.jface.viewers.LabelProvider;
+import org.eclipse.jface.viewers.Viewer;
+import org.eclipse.swt.SWT;
+import org.eclipse.swt.events.SelectionEvent;
+import org.eclipse.swt.events.SelectionListener;
+import org.eclipse.swt.graphics.Image;
+import org.eclipse.swt.layout.GridData;
+import org.eclipse.swt.layout.GridLayout;
+import org.eclipse.swt.widgets.Button;
+import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Label;
+import org.eclipse.swt.widgets.Menu;
+import org.eclipse.swt.widgets.Tree;
+import org.eclipse.ui.PlatformUI;
+import org.eclipse.ui.ide.IDE;
+
+import org.eclipse.m2e.core.internal.IMavenConstants;
+import org.eclipse.m2e.core.ui.internal.MavenImages;
+import org.eclipse.m2e.core.ui.internal.Messages;
+
+
+public class NestedProjectsComposite extends Composite implements IMenuListener {
+
+  private static final Logger log = LoggerFactory.getLogger(NestedProjectsComposite.class);
+
+  private static final String SEPARATOR = System.getProperty("file.separator"); //$NON-NLS-1$
+
+  private final IProject[] initialSelection;
+
+  private CheckboxTreeViewer codebaseViewer;
+
+  private Map<String, IProject> projectPaths;
+
+  private Collection<IProject> projects;
+
+  public NestedProjectsComposite(Composite parent, int style, IProject[] initialSelection) {
+    super(parent, style);
+    this.initialSelection = initialSelection;
+
+    setLayout(new GridLayout(2, false));
+
+    Label lblAvailable = new Label(this, SWT.NONE);
+    lblAvailable.setText(Messages.UpdateDepenciesDialog_availableCodebasesLabel);
+    lblAvailable.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, true, false, 2, 1));
+
+    codebaseViewer = new CheckboxTreeViewer(this, SWT.BORDER);
+    codebaseViewer.setContentProvider(new ITreeContentProvider() {
+
+      public void dispose() {
+      }
+
+      public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
+      }
+
+      public Object[] getElements(Object element) {
+        if(element instanceof Collection) {
+          return ((Collection) element).toArray();
+        }
+        return null;
+      }
+
+      public Object[] getChildren(Object parentElement) {
+        if(parentElement instanceof IProject) {
+          String elePath = getElePath(parentElement);
+          String prevPath = null;
+          List<IProject> children = new ArrayList<IProject>();
+          for(String path : projectPaths.keySet()) {
+            if(path.length() != elePath.length() && path.startsWith(elePath)) {
+              if(prevPath == null || !path.startsWith(prevPath)) {
+                prevPath = path;
+                children.add(getProject(path));
+              }
+            }
+          }
+          return children.toArray();
+        } else if(parentElement instanceof Collection) {
+          return ((Collection) parentElement).toArray();
+        }
+        return null;
+      }
+
+      public Object getParent(Object element) {
+        String elePath = getElePath(element);
+        String prevPath = null;
+        for(String path : projectPaths.keySet()) {
+          if(elePath.length() != path.length() && elePath.startsWith(path)
+              && (prevPath == null || prevPath.length() < path.length())) {
+            prevPath = path;
+          }
+        }
+        return prevPath == null ? projects : getProject(prevPath);
+      }
+
+      public boolean hasChildren(Object element) {
+        if(element instanceof IProject) {
+          String elePath = getElePath(element);
+          for(String path : projectPaths.keySet()) {
+            if(elePath.length() != path.length() && path.startsWith(elePath)) {
+              return true;
+            }
+          }
+        } else if(element instanceof Collection) {
+          return !((Collection) element).isEmpty();
+        }
+        return false;
+      }
+    });
+    codebaseViewer.setLabelProvider(new LabelProvider() {
+      public Image getImage(Object element) {
+        return MavenImages.createOverlayImage(MavenImages.MVN_PROJECT, PlatformUI.getWorkbench().getSharedImages()
+            .getImage(IDE.SharedImages.IMG_OBJ_PROJECT), MavenImages.MAVEN_OVERLAY, IDecoration.TOP_LEFT);
+      }
+
+      public String getText(Object element) {
+        return element instanceof IProject ? ((IProject) element).getName() : ""; //$NON-NLS-1$
+      }
+    });
+    projects = getMavenCodebases();
+    codebaseViewer.setInput(projects);
+    codebaseViewer.expandAll();
+    for(IProject project : initialSelection) {
+      codebaseViewer.setSubtreeChecked(project, true);
+    }
+
+    // Reveal the first element
+    if(initialSelection.length > 0) {
+      codebaseViewer.reveal(initialSelection[0]);
+    }
+
+    Tree tree = codebaseViewer.getTree();
+    GridData gd = new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1);
+    gd.heightHint = 300;
+    gd.widthHint = 300;
+    tree.setLayoutData(gd);
+
+    GridLayout layout = new GridLayout(2, false);
+    layout.marginLeft = 10;
+
+    Composite selectionActionComposite = new Composite(this, SWT.NONE);
+    selectionActionComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false, 1, 1));
+    GridLayout gl_selectionActionComposite = new GridLayout(1, false);
+    gl_selectionActionComposite.marginWidth = 0;
+    gl_selectionActionComposite.marginHeight = 0;
+    selectionActionComposite.setLayout(gl_selectionActionComposite);
+
+    Button selectAllBtn = new Button(selectionActionComposite, SWT.NONE);
+    selectAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, false, false, 1, 1));
+    selectAllBtn.setText(Messages.UpdateDepenciesDialog_selectAll);
+    selectAllBtn.addSelectionListener(new SelectionListener() {
+      public void widgetSelected(SelectionEvent e) {
+        for(IProject project : projects) {
+          codebaseViewer.setSubtreeChecked(project, true);
+        }
+      }
+
+      public void widgetDefaultSelected(SelectionEvent e) {
+
+      }
+    });
+
+    Button deselectAllBtn = new Button(selectionActionComposite, SWT.NONE);
+    deselectAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false, 1, 1));
+    deselectAllBtn.setText(Messages.UpdateDepenciesDialog_deselectAll);
+    deselectAllBtn.addSelectionListener(new SelectionListener() {
+      public void widgetSelected(SelectionEvent e) {
+        for(IProject project : projects) {
+          codebaseViewer.setSubtreeChecked(project, false);
+        }
+      }
+
+      public void widgetDefaultSelected(SelectionEvent e) {
+
+      }
+    });
+
+    Button expandAllBtn = new Button(selectionActionComposite, SWT.NONE);
+    expandAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, false, false, 1, 1));
+    expandAllBtn.setText(Messages.UpdateDepenciesDialog_expandAll);
+    expandAllBtn.addSelectionListener(new SelectionListener() {
+      public void widgetSelected(SelectionEvent e) {
+        codebaseViewer.expandAll();
+      }
+
+      public void widgetDefaultSelected(SelectionEvent e) {
+      }
+    });
+
+    Button collapseAllBtn = new Button(selectionActionComposite, SWT.NONE);
+    collapseAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, false, false, 1, 1));
+    collapseAllBtn.setText(Messages.UpdateDepenciesDialog_collapseAll);
+    collapseAllBtn.addSelectionListener(new SelectionListener() {
+      public void widgetSelected(SelectionEvent e) {
+        codebaseViewer.collapseAll();
+      }
+
+      public void widgetDefaultSelected(SelectionEvent e) {
+      }
+    });
+
+    createMenu();
+  }
+
+  private String getElePath(Object element) {
+    if(element instanceof IProject) {
+      IProject project = (IProject) element;
+      URI locationURI = project.getLocationURI();
+
+      try {
+        IFileStore store = EFS.getStore(locationURI);
+        File file = store.toLocalFile(0, null);
+        return file.toString() + SEPARATOR;
+      } catch(CoreException ex) {
+        log.error(ex.getMessage(), ex);
+      }
+    }
+    return null;
+  }
+
+  private IProject getProject(String path) {
+    return projectPaths.get(path);
+  }
+
+  private Collection<IProject> getMavenCodebases() {
+    projectPaths = new TreeMap<String, IProject>();
+
+    for(IProject project : ResourcesPlugin.getWorkspace().getRoot().getProjects()) {
+      try {
+        if(project.isAccessible() && project.hasNature(IMavenConstants.NATURE_ID)) {
+          if(project.getLocationURI() != null) {
+            String path = getElePath(project);
+            if(path != null) {
+              projectPaths.put(path, project);
+            }
+          }
+        }
+      } catch(CoreException ex) {
+        log.error(ex.getMessage(), ex);
+      }
+    }
+
+    if(projectPaths.isEmpty()) {
+      return Collections.<IProject> emptyList();
+    }
+    projects = new ArrayList<IProject>();
+    String previous = projectPaths.keySet().iterator().next();
+    addProject(projects, previous);
+    for(String path : projectPaths.keySet()) {
+      if(!path.startsWith(previous)) {
+        previous = path;
+        IProject project = getProject(path);
+        if(project != null) {
+          projects.add(project);
+        }
+      }
+    }
+    return projects;
+  }
+
+  private static void addProject(Collection<IProject> projects, String location) {
+    IWorkspaceRoot root = ResourcesPlugin.getWorkspace().getRoot();
+    for(IContainer container : root.findContainersForLocationURI(new File(location).toURI())) {
+      if(container instanceof IProject) {
+        projects.add((IProject) container);
+        break;
+      }
+    }
+  }
+
+  private void createMenu() {
+    MenuManager menuMgr = new MenuManager();
+    Menu contextMenu = menuMgr.createContextMenu(codebaseViewer.getControl());
+    menuMgr.addMenuListener(this);
+    codebaseViewer.getControl().setMenu(contextMenu);
+    menuMgr.setRemoveAllWhenShown(true);
+  }
+
+  public void menuAboutToShow(IMenuManager manager) {
+    if(codebaseViewer.getSelection().isEmpty()) {
+      return;
+    }
+
+    if(codebaseViewer.getSelection() instanceof IStructuredSelection) {
+      manager.add(selectTree);
+      manager.add(deselectTree);
+    }
+  }
+
+  private final Action selectTree = new Action(Messages.UpdateDepenciesDialog_selectTree) {
+    public void run() {
+      codebaseViewer.setSubtreeChecked(getSelection(), true);
+    }
+  };
+
+  private final Action deselectTree = new Action(Messages.UpdateDepenciesDialog_deselectTree) {
+    public void run() {
+      codebaseViewer.setSubtreeChecked(getSelection(), false);
+    }
+  };
+
+  private IProject getSelection() {
+    ISelection selection = codebaseViewer.getSelection();
+    if(selection instanceof IStructuredSelection) {
+      return (IProject) ((IStructuredSelection) selection).getFirstElement();
+    }
+    return null;
+  }
+
+  public IProject[] getSelectedProjects() {
+    Object[] obj = codebaseViewer.getCheckedElements();
+    IProject[] projects = new IProject[obj.length];
+    for(int i = 0; i < obj.length; i++ ) {
+      projects[i] = (IProject) obj[i];
+    }
+    return projects;
+  }
+}
diff --git a/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/dialogs/UpdateMavenProjectsDialog.java b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/dialogs/UpdateMavenProjectsDialog.java
index a6bbf8d6..2fc63a9d 100644
--- a/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/dialogs/UpdateMavenProjectsDialog.java
+++ b/org.eclipse.m2e.core.ui/src/org/eclipse/m2e/core/ui/internal/dialogs/UpdateMavenProjectsDialog.java
@@ -11,76 +11,30 @@
 
 package org.eclipse.m2e.core.ui.internal.dialogs;
 
-import java.io.File;
-import java.net.URI;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.Collections;
-import java.util.List;
-import java.util.Map;
-import java.util.TreeMap;
-
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import org.eclipse.core.filesystem.EFS;
-import org.eclipse.core.filesystem.IFileStore;
-import org.eclipse.core.resources.IContainer;
 import org.eclipse.core.resources.IProject;
-import org.eclipse.core.resources.IWorkspaceRoot;
-import org.eclipse.core.resources.ResourcesPlugin;
-import org.eclipse.core.runtime.CoreException;
-import org.eclipse.jface.action.Action;
-import org.eclipse.jface.action.IMenuListener;
-import org.eclipse.jface.action.IMenuManager;
-import org.eclipse.jface.action.MenuManager;
 import org.eclipse.jface.dialogs.IDialogConstants;
 import org.eclipse.jface.dialogs.TitleAreaDialog;
-import org.eclipse.jface.viewers.CheckboxTreeViewer;
-import org.eclipse.jface.viewers.IDecoration;
-import org.eclipse.jface.viewers.ISelection;
-import org.eclipse.jface.viewers.IStructuredSelection;
-import org.eclipse.jface.viewers.ITreeContentProvider;
-import org.eclipse.jface.viewers.LabelProvider;
-import org.eclipse.jface.viewers.Viewer;
 import org.eclipse.swt.SWT;
-import org.eclipse.swt.events.SelectionEvent;
-import org.eclipse.swt.events.SelectionListener;
-import org.eclipse.swt.graphics.Image;
 import org.eclipse.swt.layout.GridData;
 import org.eclipse.swt.layout.GridLayout;
 import org.eclipse.swt.widgets.Button;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Control;
-import org.eclipse.swt.widgets.Label;
-import org.eclipse.swt.widgets.Menu;
 import org.eclipse.swt.widgets.Shell;
-import org.eclipse.swt.widgets.Tree;
-import org.eclipse.ui.PlatformUI;
-import org.eclipse.ui.ide.IDE;
 
 import org.eclipse.m2e.core.MavenPlugin;
-import org.eclipse.m2e.core.internal.IMavenConstants;
-import org.eclipse.m2e.core.ui.internal.MavenImages;
 import org.eclipse.m2e.core.ui.internal.Messages;
+import org.eclipse.m2e.core.ui.internal.components.NestedProjectsComposite;
 
 
-public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuListener {
+public class UpdateMavenProjectsDialog extends TitleAreaDialog {
 
-  private static final Logger log = LoggerFactory.getLogger(UpdateMavenProjectsDialog.class);
-
-  private static final String SEPARATOR = System.getProperty("file.separator"); //$NON-NLS-1$
-
-  private CheckboxTreeViewer codebaseViewer;
-
-  private Collection<IProject> projects;
+  private NestedProjectsComposite nestedProjectsComposite;
 
   private Button offlineModeBtn;
 
   private Button forceUpdateBtn;
 
-  private Map<String, IProject> projectPaths;
-
   private final IProject[] initialSelection;
 
   private IProject[] selectedProjects;
@@ -125,195 +79,25 @@ public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuL
     shell.setText(getDialogTitle());
   }
 
-  private String getElePath(Object element) {
-    if(element instanceof IProject) {
-      IProject project = (IProject) element;
-      URI locationURI = project.getLocationURI();
-
-      try {
-        IFileStore store = EFS.getStore(locationURI);
-        File file = store.toLocalFile(0, null);
-        return file.toString() + SEPARATOR;
-      } catch(CoreException ex) {
-        log.error(ex.getMessage(), ex);
-      }
-    }
-    return null;
-  }
-
   /**
    * Create contents of the dialog.
    * 
    * @param parent
    */
   @Override
-  @SuppressWarnings("rawtypes")
   protected Control createDialogArea(Composite parent) {
     Composite area = (Composite) super.createDialogArea(parent);
-    Composite container = new Composite(area, SWT.NONE);
-
-    GridLayout layout = new GridLayout(2, false);
-    layout.marginLeft = 10;
-    container.setLayout(layout);
-    container.setLayoutData(new GridData(GridData.FILL_BOTH));
-
-    Label lblAvailable = new Label(container, SWT.NONE);
-    lblAvailable.setText(Messages.UpdateDepenciesDialog_availableCodebasesLabel);
-    lblAvailable.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, true, false, 2, 1));
-
-    codebaseViewer = new CheckboxTreeViewer(container, SWT.BORDER);
-    codebaseViewer.setContentProvider(new ITreeContentProvider() {
-
-      public void dispose() {
-      }
-
-      public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
-      }
-
-      public Object[] getElements(Object element) {
-        if(element instanceof Collection) {
-          return ((Collection) element).toArray();
-        }
-        return null;
-      }
-
-      public Object[] getChildren(Object parentElement) {
-        if(parentElement instanceof IProject) {
-          String elePath = getElePath(parentElement);
-          String prevPath = null;
-          List<IProject> children = new ArrayList<IProject>();
-          for(String path : projectPaths.keySet()) {
-            if(path.length() != elePath.length() && path.startsWith(elePath)) {
-              if(prevPath == null || !path.startsWith(prevPath)) {
-                prevPath = path;
-                children.add(getProject(path));
-              }
-            }
-          }
-          return children.toArray();
-        } else if(parentElement instanceof Collection) {
-          return ((Collection) parentElement).toArray();
-        }
-        return null;
-      }
-
-      public Object getParent(Object element) {
-        String elePath = getElePath(element);
-        String prevPath = null;
-        for(String path : projectPaths.keySet()) {
-          if(elePath.length() != path.length() && elePath.startsWith(path)
-              && (prevPath == null || prevPath.length() < path.length())) {
-            prevPath = path;
-          }
-        }
-        return prevPath == null ? projects : getProject(prevPath);
-      }
-
-      public boolean hasChildren(Object element) {
-        if(element instanceof IProject) {
-          String elePath = getElePath(element);
-          for(String path : projectPaths.keySet()) {
-            if(elePath.length() != path.length() && path.startsWith(elePath)) {
-              return true;
-            }
-          }
-        } else if(element instanceof Collection) {
-          return !((Collection) element).isEmpty();
-        }
-        return false;
-      }
-    });
-    codebaseViewer.setLabelProvider(new LabelProvider() {
-      public Image getImage(Object element) {
-        return MavenImages.createOverlayImage(MavenImages.MVN_PROJECT, PlatformUI.getWorkbench().getSharedImages()
-            .getImage(IDE.SharedImages.IMG_OBJ_PROJECT), MavenImages.MAVEN_OVERLAY, IDecoration.TOP_LEFT);
-      }
-
-      public String getText(Object element) {
-        return element instanceof IProject ? ((IProject) element).getName() : ""; //$NON-NLS-1$
-      }
-    });
-    projects = getMavenCodebases();
-    codebaseViewer.setInput(projects);
-    codebaseViewer.expandAll();
-    for(IProject project : initialSelection) {
-      codebaseViewer.setSubtreeChecked(project, true);
-    }
+    GridLayout gridLayout = (GridLayout) area.getLayout();
+    gridLayout.verticalSpacing = 5;
+    gridLayout.marginBottom = 5;
+    gridLayout.marginRight = 5;
+    gridLayout.marginLeft = 5;
 
-    // Reveal the first element
-    if(initialSelection.length > 0) {
-      codebaseViewer.reveal(initialSelection[0]);
-    }
+    nestedProjectsComposite = new NestedProjectsComposite(area, SWT.NONE, initialSelection);
+    nestedProjectsComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1));
 
-    Tree tree = codebaseViewer.getTree();
-    GridData gd = new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1);
-    gd.heightHint = 300;
-    gd.widthHint = 300;
-    tree.setLayoutData(gd);
-
-    Composite selectionActionComposite = new Composite(container, SWT.NONE);
-    selectionActionComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false, 1, 1));
-    GridLayout gl_selectionActionComposite = new GridLayout(1, false);
-    gl_selectionActionComposite.marginWidth = 0;
-    gl_selectionActionComposite.marginHeight = 0;
-    selectionActionComposite.setLayout(gl_selectionActionComposite);
-
-    Button selectAllBtn = new Button(selectionActionComposite, SWT.NONE);
-    selectAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, false, false, 1, 1));
-    selectAllBtn.setText(Messages.UpdateDepenciesDialog_selectAll);
-    selectAllBtn.addSelectionListener(new SelectionListener() {
-      public void widgetSelected(SelectionEvent e) {
-        for(IProject project : projects) {
-          codebaseViewer.setSubtreeChecked(project, true);
-        }
-      }
-
-      public void widgetDefaultSelected(SelectionEvent e) {
-
-      }
-    });
-
-    Button deselectAllBtn = new Button(selectionActionComposite, SWT.NONE);
-    deselectAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false, 1, 1));
-    deselectAllBtn.setText(Messages.UpdateDepenciesDialog_deselectAll);
-    deselectAllBtn.addSelectionListener(new SelectionListener() {
-      public void widgetSelected(SelectionEvent e) {
-        for(IProject project : projects) {
-          codebaseViewer.setSubtreeChecked(project, false);
-        }
-      }
-
-      public void widgetDefaultSelected(SelectionEvent e) {
-
-      }
-    });
-
-    Button expandAllBtn = new Button(selectionActionComposite, SWT.NONE);
-    expandAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, false, false, 1, 1));
-    expandAllBtn.setText(Messages.UpdateDepenciesDialog_expandAll);
-    expandAllBtn.addSelectionListener(new SelectionListener() {
-      public void widgetSelected(SelectionEvent e) {
-        codebaseViewer.expandAll();
-      }
-
-      public void widgetDefaultSelected(SelectionEvent e) {
-      }
-    });
-
-    Button collapseAllBtn = new Button(selectionActionComposite, SWT.NONE);
-    collapseAllBtn.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, false, false, 1, 1));
-    collapseAllBtn.setText(Messages.UpdateDepenciesDialog_collapseAll);
-    collapseAllBtn.addSelectionListener(new SelectionListener() {
-      public void widgetSelected(SelectionEvent e) {
-        codebaseViewer.collapseAll();
-      }
-
-      public void widgetDefaultSelected(SelectionEvent e) {
-      }
-    });
-
-    Composite optionsComposite = new Composite(container, SWT.NONE);
-    optionsComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false, 2, 1));
+    Composite optionsComposite = new Composite(area, SWT.NONE);
+    optionsComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false, 1, 1));
     GridLayout gl_optionsComposite = new GridLayout(1, false);
     gl_optionsComposite.marginHeight = 0;
     gl_optionsComposite.marginWidth = 0;
@@ -348,7 +132,7 @@ public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuL
 
     setTitle(getDialogTitle());
     setMessage(getDialogMessage());
-    createMenu();
+
     return area;
   }
 
@@ -364,12 +148,7 @@ public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuL
   }
 
   protected void okPressed() {
-    Object[] obj = codebaseViewer.getCheckedElements();
-    IProject[] projects = new IProject[obj.length];
-    for(int i = 0; i < obj.length; i++ ) {
-      projects[i] = (IProject) obj[i];
-    }
-    selectedProjects = projects;
+    selectedProjects = nestedProjectsComposite.getSelectedProjects();
 
     offlineMode = offlineModeBtn.getSelection();
     forceUpdateDependencies = forceUpdateBtn.getSelection();
@@ -379,52 +158,6 @@ public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuL
     super.okPressed();
   }
 
-  private Collection<IProject> getMavenCodebases() {
-    projectPaths = new TreeMap<String, IProject>();
-
-    for(IProject project : ResourcesPlugin.getWorkspace().getRoot().getProjects()) {
-      try {
-        if(project.isAccessible() && project.hasNature(IMavenConstants.NATURE_ID)) {
-          if(project.getLocationURI() != null) {
-            String path = getElePath(project);
-            if(path != null) {
-              projectPaths.put(path, project);
-            }
-          }
-        }
-      } catch(CoreException ex) {
-        log.error(ex.getMessage(), ex);
-      }
-    }
-
-    if(projectPaths.isEmpty()) {
-      return Collections.<IProject> emptyList();
-    }
-    projects = new ArrayList<IProject>();
-    String previous = projectPaths.keySet().iterator().next();
-    addProject(projects, previous);
-    for(String path : projectPaths.keySet()) {
-      if(!path.startsWith(previous)) {
-        previous = path;
-        IProject project = getProject(path);
-        if(project != null) {
-          projects.add(project);
-        }
-      }
-    }
-    return projects;
-  }
-
-  private static void addProject(Collection<IProject> projects, String location) {
-    IWorkspaceRoot root = ResourcesPlugin.getWorkspace().getRoot();
-    for(IContainer container : root.findContainersForLocationURI(new File(location).toURI())) {
-      if(container instanceof IProject) {
-        projects.add((IProject) container);
-        break;
-      }
-    }
-  }
-
   public IProject[] getSelectedProjects() {
     return selectedProjects;
   }
@@ -449,52 +182,6 @@ public class UpdateMavenProjectsDialog extends TitleAreaDialog implements IMenuL
     return refreshFromLocal;
   }
 
-  private IProject getProject(String path) {
-    return projectPaths.get(path);
-  }
-
-  private void createMenu() {
-    MenuManager menuMgr = new MenuManager();
-    Menu contextMenu = menuMgr.createContextMenu(codebaseViewer.getControl());
-    menuMgr.addMenuListener(this);
-    codebaseViewer.getControl().setMenu(contextMenu);
-    menuMgr.setRemoveAllWhenShown(true);
-  }
-
-  private IProject getSelection() {
-    ISelection selection = codebaseViewer.getSelection();
-    if(selection instanceof IStructuredSelection) {
-      return (IProject) ((IStructuredSelection) selection).getFirstElement();
-    }
-    return null;
-  }
-
-  /* (non-Javadoc)
-   * @see org.eclipse.jface.action.IMenuListener#menuAboutToShow(org.eclipse.jface.action.IMenuManager)
-   */
-  public void menuAboutToShow(IMenuManager manager) {
-    if(codebaseViewer.getSelection().isEmpty()) {
-      return;
-    }
-
-    if(codebaseViewer.getSelection() instanceof IStructuredSelection) {
-      manager.add(selectTree);
-      manager.add(deselectTree);
-    }
-  }
-
-  private final Action selectTree = new Action(Messages.UpdateDepenciesDialog_selectTree) {
-    public void run() {
-      codebaseViewer.setSubtreeChecked(getSelection(), true);
-    }
-  };
-
-  private final Action deselectTree = new Action(Messages.UpdateDepenciesDialog_deselectTree) {
-    public void run() {
-      codebaseViewer.setSubtreeChecked(getSelection(), false);
-    }
-  };
-
   private Button btnUpdateProjectConfiguration;
 
   private Button btnCleanProjects;