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