annotate mercurial-server/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/MercurialVcsSupport.java @ 27:7944e8985ebd

prepare modules structure for agent side checkout
author Pavel.Sher
date Wed, 23 Jul 2008 09:18:03 +0400
parents mercurial/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/MercurialVcsSupport.java@7047f643747f
children a7cab5083ada
rev   line source
25
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
1 /*
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
2 * Copyright 2000-2007 JetBrains s.r.o.
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
3 *
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
4 * Licensed under the Apache License, Version 2.0 (the "License");
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
5 * you may not use this file except in compliance with the License.
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
6 * You may obtain a copy of the License at
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
7 *
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
8 * http://www.apache.org/licenses/LICENSE-2.0
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
9 *
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
10 * Unless required by applicable law or agreed to in writing, software
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
11 * distributed under the License is distributed on an "AS IS" BASIS,
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
13 * See the License for the specific language governing permissions and
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
14 * limitations under the License.
7047f643747f license added, preamble added to source code
Pavel.Sher
parents: 23
diff changeset
15 */
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
16 package jetbrains.buildServer.buildTriggers.vcs.mercurial;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
17
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
18 import jetbrains.buildServer.CollectChangesByIncludeRule;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
19 import jetbrains.buildServer.Used;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
20 import jetbrains.buildServer.buildTriggers.vcs.AbstractVcsPropertiesProcessor;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
21 import jetbrains.buildServer.buildTriggers.vcs.mercurial.command.*;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
22 import jetbrains.buildServer.log.Loggers;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
23 import jetbrains.buildServer.serverSide.InvalidProperty;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
24 import jetbrains.buildServer.serverSide.PropertiesProcessor;
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
25 import jetbrains.buildServer.serverSide.SBuildServer;
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
26 import jetbrains.buildServer.serverSide.ServerPaths;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
27 import jetbrains.buildServer.util.FileUtil;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
28 import jetbrains.buildServer.vcs.*;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
29 import jetbrains.buildServer.vcs.patches.PatchBuilder;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
30 import org.jetbrains.annotations.NotNull;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
31 import org.jetbrains.annotations.Nullable;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
32
1
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
33 import java.io.File;
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
34 import java.io.FileFilter;
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
35 import java.io.FileInputStream;
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
36 import java.io.IOException;
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
37 import java.util.*;
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
38 import java.util.concurrent.ConcurrentHashMap;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
39 import java.util.concurrent.ConcurrentMap;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
40 import java.util.concurrent.TimeUnit;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
41 import java.util.concurrent.locks.Lock;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
42 import java.util.concurrent.locks.ReentrantLock;
1
56e7f34ca887 added TeamCity libs, fix tests
Pavel.Sher
parents: 0
diff changeset
43
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
44 /**
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
45 * Mercurial VCS plugin for TeamCity works as follows:
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
46 * <ul>
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
47 * <li>clones repository to internal storage
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
48 * <li>before any operation with working copy of repository pulls changes from the original repository
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
49 * <li>executes corresponding hg command
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
50 * </ul>
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
51 *
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
52 * <p>Working copy of repository is created in the $TEAMCITY_DATA_PATH/system/caches/mercurial folder.
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
53 * <p>Personal builds (remote runs) are not yet supported, they require corresponding functionality from the IDE.
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
54 * <p>Checkout on agent mode is not yet supported too.
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
55 */
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
56 public class MercurialVcsSupport extends VcsSupport implements CollectChangesByIncludeRule {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
57 private ServerPaths myServerPaths;
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
58 private ConcurrentMap<String, Lock> myWorkDirLocks= new ConcurrentHashMap<String, Lock>();
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
59 private static final int OLD_WORK_DIRS_CLEANUP_PERIOD = 600;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
60 private VcsManager myVcsManager;
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
61 private File myDefaultWorkFolderParent;
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
62
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
63 public MercurialVcsSupport(@NotNull final VcsManager vcsManager,
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
64 @NotNull ServerPaths paths,
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
65 @NotNull SBuildServer server) {
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
66 vcsManager.registerVcsSupport(this);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
67 myServerPaths = paths;
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
68 myVcsManager = vcsManager;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
69 server.getExecutor().scheduleAtFixedRate(new Runnable() {
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
70 public void run() {
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
71 removeOldWorkFolders();
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
72 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
73 }, 0, OLD_WORK_DIRS_CLEANUP_PERIOD, TimeUnit.SECONDS);
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
74 myDefaultWorkFolderParent = new File(paths.getCachesDir());
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
75 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
76
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
77 public List<ModificationData> collectBuildChanges(final VcsRoot root,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
78 @NotNull final String fromVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
79 @NotNull final String currentVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
80 final CheckoutRules checkoutRules) throws VcsException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
81 updateWorkingDirectory(root);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
82 return VcsSupportUtil.collectBuildChanges(root, fromVersion, currentVersion, checkoutRules, this);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
83 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
84
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
85 public List<ModificationData> collectBuildChanges(final VcsRoot root,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
86 final String fromVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
87 final String currentVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
88 final IncludeRule includeRule) throws VcsException {
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
89 // first obtain changes between specified versions
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
90 List<ModificationData> result = new ArrayList<ModificationData>();
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
91 Settings settings = new Settings(myDefaultWorkFolderParent, root);
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
92 LogCommand lc = new LogCommand(settings);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
93 lc.setFromRevId(new ChangeSet(fromVersion).getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
94 lc.setToRevId(new ChangeSet(currentVersion).getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
95 List<ChangeSet> changeSets = lc.execute();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
96 if (changeSets.isEmpty()) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
97 return result;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
98 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
99
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
100 // invoke status command for each changeset and determine what files were modified in these changesets
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
101 Iterator<ChangeSet> it = changeSets.iterator();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
102 ChangeSet prev = it.next(); // skip first changeset (cause it was already reported)
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
103 StatusCommand st = new StatusCommand(settings);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
104 while (it.hasNext()) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
105 ChangeSet cur = it.next();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
106 st.setFromRevId(prev.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
107 st.setToRevId(cur.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
108 List<ModifiedFile> modifiedFiles = st.execute();
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
109 // changeset full version will be set into VcsChange structure and
20
90f5e574fb73 comments
Pavel.Sher
parents: 19
diff changeset
110 // stored in database (note that getContent method will be invoked with this version)
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
111 List<VcsChange> files = toVcsChanges(modifiedFiles, prev.getFullVersion(), cur.getFullVersion(), includeRule);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
112 if (files.isEmpty()) continue;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
113 ModificationData md = new ModificationData(cur.getTimestamp(), files, cur.getSummary(), cur.getUser(), root, cur.getFullVersion(), cur.getFullVersion());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
114 result.add(md);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
115 prev = cur;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
116 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
117
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
118 return result;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
119 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
120
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
121 private List<VcsChange> toVcsChanges(final List<ModifiedFile> modifiedFiles, String prevVer, String curVer, final IncludeRule includeRule) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
122 List<VcsChange> files = new ArrayList<VcsChange>();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
123 for (ModifiedFile mf: modifiedFiles) {
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
124 String normalizedPath = PathUtil.normalizeSeparator(mf.getPath());
12
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
125 if (!normalizedPath.startsWith(includeRule.getFrom())) continue; // skip files which do not match include rule
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
126
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
127 VcsChangeInfo.Type changeType = getChangeType(mf.getStatus());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
128 if (changeType == null) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
129 Loggers.VCS.warn("Unable to convert status: " + mf.getStatus() + " to VCS change type");
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
130 changeType = VcsChangeInfo.Type.NOT_CHANGED;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
131 }
12
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
132 files.add(new VcsChange(changeType, mf.getStatus().getName(), normalizedPath, normalizedPath, prevVer, curVer));
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
133 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
134 return files;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
135 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
136
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
137 private VcsChangeInfo.Type getChangeType(final ModifiedFile.Status status) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
138 switch (status) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
139 case ADDED:return VcsChangeInfo.Type.ADDED;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
140 case MODIFIED:return VcsChangeInfo.Type.CHANGED;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
141 case REMOVED:return VcsChangeInfo.Type.REMOVED;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
142 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
143 return null;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
144 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
145
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
146 @NotNull
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
147 public byte[] getContent(final VcsModification vcsModification,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
148 final VcsChangeInfo change,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
149 final VcsChangeInfo.ContentType contentType,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
150 final VcsRoot vcsRoot) throws VcsException {
12
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
151 updateWorkingDirectory(vcsRoot);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
152 String version = contentType == VcsChangeInfo.ContentType.AFTER_CHANGE ? change.getAfterChangeRevisionNumber() : change.getBeforeChangeRevisionNumber();
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
153 return getContent(change.getRelativeFileName(), vcsRoot, version);
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
154 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
155
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
156 @NotNull
12
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
157 public byte[] getContent(final String filePath, final VcsRoot vcsRoot, final String version) throws VcsException {
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
158 updateWorkingDirectory(vcsRoot);
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
159 Settings settings = new Settings(myDefaultWorkFolderParent, vcsRoot);
12
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
160 CatCommand cc = new CatCommand(settings);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
161 ChangeSet cs = new ChangeSet(version);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
162 cc.setRevId(cs.getId());
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
163 File parentDir = cc.execute(Collections.singletonList(filePath));
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
164 File file = new File(parentDir, filePath);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
165 if (file.isFile()) {
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
166 try {
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
167 return FileUtil.loadFileBytes(file);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
168 } catch (IOException e) {
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
169 throw new VcsException("Failed to load content of file: " + file.getAbsolutePath(), e);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
170 }
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
171 } else {
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
172 Loggers.VCS.warn("Unable to obtain content of the file: " + filePath);
6989ac0f8cac getContent supported
Pavel.Sher
parents: 11
diff changeset
173 }
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
174 return new byte[0];
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
175 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
176
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
177 public String getName() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
178 return "mercurial";
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
179 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
180
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
181 @Used("jsp")
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
182 public String getDisplayName() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
183 return "Mercurial";
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
184 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
185
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
186 @Nullable
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
187 public PropertiesProcessor getVcsPropertiesProcessor() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
188 return new AbstractVcsPropertiesProcessor() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
189 public Collection<InvalidProperty> process(final Map<String, String> properties) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
190 List<InvalidProperty> result = new ArrayList<InvalidProperty>();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
191 if (isEmpty(properties.get(Constants.HG_COMMAND_PATH_PROP))) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
192 result.add(new InvalidProperty(Constants.HG_COMMAND_PATH_PROP, "Path to 'hg' command must be specified"));
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
193 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
194 if (isEmpty(properties.get(Constants.REPOSITORY_PROP))) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
195 result.add(new InvalidProperty(Constants.REPOSITORY_PROP, "Repository must be specified"));
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
196 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
197 return result;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
198 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
199 };
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
200 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
201
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
202 public String getVcsSettingsJspFilePath() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
203 return "mercurialSettings.jsp";
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
204 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
205
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
206 @NotNull
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
207 public String getCurrentVersion(final VcsRoot root) throws VcsException {
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
208 // we will return full version of the most recent change as current version
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
209 updateWorkingDirectory(root);
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
210 Settings settings = new Settings(myDefaultWorkFolderParent, root);
9
7dadebd03515 use tip command instead of log in getCurrentVersion method
Pavel.Sher
parents: 8
diff changeset
211 TipCommand lc = new TipCommand(settings);
7dadebd03515 use tip command instead of log in getCurrentVersion method
Pavel.Sher
parents: 8
diff changeset
212 ChangeSet changeSet = lc.execute();
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
213 return changeSet.getFullVersion();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
214 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
215
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
216 public String describeVcsRoot(final VcsRoot vcsRoot) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
217 return "mercurial: " + vcsRoot.getProperty(Constants.REPOSITORY_PROP);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
218 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
219
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
220 public boolean isTestConnectionSupported() {
16
7aa397165fa0 identify command added, test connection now uses identify command
Pavel.Sher
parents: 13
diff changeset
221 return true;
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
222 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
223
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
224 @Nullable
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
225 public String testConnection(final VcsRoot vcsRoot) throws VcsException {
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
226 Settings settings = new Settings(myDefaultWorkFolderParent, vcsRoot);
16
7aa397165fa0 identify command added, test connection now uses identify command
Pavel.Sher
parents: 13
diff changeset
227 IdentifyCommand id = new IdentifyCommand(settings);
7aa397165fa0 identify command added, test connection now uses identify command
Pavel.Sher
parents: 13
diff changeset
228 StringBuilder res = new StringBuilder();
17
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
229 res.append(quoteIfNeeded(settings.getHgCommandPath()));
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
230 res.append(" identify ");
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
231 res.append(quoteIfNeeded(settings.getRepository()));
16
7aa397165fa0 identify command added, test connection now uses identify command
Pavel.Sher
parents: 13
diff changeset
232 res.append('\n').append(id.execute());
7aa397165fa0 identify command added, test connection now uses identify command
Pavel.Sher
parents: 13
diff changeset
233 return res.toString();
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
234 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
235
17
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
236 private String quoteIfNeeded(@NotNull String str) {
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
237 if (str.indexOf(' ') != -1) {
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
238 return "\"" + str + "\"";
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
239 }
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
240
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
241 return str;
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
242 }
21b5b1c5dd74 test connection minor fix
Pavel.Sher
parents: 16
diff changeset
243
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
244 @Nullable
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
245 public Map<String, String> getDefaultVcsProperties() {
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
246 return Collections.singletonMap(Constants.HG_COMMAND_PATH_PROP, "hg");
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
247 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
248
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
249 public String getVersionDisplayName(final String version, final VcsRoot root) throws VcsException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
250 return version;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
251 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
252
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
253 @NotNull
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
254 public Comparator<String> getVersionComparator() {
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
255 // comparator is called when TeamCity needs to sort modifications in the order of their appearance,
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
256 // currently we sort changes by revision number, not sure however that this is a good idea,
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
257 // probably it would be better to sort them by timestamp (and to add timestamp into the version).
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
258 return new Comparator<String>() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
259 public int compare(final String o1, final String o2) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
260 try {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
261 return new ChangeSet(o1).getRevNumber() - new ChangeSet(o2).getRevNumber();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
262 } catch (Exception e) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
263 return 1;
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
264 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
265 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
266 };
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
267 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
268
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
269 public void buildPatch(final VcsRoot root,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
270 @Nullable final String fromVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
271 @NotNull final String toVersion,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
272 final PatchBuilder builder,
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
273 final CheckoutRules checkoutRules) throws IOException, VcsException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
274 updateWorkingDirectory(root);
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
275 Settings settings = new Settings(myDefaultWorkFolderParent, root);
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
276 if (fromVersion == null) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
277 buildFullPatch(settings, new ChangeSet(toVersion), builder);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
278 } else {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
279 buildIncrementalPatch(settings, new ChangeSet(fromVersion), new ChangeSet(toVersion), builder);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
280 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
281 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
282
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
283 // builds patch from version to version
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
284 private void buildIncrementalPatch(final Settings settings, @NotNull final ChangeSet fromVer, @NotNull final ChangeSet toVer, final PatchBuilder builder)
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
285 throws VcsException, IOException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
286 StatusCommand st = new StatusCommand(settings);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
287 st.setFromRevId(fromVer.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
288 st.setToRevId(toVer.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
289 List<ModifiedFile> modifiedFiles = st.execute();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
290 List<String> notDeletedFiles = new ArrayList<String>();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
291 for (ModifiedFile f: modifiedFiles) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
292 if (f.getStatus() != ModifiedFile.Status.REMOVED) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
293 notDeletedFiles.add(f.getPath());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
294 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
295 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
296
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
297 CatCommand cc = new CatCommand(settings);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
298 cc.setRevId(toVer.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
299 File parentDir = cc.execute(notDeletedFiles);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
300
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
301 try {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
302 for (ModifiedFile f: modifiedFiles) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
303 final File virtualFile = new File(f.getPath());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
304 if (f.getStatus() == ModifiedFile.Status.REMOVED) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
305 builder.deleteFile(virtualFile, true);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
306 } else {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
307 File realFile = new File(parentDir, f.getPath());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
308 FileInputStream is = new FileInputStream(realFile);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
309 try {
11
568707240741 fix patch
Pavel.Sher
parents: 9
diff changeset
310 builder.changeOrCreateBinaryFile(virtualFile, null, is, realFile.length());
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
311 } finally {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
312 is.close();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
313 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
314 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
315 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
316 } finally {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
317 FileUtil.delete(parentDir);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
318 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
319 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
320
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
321 // builds patch by exporting files using specified version
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
322 private void buildFullPatch(final Settings settings, @NotNull final ChangeSet toVer, final PatchBuilder builder)
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
323 throws IOException, VcsException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
324 CloneCommand cl = new CloneCommand(settings);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
325 cl.setToId(toVer.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
326 File tempDir = FileUtil.createTempDirectory("mercurial", toVer.getId());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
327 try {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
328 final File repRoot = new File(tempDir, "rep");
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
329 cl.setDestDir(repRoot.getAbsolutePath());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
330 cl.execute();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
331 buildPatchFromDirectory(builder, repRoot, new FileFilter() {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
332 public boolean accept(final File file) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
333 return !(file.isDirectory() && ".hg".equals(file.getName()));
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
334 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
335 });
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
336 } finally {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
337 FileUtil.delete(tempDir);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
338 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
339 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
340
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
341 private void buildPatchFromDirectory(final PatchBuilder builder, final File repRoot, final FileFilter filter) throws IOException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
342 buildPatchFromDirectory(repRoot, builder, repRoot, filter);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
343 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
344
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
345 private void buildPatchFromDirectory(File curDir, final PatchBuilder builder, final File repRoot, final FileFilter filter) throws IOException {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
346 File[] files = curDir.listFiles(filter);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
347 if (files != null) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
348 for (File realFile: files) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
349 String relPath = realFile.getAbsolutePath().substring(repRoot.getAbsolutePath().length());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
350 final File virtualFile = new File(relPath);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
351 if (realFile.isDirectory()) {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
352 builder.createDirectory(virtualFile);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
353 buildPatchFromDirectory(realFile, builder, repRoot, filter);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
354 } else {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
355 final FileInputStream is = new FileInputStream(realFile);
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
356 try {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
357 builder.createBinaryFile(virtualFile, null, is, realFile.length());
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
358 } finally {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
359 is.close();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
360 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
361 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
362 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
363 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
364 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
365
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
366 // updates current working copy of repository by pulling changes from the repository specified in VCS root
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
367 private void updateWorkingDirectory(final VcsRoot root) throws VcsException {
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
368 Settings settings = new Settings(myDefaultWorkFolderParent, root);
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
369 File workDir = settings.getWorkingDir();
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
370 lockWorkDir(workDir);
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
371 try {
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
372 if (hasRepositoryCopy(workDir)) {
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
373 // update
8
2cb2df5a0dcd fix working directory update (use pull command)
Pavel.Sher
parents: 1
diff changeset
374 PullCommand pull = new PullCommand(settings);
2cb2df5a0dcd fix working directory update (use pull command)
Pavel.Sher
parents: 1
diff changeset
375 pull.execute();
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
376 } else {
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
377 // clone
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
378 CloneCommand cl = new CloneCommand(settings);
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
379 cl.setDestDir(workDir.getAbsolutePath());
18
d787c696225c do not update local working directory
Pavel.Sher
parents: 17
diff changeset
380 cl.setUpdateWorkingDir(false);
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
381 cl.execute();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
382 }
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
383 } finally {
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
384 unlockWorkDir(workDir);
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
385 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
386 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
387
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
388 private void lockWorkDir(@NotNull File workDir) {
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
389 getWorkDirLock(workDir).lock();
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
390 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
391
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
392 private void unlockWorkDir(@NotNull File workDir) {
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
393 getWorkDirLock(workDir).unlock();
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
394 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
395
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
396 private Lock getWorkDirLock(final File workDir) {
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
397 String path = workDir.getAbsolutePath();
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
398 Lock lock = myWorkDirLocks.get(path);
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
399 if (lock == null) {
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
400 lock = new ReentrantLock();
22
0d6f27953b30 synchronization fixes
Pavel.Sher
parents: 21
diff changeset
401 Lock curLock = myWorkDirLocks.putIfAbsent(path, lock);
21
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
402 if (curLock != null) {
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
403 lock = curLock;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
404 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
405 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
406 return lock;
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
407 }
c76d6a2b27f6 proper synchronization for working directories
Pavel.Sher
parents: 20
diff changeset
408
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
409 private boolean hasRepositoryCopy(final File workDir) {
19
40b2cf04cd4b some comments added
Pavel.Sher
parents: 18
diff changeset
410 // need better way to check that repository copy is ok
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
411 return workDir.isDirectory() && new File(workDir, ".hg").isDirectory();
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
412 }
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
413
23
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
414 private void removeOldWorkFolders() {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
415 File workFoldersParent = new File(myServerPaths.getCachesDir(), "mercurial");
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
416 if (!workFoldersParent.isDirectory()) return;
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
417
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
418 Set<File> workDirs = new HashSet<File>();
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
419 File[] files = workFoldersParent.listFiles(new FileFilter() {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
420 public boolean accept(final File file) {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
421 return file.isDirectory() && file.getName().startsWith(Settings.DEFAULT_WORK_DIR_PREFIX);
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
422 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
423 });
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
424 if (files != null) {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
425 for (File f: files) {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
426 workDirs.add(FileUtil.getCanonicalFile(f));
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
427 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
428 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
429
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
430 for (VcsRoot vcsRoot: myVcsManager.getAllRegisteredVcsRoots()) {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
431 if (getName().equals(vcsRoot.getVcsName())) {
27
7944e8985ebd prepare modules structure for agent side checkout
Pavel.Sher
parents: 25
diff changeset
432 Settings s = new Settings(myDefaultWorkFolderParent, vcsRoot);
23
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
433 workDirs.remove(FileUtil.getCanonicalFile(s.getWorkingDir()));
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
434 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
435 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
436
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
437 for (File f: workDirs) {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
438 lockWorkDir(f);
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
439 try {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
440 FileUtil.delete(f);
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
441 } finally {
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
442 unlockWorkDir(f);
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
443 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
444 }
3ddf410c2fd6 minor change
Pavel.Sher
parents: 22
diff changeset
445 }
0
a530ea876f55 mercurial support sources added
Pavel.Sher
parents:
diff changeset
446 }