org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/repository/tree/command/SubmoduleUpdateCommand.java

   1 /******************************************************************************
   2  *  Copyright (c) 2012 GitHub Inc.
   3  *  All rights reserved. This program and the accompanying materials
   4  *  are made available under the terms of the Eclipse Public License v1.0
   5  *  which accompanies this distribution, and is available at
   6  *  http://www.eclipse.org/legal/epl-v10.html
   7  *
   8  *  Contributors:
   9  *    Kevin Sawicki (GitHub Inc.) - initial API and implementation
  10  *****************************************************************************/
  11 package org.eclipse.egit.ui.internal.repository.tree.command;
  12
  13 import java.text.MessageFormat;
  14 import java.util.ArrayList;
  15 import java.util.List;
  16 import java.util.Map;
  17 import java.util.Map.Entry;
  18
  19 import org.eclipse.core.commands.ExecutionEvent;
  20 import org.eclipse.core.commands.ExecutionException;
  21 import org.eclipse.core.resources.ResourcesPlugin;
  22 import org.eclipse.core.resources.WorkspaceJob;
  23 import org.eclipse.core.runtime.CoreException;
  24 import org.eclipse.core.runtime.IProgressMonitor;
  25 import org.eclipse.core.runtime.IStatus;
  26 import org.eclipse.core.runtime.Status;
  27 import org.eclipse.core.runtime.SubMonitor;
  28 import org.eclipse.core.runtime.jobs.Job;
  29 import org.eclipse.egit.core.op.SubmoduleUpdateOperation;
  30 import org.eclipse.egit.ui.Activator;
  31 import org.eclipse.egit.ui.JobFamilies;
  32 import org.eclipse.egit.ui.internal.UIRepositoryUtils;
  33 import org.eclipse.egit.ui.internal.UIText;
  34 import org.eclipse.egit.ui.internal.repository.tree.RepositoryTreeNode;
  35 import org.eclipse.egit.ui.internal.repository.tree.RepositoryTreeNodeType;
  36 import org.eclipse.jgit.lib.Repository;
  37 import org.eclipse.jgit.submodule.SubmoduleWalk;
  38 import org.eclipse.swt.widgets.Shell;
  39
  40 /**
  41  * Command to update selected submodules
  42  */
  43 public class SubmoduleUpdateCommand extends
  44                 SubmoduleCommand<RepositoryTreeNode<?>> {
  45
  46         @Override
  47         public Object execute(ExecutionEvent event) throws ExecutionException {
  48                 final Map<Repository, List<String>> repoPaths = getSubmodules(getSelectedNodes(event));
  49                 if (!repoPaths.isEmpty()) {
  50                         // Check for uncommitted changes in submodules.
  51                         try {
  52                                 boolean submodulesNodeSelected = false;
  53                                 List<Repository> subRepos = new ArrayList<>();
  54                                 // If Submodules node is selected, check all submodules.
  55                                 for (RepositoryTreeNode<?> node : getSelectedNodes(event)) {
  56                                         if (node.getType() == RepositoryTreeNodeType.SUBMODULES) {
  57                                                 submodulesNodeSelected = true;
  58                                                 SubmoduleWalk walk = SubmoduleWalk
  59                                                                 .forIndex(node.getRepository());
  60                                                 while (walk.next()) {
  61                                                         Repository subRepo = walk.getRepository();
  62                                                         if (subRepo != null) {
  63                                                                 subRepos.add(subRepo);
  64                                                         }
  65                                                 }
  66                                                 break;
  67                                         }
  68                                 }
  69                                 // If Submodule node is not selected, check the selected
  70                                 // submodules.
  71                                 if (!submodulesNodeSelected) {
  72                                         for (Entry<Repository, List<String>> entry : repoPaths
  73                                                         .entrySet()) {
  74                                                 if (entry.getValue() != null) {
  75                                                         for (String path : entry.getValue()) {
  76                                                                 Repository subRepo;
  77                                                                 subRepo = SubmoduleWalk.getSubmoduleRepository(
  78                                                                                 entry.getKey(), path);
  79                                                                 if (subRepo != null) {
  80                                                                         subRepos.add(subRepo);
  81                                                                 }
  82                                                         }
  83                                                 }
  84                                         }
  85                                 }
  86                                 Shell parent = getActiveShell(event);
  87                                 for (Repository subRepo : subRepos) {
  88                                         String repoName = Activator.getDefault().getRepositoryUtil()
  89                                                         .getRepositoryName(subRepo);
  90                                         if (!UIRepositoryUtils.handleUncommittedFiles(subRepo,
  91                                                         parent,
  92                                                         MessageFormat.format(
  93                                                                         UIText.SubmoduleUpdateCommand_UncommittedChanges,
  94                                                                         repoName))) {
  95                                                 return null;
  96                                         }
  97                                 }
  98                         } catch (Exception e) {
  99                                 Activator.handleError(UIText.SubmoduleUpdateCommand_UpdateError,
 100                                                 e, true);
 101                                 return null;
 102                         }
 103
 104                         Job job = new WorkspaceJob(UIText.SubmoduleUpdateCommand_Title) {
 105
 106                                 @Override
 107                                 public IStatus runInWorkspace(IProgressMonitor monitor) {
 108                                         SubMonitor progress = SubMonitor.convert(monitor,
 109                                                         repoPaths.size());
 110                                         try {
 111                                                 for (Entry<Repository, List<String>> entry : repoPaths
 112                                                                 .entrySet()) {
 113                                                         if (progress.isCanceled()) {
 114                                                                 return Status.CANCEL_STATUS;
 115                                                         }
 116                                                         SubmoduleUpdateOperation op = new SubmoduleUpdateOperation(
 117                                                                         entry.getKey());
 118                                                         if (entry.getValue() != null) {
 119                                                                 for (String path : entry.getValue()) {
 120                                                                         op.addPath(path);
 121                                                                 }
 122                                                         }
 123                                                         op.execute(progress.newChild(1));
 124                                                 }
 125                                         } catch (CoreException e) {
 126                                                 Activator.logError(
 127                                                                 UIText.SubmoduleUpdateCommand_UpdateError, e);
 128                                         }
 129                                         return Status.OK_STATUS;
 130                                 }
 131
 132                                 @Override
 133                                 public boolean belongsTo(Object family) {
 134                                         if (JobFamilies.SUBMODULE_UPDATE.equals(family))
 135                                                 return true;
 136                                         return super.belongsTo(family);
 137                                 }
 138                         };
 139                         job.setUser(true);
 140                         job.setRule(ResourcesPlugin.getWorkspace().getRoot());
 141                         job.schedule();
 142                 }
 143                 return null;
 144         }
 145 }