// 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.openapi.project.Project; import com.intellij.openapi.ui.Messages; import com.intellij.openapi.util.text.StringUtil; 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.HgVcs; import org.zmlx.hg4idea.execution.HgCommandResult; import org.zmlx.hg4idea.execution.HgPromptCommandExecutor; import java.util.LinkedList; import java.util.List; import java.util.concurrent.atomic.AtomicInteger; import static org.zmlx.hg4idea.util.HgErrorUtil.hasUncommittedChangesConflict; public class HgUpdateCommand { private final Project project; private final VirtualFile repo; private String branch; private String revision; private boolean clean; public HgUpdateCommand(@NotNull Project project, @NotNull VirtualFile repo) { this.project = project; this.repo = repo; } public void setBranch(String branch) { this.branch = branch; } public void setRevision(String revision) { this.revision = revision; } public void setClean(boolean clean) { this.clean = clean; } @Nullable public HgCommandResult execute() { List arguments = new LinkedList(); if (clean) { arguments.add("--clean"); } if (!StringUtil.isEmptyOrSpaces(revision)) { arguments.add("--rev"); arguments.add(revision); } else if (!StringUtil.isEmptyOrSpaces(branch)) { arguments.add(branch); } final HgPromptCommandExecutor executor = new HgPromptCommandExecutor(project); executor.setShowOutput(true); HgCommandResult result = executor.executeInCurrentThread(repo, "update", arguments); if (!clean && hasUncommittedChangesConflict(result)) { final String message = "Your uncommitted changes couldn't be merged into the requested changeset.
" + "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); } } project.getMessageBus().syncPublisher(HgVcs.BRANCH_TOPIC).update(project, null); VfsUtil.markDirtyAndRefresh(true, true, false, repo); return result; } public static int showDiscardChangesConfirmation(@NotNull final Project project, @NotNull final String confirmationMessage) { final AtomicInteger exitCode = new AtomicInteger(); UIUtil.invokeAndWaitIfNeeded(new Runnable() { @Override public void run() { exitCode.set(Messages.showOkCancelDialog(project, confirmationMessage, "Uncommitted Changes Problem", "&Discard Changes", "&Cancel", Messages.getWarningIcon())); } }); return exitCode.get(); } }