/src/org/bitbucket/connectors/jetbrains/BitbucketShareAction.java

package org.bitbucket.connectors.jetbrains;

import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.PlatformDataKeys;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.project.DumbAwareAction;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.vcs.AbstractVcs;
import com.intellij.openapi.vcs.ProjectLevelVcsManager;
import com.intellij.openapi.vcs.VcsDirectoryMapping;
import com.intellij.openapi.vcs.changes.VcsDirtyScopeManager;
import com.intellij.openapi.vfs.VirtualFile;
import org.bitbucket.connectors.jetbrains.ui.BitbucketBundle;
import org.bitbucket.connectors.jetbrains.ui.BitbucketShareDialog;
import org.bitbucket.connectors.jetbrains.vcs.GitHandler;
import org.bitbucket.connectors.jetbrains.vcs.HgHandler;
import org.bitbucket.connectors.jetbrains.vcs.VcsHandler;
import org.jetbrains.annotations.NotNull;

import java.util.*;

/**
 * User: leha2000
 * Date: Apr 15, 2011
 * Time: 12:16:00 PM
 */
public class BitbucketShareAction extends DumbAwareAction {

    public BitbucketShareAction() {
        super(BitbucketBundle.message("share-on-bitbucket"), BitbucketBundle.message("share-on-bitbucket"), BitbucketUtil.ICON);
    }

    public void update(AnActionEvent e) {
        Project project = e.getData(PlatformDataKeys.PROJECT);
        boolean enabled = project != null && !project.isDefault();
        e.getPresentation().setVisible(enabled);
        e.getPresentation().setEnabled(enabled);
    }

    @Override
    public void actionPerformed(AnActionEvent e) {
        ApplicationManager.getApplication().runWriteAction(new Runnable() {
            public void run() {
                FileDocumentManager.getInstance().saveAllDocuments();
            }
        });

        Project project = e.getData(PlatformDataKeys.PROJECT);
        Set<String> names = new HashSet<String>();
        for (RepositoryInfo r : BitbucketUtil.getRepositories(project, true)) {
            names.add(r.getName());
        }

        BitbucketShareDialog dialog = new BitbucketShareDialog(project, names, true);
        final VirtualFile root = project.getBaseDir();
        boolean hg = new HgHandler().getRepositoryRoot(root) != null;
        boolean git = new GitHandler().getRepositoryRoot(root) != null;
        if (hg) {
            dialog.enableGit(false);
        } else if (git) {
            dialog.setGit(true);
            dialog.enableGit(false);
        }

        dialog.show();
        if (!dialog.isOK()) {
            return;
        }

        share(project, dialog.getRepositoryName(), dialog.getDescription(), dialog.isSshRepositoryAccess(), dialog.isGit());
    }

    private void share(final Project project, String name, String description, boolean ssh, boolean git) {
        final VirtualFile root = project.getBaseDir();
        final VcsHandler vcsHandler = git ? new GitHandler() : new HgHandler();

        VirtualFile vcsRoot = vcsHandler.getRepositoryRoot(root);
        if (vcsRoot == null) {
            vcsHandler.initRepository(project, root);
        } else if (vcsRoot != root) {
            Messages.showErrorDialog(project, BitbucketBundle.message("publish-part-err"), BitbucketBundle.message("share-project-on-bitbucket"));
            return;
        }

        refreshAndConfigureVcsMappings(project, root, "", vcsHandler.getVcs(project));

        BitbucketUtil.share(project, root, name, description, ssh, git, vcsHandler);
    }

    public static void refreshAndConfigureVcsMappings(final Project project, final VirtualFile root, final String path, final AbstractVcs vcs) {
        root.refresh(false, false);
        ProjectLevelVcsManager vcsManager = ProjectLevelVcsManager.getInstance(project);
        final List<VcsDirectoryMapping> vcsDirectoryMappings = new ArrayList<VcsDirectoryMapping>(vcsManager.getDirectoryMappings());
        VcsDirectoryMapping mapping = new VcsDirectoryMapping(path, vcs.getName());
        for (int i = 0; i < vcsDirectoryMappings.size(); i++) {
            VcsDirectoryMapping m = vcsDirectoryMappings.get(i);
            if (m.getDirectory().equals(path)) {
                if (m.getVcs().length() == 0) {
                    vcsDirectoryMappings.set(i, mapping);
                    mapping = null;
                    break;
                } else if (m.getVcs().equals(mapping.getVcs())) {
                    mapping = null;
                    break;
                }
            }
        }
        if (mapping != null) {
            vcsDirectoryMappings.add(mapping);
        }
        vcsManager.setDirectoryMappings(vcsDirectoryMappings);
        vcsManager.updateActiveVcss();
        refreshFiles(project, Collections.singleton(root));
    }

    public static void refreshFiles(@NotNull final Project project, @NotNull final Collection<VirtualFile> affectedFiles) {
        final VcsDirtyScopeManager dirty = VcsDirtyScopeManager.getInstance(project);
        for (VirtualFile file : affectedFiles) {
            if (!file.isValid()) {
                continue;
            }
            file.refresh(false, true);
            if (file.isDirectory()) {
                dirty.dirDirtyRecursively(file);
            } else {
                dirty.fileDirty(file);
            }
        }
    }

}