// Copyright 2008-2010 Victor Iacoban
//
// 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.command;
import com.intellij.dvcs.DvcsUtil;
import com.intellij.openapi.application.AccessToken;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vcs.VcsNotifier;
import com.intellij.openapi.vfs.VfsUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.util.ui.UIUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.zmlx.hg4idea.HgVcsMessages;
import org.zmlx.hg4idea.action.HgCommandResultNotifier;
import org.zmlx.hg4idea.execution.HgCommandResult;
import org.zmlx.hg4idea.execution.HgPromptCommandExecutor;
import org.zmlx.hg4idea.provider.update.HgConflictResolver;
import org.zmlx.hg4idea.repo.HgRepository;
import org.zmlx.hg4idea.util.HgErrorUtil;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import static org.zmlx.hg4idea.util.HgErrorUtil.hasUncommittedChangesConflict;
import static org.zmlx.hg4idea.util.HgUtil.getRepositoryManager;
public class HgUpdateCommand {
private final Project project;
private final VirtualFile repo;
private String revision;
private boolean clean;
public HgUpdateCommand(@NotNull Project project, @NotNull VirtualFile repo) {
this.project = project;
this.repo = repo;
}
public void setRevision(String revision) {
this.revision = revision;
}
public void setClean(boolean clean) {
this.clean = clean;
}
@Nullable
public HgCommandResult execute() {
List<String> arguments = new LinkedList<>();
if (clean) {
arguments.add("--clean");
}
if (!StringUtil.isEmptyOrSpaces(revision)) {
arguments.add("--rev");
arguments.add(revision);
}
final HgPromptCommandExecutor executor = new HgPromptCommandExecutor(project);
executor.setShowOutput(true);
HgCommandResult result;
AccessToken token = DvcsUtil.workingTreeChangeStarted(project);
try {
result =
executor.executeInCurrentThread(repo, "update", arguments);
if (!clean && hasUncommittedChangesConflict(result)) {
final String message = "<html>Your uncommitted changes couldn't be merged into the requested changeset.<br>" +
"Would you like to perform force update and discard them?";
if (showDiscardChangesConfirmation(project, message) == Messages.OK) {
arguments.add("-C");
result = executor.executeInCurrentThread(repo, "update", arguments);
}
}
}
finally {
token.finish();
}
VfsUtil.markDirtyAndRefresh(false, true, false, repo);
return result;
}
public static int showDiscardChangesConfirmation(@NotNull final Project project, @NotNull final String confirmationMessage) {
final AtomicInteger exitCode = new AtomicInteger();
UIUtil.invokeAndWaitIfNeeded(
(Runnable)() -> exitCode.set(Messages.showOkCancelDialog(project, confirmationMessage, "Uncommitted Changes Problem",
"&Discard Changes", "&Cancel", Messages.getWarningIcon())));
return exitCode.get();
}
public static void updateTo(@NotNull final String targetRevision, @NotNull List<HgRepository> repos, @Nullable final Runnable callInAwtLater) {
FileDocumentManager.getInstance().saveAllDocuments();
for (HgRepository repo : repos) {
final VirtualFile repository = repo.getRoot();
Project project = repo.getProject();
updateRepoTo(project, repository, targetRevision, callInAwtLater);
}
}
public static void updateRepoTo(@NotNull final Project project,
@NotNull final VirtualFile repository,
@NotNull final String targetRevision,
@Nullable final Runnable callInAwtLater) {
updateRepoTo(project, repository, targetRevision, false, callInAwtLater);
}
public static void updateRepoTo(@NotNull final Project project,
@NotNull final VirtualFile repository,
@NotNull final String targetRevision,
final boolean clean,
@Nullable final Runnable callInAwtLater) {
new Task.Backgroundable(project, HgVcsMessages.message("action.hg4idea.updateTo.description", targetRevision)) {
@Override
public void onSuccess() {
if (callInAwtLater != null) {
callInAwtLater.run();
}
}
@Override
public void run(@NotNull ProgressIndicator indicator) {
updateRepoToInCurrentThread(project, repository, targetRevision, clean);
}
}.queue();
}
public static boolean updateRepoToInCurrentThread(@NotNull final Project project,
@NotNull final VirtualFile repository,
@NotNull final String targetRevision,
final boolean clean) {
final HgUpdateCommand hgUpdateCommand = new HgUpdateCommand(project, repository);
hgUpdateCommand.setRevision(targetRevision);
hgUpdateCommand.setClean(clean);
HgCommandResult result = hgUpdateCommand.execute();
new HgConflictResolver(project).resolve(repository);
boolean success = !HgErrorUtil.isCommandExecutionFailed(result);
boolean hasUnresolvedConflicts = HgConflictResolver.hasConflicts(project, repository);
if (!success) {
new HgCommandResultNotifier(project).notifyError(result, "", "Update failed");
}
else if (hasUnresolvedConflicts) {
new VcsNotifier(project)
.notifyImportantWarning("Unresolved conflicts.",
HgVcsMessages.message("hg4idea.update.warning.merge.conflicts", repository.getPath()));
}
getRepositoryManager(project).updateRepository(repository);
HgErrorUtil.markDirtyAndHandleErrors(project, repository);
return success;
}
}