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

import com.intellij.execution.configurations.GeneralCommandLine;
import jetbrains.buildServer.util.StringUtil;
import org.jetbrains.annotations.NotNull;

import java.util.Set;

public class MercurialCommandLine extends GeneralCommandLine {

  private final Set<String> myPrivateData;

  public MercurialCommandLine(@NotNull Set<String> privateData) {
    myPrivateData = privateData;
  }

  @Override
  public String getCommandLineString() {
    String original = super.getCommandLineString();
    return CommandUtil.removePrivateData(original, myPrivateData);
  }

  @Override
  public void addParameter(@NotNull String parameter) {
    String escaped = escape(parameter);
    super.addParameter(escaped);
  }

  private String escape(String s) {
    return StringUtil.escapeQuotesIfWindows(s);
  }
}