##// END OF EJS Templates
Merged r6245 from trunk....
Toshi MARUYAMA -
r6133:94f5c737465e
parent child
Show More
@@ -1,327 +1,331
1 1 # Redmine - project management software
2 2 # Copyright (C) 2006-2011 Jean-Philippe Lang
3 3 #
4 4 # This program is free software; you can redistribute it and/or
5 5 # modify it under the terms of the GNU General Public License
6 6 # as published by the Free Software Foundation; either version 2
7 7 # of the License, or (at your option) any later version.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU General Public License
15 15 # along with this program; if not, write to the Free Software
16 16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17 17
18 18 require 'redmine/scm/adapters/abstract_adapter'
19 19 require 'cgi'
20 20
21 21 module Redmine
22 22 module Scm
23 23 module Adapters
24 24 class MercurialAdapter < AbstractAdapter
25 25
26 26 # Mercurial executable name
27 27 HG_BIN = Redmine::Configuration['scm_mercurial_command'] || "hg"
28 28 HELPERS_DIR = File.dirname(__FILE__) + "/mercurial"
29 29 HG_HELPER_EXT = "#{HELPERS_DIR}/redminehelper.py"
30 30 TEMPLATE_NAME = "hg-template"
31 31 TEMPLATE_EXTENSION = "tmpl"
32 32
33 33 # raised if hg command exited with error, e.g. unknown revision.
34 34 class HgCommandAborted < CommandFailed; end
35 35
36 36 class << self
37 37 def client_command
38 38 @@bin ||= HG_BIN
39 39 end
40 40
41 41 def sq_bin
42 42 @@sq_bin ||= shell_quote(HG_BIN)
43 43 end
44 44
45 45 def client_version
46 46 @@client_version ||= (hgversion || [])
47 47 end
48 48
49 49 def client_available
50 50 client_version_above?([0, 9, 5])
51 51 end
52 52
53 53 def hgversion
54 54 # The hg version is expressed either as a
55 55 # release number (eg 0.9.5 or 1.0) or as a revision
56 56 # id composed of 12 hexa characters.
57 57 theversion = hgversion_from_command_line.dup
58 58 if theversion.respond_to?(:force_encoding)
59 59 theversion.force_encoding('ASCII-8BIT')
60 60 end
61 61 if m = theversion.match(%r{\A(.*?)((\d+\.)+\d+)})
62 62 m[2].scan(%r{\d+}).collect(&:to_i)
63 63 end
64 64 end
65 65
66 66 def hgversion_from_command_line
67 67 shellout("#{sq_bin} --version") { |io| io.read }.to_s
68 68 end
69 69
70 70 def template_path
71 71 @@template_path ||= template_path_for(client_version)
72 72 end
73 73
74 74 def template_path_for(version)
75 75 if ((version <=> [0,9,5]) > 0) || version.empty?
76 76 ver = "1.0"
77 77 else
78 78 ver = "0.9.5"
79 79 end
80 80 "#{HELPERS_DIR}/#{TEMPLATE_NAME}-#{ver}.#{TEMPLATE_EXTENSION}"
81 81 end
82 82 end
83 83
84 84 def initialize(url, root_url=nil, login=nil, password=nil, path_encoding=nil)
85 85 super
86 86 @path_encoding = path_encoding.blank? ? 'UTF-8' : path_encoding
87 87 end
88 88
89 89 def path_encoding
90 90 @path_encoding
91 91 end
92 92
93 93 def info
94 94 tip = summary['repository']['tip']
95 95 Info.new(:root_url => CGI.unescape(summary['repository']['root']),
96 96 :lastrev => Revision.new(:revision => tip['revision'],
97 97 :scmid => tip['node']))
98 # rescue HgCommandAborted
99 rescue Exception => e
100 logger.error "hg: error during getting info: #{e.message}"
101 nil
98 102 end
99 103
100 104 def tags
101 105 as_ary(summary['repository']['tag']).map { |e| e['name'] }
102 106 end
103 107
104 108 # Returns map of {'tag' => 'nodeid', ...}
105 109 def tagmap
106 110 alist = as_ary(summary['repository']['tag']).map do |e|
107 111 e.values_at('name', 'node')
108 112 end
109 113 Hash[*alist.flatten]
110 114 end
111 115
112 116 def branches
113 117 as_ary(summary['repository']['branch']).map { |e| e['name'] }
114 118 end
115 119
116 120 # Returns map of {'branch' => 'nodeid', ...}
117 121 def branchmap
118 122 alist = as_ary(summary['repository']['branch']).map do |e|
119 123 e.values_at('name', 'node')
120 124 end
121 125 Hash[*alist.flatten]
122 126 end
123 127
124 128 def summary
125 129 return @summary if @summary
126 130 hg 'rhsummary' do |io|
127 131 output = io.read
128 132 if output.respond_to?(:force_encoding)
129 133 output.force_encoding('UTF-8')
130 134 end
131 135 begin
132 136 @summary = ActiveSupport::XmlMini.parse(output)['rhsummary']
133 137 rescue
134 138 end
135 139 end
136 140 end
137 141 private :summary
138 142
139 143 def entries(path=nil, identifier=nil, options={})
140 144 p1 = scm_iconv(@path_encoding, 'UTF-8', path)
141 145 manifest = hg('rhmanifest', '-r', CGI.escape(hgrev(identifier)),
142 146 CGI.escape(without_leading_slash(p1.to_s))) do |io|
143 147 output = io.read
144 148 if output.respond_to?(:force_encoding)
145 149 output.force_encoding('UTF-8')
146 150 end
147 151 begin
148 152 ActiveSupport::XmlMini.parse(output)['rhmanifest']['repository']['manifest']
149 153 rescue
150 154 end
151 155 end
152 156 path_prefix = path.blank? ? '' : with_trailling_slash(path)
153 157
154 158 entries = Entries.new
155 159 as_ary(manifest['dir']).each do |e|
156 160 n = scm_iconv('UTF-8', @path_encoding, CGI.unescape(e['name']))
157 161 p = "#{path_prefix}#{n}"
158 162 entries << Entry.new(:name => n, :path => p, :kind => 'dir')
159 163 end
160 164
161 165 as_ary(manifest['file']).each do |e|
162 166 n = scm_iconv('UTF-8', @path_encoding, CGI.unescape(e['name']))
163 167 p = "#{path_prefix}#{n}"
164 168 lr = Revision.new(:revision => e['revision'], :scmid => e['node'],
165 169 :identifier => e['node'],
166 170 :time => Time.at(e['time'].to_i))
167 171 entries << Entry.new(:name => n, :path => p, :kind => 'file',
168 172 :size => e['size'].to_i, :lastrev => lr)
169 173 end
170 174
171 175 entries
172 176 rescue HgCommandAborted
173 177 nil # means not found
174 178 end
175 179
176 180 def revisions(path=nil, identifier_from=nil, identifier_to=nil, options={})
177 181 revs = Revisions.new
178 182 each_revision(path, identifier_from, identifier_to, options) { |e| revs << e }
179 183 revs
180 184 end
181 185
182 186 # Iterates the revisions by using a template file that
183 187 # makes Mercurial produce a xml output.
184 188 def each_revision(path=nil, identifier_from=nil, identifier_to=nil, options={})
185 189 hg_args = ['log', '--debug', '-C', '--style', self.class.template_path]
186 190 hg_args << '-r' << "#{hgrev(identifier_from)}:#{hgrev(identifier_to)}"
187 191 hg_args << '--limit' << options[:limit] if options[:limit]
188 192 hg_args << hgtarget(path) unless path.blank?
189 193 log = hg(*hg_args) do |io|
190 194 output = io.read
191 195 if output.respond_to?(:force_encoding)
192 196 output.force_encoding('UTF-8')
193 197 end
194 198 begin
195 199 # Mercurial < 1.5 does not support footer template for '</log>'
196 200 ActiveSupport::XmlMini.parse("#{output}</log>")['log']
197 201 rescue
198 202 end
199 203 end
200 204 as_ary(log['logentry']).each do |le|
201 205 cpalist = as_ary(le['paths']['path-copied']).map do |e|
202 206 [e['__content__'], e['copyfrom-path']].map do |s|
203 207 scm_iconv('UTF-8', @path_encoding, CGI.unescape(s))
204 208 end
205 209 end
206 210 cpmap = Hash[*cpalist.flatten]
207 211 paths = as_ary(le['paths']['path']).map do |e|
208 212 p = scm_iconv('UTF-8', @path_encoding, CGI.unescape(e['__content__']) )
209 213 {:action => e['action'],
210 214 :path => with_leading_slash(p),
211 215 :from_path => (cpmap.member?(p) ? with_leading_slash(cpmap[p]) : nil),
212 216 :from_revision => (cpmap.member?(p) ? le['node'] : nil)}
213 217 end.sort { |a, b| a[:path] <=> b[:path] }
214 218 yield Revision.new(:revision => le['revision'],
215 219 :scmid => le['node'],
216 220 :author => (le['author']['__content__'] rescue ''),
217 221 :time => Time.parse(le['date']['__content__']),
218 222 :message => le['msg']['__content__'],
219 223 :paths => paths)
220 224 end
221 225 self
222 226 end
223 227
224 228 # Returns list of nodes in the specified branch
225 229 def nodes_in_branch(branch, options={})
226 230 hg_args = ['rhlog', '--template', '{node|short}\n', '--rhbranch', CGI.escape(branch)]
227 231 hg_args << '--from' << CGI.escape(branch)
228 232 hg_args << '--to' << '0'
229 233 hg_args << '--limit' << options[:limit] if options[:limit]
230 234 hg(*hg_args) { |io| io.readlines.map { |e| e.chomp } }
231 235 end
232 236
233 237 def diff(path, identifier_from, identifier_to=nil)
234 238 hg_args = %w|rhdiff|
235 239 if identifier_to
236 240 hg_args << '-r' << hgrev(identifier_to) << '-r' << hgrev(identifier_from)
237 241 else
238 242 hg_args << '-c' << hgrev(identifier_from)
239 243 end
240 244 unless path.blank?
241 245 p = scm_iconv(@path_encoding, 'UTF-8', path)
242 246 hg_args << CGI.escape(hgtarget(p))
243 247 end
244 248 diff = []
245 249 hg *hg_args do |io|
246 250 io.each_line do |line|
247 251 diff << line
248 252 end
249 253 end
250 254 diff
251 255 rescue HgCommandAborted
252 256 nil # means not found
253 257 end
254 258
255 259 def cat(path, identifier=nil)
256 260 p = CGI.escape(scm_iconv(@path_encoding, 'UTF-8', path))
257 261 hg 'rhcat', '-r', CGI.escape(hgrev(identifier)), hgtarget(p) do |io|
258 262 io.binmode
259 263 io.read
260 264 end
261 265 rescue HgCommandAborted
262 266 nil # means not found
263 267 end
264 268
265 269 def annotate(path, identifier=nil)
266 270 p = CGI.escape(scm_iconv(@path_encoding, 'UTF-8', path))
267 271 blame = Annotate.new
268 272 hg 'rhannotate', '-ncu', '-r', CGI.escape(hgrev(identifier)), hgtarget(p) do |io|
269 273 io.each_line do |line|
270 274 line.force_encoding('ASCII-8BIT') if line.respond_to?(:force_encoding)
271 275 next unless line =~ %r{^([^:]+)\s(\d+)\s([0-9a-f]+):\s(.*)$}
272 276 r = Revision.new(:author => $1.strip, :revision => $2, :scmid => $3,
273 277 :identifier => $3)
274 278 blame.add_line($4.rstrip, r)
275 279 end
276 280 end
277 281 blame
278 282 rescue HgCommandAborted
279 283 # means not found or cannot be annotated
280 284 Annotate.new
281 285 end
282 286
283 287 class Revision < Redmine::Scm::Adapters::Revision
284 288 # Returns the readable identifier
285 289 def format_identifier
286 290 "#{revision}:#{scmid}"
287 291 end
288 292 end
289 293
290 294 # Runs 'hg' command with the given args
291 295 def hg(*args, &block)
292 296 repo_path = root_url || url
293 297 full_args = [HG_BIN, '-R', repo_path, '--encoding', 'utf-8']
294 298 full_args << '--config' << "extensions.redminehelper=#{HG_HELPER_EXT}"
295 299 full_args << '--config' << 'diff.git=false'
296 300 full_args += args
297 301 ret = shellout(full_args.map { |e| shell_quote e.to_s }.join(' '), &block)
298 302 if $? && $?.exitstatus != 0
299 303 raise HgCommandAborted, "hg exited with non-zero status: #{$?.exitstatus}"
300 304 end
301 305 ret
302 306 end
303 307 private :hg
304 308
305 309 # Returns correct revision identifier
306 310 def hgrev(identifier, sq=false)
307 311 rev = identifier.blank? ? 'tip' : identifier.to_s
308 312 rev = shell_quote(rev) if sq
309 313 rev
310 314 end
311 315 private :hgrev
312 316
313 317 def hgtarget(path)
314 318 path ||= ''
315 319 root_url + '/' + without_leading_slash(path)
316 320 end
317 321 private :hgtarget
318 322
319 323 def as_ary(o)
320 324 return [] unless o
321 325 o.is_a?(Array) ? o : Array[o]
322 326 end
323 327 private :as_ary
324 328 end
325 329 end
326 330 end
327 331 end
General Comments 0
You need to be logged in to leave comments. Login now