package jetbrains.buildServer.buildTriggers.vcs.mercurial.command;

import com.intellij.execution.configurations.GeneralCommandLine;
import jetbrains.buildServer.ExecResult;
import jetbrains.buildServer.vcs.VcsException;
import org.jetbrains.annotations.NotNull;

import java.io.File;
import java.util.Collections;
import java.util.Set;

/**
 * @author dmitry.neverov
 */
public class VcsRootCommand extends BaseCommand {

  private final Settings mySettings;


  public VcsRootCommand(@NotNull final Settings settings, @NotNull final File workDir) {
    super(settings.getHgCommandPath(), workDir);
    mySettings = settings;
  }


  public Settings getSettings() {
    return mySettings;
  }


  public Set<String> getPrivateData() {
    return Collections.singleton(mySettings.getPassword());
  }


  protected ExecResult runCommand(@NotNull GeneralCommandLine cli) throws VcsException {
    return CommandUtil.runCommand(cli, getPrivateData());
  }


  protected ExecResult runCommand(@NotNull GeneralCommandLine cli, int executionTimeout) throws VcsException {
    return CommandUtil.runCommand(cli, executionTimeout, getPrivateData());
  }


  protected ExecResult runCommand(@NotNull GeneralCommandLine cli, boolean checkFailure) throws VcsException {
    return CommandUtil.runCommand(cli, getPrivateData(), checkFailure);
  }
}