# HG changeset patch
# User Dmitry Neverov <dmitry.neverov@jetbrains.com>
# Date 1417629489 -3600
# Node ID b467eb363d46fa54da9ba69109f6770333af6b40
# Parent  884c3d03f6cac6c538d21c1eccfd888319eeaf8d
Support profile for hg update command on agent

diff -r 884c3d03f6ca -r b467eb363d46 mercurial-agent/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/MercurialIncludeRuleUpdater.java
--- a/mercurial-agent/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/MercurialIncludeRuleUpdater.java	Fri Nov 28 16:28:47 2014 +0100
+++ b/mercurial-agent/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/MercurialIncludeRuleUpdater.java	Wed Dec 03 18:58:09 2014 +0100
@@ -247,7 +247,7 @@
     for (BeforeWorkingDirUpdateExtension e : getExtensions(BeforeWorkingDirUpdateExtension.class)) {
       e.call(repo, revision);
     }
-    repo.update().withTraceback(myUseTraceback).toRevision(revision).call();
+    repo.update().withTraceback(myUseTraceback).withProfile(myProfile).toRevision(revision).call();
   }
 
 
diff -r 884c3d03f6ca -r b467eb363d46 mercurial-common/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/command/UpdateCommand.java
--- a/mercurial-common/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/command/UpdateCommand.java	Fri Nov 28 16:28:47 2014 +0100
+++ b/mercurial-common/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/command/UpdateCommand.java	Wed Dec 03 18:58:09 2014 +0100
@@ -29,6 +29,7 @@
   private String myToId;
   private String myBranchName;
   private boolean myTraceback;
+  private boolean myProfile;
 
   public UpdateCommand(@NotNull CommandSettings commandSettings,
                        @NotNull String hgPath,
@@ -56,6 +57,11 @@
     return this;
   }
 
+  public UpdateCommand withProfile(boolean runWithProfile) {
+    myProfile = runWithProfile;
+    return this;
+  }
+
   public void call() throws VcsException {
     ensureWorkingDirIsNotLocked();
 
@@ -63,6 +69,8 @@
     cli.addParameter("update");
     if (myTraceback)
       cli.addParameter("--traceback");
+    if (myProfile)
+      cli.addParameter("--profile");
     addHttpAuthParams(cli);
     cli.addParameter("-C");
     cli.addParameter("-r");