1 files changed, 90 insertions, 0 deletions
diff --git a/plugins/hg4idea/src/org/zmlx/hg4idea/push/HgOutgoingCommitsProvider.java b/plugins/hg4idea/src/org/zmlx/hg4idea/push/HgOutgoingCommitsProvider.java
new file mode 100644
index 000000000000..70055cafe1a1
--- /dev/null
+++ b/plugins/hg4idea/src/org/zmlx/hg4idea/push/HgOutgoingCommitsProvider.java
@@ -0,0 +1,90 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.zmlx.hg4idea.push;
+
+import com.intellij.dvcs.push.*;
+import com.intellij.dvcs.repo.Repository;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.vcs.log.VcsFullCommitDetails;
+import org.jetbrains.annotations.NotNull;
+import org.zmlx.hg4idea.HgVcs;
+import org.zmlx.hg4idea.command.HgOutgoingCommand;
+import org.zmlx.hg4idea.execution.HgCommandResult;
+import org.zmlx.hg4idea.log.HgBaseLogParser;
+import org.zmlx.hg4idea.log.HgHistoryUtil;
+import org.zmlx.hg4idea.util.HgChangesetUtil;
+import org.zmlx.hg4idea.util.HgErrorUtil;
+import org.zmlx.hg4idea.util.HgVersion;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+public class HgOutgoingCommitsProvider extends OutgoingCommitsProvider {
+
+
+  private static final Logger LOG = Logger.getInstance(HgOutgoingCommitsProvider.class);
+  private static final String LOGIN_AND_REFRESH_LINK = "Enter Password & Refresh";
+
+  @NotNull
+  @Override
+  public OutgoingResult getOutgoingCommits(@NotNull final Repository repository,
+                                           @NotNull final PushSpec pushSpec,
+                                           boolean initial) {
+    final Project project = repository.getProject();
+    HgVcs hgvcs = HgVcs.getInstance(project);
+    assert hgvcs != null;
+    final HgVersion version = hgvcs.getVersion();
+    String[] templates = HgBaseLogParser.constructFullTemplateArgument(true, version);
+    HgOutgoingCommand hgOutgoingCommand = new HgOutgoingCommand(project);
+    HgTarget hgTarget = (HgTarget)pushSpec.getTarget();
+    List<VcsError> errors = new ArrayList<VcsError>();
+    if (hgTarget == null || StringUtil.isEmptyOrSpaces(hgTarget.myTarget)) {
+      errors.add(new VcsError("Hg push path could not be empty."));
+      return new OutgoingResult(Collections.<VcsFullCommitDetails>emptyList(), errors);
+    }
+    HgCommandResult result = hgOutgoingCommand
+      .execute(repository.getRoot(), HgChangesetUtil.makeTemplate(templates), pushSpec.getSource().getPresentation(),
+               hgTarget.myTarget, initial);
+    if (result == null) {
+      errors.add(new VcsError("Couldn't execute hg outgoing command for " + repository));
+      return new OutgoingResult(Collections.<VcsFullCommitDetails>emptyList(), errors);
+    }
+    List<String> resultErrors = result.getErrorLines();
+    if (resultErrors != null && !resultErrors.isEmpty() && result.getExitValue() != 0) {
+      for (String error : resultErrors) {
+        if (HgErrorUtil.isAbortLine(error)) {
+          if (HgErrorUtil.isAuthorizationError(error)) {
+            VcsError authorizationError =
+              new VcsError(error + "<a href='authenticate'>" + LOGIN_AND_REFRESH_LINK + "</a>", new VcsErrorHandler() {
+                public void handleError(@NotNull CommitLoader commitLoader) {
+                  commitLoader.reloadCommits();
+                }
+              });
+            errors.add(authorizationError);
+          }
+          else {
+            errors.add(new VcsError(error));
+          }
+        }
+      }
+      LOG.warn(resultErrors.toString());
+    }
+    return new OutgoingResult(HgHistoryUtil.createFullCommitsFromResult(project, repository.getRoot(), result, version, true), errors);
+  }
+}