CheckoutProvider.java

/*
 * Copyright 2000-2009 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.community.intellij.plugins.communitycase.checkout;

import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.vcs.changes.VcsDirtyScopeManager;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;
import org.community.intellij.plugins.communitycase.Vcs;
import org.community.intellij.plugins.communitycase.actions.BasicAction;
import org.community.intellij.plugins.communitycase.commands.Command;
import org.community.intellij.plugins.communitycase.commands.HandlerUtil;
import org.community.intellij.plugins.communitycase.commands.LineHandler;
import org.community.intellij.plugins.communitycase.config.Version;
import org.community.intellij.plugins.communitycase.i18n.Bundle;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.File;

/**
 * Checkout provider for the Git
 */
public class CheckoutProvider implements com.intellij.openapi.vcs.CheckoutProvider {
  /**
   * The version number since which "-v" options is supported.
   */
  // TODO check if they will actually support the switch in the released 1.6.0.5
  private static final Version VERBOSE_CLONE_SUPPORTED = new Version(1, 6, 0, 5);

  /**
   * {@inheritDoc}
   */
  public void doCheckout(@NotNull final Project project, @Nullable final Listener listener) {
    BasicAction.saveAll();
    CloneDialog dialog = new CloneDialog(project);
    dialog.show();
    if (!dialog.isOK()) {
      return;
    }
    final VirtualFile destinationParent = LocalFileSystem.getInstance().findFileByIoFile(new File(dialog.getParentDirectory()));
    if (destinationParent == null) {
      return;
    }
    final String sourceRepositoryURL = dialog.getSourceRepositoryURL();
    final String directoryName = dialog.getDirectoryName();
    final String originName = dialog.getOriginName();
    final String parentDirectory = dialog.getParentDirectory();
    checkout(project, listener, destinationParent, sourceRepositoryURL, directoryName, originName, parentDirectory);
  }

  public static void checkout(final Project project,
                              final Listener listener,
                              final VirtualFile destinationParent,
                              final String sourceRepositoryURL,
                              final String directoryName,
                              final String originName,
                              final String parentDirectory) {
    final LineHandler handler = clone(project, sourceRepositoryURL, new File(parentDirectory), directoryName, originName);

    handler.addLineListener(new HandlerUtil.LineHandlerListenerProgress(ProgressManager.getInstance().getProgressIndicator(), handler, "git clone", true));
    new Task.Backgroundable(project, Bundle.message("cloning.repository", sourceRepositoryURL), true) {
      @Override public void run(@NotNull ProgressIndicator indicator) {
        HandlerUtil.runInCurrentThread(handler, indicator, true, "git clone");
      }

      @Override public void onSuccess() {
        destinationParent.refresh(true, true, new Runnable() {
          public void run() {
            if (project.isOpen() && (! project.isDisposed()) && (! project.isDefault())) {
              final VcsDirtyScopeManager mgr = VcsDirtyScopeManager.getInstance(project);
              mgr.fileDirty(destinationParent);
            }
          }
        });
        listener.directoryCheckedOut(new File(parentDirectory, directoryName), Vcs.getKey());
        listener.checkoutCompleted();
      }
    }.queue();
  }

  /**
   * {@inheritDoc}
   */
  public String getVcsName() {
    return "_Git";
  }

  /**
   * Prepare clone handler
   *
   * @param project    a project
   * @param url        an url
   * @param directory  a base directory
   * @param name       a name to checkout
   * @param originName origin name (ignored if null or empty string)
   * @return a handler for clone operation
   */
  public static LineHandler clone(Project project, final String url, final File directory, final String name, final String originName) {
    LineHandler handler = new LineHandler(project, directory, Command.CLONE);
    if (VERBOSE_CLONE_SUPPORTED.isLessOrEqual(Vcs.getInstance(project).version())) {
      handler.addParameters("-v");
    }
    if (originName != null && originName.length() > 0) {
      handler.addParameters("-o", originName);
    }
    handler.addParameters(url, name);
    return handler;
  }
}