project.py 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478
  1. # 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. import errno
  15. import filecmp
  16. import os
  17. import re
  18. import shutil
  19. import stat
  20. import sys
  21. import urllib2
  22. from color import Coloring
  23. from git_command import GitCommand
  24. from git_config import GitConfig, IsId
  25. from error import GitError, ImportError, UploadError
  26. from error import ManifestInvalidRevisionError
  27. from remote import Remote
  28. from git_refs import GitRefs, HEAD, R_HEADS, R_TAGS, R_PUB, R_M
  29. def _lwrite(path, content):
  30. lock = '%s.lock' % path
  31. fd = open(lock, 'wb')
  32. try:
  33. fd.write(content)
  34. finally:
  35. fd.close()
  36. try:
  37. os.rename(lock, path)
  38. except OSError:
  39. os.remove(lock)
  40. raise
  41. def _error(fmt, *args):
  42. msg = fmt % args
  43. print >>sys.stderr, 'error: %s' % msg
  44. def not_rev(r):
  45. return '^' + r
  46. def sq(r):
  47. return "'" + r.replace("'", "'\''") + "'"
  48. hook_list = None
  49. def repo_hooks():
  50. global hook_list
  51. if hook_list is None:
  52. d = os.path.abspath(os.path.dirname(__file__))
  53. d = os.path.join(d , 'hooks')
  54. hook_list = map(lambda x: os.path.join(d, x), os.listdir(d))
  55. return hook_list
  56. def relpath(dst, src):
  57. src = os.path.dirname(src)
  58. top = os.path.commonprefix([dst, src])
  59. if top.endswith('/'):
  60. top = top[:-1]
  61. else:
  62. top = os.path.dirname(top)
  63. tmp = src
  64. rel = ''
  65. while top != tmp:
  66. rel += '../'
  67. tmp = os.path.dirname(tmp)
  68. return rel + dst[len(top) + 1:]
  69. class DownloadedChange(object):
  70. _commit_cache = None
  71. def __init__(self, project, base, change_id, ps_id, commit):
  72. self.project = project
  73. self.base = base
  74. self.change_id = change_id
  75. self.ps_id = ps_id
  76. self.commit = commit
  77. @property
  78. def commits(self):
  79. if self._commit_cache is None:
  80. self._commit_cache = self.project.bare_git.rev_list(
  81. '--abbrev=8',
  82. '--abbrev-commit',
  83. '--pretty=oneline',
  84. '--reverse',
  85. '--date-order',
  86. not_rev(self.base),
  87. self.commit,
  88. '--')
  89. return self._commit_cache
  90. class ReviewableBranch(object):
  91. _commit_cache = None
  92. def __init__(self, project, branch, base):
  93. self.project = project
  94. self.branch = branch
  95. self.base = base
  96. self.replace_changes = None
  97. @property
  98. def name(self):
  99. return self.branch.name
  100. @property
  101. def commits(self):
  102. if self._commit_cache is None:
  103. self._commit_cache = self.project.bare_git.rev_list(
  104. '--abbrev=8',
  105. '--abbrev-commit',
  106. '--pretty=oneline',
  107. '--reverse',
  108. '--date-order',
  109. not_rev(self.base),
  110. R_HEADS + self.name,
  111. '--')
  112. return self._commit_cache
  113. @property
  114. def unabbrev_commits(self):
  115. r = dict()
  116. for commit in self.project.bare_git.rev_list(
  117. not_rev(self.base),
  118. R_HEADS + self.name,
  119. '--'):
  120. r[commit[0:8]] = commit
  121. return r
  122. @property
  123. def date(self):
  124. return self.project.bare_git.log(
  125. '--pretty=format:%cd',
  126. '-n', '1',
  127. R_HEADS + self.name,
  128. '--')
  129. def UploadForReview(self, people):
  130. self.project.UploadForReview(self.name,
  131. self.replace_changes,
  132. people)
  133. class StatusColoring(Coloring):
  134. def __init__(self, config):
  135. Coloring.__init__(self, config, 'status')
  136. self.project = self.printer('header', attr = 'bold')
  137. self.branch = self.printer('header', attr = 'bold')
  138. self.nobranch = self.printer('nobranch', fg = 'red')
  139. self.important = self.printer('important', fg = 'red')
  140. self.added = self.printer('added', fg = 'green')
  141. self.changed = self.printer('changed', fg = 'red')
  142. self.untracked = self.printer('untracked', fg = 'red')
  143. class DiffColoring(Coloring):
  144. def __init__(self, config):
  145. Coloring.__init__(self, config, 'diff')
  146. self.project = self.printer('header', attr = 'bold')
  147. class _CopyFile:
  148. def __init__(self, src, dest, abssrc, absdest):
  149. self.src = src
  150. self.dest = dest
  151. self.abs_src = abssrc
  152. self.abs_dest = absdest
  153. def _Copy(self):
  154. src = self.abs_src
  155. dest = self.abs_dest
  156. # copy file if it does not exist or is out of date
  157. if not os.path.exists(dest) or not filecmp.cmp(src, dest):
  158. try:
  159. # remove existing file first, since it might be read-only
  160. if os.path.exists(dest):
  161. os.remove(dest)
  162. shutil.copy(src, dest)
  163. # make the file read-only
  164. mode = os.stat(dest)[stat.ST_MODE]
  165. mode = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)
  166. os.chmod(dest, mode)
  167. except IOError:
  168. _error('Cannot copy file %s to %s', src, dest)
  169. class Project(object):
  170. def __init__(self,
  171. manifest,
  172. name,
  173. remote,
  174. gitdir,
  175. worktree,
  176. relpath,
  177. revision):
  178. self.manifest = manifest
  179. self.name = name
  180. self.remote = remote
  181. self.gitdir = gitdir
  182. self.worktree = worktree
  183. self.relpath = relpath
  184. self.revision = revision
  185. self.snapshots = {}
  186. self.extraRemotes = {}
  187. self.copyfiles = []
  188. self.config = GitConfig.ForRepository(
  189. gitdir = self.gitdir,
  190. defaults = self.manifest.globalConfig)
  191. if self.worktree:
  192. self.work_git = self._GitGetByExec(self, bare=False)
  193. else:
  194. self.work_git = None
  195. self.bare_git = self._GitGetByExec(self, bare=True)
  196. self.bare_ref = GitRefs(gitdir)
  197. @property
  198. def Exists(self):
  199. return os.path.isdir(self.gitdir)
  200. @property
  201. def CurrentBranch(self):
  202. """Obtain the name of the currently checked out branch.
  203. The branch name omits the 'refs/heads/' prefix.
  204. None is returned if the project is on a detached HEAD.
  205. """
  206. b = self.work_git.GetHead()
  207. if b.startswith(R_HEADS):
  208. return b[len(R_HEADS):]
  209. return None
  210. def IsRebaseInProgress(self):
  211. w = self.worktree
  212. g = os.path.join(w, '.git')
  213. return os.path.exists(os.path.join(g, 'rebase-apply')) \
  214. or os.path.exists(os.path.join(g, 'rebase-merge')) \
  215. or os.path.exists(os.path.join(w, '.dotest'))
  216. def IsDirty(self, consider_untracked=True):
  217. """Is the working directory modified in some way?
  218. """
  219. self.work_git.update_index('-q',
  220. '--unmerged',
  221. '--ignore-missing',
  222. '--refresh')
  223. if self.work_git.DiffZ('diff-index','-M','--cached',HEAD):
  224. return True
  225. if self.work_git.DiffZ('diff-files'):
  226. return True
  227. if consider_untracked and self.work_git.LsOthers():
  228. return True
  229. return False
  230. _userident_name = None
  231. _userident_email = None
  232. @property
  233. def UserName(self):
  234. """Obtain the user's personal name.
  235. """
  236. if self._userident_name is None:
  237. self._LoadUserIdentity()
  238. return self._userident_name
  239. @property
  240. def UserEmail(self):
  241. """Obtain the user's email address. This is very likely
  242. to be their Gerrit login.
  243. """
  244. if self._userident_email is None:
  245. self._LoadUserIdentity()
  246. return self._userident_email
  247. def _LoadUserIdentity(self):
  248. u = self.bare_git.var('GIT_COMMITTER_IDENT')
  249. m = re.compile("^(.*) <([^>]*)> ").match(u)
  250. if m:
  251. self._userident_name = m.group(1)
  252. self._userident_email = m.group(2)
  253. else:
  254. self._userident_name = ''
  255. self._userident_email = ''
  256. def GetRemote(self, name):
  257. """Get the configuration for a single remote.
  258. """
  259. return self.config.GetRemote(name)
  260. def GetBranch(self, name):
  261. """Get the configuration for a single branch.
  262. """
  263. return self.config.GetBranch(name)
  264. def GetBranches(self):
  265. """Get all existing local branches.
  266. """
  267. current = self.CurrentBranch
  268. all = self._allrefs
  269. heads = {}
  270. pubd = {}
  271. for name, id in all.iteritems():
  272. if name.startswith(R_HEADS):
  273. name = name[len(R_HEADS):]
  274. b = self.GetBranch(name)
  275. b.current = name == current
  276. b.published = None
  277. b.revision = id
  278. heads[name] = b
  279. for name, id in all.iteritems():
  280. if name.startswith(R_PUB):
  281. name = name[len(R_PUB):]
  282. b = heads.get(name)
  283. if b:
  284. b.published = id
  285. return heads
  286. ## Status Display ##
  287. def PrintWorkTreeStatus(self):
  288. """Prints the status of the repository to stdout.
  289. """
  290. if not os.path.isdir(self.worktree):
  291. print ''
  292. print 'project %s/' % self.relpath
  293. print ' missing (run "repo sync")'
  294. return
  295. self.work_git.update_index('-q',
  296. '--unmerged',
  297. '--ignore-missing',
  298. '--refresh')
  299. rb = self.IsRebaseInProgress()
  300. di = self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD)
  301. df = self.work_git.DiffZ('diff-files')
  302. do = self.work_git.LsOthers()
  303. if not rb and not di and not df and not do:
  304. return 'CLEAN'
  305. out = StatusColoring(self.config)
  306. out.project('project %-40s', self.relpath + '/')
  307. branch = self.CurrentBranch
  308. if branch is None:
  309. out.nobranch('(*** NO BRANCH ***)')
  310. else:
  311. out.branch('branch %s', branch)
  312. out.nl()
  313. if rb:
  314. out.important('prior sync failed; rebase still in progress')
  315. out.nl()
  316. paths = list()
  317. paths.extend(di.keys())
  318. paths.extend(df.keys())
  319. paths.extend(do)
  320. paths = list(set(paths))
  321. paths.sort()
  322. for p in paths:
  323. try: i = di[p]
  324. except KeyError: i = None
  325. try: f = df[p]
  326. except KeyError: f = None
  327. if i: i_status = i.status.upper()
  328. else: i_status = '-'
  329. if f: f_status = f.status.lower()
  330. else: f_status = '-'
  331. if i and i.src_path:
  332. line = ' %s%s\t%s => %s (%s%%)' % (i_status, f_status,
  333. i.src_path, p, i.level)
  334. else:
  335. line = ' %s%s\t%s' % (i_status, f_status, p)
  336. if i and not f:
  337. out.added('%s', line)
  338. elif (i and f) or (not i and f):
  339. out.changed('%s', line)
  340. elif not i and not f:
  341. out.untracked('%s', line)
  342. else:
  343. out.write('%s', line)
  344. out.nl()
  345. return 'DIRTY'
  346. def PrintWorkTreeDiff(self):
  347. """Prints the status of the repository to stdout.
  348. """
  349. out = DiffColoring(self.config)
  350. cmd = ['diff']
  351. if out.is_on:
  352. cmd.append('--color')
  353. cmd.append(HEAD)
  354. cmd.append('--')
  355. p = GitCommand(self,
  356. cmd,
  357. capture_stdout = True,
  358. capture_stderr = True)
  359. has_diff = False
  360. for line in p.process.stdout:
  361. if not has_diff:
  362. out.nl()
  363. out.project('project %s/' % self.relpath)
  364. out.nl()
  365. has_diff = True
  366. print line[:-1]
  367. p.Wait()
  368. ## Publish / Upload ##
  369. def WasPublished(self, branch, all=None):
  370. """Was the branch published (uploaded) for code review?
  371. If so, returns the SHA-1 hash of the last published
  372. state for the branch.
  373. """
  374. key = R_PUB + branch
  375. if all is None:
  376. try:
  377. return self.bare_git.rev_parse(key)
  378. except GitError:
  379. return None
  380. else:
  381. try:
  382. return all[key]
  383. except KeyError:
  384. return None
  385. def CleanPublishedCache(self, all=None):
  386. """Prunes any stale published refs.
  387. """
  388. if all is None:
  389. all = self._allrefs
  390. heads = set()
  391. canrm = {}
  392. for name, id in all.iteritems():
  393. if name.startswith(R_HEADS):
  394. heads.add(name)
  395. elif name.startswith(R_PUB):
  396. canrm[name] = id
  397. for name, id in canrm.iteritems():
  398. n = name[len(R_PUB):]
  399. if R_HEADS + n not in heads:
  400. self.bare_git.DeleteRef(name, id)
  401. def GetUploadableBranches(self):
  402. """List any branches which can be uploaded for review.
  403. """
  404. heads = {}
  405. pubed = {}
  406. for name, id in self._allrefs.iteritems():
  407. if name.startswith(R_HEADS):
  408. heads[name[len(R_HEADS):]] = id
  409. elif name.startswith(R_PUB):
  410. pubed[name[len(R_PUB):]] = id
  411. ready = []
  412. for branch, id in heads.iteritems():
  413. if branch in pubed and pubed[branch] == id:
  414. continue
  415. rb = self.GetUploadableBranch(branch)
  416. if rb:
  417. ready.append(rb)
  418. return ready
  419. def GetUploadableBranch(self, branch_name):
  420. """Get a single uploadable branch, or None.
  421. """
  422. branch = self.GetBranch(branch_name)
  423. base = branch.LocalMerge
  424. if branch.LocalMerge:
  425. rb = ReviewableBranch(self, branch, base)
  426. if rb.commits:
  427. return rb
  428. return None
  429. def UploadForReview(self, branch=None, replace_changes=None, people=([],[])):
  430. """Uploads the named branch for code review.
  431. """
  432. if branch is None:
  433. branch = self.CurrentBranch
  434. if branch is None:
  435. raise GitError('not currently on a branch')
  436. branch = self.GetBranch(branch)
  437. if not branch.LocalMerge:
  438. raise GitError('branch %s does not track a remote' % branch.name)
  439. if not branch.remote.review:
  440. raise GitError('remote %s has no review url' % branch.remote.name)
  441. dest_branch = branch.merge
  442. if not dest_branch.startswith(R_HEADS):
  443. dest_branch = R_HEADS + dest_branch
  444. if not branch.remote.projectname:
  445. branch.remote.projectname = self.name
  446. branch.remote.Save()
  447. if branch.remote.ReviewProtocol == 'ssh':
  448. if dest_branch.startswith(R_HEADS):
  449. dest_branch = dest_branch[len(R_HEADS):]
  450. rp = ['gerrit receive-pack']
  451. for e in people[0]:
  452. rp.append('--reviewer=%s' % sq(e))
  453. for e in people[1]:
  454. rp.append('--cc=%s' % sq(e))
  455. cmd = ['push']
  456. cmd.append('--receive-pack=%s' % " ".join(rp))
  457. cmd.append(branch.remote.SshReviewUrl(self.UserEmail))
  458. cmd.append('%s:refs/for/%s' % (R_HEADS + branch.name, dest_branch))
  459. if replace_changes:
  460. for change_id,commit_id in replace_changes.iteritems():
  461. cmd.append('%s:refs/changes/%s/new' % (commit_id, change_id))
  462. if GitCommand(self, cmd, bare = True).Wait() != 0:
  463. raise UploadError('Upload failed')
  464. else:
  465. raise UploadError('Unsupported protocol %s' \
  466. % branch.remote.review)
  467. msg = "posted to %s for %s" % (branch.remote.review, dest_branch)
  468. self.bare_git.UpdateRef(R_PUB + branch.name,
  469. R_HEADS + branch.name,
  470. message = msg)
  471. ## Sync ##
  472. def Sync_NetworkHalf(self):
  473. """Perform only the network IO portion of the sync process.
  474. Local working directory/branch state is not affected.
  475. """
  476. if not self.Exists:
  477. print >>sys.stderr
  478. print >>sys.stderr, 'Initializing project %s ...' % self.name
  479. self._InitGitDir()
  480. self._InitRemote()
  481. for r in self.extraRemotes.values():
  482. if not self._RemoteFetch(r.name):
  483. return False
  484. if not self._RemoteFetch():
  485. return False
  486. if self.worktree:
  487. self._InitMRef()
  488. else:
  489. self._InitMirrorHead()
  490. try:
  491. os.remove(os.path.join(self.gitdir, 'FETCH_HEAD'))
  492. except OSError:
  493. pass
  494. return True
  495. def PostRepoUpgrade(self):
  496. self._InitHooks()
  497. def _CopyFiles(self):
  498. for file in self.copyfiles:
  499. file._Copy()
  500. def Sync_LocalHalf(self, syncbuf):
  501. """Perform only the local IO portion of the sync process.
  502. Network access is not required.
  503. """
  504. self._InitWorkTree()
  505. all = self.bare_ref.all
  506. self.CleanPublishedCache(all)
  507. rem = self.GetRemote(self.remote.name)
  508. rev = rem.ToLocal(self.revision)
  509. if rev in all:
  510. revid = all[rev]
  511. elif IsId(rev):
  512. revid = rev
  513. else:
  514. try:
  515. revid = self.bare_git.rev_parse('--verify', '%s^0' % rev)
  516. except GitError:
  517. raise ManifestInvalidRevisionError(
  518. 'revision %s in %s not found' % (self.revision, self.name))
  519. head = self.work_git.GetHead()
  520. if head.startswith(R_HEADS):
  521. branch = head[len(R_HEADS):]
  522. try:
  523. head = all[head]
  524. except KeyError:
  525. head = None
  526. else:
  527. branch = None
  528. if branch is None or syncbuf.detach_head:
  529. # Currently on a detached HEAD. The user is assumed to
  530. # not have any local modifications worth worrying about.
  531. #
  532. if self.IsRebaseInProgress():
  533. syncbuf.fail(self, _PriorSyncFailedError())
  534. return
  535. if head == revid:
  536. # No changes; don't do anything further.
  537. #
  538. return
  539. lost = self._revlist(not_rev(rev), HEAD)
  540. if lost:
  541. syncbuf.info(self, "discarding %d commits", len(lost))
  542. try:
  543. self._Checkout(rev, quiet=True)
  544. except GitError, e:
  545. syncbuf.fail(self, e)
  546. return
  547. self._CopyFiles()
  548. return
  549. if head == revid:
  550. # No changes; don't do anything further.
  551. #
  552. return
  553. branch = self.GetBranch(branch)
  554. merge = branch.LocalMerge
  555. if not merge:
  556. # The current branch has no tracking configuration.
  557. # Jump off it to a deatched HEAD.
  558. #
  559. syncbuf.info(self,
  560. "leaving %s; does not track upstream",
  561. branch.name)
  562. try:
  563. self._Checkout(rev, quiet=True)
  564. except GitError, e:
  565. syncbuf.fail(self, e)
  566. return
  567. self._CopyFiles()
  568. return
  569. upstream_gain = self._revlist(not_rev(HEAD), rev)
  570. pub = self.WasPublished(branch.name, all)
  571. if pub:
  572. not_merged = self._revlist(not_rev(rev), pub)
  573. if not_merged:
  574. if upstream_gain:
  575. # The user has published this branch and some of those
  576. # commits are not yet merged upstream. We do not want
  577. # to rewrite the published commits so we punt.
  578. #
  579. syncbuf.info(self,
  580. "branch %s is published but is now %d commits behind",
  581. branch.name,
  582. len(upstream_gain))
  583. return
  584. elif upstream_gain:
  585. # We can fast-forward safely.
  586. #
  587. def _doff():
  588. self._FastForward(rev)
  589. self._CopyFiles()
  590. syncbuf.later1(self, _doff)
  591. return
  592. else:
  593. # Trivially no changes in the upstream.
  594. #
  595. return
  596. if merge == rev:
  597. try:
  598. old_merge = self.bare_git.rev_parse('%s@{1}' % merge)
  599. except GitError:
  600. old_merge = merge
  601. if old_merge == '0000000000000000000000000000000000000000' \
  602. or old_merge == '':
  603. old_merge = merge
  604. else:
  605. # The upstream switched on us. Time to cross our fingers
  606. # and pray that the old upstream also wasn't in the habit
  607. # of rebasing itself.
  608. #
  609. syncbuf.info(self, "manifest switched %s...%s", merge, rev)
  610. old_merge = merge
  611. if rev == old_merge:
  612. upstream_lost = []
  613. else:
  614. upstream_lost = self._revlist(not_rev(rev), old_merge)
  615. if not upstream_lost and not upstream_gain:
  616. # Trivially no changes caused by the upstream.
  617. #
  618. return
  619. if self.IsDirty(consider_untracked=False):
  620. syncbuf.fail(self, _DirtyError())
  621. return
  622. if upstream_lost:
  623. # Upstream rebased. Not everything in HEAD
  624. # may have been caused by the user.
  625. #
  626. syncbuf.info(self,
  627. "discarding %d commits removed from upstream",
  628. len(upstream_lost))
  629. branch.remote = rem
  630. branch.merge = self.revision
  631. branch.Save()
  632. my_changes = self._revlist(not_rev(old_merge), HEAD)
  633. if my_changes:
  634. def _dorebase():
  635. self._Rebase(upstream = old_merge, onto = rev)
  636. self._CopyFiles()
  637. syncbuf.later2(self, _dorebase)
  638. elif upstream_lost:
  639. try:
  640. self._ResetHard(rev)
  641. self._CopyFiles()
  642. except GitError, e:
  643. syncbuf.fail(self, e)
  644. return
  645. else:
  646. def _doff():
  647. self._FastForward(rev)
  648. self._CopyFiles()
  649. syncbuf.later1(self, _doff)
  650. def AddCopyFile(self, src, dest, absdest):
  651. # dest should already be an absolute path, but src is project relative
  652. # make src an absolute path
  653. abssrc = os.path.join(self.worktree, src)
  654. self.copyfiles.append(_CopyFile(src, dest, abssrc, absdest))
  655. def DownloadPatchSet(self, change_id, patch_id):
  656. """Download a single patch set of a single change to FETCH_HEAD.
  657. """
  658. remote = self.GetRemote(self.remote.name)
  659. cmd = ['fetch', remote.name]
  660. cmd.append('refs/changes/%2.2d/%d/%d' \
  661. % (change_id % 100, change_id, patch_id))
  662. cmd.extend(map(lambda x: str(x), remote.fetch))
  663. if GitCommand(self, cmd, bare=True).Wait() != 0:
  664. return None
  665. return DownloadedChange(self,
  666. remote.ToLocal(self.revision),
  667. change_id,
  668. patch_id,
  669. self.bare_git.rev_parse('FETCH_HEAD'))
  670. ## Branch Management ##
  671. def StartBranch(self, name):
  672. """Create a new branch off the manifest's revision.
  673. """
  674. head = self.work_git.GetHead()
  675. if head == (R_HEADS + name):
  676. return True
  677. all = self.bare_ref.all
  678. if (R_HEADS + name) in all:
  679. return GitCommand(self,
  680. ['checkout', name, '--'],
  681. capture_stdout = True,
  682. capture_stderr = True).Wait() == 0
  683. branch = self.GetBranch(name)
  684. branch.remote = self.GetRemote(self.remote.name)
  685. branch.merge = self.revision
  686. rev = branch.LocalMerge
  687. if rev in all:
  688. revid = all[rev]
  689. elif IsId(rev):
  690. revid = rev
  691. else:
  692. revid = None
  693. if head.startswith(R_HEADS):
  694. try:
  695. head = all[head]
  696. except KeyError:
  697. head = None
  698. if revid and head and revid == head:
  699. ref = os.path.join(self.gitdir, R_HEADS + name)
  700. try:
  701. os.makedirs(os.path.dirname(ref))
  702. except OSError:
  703. pass
  704. _lwrite(ref, '%s\n' % revid)
  705. _lwrite(os.path.join(self.worktree, '.git', HEAD),
  706. 'ref: %s%s\n' % (R_HEADS, name))
  707. branch.Save()
  708. return True
  709. if GitCommand(self,
  710. ['checkout', '-b', branch.name, rev],
  711. capture_stdout = True,
  712. capture_stderr = True).Wait() == 0:
  713. branch.Save()
  714. return True
  715. return False
  716. def CheckoutBranch(self, name):
  717. """Checkout a local topic branch.
  718. """
  719. rev = R_HEADS + name
  720. head = self.work_git.GetHead()
  721. if head == rev:
  722. # Already on the branch
  723. #
  724. return True
  725. all = self.bare_ref.all
  726. try:
  727. revid = all[rev]
  728. except KeyError:
  729. # Branch does not exist in this project
  730. #
  731. return False
  732. if head.startswith(R_HEADS):
  733. try:
  734. head = all[head]
  735. except KeyError:
  736. head = None
  737. if head == revid:
  738. # Same revision; just update HEAD to point to the new
  739. # target branch, but otherwise take no other action.
  740. #
  741. _lwrite(os.path.join(self.worktree, '.git', HEAD),
  742. 'ref: %s%s\n' % (R_HEADS, name))
  743. return True
  744. return GitCommand(self,
  745. ['checkout', name, '--'],
  746. capture_stdout = True,
  747. capture_stderr = True).Wait() == 0
  748. def AbandonBranch(self, name):
  749. """Destroy a local topic branch.
  750. """
  751. rev = R_HEADS + name
  752. all = self.bare_ref.all
  753. if rev not in all:
  754. # Doesn't exist; assume already abandoned.
  755. #
  756. return True
  757. head = self.work_git.GetHead()
  758. if head == rev:
  759. # We can't destroy the branch while we are sitting
  760. # on it. Switch to a detached HEAD.
  761. #
  762. head = all[head]
  763. rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
  764. if rev in all:
  765. revid = all[rev]
  766. elif IsId(rev):
  767. revid = rev
  768. else:
  769. revid = None
  770. if revid and head == revid:
  771. _lwrite(os.path.join(self.worktree, '.git', HEAD),
  772. '%s\n' % revid)
  773. else:
  774. self._Checkout(rev, quiet=True)
  775. return GitCommand(self,
  776. ['branch', '-D', name],
  777. capture_stdout = True,
  778. capture_stderr = True).Wait() == 0
  779. def PruneHeads(self):
  780. """Prune any topic branches already merged into upstream.
  781. """
  782. cb = self.CurrentBranch
  783. kill = []
  784. left = self._allrefs
  785. for name in left.keys():
  786. if name.startswith(R_HEADS):
  787. name = name[len(R_HEADS):]
  788. if cb is None or name != cb:
  789. kill.append(name)
  790. rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
  791. if cb is not None \
  792. and not self._revlist(HEAD + '...' + rev) \
  793. and not self.IsDirty(consider_untracked = False):
  794. self.work_git.DetachHead(HEAD)
  795. kill.append(cb)
  796. if kill:
  797. old = self.bare_git.GetHead()
  798. if old is None:
  799. old = 'refs/heads/please_never_use_this_as_a_branch_name'
  800. try:
  801. self.bare_git.DetachHead(rev)
  802. b = ['branch', '-d']
  803. b.extend(kill)
  804. b = GitCommand(self, b, bare=True,
  805. capture_stdout=True,
  806. capture_stderr=True)
  807. b.Wait()
  808. finally:
  809. self.bare_git.SetHead(old)
  810. left = self._allrefs
  811. for branch in kill:
  812. if (R_HEADS + branch) not in left:
  813. self.CleanPublishedCache()
  814. break
  815. if cb and cb not in kill:
  816. kill.append(cb)
  817. kill.sort()
  818. kept = []
  819. for branch in kill:
  820. if (R_HEADS + branch) in left:
  821. branch = self.GetBranch(branch)
  822. base = branch.LocalMerge
  823. if not base:
  824. base = rev
  825. kept.append(ReviewableBranch(self, branch, base))
  826. return kept
  827. ## Direct Git Commands ##
  828. def _RemoteFetch(self, name=None):
  829. if not name:
  830. name = self.remote.name
  831. ssh_proxy = False
  832. if self.GetRemote(name).PreConnectFetch():
  833. ssh_proxy = True
  834. cmd = ['fetch']
  835. if not self.worktree:
  836. cmd.append('--update-head-ok')
  837. cmd.append(name)
  838. return GitCommand(self,
  839. cmd,
  840. bare = True,
  841. ssh_proxy = ssh_proxy).Wait() == 0
  842. def _Checkout(self, rev, quiet=False):
  843. cmd = ['checkout']
  844. if quiet:
  845. cmd.append('-q')
  846. cmd.append(rev)
  847. cmd.append('--')
  848. if GitCommand(self, cmd).Wait() != 0:
  849. if self._allrefs:
  850. raise GitError('%s checkout %s ' % (self.name, rev))
  851. def _ResetHard(self, rev, quiet=True):
  852. cmd = ['reset', '--hard']
  853. if quiet:
  854. cmd.append('-q')
  855. cmd.append(rev)
  856. if GitCommand(self, cmd).Wait() != 0:
  857. raise GitError('%s reset --hard %s ' % (self.name, rev))
  858. def _Rebase(self, upstream, onto = None):
  859. cmd = ['rebase']
  860. if onto is not None:
  861. cmd.extend(['--onto', onto])
  862. cmd.append(upstream)
  863. if GitCommand(self, cmd).Wait() != 0:
  864. raise GitError('%s rebase %s ' % (self.name, upstream))
  865. def _FastForward(self, head):
  866. cmd = ['merge', head]
  867. if GitCommand(self, cmd).Wait() != 0:
  868. raise GitError('%s merge %s ' % (self.name, head))
  869. def _InitGitDir(self):
  870. if not os.path.exists(self.gitdir):
  871. os.makedirs(self.gitdir)
  872. self.bare_git.init()
  873. if self.manifest.IsMirror:
  874. self.config.SetString('core.bare', 'true')
  875. else:
  876. self.config.SetString('core.bare', None)
  877. hooks = self._gitdir_path('hooks')
  878. try:
  879. to_rm = os.listdir(hooks)
  880. except OSError:
  881. to_rm = []
  882. for old_hook in to_rm:
  883. os.remove(os.path.join(hooks, old_hook))
  884. self._InitHooks()
  885. m = self.manifest.manifestProject.config
  886. for key in ['user.name', 'user.email']:
  887. if m.Has(key, include_defaults = False):
  888. self.config.SetString(key, m.GetString(key))
  889. def _InitHooks(self):
  890. hooks = self._gitdir_path('hooks')
  891. if not os.path.exists(hooks):
  892. os.makedirs(hooks)
  893. for stock_hook in repo_hooks():
  894. dst = os.path.join(hooks, os.path.basename(stock_hook))
  895. try:
  896. os.symlink(relpath(stock_hook, dst), dst)
  897. except OSError, e:
  898. if e.errno == errno.EEXIST:
  899. pass
  900. elif e.errno == errno.EPERM:
  901. raise GitError('filesystem must support symlinks')
  902. else:
  903. raise
  904. def _InitRemote(self):
  905. if self.remote.fetchUrl:
  906. remote = self.GetRemote(self.remote.name)
  907. url = self.remote.fetchUrl
  908. while url.endswith('/'):
  909. url = url[:-1]
  910. url += '/%s.git' % self.name
  911. remote.url = url
  912. remote.review = self.remote.reviewUrl
  913. if remote.projectname is None:
  914. remote.projectname = self.name
  915. if self.worktree:
  916. remote.ResetFetch(mirror=False)
  917. else:
  918. remote.ResetFetch(mirror=True)
  919. remote.Save()
  920. for r in self.extraRemotes.values():
  921. remote = self.GetRemote(r.name)
  922. remote.url = r.fetchUrl
  923. remote.review = r.reviewUrl
  924. if r.projectName:
  925. remote.projectname = r.projectName
  926. elif remote.projectname is None:
  927. remote.projectname = self.name
  928. remote.ResetFetch()
  929. remote.Save()
  930. def _InitMRef(self):
  931. if self.manifest.branch:
  932. msg = 'manifest set to %s' % self.revision
  933. ref = R_M + self.manifest.branch
  934. cur = self.bare_ref.symref(ref)
  935. if IsId(self.revision):
  936. if cur != '' or self.bare_ref.get(ref) != self.revision:
  937. dst = self.revision + '^0'
  938. self.bare_git.UpdateRef(ref, dst, message = msg, detach = True)
  939. else:
  940. remote = self.GetRemote(self.remote.name)
  941. dst = remote.ToLocal(self.revision)
  942. if cur != dst:
  943. self.bare_git.symbolic_ref('-m', msg, ref, dst)
  944. def _InitMirrorHead(self):
  945. dst = self.GetRemote(self.remote.name).ToLocal(self.revision)
  946. msg = 'manifest set to %s' % self.revision
  947. self.bare_git.SetHead(dst, message=msg)
  948. def _InitWorkTree(self):
  949. dotgit = os.path.join(self.worktree, '.git')
  950. if not os.path.exists(dotgit):
  951. os.makedirs(dotgit)
  952. for name in ['config',
  953. 'description',
  954. 'hooks',
  955. 'info',
  956. 'logs',
  957. 'objects',
  958. 'packed-refs',
  959. 'refs',
  960. 'rr-cache',
  961. 'svn']:
  962. try:
  963. src = os.path.join(self.gitdir, name)
  964. dst = os.path.join(dotgit, name)
  965. os.symlink(relpath(src, dst), dst)
  966. except OSError, e:
  967. if e.errno == errno.EPERM:
  968. raise GitError('filesystem must support symlinks')
  969. else:
  970. raise
  971. rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
  972. rev = self.bare_git.rev_parse('%s^0' % rev)
  973. _lwrite(os.path.join(dotgit, HEAD), '%s\n' % rev)
  974. cmd = ['read-tree', '--reset', '-u']
  975. cmd.append('-v')
  976. cmd.append('HEAD')
  977. if GitCommand(self, cmd).Wait() != 0:
  978. raise GitError("cannot initialize work tree")
  979. def _gitdir_path(self, path):
  980. return os.path.join(self.gitdir, path)
  981. def _revlist(self, *args):
  982. cmd = []
  983. cmd.extend(args)
  984. cmd.append('--')
  985. return self.work_git.rev_list(*args)
  986. @property
  987. def _allrefs(self):
  988. return self.bare_ref.all
  989. class _GitGetByExec(object):
  990. def __init__(self, project, bare):
  991. self._project = project
  992. self._bare = bare
  993. def LsOthers(self):
  994. p = GitCommand(self._project,
  995. ['ls-files',
  996. '-z',
  997. '--others',
  998. '--exclude-standard'],
  999. bare = False,
  1000. capture_stdout = True,
  1001. capture_stderr = True)
  1002. if p.Wait() == 0:
  1003. out = p.stdout
  1004. if out:
  1005. return out[:-1].split("\0")
  1006. return []
  1007. def DiffZ(self, name, *args):
  1008. cmd = [name]
  1009. cmd.append('-z')
  1010. cmd.extend(args)
  1011. p = GitCommand(self._project,
  1012. cmd,
  1013. bare = False,
  1014. capture_stdout = True,
  1015. capture_stderr = True)
  1016. try:
  1017. out = p.process.stdout.read()
  1018. r = {}
  1019. if out:
  1020. out = iter(out[:-1].split('\0'))
  1021. while out:
  1022. try:
  1023. info = out.next()
  1024. path = out.next()
  1025. except StopIteration:
  1026. break
  1027. class _Info(object):
  1028. def __init__(self, path, omode, nmode, oid, nid, state):
  1029. self.path = path
  1030. self.src_path = None
  1031. self.old_mode = omode
  1032. self.new_mode = nmode
  1033. self.old_id = oid
  1034. self.new_id = nid
  1035. if len(state) == 1:
  1036. self.status = state
  1037. self.level = None
  1038. else:
  1039. self.status = state[:1]
  1040. self.level = state[1:]
  1041. while self.level.startswith('0'):
  1042. self.level = self.level[1:]
  1043. info = info[1:].split(' ')
  1044. info =_Info(path, *info)
  1045. if info.status in ('R', 'C'):
  1046. info.src_path = info.path
  1047. info.path = out.next()
  1048. r[info.path] = info
  1049. return r
  1050. finally:
  1051. p.Wait()
  1052. def GetHead(self):
  1053. if self._bare:
  1054. path = os.path.join(self._project.gitdir, HEAD)
  1055. else:
  1056. path = os.path.join(self._project.worktree, '.git', HEAD)
  1057. fd = open(path, 'rb')
  1058. try:
  1059. line = fd.read()
  1060. finally:
  1061. fd.close()
  1062. if line.startswith('ref: '):
  1063. return line[5:-1]
  1064. return line[:-1]
  1065. def SetHead(self, ref, message=None):
  1066. cmdv = []
  1067. if message is not None:
  1068. cmdv.extend(['-m', message])
  1069. cmdv.append(HEAD)
  1070. cmdv.append(ref)
  1071. self.symbolic_ref(*cmdv)
  1072. def DetachHead(self, new, message=None):
  1073. cmdv = ['--no-deref']
  1074. if message is not None:
  1075. cmdv.extend(['-m', message])
  1076. cmdv.append(HEAD)
  1077. cmdv.append(new)
  1078. self.update_ref(*cmdv)
  1079. def UpdateRef(self, name, new, old=None,
  1080. message=None,
  1081. detach=False):
  1082. cmdv = []
  1083. if message is not None:
  1084. cmdv.extend(['-m', message])
  1085. if detach:
  1086. cmdv.append('--no-deref')
  1087. cmdv.append(name)
  1088. cmdv.append(new)
  1089. if old is not None:
  1090. cmdv.append(old)
  1091. self.update_ref(*cmdv)
  1092. def DeleteRef(self, name, old=None):
  1093. if not old:
  1094. old = self.rev_parse(name)
  1095. self.update_ref('-d', name, old)
  1096. self._project.bare_ref.deleted(name)
  1097. def rev_list(self, *args):
  1098. cmdv = ['rev-list']
  1099. cmdv.extend(args)
  1100. p = GitCommand(self._project,
  1101. cmdv,
  1102. bare = self._bare,
  1103. capture_stdout = True,
  1104. capture_stderr = True)
  1105. r = []
  1106. for line in p.process.stdout:
  1107. r.append(line[:-1])
  1108. if p.Wait() != 0:
  1109. raise GitError('%s rev-list %s: %s' % (
  1110. self._project.name,
  1111. str(args),
  1112. p.stderr))
  1113. return r
  1114. def __getattr__(self, name):
  1115. name = name.replace('_', '-')
  1116. def runner(*args):
  1117. cmdv = [name]
  1118. cmdv.extend(args)
  1119. p = GitCommand(self._project,
  1120. cmdv,
  1121. bare = self._bare,
  1122. capture_stdout = True,
  1123. capture_stderr = True)
  1124. if p.Wait() != 0:
  1125. raise GitError('%s %s: %s' % (
  1126. self._project.name,
  1127. name,
  1128. p.stderr))
  1129. r = p.stdout
  1130. if r.endswith('\n') and r.index('\n') == len(r) - 1:
  1131. return r[:-1]
  1132. return r
  1133. return runner
  1134. class _PriorSyncFailedError(Exception):
  1135. def __str__(self):
  1136. return 'prior sync failed; rebase still in progress'
  1137. class _DirtyError(Exception):
  1138. def __str__(self):
  1139. return 'contains uncommitted changes'
  1140. class _InfoMessage(object):
  1141. def __init__(self, project, text):
  1142. self.project = project
  1143. self.text = text
  1144. def Print(self, syncbuf):
  1145. syncbuf.out.info('%s/: %s', self.project.relpath, self.text)
  1146. syncbuf.out.nl()
  1147. class _Failure(object):
  1148. def __init__(self, project, why):
  1149. self.project = project
  1150. self.why = why
  1151. def Print(self, syncbuf):
  1152. syncbuf.out.fail('error: %s/: %s',
  1153. self.project.relpath,
  1154. str(self.why))
  1155. syncbuf.out.nl()
  1156. class _Later(object):
  1157. def __init__(self, project, action):
  1158. self.project = project
  1159. self.action = action
  1160. def Run(self, syncbuf):
  1161. out = syncbuf.out
  1162. out.project('project %s/', self.project.relpath)
  1163. out.nl()
  1164. try:
  1165. self.action()
  1166. out.nl()
  1167. return True
  1168. except GitError, e:
  1169. out.nl()
  1170. return False
  1171. class _SyncColoring(Coloring):
  1172. def __init__(self, config):
  1173. Coloring.__init__(self, config, 'reposync')
  1174. self.project = self.printer('header', attr = 'bold')
  1175. self.info = self.printer('info')
  1176. self.fail = self.printer('fail', fg='red')
  1177. class SyncBuffer(object):
  1178. def __init__(self, config, detach_head=False):
  1179. self._messages = []
  1180. self._failures = []
  1181. self._later_queue1 = []
  1182. self._later_queue2 = []
  1183. self.out = _SyncColoring(config)
  1184. self.out.redirect(sys.stderr)
  1185. self.detach_head = detach_head
  1186. self.clean = True
  1187. def info(self, project, fmt, *args):
  1188. self._messages.append(_InfoMessage(project, fmt % args))
  1189. def fail(self, project, err=None):
  1190. self._failures.append(_Failure(project, err))
  1191. self.clean = False
  1192. def later1(self, project, what):
  1193. self._later_queue1.append(_Later(project, what))
  1194. def later2(self, project, what):
  1195. self._later_queue2.append(_Later(project, what))
  1196. def Finish(self):
  1197. self._PrintMessages()
  1198. self._RunLater()
  1199. self._PrintMessages()
  1200. return self.clean
  1201. def _RunLater(self):
  1202. for q in ['_later_queue1', '_later_queue2']:
  1203. if not self._RunQueue(q):
  1204. return
  1205. def _RunQueue(self, queue):
  1206. for m in getattr(self, queue):
  1207. if not m.Run(self):
  1208. self.clean = False
  1209. return False
  1210. setattr(self, queue, [])
  1211. return True
  1212. def _PrintMessages(self):
  1213. for m in self._messages:
  1214. m.Print(self)
  1215. for m in self._failures:
  1216. m.Print(self)
  1217. self._messages = []
  1218. self._failures = []
  1219. class MetaProject(Project):
  1220. """A special project housed under .repo.
  1221. """
  1222. def __init__(self, manifest, name, gitdir, worktree):
  1223. repodir = manifest.repodir
  1224. Project.__init__(self,
  1225. manifest = manifest,
  1226. name = name,
  1227. gitdir = gitdir,
  1228. worktree = worktree,
  1229. remote = Remote('origin'),
  1230. relpath = '.repo/%s' % name,
  1231. revision = 'refs/heads/master')
  1232. def PreSync(self):
  1233. if self.Exists:
  1234. cb = self.CurrentBranch
  1235. if cb:
  1236. base = self.GetBranch(cb).merge
  1237. if base:
  1238. self.revision = base
  1239. @property
  1240. def LastFetch(self):
  1241. try:
  1242. fh = os.path.join(self.gitdir, 'FETCH_HEAD')
  1243. return os.path.getmtime(fh)
  1244. except OSError:
  1245. return 0
  1246. @property
  1247. def HasChanges(self):
  1248. """Has the remote received new commits not yet checked out?
  1249. """
  1250. if not self.remote or not self.revision:
  1251. return False
  1252. all = self.bare_ref.all
  1253. rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
  1254. if rev in all:
  1255. revid = all[rev]
  1256. else:
  1257. revid = rev
  1258. head = self.work_git.GetHead()
  1259. if head.startswith(R_HEADS):
  1260. try:
  1261. head = all[head]
  1262. except KeyError:
  1263. head = None
  1264. if revid == head:
  1265. return False
  1266. elif self._revlist(not_rev(HEAD), rev):
  1267. return True
  1268. return False