blob: d38af12b24897e5c05aaaff1f0304ca122739c9d [file] [log] [blame]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001# Copyright (C) 2008 The Android Open Source Project
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
15import filecmp
16import os
17import re
18import shutil
19import stat
20import sys
21import urllib2
22
23from color import Coloring
24from git_command import GitCommand
25from git_config import GitConfig, IsId
26from gerrit_upload import UploadBundle
27from error import GitError, ImportError, UploadError
28from remote import Remote
29from codereview import proto_client
30
31HEAD = 'HEAD'
32R_HEADS = 'refs/heads/'
33R_TAGS = 'refs/tags/'
34R_PUB = 'refs/published/'
35R_M = 'refs/remotes/m/'
36
37def _warn(fmt, *args):
38 msg = fmt % args
39 print >>sys.stderr, 'warn: %s' % msg
40
41def _info(fmt, *args):
42 msg = fmt % args
43 print >>sys.stderr, 'info: %s' % msg
44
45def not_rev(r):
46 return '^' + r
47
Shawn O. Pearce632768b2008-10-23 11:58:52 -070048class DownloadedChange(object):
49 _commit_cache = None
50
51 def __init__(self, project, base, change_id, ps_id, commit):
52 self.project = project
53 self.base = base
54 self.change_id = change_id
55 self.ps_id = ps_id
56 self.commit = commit
57
58 @property
59 def commits(self):
60 if self._commit_cache is None:
61 self._commit_cache = self.project.bare_git.rev_list(
62 '--abbrev=8',
63 '--abbrev-commit',
64 '--pretty=oneline',
65 '--reverse',
66 '--date-order',
67 not_rev(self.base),
68 self.commit,
69 '--')
70 return self._commit_cache
71
72
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070073class ReviewableBranch(object):
74 _commit_cache = None
75
76 def __init__(self, project, branch, base):
77 self.project = project
78 self.branch = branch
79 self.base = base
80
81 @property
82 def name(self):
83 return self.branch.name
84
85 @property
86 def commits(self):
87 if self._commit_cache is None:
88 self._commit_cache = self.project.bare_git.rev_list(
89 '--abbrev=8',
90 '--abbrev-commit',
91 '--pretty=oneline',
92 '--reverse',
93 '--date-order',
94 not_rev(self.base),
95 R_HEADS + self.name,
96 '--')
97 return self._commit_cache
98
99 @property
100 def date(self):
101 return self.project.bare_git.log(
102 '--pretty=format:%cd',
103 '-n', '1',
104 R_HEADS + self.name,
105 '--')
106
107 def UploadForReview(self):
108 self.project.UploadForReview(self.name)
109
110 @property
111 def tip_url(self):
112 me = self.project.GetBranch(self.name)
113 commit = self.project.bare_git.rev_parse(R_HEADS + self.name)
114 return 'http://%s/r/%s' % (me.remote.review, commit[0:12])
115
Shawn O. Pearce0758d2f2008-10-22 13:13:40 -0700116 @property
117 def owner_email(self):
118 return self.project.UserEmail
119
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700120
121class StatusColoring(Coloring):
122 def __init__(self, config):
123 Coloring.__init__(self, config, 'status')
124 self.project = self.printer('header', attr = 'bold')
125 self.branch = self.printer('header', attr = 'bold')
126 self.nobranch = self.printer('nobranch', fg = 'red')
127
128 self.added = self.printer('added', fg = 'green')
129 self.changed = self.printer('changed', fg = 'red')
130 self.untracked = self.printer('untracked', fg = 'red')
131
132
133class DiffColoring(Coloring):
134 def __init__(self, config):
135 Coloring.__init__(self, config, 'diff')
136 self.project = self.printer('header', attr = 'bold')
137
138
139class _CopyFile:
140 def __init__(self, src, dest):
141 self.src = src
142 self.dest = dest
143
144 def _Copy(self):
145 src = self.src
146 dest = self.dest
147 # copy file if it does not exist or is out of date
148 if not os.path.exists(dest) or not filecmp.cmp(src, dest):
149 try:
150 # remove existing file first, since it might be read-only
151 if os.path.exists(dest):
152 os.remove(dest)
153 shutil.copy(src, dest)
154 # make the file read-only
155 mode = os.stat(dest)[stat.ST_MODE]
156 mode = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)
157 os.chmod(dest, mode)
158 except IOError:
159 print >>sys.stderr, \
160 'error: Cannot copy file %s to %s' \
161 % (src, dest)
162
163
164class Project(object):
165 def __init__(self,
166 manifest,
167 name,
168 remote,
169 gitdir,
170 worktree,
171 relpath,
172 revision):
173 self.manifest = manifest
174 self.name = name
175 self.remote = remote
176 self.gitdir = gitdir
177 self.worktree = worktree
178 self.relpath = relpath
179 self.revision = revision
180 self.snapshots = {}
181 self.extraRemotes = {}
182 self.copyfiles = []
183 self.config = GitConfig.ForRepository(
184 gitdir = self.gitdir,
185 defaults = self.manifest.globalConfig)
186
187 self.work_git = self._GitGetByExec(self, bare=False)
188 self.bare_git = self._GitGetByExec(self, bare=True)
189
190 @property
191 def Exists(self):
192 return os.path.isdir(self.gitdir)
193
194 @property
195 def CurrentBranch(self):
196 """Obtain the name of the currently checked out branch.
197 The branch name omits the 'refs/heads/' prefix.
198 None is returned if the project is on a detached HEAD.
199 """
200 try:
201 b = self.work_git.GetHead()
202 except GitError:
203 return None
204 if b.startswith(R_HEADS):
205 return b[len(R_HEADS):]
206 return None
207
208 def IsDirty(self, consider_untracked=True):
209 """Is the working directory modified in some way?
210 """
211 self.work_git.update_index('-q',
212 '--unmerged',
213 '--ignore-missing',
214 '--refresh')
215 if self.work_git.DiffZ('diff-index','-M','--cached',HEAD):
216 return True
217 if self.work_git.DiffZ('diff-files'):
218 return True
219 if consider_untracked and self.work_git.LsOthers():
220 return True
221 return False
222
223 _userident_name = None
224 _userident_email = None
225
226 @property
227 def UserName(self):
228 """Obtain the user's personal name.
229 """
230 if self._userident_name is None:
231 self._LoadUserIdentity()
232 return self._userident_name
233
234 @property
235 def UserEmail(self):
236 """Obtain the user's email address. This is very likely
237 to be their Gerrit login.
238 """
239 if self._userident_email is None:
240 self._LoadUserIdentity()
241 return self._userident_email
242
243 def _LoadUserIdentity(self):
244 u = self.bare_git.var('GIT_COMMITTER_IDENT')
245 m = re.compile("^(.*) <([^>]*)> ").match(u)
246 if m:
247 self._userident_name = m.group(1)
248 self._userident_email = m.group(2)
249 else:
250 self._userident_name = ''
251 self._userident_email = ''
252
253 def GetRemote(self, name):
254 """Get the configuration for a single remote.
255 """
256 return self.config.GetRemote(name)
257
258 def GetBranch(self, name):
259 """Get the configuration for a single branch.
260 """
261 return self.config.GetBranch(name)
262
263
264## Status Display ##
265
266 def PrintWorkTreeStatus(self):
267 """Prints the status of the repository to stdout.
268 """
269 if not os.path.isdir(self.worktree):
270 print ''
271 print 'project %s/' % self.relpath
272 print ' missing (run "repo sync")'
273 return
274
275 self.work_git.update_index('-q',
276 '--unmerged',
277 '--ignore-missing',
278 '--refresh')
279 di = self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD)
280 df = self.work_git.DiffZ('diff-files')
281 do = self.work_git.LsOthers()
282 if not di and not df and not do:
283 return
284
285 out = StatusColoring(self.config)
286 out.project('project %-40s', self.relpath + '/')
287
288 branch = self.CurrentBranch
289 if branch is None:
290 out.nobranch('(*** NO BRANCH ***)')
291 else:
292 out.branch('branch %s', branch)
293 out.nl()
294
295 paths = list()
296 paths.extend(di.keys())
297 paths.extend(df.keys())
298 paths.extend(do)
299
300 paths = list(set(paths))
301 paths.sort()
302
303 for p in paths:
304 try: i = di[p]
305 except KeyError: i = None
306
307 try: f = df[p]
308 except KeyError: f = None
309
310 if i: i_status = i.status.upper()
311 else: i_status = '-'
312
313 if f: f_status = f.status.lower()
314 else: f_status = '-'
315
316 if i and i.src_path:
317 line = ' %s%s\t%s => (%s%%)' % (i_status, f_status,
318 i.src_path, p, i.level)
319 else:
320 line = ' %s%s\t%s' % (i_status, f_status, p)
321
322 if i and not f:
323 out.added('%s', line)
324 elif (i and f) or (not i and f):
325 out.changed('%s', line)
326 elif not i and not f:
327 out.untracked('%s', line)
328 else:
329 out.write('%s', line)
330 out.nl()
331
332 def PrintWorkTreeDiff(self):
333 """Prints the status of the repository to stdout.
334 """
335 out = DiffColoring(self.config)
336 cmd = ['diff']
337 if out.is_on:
338 cmd.append('--color')
339 cmd.append(HEAD)
340 cmd.append('--')
341 p = GitCommand(self,
342 cmd,
343 capture_stdout = True,
344 capture_stderr = True)
345 has_diff = False
346 for line in p.process.stdout:
347 if not has_diff:
348 out.nl()
349 out.project('project %s/' % self.relpath)
350 out.nl()
351 has_diff = True
352 print line[:-1]
353 p.Wait()
354
355
356## Publish / Upload ##
357
358 def WasPublished(self, branch):
359 """Was the branch published (uploaded) for code review?
360 If so, returns the SHA-1 hash of the last published
361 state for the branch.
362 """
363 try:
364 return self.bare_git.rev_parse(R_PUB + branch)
365 except GitError:
366 return None
367
368 def CleanPublishedCache(self):
369 """Prunes any stale published refs.
370 """
371 heads = set()
372 canrm = {}
373 for name, id in self._allrefs.iteritems():
374 if name.startswith(R_HEADS):
375 heads.add(name)
376 elif name.startswith(R_PUB):
377 canrm[name] = id
378
379 for name, id in canrm.iteritems():
380 n = name[len(R_PUB):]
381 if R_HEADS + n not in heads:
382 self.bare_git.DeleteRef(name, id)
383
384 def GetUploadableBranches(self):
385 """List any branches which can be uploaded for review.
386 """
387 heads = {}
388 pubed = {}
389
390 for name, id in self._allrefs.iteritems():
391 if name.startswith(R_HEADS):
392 heads[name[len(R_HEADS):]] = id
393 elif name.startswith(R_PUB):
394 pubed[name[len(R_PUB):]] = id
395
396 ready = []
397 for branch, id in heads.iteritems():
398 if branch in pubed and pubed[branch] == id:
399 continue
400
401 branch = self.GetBranch(branch)
402 base = branch.LocalMerge
403 if branch.LocalMerge:
404 rb = ReviewableBranch(self, branch, base)
405 if rb.commits:
406 ready.append(rb)
407 return ready
408
409 def UploadForReview(self, branch=None):
410 """Uploads the named branch for code review.
411 """
412 if branch is None:
413 branch = self.CurrentBranch
414 if branch is None:
415 raise GitError('not currently on a branch')
416
417 branch = self.GetBranch(branch)
418 if not branch.LocalMerge:
419 raise GitError('branch %s does not track a remote' % branch.name)
420 if not branch.remote.review:
421 raise GitError('remote %s has no review url' % branch.remote.name)
422
423 dest_branch = branch.merge
424 if not dest_branch.startswith(R_HEADS):
425 dest_branch = R_HEADS + dest_branch
426
427 base_list = []
428 for name, id in self._allrefs.iteritems():
429 if branch.remote.WritesTo(name):
430 base_list.append(not_rev(name))
431 if not base_list:
432 raise GitError('no base refs, cannot upload %s' % branch.name)
433
434 print >>sys.stderr, ''
435 _info("Uploading %s to %s:", branch.name, self.name)
436 try:
437 UploadBundle(project = self,
438 server = branch.remote.review,
439 email = self.UserEmail,
440 dest_project = self.name,
441 dest_branch = dest_branch,
442 src_branch = R_HEADS + branch.name,
443 bases = base_list)
444 except proto_client.ClientLoginError:
445 raise UploadError('Login failure')
446 except urllib2.HTTPError, e:
447 raise UploadError('HTTP error %d' % e.code)
448
449 msg = "posted to %s for %s" % (branch.remote.review, dest_branch)
450 self.bare_git.UpdateRef(R_PUB + branch.name,
451 R_HEADS + branch.name,
452 message = msg)
453
454
455## Sync ##
456
457 def Sync_NetworkHalf(self):
458 """Perform only the network IO portion of the sync process.
459 Local working directory/branch state is not affected.
460 """
461 if not self.Exists:
462 print >>sys.stderr
463 print >>sys.stderr, 'Initializing project %s ...' % self.name
464 self._InitGitDir()
465 self._InitRemote()
466 for r in self.extraRemotes.values():
467 if not self._RemoteFetch(r.name):
468 return False
469 if not self._SnapshotDownload():
470 return False
471 if not self._RemoteFetch():
472 return False
Shawn O. Pearce329c31d2008-10-24 09:17:25 -0700473 self._RepairAndroidImportErrors()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700474 self._InitMRef()
475 return True
476
477 def _CopyFiles(self):
478 for file in self.copyfiles:
479 file._Copy()
480
Shawn O. Pearce329c31d2008-10-24 09:17:25 -0700481 def _RepairAndroidImportErrors(self):
482 if self.name in ['platform/external/iptables',
483 'platform/external/libpcap',
484 'platform/external/tcpdump',
485 'platform/external/webkit',
486 'platform/system/wlan/ti']:
487 # I hate myself for doing this...
488 #
489 # In the initial Android 1.0 release these projects were
490 # shipped, some users got them, and then the history had
491 # to be rewritten to correct problems with their imports.
492 # The 'android-1.0' tag may still be pointing at the old
493 # history, so we need to drop the tag and fetch it again.
494 #
495 try:
496 remote = self.GetRemote(self.remote.name)
497 relname = remote.ToLocal(R_HEADS + 'release-1.0')
498 tagname = R_TAGS + 'android-1.0'
499 if self._revlist(not_rev(relname), tagname):
500 cmd = ['fetch', remote.name, '+%s:%s' % (tagname, tagname)]
501 GitCommand(self, cmd, bare = True).Wait()
502 except GitError:
503 pass
504
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700505 def Sync_LocalHalf(self):
506 """Perform only the local IO portion of the sync process.
507 Network access is not required.
508
509 Return:
510 True: the sync was successful
511 False: the sync requires user input
512 """
513 self._InitWorkTree()
514 self.CleanPublishedCache()
515
516 rem = self.GetRemote(self.remote.name)
517 rev = rem.ToLocal(self.revision)
518 branch = self.CurrentBranch
519
520 if branch is None:
521 # Currently on a detached HEAD. The user is assumed to
522 # not have any local modifications worth worrying about.
523 #
524 lost = self._revlist(not_rev(rev), HEAD)
525 if lost:
526 _info("[%s] Discarding %d commits", self.name, len(lost))
527 try:
528 self._Checkout(rev, quiet=True)
529 except GitError:
530 return False
531 self._CopyFiles()
532 return True
533
534 branch = self.GetBranch(branch)
535 merge = branch.LocalMerge
536
537 if not merge:
538 # The current branch has no tracking configuration.
539 # Jump off it to a deatched HEAD.
540 #
541 _info("[%s] Leaving %s"
542 " (does not track any upstream)",
543 self.name,
544 branch.name)
545 try:
546 self._Checkout(rev, quiet=True)
547 except GitError:
548 return False
549 self._CopyFiles()
550 return True
551
552 upstream_gain = self._revlist(not_rev(HEAD), rev)
553 pub = self.WasPublished(branch.name)
554 if pub:
555 not_merged = self._revlist(not_rev(rev), pub)
556 if not_merged:
557 if upstream_gain:
558 # The user has published this branch and some of those
559 # commits are not yet merged upstream. We do not want
560 # to rewrite the published commits so we punt.
561 #
562 _info("[%s] Branch %s is published,"
563 " but is now %d commits behind.",
564 self.name, branch.name, len(upstream_gain))
565 _info("[%s] Consider merging or rebasing the"
566 " unpublished commits.", self.name)
567 return True
568
569 if merge == rev:
570 try:
571 old_merge = self.bare_git.rev_parse('%s@{1}' % merge)
572 except GitError:
573 old_merge = merge
Shawn O. Pearce07346002008-10-21 07:09:27 -0700574 if old_merge == '0000000000000000000000000000000000000000' \
575 or old_merge == '':
576 old_merge = merge
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700577 else:
578 # The upstream switched on us. Time to cross our fingers
579 # and pray that the old upstream also wasn't in the habit
580 # of rebasing itself.
581 #
582 _info("[%s] Manifest switched from %s to %s",
583 self.name, merge, rev)
584 old_merge = merge
585
586 if rev == old_merge:
587 upstream_lost = []
588 else:
589 upstream_lost = self._revlist(not_rev(rev), old_merge)
590
591 if not upstream_lost and not upstream_gain:
592 # Trivially no changes caused by the upstream.
593 #
594 return True
595
596 if self.IsDirty(consider_untracked=False):
597 _warn('[%s] commit (or discard) uncommitted changes'
598 ' before sync', self.name)
599 return False
600
601 if upstream_lost:
602 # Upstream rebased. Not everything in HEAD
603 # may have been caused by the user.
604 #
605 _info("[%s] Discarding %d commits removed from upstream",
606 self.name, len(upstream_lost))
607
608 branch.remote = rem
609 branch.merge = self.revision
610 branch.Save()
611
612 my_changes = self._revlist(not_rev(old_merge), HEAD)
613 if my_changes:
614 try:
615 self._Rebase(upstream = old_merge, onto = rev)
616 except GitError:
617 return False
618 elif upstream_lost:
619 try:
620 self._ResetHard(rev)
621 except GitError:
622 return False
623 else:
624 try:
625 self._FastForward(rev)
626 except GitError:
627 return False
628
629 self._CopyFiles()
630 return True
631
632 def _SnapshotDownload(self):
633 if self.snapshots:
634 have = set(self._allrefs.keys())
635 need = []
636
637 for tag, sn in self.snapshots.iteritems():
638 if tag not in have:
639 need.append(sn)
640
641 if need:
642 print >>sys.stderr, """
643 *** Downloading source(s) from a mirror site. ***
644 *** If the network hangs, kill and restart repo. ***
645"""
646 for sn in need:
647 try:
648 sn.Import()
649 except ImportError, e:
650 print >>sys.stderr, \
651 'error: Cannot import %s: %s' \
652 % (self.name, e)
653 return False
654 cmd = ['repack', '-a', '-d', '-f', '-l']
655 if GitCommand(self, cmd, bare = True).Wait() != 0:
656 return False
657 return True
658
659 def AddCopyFile(self, src, dest):
660 # dest should already be an absolute path, but src is project relative
661 # make src an absolute path
662 src = os.path.join(self.worktree, src)
663 self.copyfiles.append(_CopyFile(src, dest))
664
Shawn O. Pearce632768b2008-10-23 11:58:52 -0700665 def DownloadPatchSet(self, change_id, patch_id):
666 """Download a single patch set of a single change to FETCH_HEAD.
667 """
668 remote = self.GetRemote(self.remote.name)
669
670 cmd = ['fetch', remote.name]
671 cmd.append('refs/changes/%2.2d/%d/%d' \
672 % (change_id % 100, change_id, patch_id))
673 cmd.extend(map(lambda x: str(x), remote.fetch))
674 if GitCommand(self, cmd, bare=True).Wait() != 0:
675 return None
676 return DownloadedChange(self,
677 remote.ToLocal(self.revision),
678 change_id,
679 patch_id,
680 self.bare_git.rev_parse('FETCH_HEAD'))
681
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700682
683## Branch Management ##
684
685 def StartBranch(self, name):
686 """Create a new branch off the manifest's revision.
687 """
688 branch = self.GetBranch(name)
689 branch.remote = self.GetRemote(self.remote.name)
690 branch.merge = self.revision
691
692 rev = branch.LocalMerge
693 cmd = ['checkout', '-b', branch.name, rev]
694 if GitCommand(self, cmd).Wait() == 0:
695 branch.Save()
696 else:
697 raise GitError('%s checkout %s ' % (self.name, rev))
698
699 def PruneHeads(self):
700 """Prune any topic branches already merged into upstream.
701 """
702 cb = self.CurrentBranch
703 kill = []
704 for name in self._allrefs.keys():
705 if name.startswith(R_HEADS):
706 name = name[len(R_HEADS):]
707 if cb is None or name != cb:
708 kill.append(name)
709
710 rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
711 if cb is not None \
712 and not self._revlist(HEAD + '...' + rev) \
713 and not self.IsDirty(consider_untracked = False):
714 self.work_git.DetachHead(HEAD)
715 kill.append(cb)
716
717 deleted = set()
718 if kill:
719 try:
720 old = self.bare_git.GetHead()
721 except GitError:
722 old = 'refs/heads/please_never_use_this_as_a_branch_name'
723
724 rm_re = re.compile(r"^Deleted branch (.*)\.$")
725 try:
726 self.bare_git.DetachHead(rev)
727
728 b = ['branch', '-d']
729 b.extend(kill)
730 b = GitCommand(self, b, bare=True,
731 capture_stdout=True,
732 capture_stderr=True)
733 b.Wait()
734 finally:
735 self.bare_git.SetHead(old)
736
737 for line in b.stdout.split("\n"):
738 m = rm_re.match(line)
739 if m:
740 deleted.add(m.group(1))
741
742 if deleted:
743 self.CleanPublishedCache()
744
745 if cb and cb not in kill:
746 kill.append(cb)
747 kill.sort()
748
749 kept = []
750 for branch in kill:
751 if branch not in deleted:
752 branch = self.GetBranch(branch)
753 base = branch.LocalMerge
754 if not base:
755 base = rev
756 kept.append(ReviewableBranch(self, branch, base))
757 return kept
758
759
760## Direct Git Commands ##
761
762 def _RemoteFetch(self, name=None):
763 if not name:
764 name = self.remote.name
Shawn O. Pearcece03a402008-10-28 16:12:03 -0700765 return GitCommand(self,
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700766 ['fetch', name],
Shawn O. Pearcece03a402008-10-28 16:12:03 -0700767 bare = True).Wait() == 0
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700768
769 def _Checkout(self, rev, quiet=False):
770 cmd = ['checkout']
771 if quiet:
772 cmd.append('-q')
773 cmd.append(rev)
774 cmd.append('--')
775 if GitCommand(self, cmd).Wait() != 0:
776 if self._allrefs:
777 raise GitError('%s checkout %s ' % (self.name, rev))
778
779 def _ResetHard(self, rev, quiet=True):
780 cmd = ['reset', '--hard']
781 if quiet:
782 cmd.append('-q')
783 cmd.append(rev)
784 if GitCommand(self, cmd).Wait() != 0:
785 raise GitError('%s reset --hard %s ' % (self.name, rev))
786
787 def _Rebase(self, upstream, onto = None):
788 cmd = ['rebase', '-i']
789 if onto is not None:
790 cmd.extend(['--onto', onto])
791 cmd.append(upstream)
792 if GitCommand(self, cmd, disable_editor=True).Wait() != 0:
793 raise GitError('%s rebase %s ' % (self.name, upstream))
794
795 def _FastForward(self, head):
796 cmd = ['merge', head]
797 if GitCommand(self, cmd).Wait() != 0:
798 raise GitError('%s merge %s ' % (self.name, head))
799
800 def _InitGitDir(self):
801 if not os.path.exists(self.gitdir):
802 os.makedirs(self.gitdir)
803 self.bare_git.init()
804 self.config.SetString('core.bare', None)
805
806 hooks = self._gitdir_path('hooks')
807 for old_hook in os.listdir(hooks):
808 os.remove(os.path.join(hooks, old_hook))
809
810 # TODO(sop) install custom repo hooks
811
812 m = self.manifest.manifestProject.config
813 for key in ['user.name', 'user.email']:
814 if m.Has(key, include_defaults = False):
815 self.config.SetString(key, m.GetString(key))
816
817 def _InitRemote(self):
818 if self.remote.fetchUrl:
819 remote = self.GetRemote(self.remote.name)
820
821 url = self.remote.fetchUrl
822 while url.endswith('/'):
823 url = url[:-1]
824 url += '/%s.git' % self.name
825 remote.url = url
826 remote.review = self.remote.reviewUrl
827
828 remote.ResetFetch()
829 remote.Save()
830
831 for r in self.extraRemotes.values():
832 remote = self.GetRemote(r.name)
833 remote.url = r.fetchUrl
834 remote.review = r.reviewUrl
835 remote.ResetFetch()
836 remote.Save()
837
838 def _InitMRef(self):
839 if self.manifest.branch:
840 msg = 'manifest set to %s' % self.revision
841 ref = R_M + self.manifest.branch
842
843 if IsId(self.revision):
844 dst = self.revision + '^0',
845 self.bare_git.UpdateRef(ref, dst, message = msg, detach = True)
846 else:
847 remote = self.GetRemote(self.remote.name)
848 dst = remote.ToLocal(self.revision)
849 self.bare_git.symbolic_ref('-m', msg, ref, dst)
850
851 def _InitWorkTree(self):
852 dotgit = os.path.join(self.worktree, '.git')
853 if not os.path.exists(dotgit):
854 os.makedirs(dotgit)
855
856 topdir = os.path.commonprefix([self.gitdir, dotgit])
857 if topdir.endswith('/'):
858 topdir = topdir[:-1]
859 else:
860 topdir = os.path.dirname(topdir)
861
862 tmpdir = dotgit
863 relgit = ''
864 while topdir != tmpdir:
865 relgit += '../'
866 tmpdir = os.path.dirname(tmpdir)
867 relgit += self.gitdir[len(topdir) + 1:]
868
869 for name in ['config',
870 'description',
871 'hooks',
872 'info',
873 'logs',
874 'objects',
875 'packed-refs',
876 'refs',
877 'rr-cache',
878 'svn']:
879 os.symlink(os.path.join(relgit, name),
880 os.path.join(dotgit, name))
881
882 rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
883 rev = self.bare_git.rev_parse('%s^0' % rev)
884
885 f = open(os.path.join(dotgit, HEAD), 'wb')
886 f.write("%s\n" % rev)
887 f.close()
888
889 cmd = ['read-tree', '--reset', '-u']
890 cmd.append('-v')
891 cmd.append('HEAD')
892 if GitCommand(self, cmd).Wait() != 0:
893 raise GitError("cannot initialize work tree")
894
895 def _gitdir_path(self, path):
896 return os.path.join(self.gitdir, path)
897
898 def _revlist(self, *args):
899 cmd = []
900 cmd.extend(args)
901 cmd.append('--')
902 return self.work_git.rev_list(*args)
903
904 @property
905 def _allrefs(self):
906 return self.bare_git.ListRefs()
907
908 class _GitGetByExec(object):
909 def __init__(self, project, bare):
910 self._project = project
911 self._bare = bare
912
913 def ListRefs(self, *args):
914 cmdv = ['for-each-ref', '--format=%(objectname) %(refname)']
915 cmdv.extend(args)
916 p = GitCommand(self._project,
917 cmdv,
918 bare = self._bare,
919 capture_stdout = True,
920 capture_stderr = True)
921 r = {}
922 for line in p.process.stdout:
923 id, name = line[:-1].split(' ', 2)
924 r[name] = id
925 if p.Wait() != 0:
926 raise GitError('%s for-each-ref %s: %s' % (
927 self._project.name,
928 str(args),
929 p.stderr))
930 return r
931
932 def LsOthers(self):
933 p = GitCommand(self._project,
934 ['ls-files',
935 '-z',
936 '--others',
937 '--exclude-standard'],
938 bare = False,
939 capture_stdout = True,
940 capture_stderr = True)
941 if p.Wait() == 0:
942 out = p.stdout
943 if out:
944 return out[:-1].split("\0")
945 return []
946
947 def DiffZ(self, name, *args):
948 cmd = [name]
949 cmd.append('-z')
950 cmd.extend(args)
951 p = GitCommand(self._project,
952 cmd,
953 bare = False,
954 capture_stdout = True,
955 capture_stderr = True)
956 try:
957 out = p.process.stdout.read()
958 r = {}
959 if out:
960 out = iter(out[:-1].split('\0'))
961 while out:
Shawn O. Pearce02dbb6d2008-10-21 13:59:08 -0700962 try:
963 info = out.next()
964 path = out.next()
965 except StopIteration:
966 break
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700967
968 class _Info(object):
969 def __init__(self, path, omode, nmode, oid, nid, state):
970 self.path = path
971 self.src_path = None
972 self.old_mode = omode
973 self.new_mode = nmode
974 self.old_id = oid
975 self.new_id = nid
976
977 if len(state) == 1:
978 self.status = state
979 self.level = None
980 else:
981 self.status = state[:1]
982 self.level = state[1:]
983 while self.level.startswith('0'):
984 self.level = self.level[1:]
985
986 info = info[1:].split(' ')
987 info =_Info(path, *info)
988 if info.status in ('R', 'C'):
989 info.src_path = info.path
990 info.path = out.next()
991 r[info.path] = info
992 return r
993 finally:
994 p.Wait()
995
996 def GetHead(self):
997 return self.symbolic_ref(HEAD)
998
999 def SetHead(self, ref, message=None):
1000 cmdv = []
1001 if message is not None:
1002 cmdv.extend(['-m', message])
1003 cmdv.append(HEAD)
1004 cmdv.append(ref)
1005 self.symbolic_ref(*cmdv)
1006
1007 def DetachHead(self, new, message=None):
1008 cmdv = ['--no-deref']
1009 if message is not None:
1010 cmdv.extend(['-m', message])
1011 cmdv.append(HEAD)
1012 cmdv.append(new)
1013 self.update_ref(*cmdv)
1014
1015 def UpdateRef(self, name, new, old=None,
1016 message=None,
1017 detach=False):
1018 cmdv = []
1019 if message is not None:
1020 cmdv.extend(['-m', message])
1021 if detach:
1022 cmdv.append('--no-deref')
1023 cmdv.append(name)
1024 cmdv.append(new)
1025 if old is not None:
1026 cmdv.append(old)
1027 self.update_ref(*cmdv)
1028
1029 def DeleteRef(self, name, old=None):
1030 if not old:
1031 old = self.rev_parse(name)
1032 self.update_ref('-d', name, old)
1033
1034 def rev_list(self, *args):
1035 cmdv = ['rev-list']
1036 cmdv.extend(args)
1037 p = GitCommand(self._project,
1038 cmdv,
1039 bare = self._bare,
1040 capture_stdout = True,
1041 capture_stderr = True)
1042 r = []
1043 for line in p.process.stdout:
1044 r.append(line[:-1])
1045 if p.Wait() != 0:
1046 raise GitError('%s rev-list %s: %s' % (
1047 self._project.name,
1048 str(args),
1049 p.stderr))
1050 return r
1051
1052 def __getattr__(self, name):
1053 name = name.replace('_', '-')
1054 def runner(*args):
1055 cmdv = [name]
1056 cmdv.extend(args)
1057 p = GitCommand(self._project,
1058 cmdv,
1059 bare = self._bare,
1060 capture_stdout = True,
1061 capture_stderr = True)
1062 if p.Wait() != 0:
1063 raise GitError('%s %s: %s' % (
1064 self._project.name,
1065 name,
1066 p.stderr))
1067 r = p.stdout
1068 if r.endswith('\n') and r.index('\n') == len(r) - 1:
1069 return r[:-1]
1070 return r
1071 return runner
1072
1073
1074class MetaProject(Project):
1075 """A special project housed under .repo.
1076 """
1077 def __init__(self, manifest, name, gitdir, worktree):
1078 repodir = manifest.repodir
1079 Project.__init__(self,
1080 manifest = manifest,
1081 name = name,
1082 gitdir = gitdir,
1083 worktree = worktree,
1084 remote = Remote('origin'),
1085 relpath = '.repo/%s' % name,
1086 revision = 'refs/heads/master')
1087
1088 def PreSync(self):
1089 if self.Exists:
1090 cb = self.CurrentBranch
1091 if cb:
1092 base = self.GetBranch(cb).merge
1093 if base:
1094 self.revision = base
1095
1096 @property
1097 def HasChanges(self):
1098 """Has the remote received new commits not yet checked out?
1099 """
1100 rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
1101 if self._revlist(not_rev(HEAD), rev):
1102 return True
1103 return False