org.jetbrains.idea.svn.actions.MarkTreeConflictResolvedAction.java Source code

Introduction

Here is the source code for org.jetbrains.idea.svn.actions.MarkTreeConflictResolvedAction.java

Source

/*
 * Copyright 2000-2013 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.jetbrains.idea.svn.actions;

import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.actionSystem.DataContext;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.DumbAware;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.util.Ref;
import com.intellij.openapi.vcs.AbstractVcsHelper;
import com.intellij.openapi.vcs.FilePath;
import com.intellij.openapi.vcs.VcsDataKeys;
import com.intellij.openapi.vcs.VcsException;
import com.intellij.openapi.vcs.changes.BackgroundFromStartOption;
import com.intellij.openapi.vcs.changes.Change;
import com.intellij.openapi.vcs.changes.VcsDirtyScopeManager;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.idea.svn.ConflictedSvnChange;
import org.jetbrains.idea.svn.SvnBundle;
import org.jetbrains.idea.svn.SvnVcs;
import org.tmatesoft.svn.core.SVNDepth;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

public class MarkTreeConflictResolvedAction extends AnAction implements DumbAware {
    private static final String myText = SvnBundle.message("action.mark.tree.conflict.resolved.text");

    public MarkTreeConflictResolvedAction() {
        super(myText);
    }

    @Override
    public void update(AnActionEvent e) {
        final MyChecker checker = new MyChecker(e);
        e.getPresentation().setVisible(checker.isEnabled());
        e.getPresentation().setEnabled(checker.isEnabled());
        e.getPresentation().setText(myText);
    }

    private static class MyChecker {
        private final boolean myEnabled;
        private final ConflictedSvnChange myChange;
        private final Project myProject;

        public MyChecker(final AnActionEvent e) {
            final DataContext dc = e.getDataContext();
            myProject = CommonDataKeys.PROJECT.getData(dc);
            final Change[] changes = VcsDataKeys.CHANGE_LEAD_SELECTION.getData(dc);

            if (myProject == null || changes == null || changes.length != 1) {
                myEnabled = false;
                myChange = null;
                return;
            }

            final Change change = changes[0];
            myEnabled = change instanceof ConflictedSvnChange
                    && ((ConflictedSvnChange) change).getConflictState().isTree();
            if (myEnabled) {
                myChange = (ConflictedSvnChange) change;
            } else {
                myChange = null;
            }
        }

        public boolean isEnabled() {
            return myEnabled;
        }

        public ConflictedSvnChange getChange() {
            return myChange;
        }

        public Project getProject() {
            return myProject;
        }
    }

    public void actionPerformed(AnActionEvent e) {
        final MyChecker checker = new MyChecker(e);
        if (!checker.isEnabled())
            return;

        final String markText = SvnBundle.message("action.mark.tree.conflict.resolved.confirmation.title");
        final int result = Messages.showYesNoDialog(checker.getProject(),
                SvnBundle.message("action.mark.tree.conflict.resolved.confirmation.text"), markText,
                Messages.getQuestionIcon());
        if (result == Messages.YES) {
            final Ref<VcsException> exception = new Ref<VcsException>();
            ProgressManager.getInstance().run(new Task.Backgroundable(checker.getProject(), markText, true,
                    BackgroundFromStartOption.getInstance()) {
                public void run(@NotNull ProgressIndicator indicator) {
                    final ConflictedSvnChange change = checker.getChange();
                    final FilePath path = change.getTreeConflictMarkHolder();
                    SvnVcs vcs = SvnVcs.getInstance(checker.getProject());

                    try {
                        vcs.getFactory(path.getIOFile()).createConflictClient().resolve(path.getIOFile(),
                                SVNDepth.EMPTY, false, false, true);
                    } catch (VcsException e) {
                        exception.set(e);
                    }
                    VcsDirtyScopeManager.getInstance(checker.getProject()).filePathsDirty(getDistinctFiles(change),
                            null);
                }
            });
            if (!exception.isNull()) {
                AbstractVcsHelper.getInstance(checker.getProject()).showError(exception.get(), markText);
            }
        }
    }

    private Collection<FilePath> getDistinctFiles(final Change change) {
        final List<FilePath> result = new ArrayList<FilePath>(2);
        if (change.getBeforeRevision() != null) {
            result.add(change.getBeforeRevision().getFile());
        }
        if (change.getAfterRevision() != null) {
            if (change.getBeforeRevision() == null
                    || change.getBeforeRevision() != null && (change.isMoved() || change.isRenamed())) {
                result.add(change.getAfterRevision().getFile());
            }
        }
        return result;
    }
}