repo 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710
  1. #!/bin/sh
  2. ## repo default configuration
  3. ##
  4. REPO_URL='https://gerrit.googlesource.com/git-repo'
  5. REPO_REV='stable'
  6. # Copyright (C) 2008 Google Inc.
  7. #
  8. # Licensed under the Apache License, Version 2.0 (the "License");
  9. # you may not use this file except in compliance with the License.
  10. # You may obtain a copy of the License at
  11. #
  12. # http://www.apache.org/licenses/LICENSE-2.0
  13. #
  14. # Unless required by applicable law or agreed to in writing, software
  15. # distributed under the License is distributed on an "AS IS" BASIS,
  16. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17. # See the License for the specific language governing permissions and
  18. # limitations under the License.
  19. magic='--calling-python-from-/bin/sh--'
  20. """exec" python -E "$0" "$@" """#$magic"
  21. if __name__ == '__main__':
  22. import sys
  23. if sys.argv[-1] == '#%s' % magic:
  24. del sys.argv[-1]
  25. del magic
  26. # increment this whenever we make important changes to this script
  27. VERSION = (1, 17)
  28. # increment this if the MAINTAINER_KEYS block is modified
  29. KEYRING_VERSION = (1,0)
  30. MAINTAINER_KEYS = """
  31. Repo Maintainer <repo@android.kernel.org>
  32. -----BEGIN PGP PUBLIC KEY BLOCK-----
  33. Version: GnuPG v1.4.2.2 (GNU/Linux)
  34. mQGiBEj3ugERBACrLJh/ZPyVSKeClMuznFIrsQ+hpNnmJGw1a9GXKYKk8qHPhAZf
  35. WKtrBqAVMNRLhL85oSlekRz98u41H5si5zcuv+IXJDF5MJYcB8f22wAy15lUqPWi
  36. VCkk1l8qqLiuW0fo+ZkPY5qOgrvc0HW1SmdH649uNwqCbcKb6CxaTxzhOwCgj3AP
  37. xI1WfzLqdJjsm1Nq98L0cLcD/iNsILCuw44PRds3J75YP0pze7YF/6WFMB6QSFGu
  38. aUX1FsTTztKNXGms8i5b2l1B8JaLRWq/jOnZzyl1zrUJhkc0JgyZW5oNLGyWGhKD
  39. Fxp5YpHuIuMImopWEMFIRQNrvlg+YVK8t3FpdI1RY0LYqha8pPzANhEYgSfoVzOb
  40. fbfbA/4ioOrxy8ifSoga7ITyZMA+XbW8bx33WXutO9N7SPKS/AK2JpasSEVLZcON
  41. ae5hvAEGVXKxVPDjJBmIc2cOe7kOKSi3OxLzBqrjS2rnjiP4o0ekhZIe4+ocwVOg
  42. e0PLlH5avCqihGRhpoqDRsmpzSHzJIxtoeb+GgGEX8KkUsVAhbQpUmVwbyBNYWlu
  43. dGFpbmVyIDxyZXBvQGFuZHJvaWQua2VybmVsLm9yZz6IYAQTEQIAIAUCSPe6AQIb
  44. AwYLCQgHAwIEFQIIAwQWAgMBAh4BAheAAAoJEBZTDV6SD1xl1GEAn0x/OKQpy7qI
  45. 6G73NJviU0IUMtftAKCFMUhGb/0bZvQ8Rm3QCUpWHyEIu7kEDQRI97ogEBAA2wI6
  46. 5fs9y/rMwD6dkD/vK9v4C9mOn1IL5JCPYMJBVSci+9ED4ChzYvfq7wOcj9qIvaE0
  47. GwCt2ar7Q56me5J+byhSb32Rqsw/r3Vo5cZMH80N4cjesGuSXOGyEWTe4HYoxnHv
  48. gF4EKI2LK7xfTUcxMtlyn52sUpkfKsCpUhFvdmbAiJE+jCkQZr1Z8u2KphV79Ou+
  49. P1N5IXY/XWOlq48Qf4MWCYlJFrB07xjUjLKMPDNDnm58L5byDrP/eHysKexpbakL
  50. xCmYyfT6DV1SWLblpd2hie0sL3YejdtuBMYMS2rI7Yxb8kGuqkz+9l1qhwJtei94
  51. 5MaretDy/d/JH/pRYkRf7L+ke7dpzrP+aJmcz9P1e6gq4NJsWejaALVASBiioqNf
  52. QmtqSVzF1wkR5avZkFHuYvj6V/t1RrOZTXxkSk18KFMJRBZrdHFCWbc5qrVxUB6e
  53. N5pja0NFIUCigLBV1c6I2DwiuboMNh18VtJJh+nwWeez/RueN4ig59gRTtkcc0PR
  54. 35tX2DR8+xCCFVW/NcJ4PSePYzCuuLvp1vEDHnj41R52Fz51hgddT4rBsp0nL+5I
  55. socSOIIezw8T9vVzMY4ArCKFAVu2IVyBcahTfBS8q5EM63mONU6UVJEozfGljiMw
  56. xuQ7JwKcw0AUEKTKG7aBgBaTAgT8TOevpvlw91cAAwUP/jRkyVi/0WAb0qlEaq/S
  57. ouWxX1faR+vU3b+Y2/DGjtXQMzG0qpetaTHC/AxxHpgt/dCkWI6ljYDnxgPLwG0a
  58. Oasm94BjZc6vZwf1opFZUKsjOAAxRxNZyjUJKe4UZVuMTk6zo27Nt3LMnc0FO47v
  59. FcOjRyquvgNOS818irVHUf12waDx8gszKxQTTtFxU5/ePB2jZmhP6oXSe4K/LG5T
  60. +WBRPDrHiGPhCzJRzm9BP0lTnGCAj3o9W90STZa65RK7IaYpC8TB35JTBEbrrNCp
  61. w6lzd74LnNEp5eMlKDnXzUAgAH0yzCQeMl7t33QCdYx2hRs2wtTQSjGfAiNmj/WW
  62. Vl5Jn+2jCDnRLenKHwVRFsBX2e0BiRWt/i9Y8fjorLCXVj4z+7yW6DawdLkJorEo
  63. p3v5ILwfC7hVx4jHSnOgZ65L9s8EQdVr1ckN9243yta7rNgwfcqb60ILMFF1BRk/
  64. 0V7wCL+68UwwiQDvyMOQuqkysKLSDCLb7BFcyA7j6KG+5hpsREstFX2wK1yKeraz
  65. 5xGrFy8tfAaeBMIQ17gvFSp/suc9DYO0ICK2BISzq+F+ZiAKsjMYOBNdH/h0zobQ
  66. HTHs37+/QLMomGEGKZMWi0dShU2J5mNRQu3Hhxl3hHDVbt5CeJBb26aQcQrFz69W
  67. zE3GNvmJosh6leayjtI9P2A6iEkEGBECAAkFAkj3uiACGwwACgkQFlMNXpIPXGWp
  68. TACbBS+Up3RpfYVfd63c1cDdlru13pQAn3NQy/SN858MkxN+zym86UBgOad2
  69. =CMiZ
  70. -----END PGP PUBLIC KEY BLOCK-----
  71. """
  72. GIT = 'git' # our git command
  73. MIN_GIT_VERSION = (1, 5, 4) # minimum supported git version
  74. repodir = '.repo' # name of repo's private directory
  75. S_repo = 'repo' # special repo reposiory
  76. S_manifests = 'manifests' # special manifest repository
  77. REPO_MAIN = S_repo + '/main.py' # main script
  78. import optparse
  79. import os
  80. import re
  81. import readline
  82. import subprocess
  83. import sys
  84. import urllib2
  85. home_dot_repo = os.path.expanduser('~/.repoconfig')
  86. gpg_dir = os.path.join(home_dot_repo, 'gnupg')
  87. extra_args = []
  88. init_optparse = optparse.OptionParser(usage="repo init -u url [options]")
  89. # Logging
  90. group = init_optparse.add_option_group('Logging options')
  91. group.add_option('-q', '--quiet',
  92. dest="quiet", action="store_true", default=False,
  93. help="be quiet")
  94. # Manifest
  95. group = init_optparse.add_option_group('Manifest options')
  96. group.add_option('-u', '--manifest-url',
  97. dest='manifest_url',
  98. help='manifest repository location', metavar='URL')
  99. group.add_option('-b', '--manifest-branch',
  100. dest='manifest_branch',
  101. help='manifest branch or revision', metavar='REVISION')
  102. group.add_option('-m', '--manifest-name',
  103. dest='manifest_name',
  104. help='initial manifest file', metavar='NAME.xml')
  105. group.add_option('--mirror',
  106. dest='mirror', action='store_true',
  107. help='mirror the forrest')
  108. group.add_option('--reference',
  109. dest='reference',
  110. help='location of mirror directory', metavar='DIR')
  111. group.add_option('--depth', type='int', default=None,
  112. dest='depth',
  113. help='create a shallow clone with given depth; see git clone')
  114. group.add_option('-g', '--groups',
  115. dest='groups', default='default',
  116. help='restrict manifest projects to ones with a specified group',
  117. metavar='GROUP')
  118. group.add_option('-p', '--platform',
  119. dest='platform', default="auto",
  120. help='restrict manifest projects to ones with a specified'
  121. 'platform group [auto|all|none|linux|darwin|...]',
  122. metavar='PLATFORM')
  123. # Tool
  124. group = init_optparse.add_option_group('repo Version options')
  125. group.add_option('--repo-url',
  126. dest='repo_url',
  127. help='repo repository location', metavar='URL')
  128. group.add_option('--repo-branch',
  129. dest='repo_branch',
  130. help='repo branch or revision', metavar='REVISION')
  131. group.add_option('--no-repo-verify',
  132. dest='no_repo_verify', action='store_true',
  133. help='do not verify repo source code')
  134. # Other
  135. group = init_optparse.add_option_group('Other options')
  136. group.add_option('--config-name',
  137. dest='config_name', action="store_true", default=False,
  138. help='Always prompt for name/e-mail')
  139. class CloneFailure(Exception):
  140. """Indicate the remote clone of repo itself failed.
  141. """
  142. def _Init(args):
  143. """Installs repo by cloning it over the network.
  144. """
  145. opt, args = init_optparse.parse_args(args)
  146. if args:
  147. init_optparse.print_usage()
  148. sys.exit(1)
  149. url = opt.repo_url
  150. if not url:
  151. url = REPO_URL
  152. extra_args.append('--repo-url=%s' % url)
  153. branch = opt.repo_branch
  154. if not branch:
  155. branch = REPO_REV
  156. extra_args.append('--repo-branch=%s' % branch)
  157. if branch.startswith('refs/heads/'):
  158. branch = branch[len('refs/heads/'):]
  159. if branch.startswith('refs/'):
  160. print >>sys.stderr, "fatal: invalid branch name '%s'" % branch
  161. raise CloneFailure()
  162. if not os.path.isdir(repodir):
  163. try:
  164. os.mkdir(repodir)
  165. except OSError, e:
  166. print >>sys.stderr, \
  167. 'fatal: cannot make %s directory: %s' % (
  168. repodir, e.strerror)
  169. # Don't faise CloneFailure; that would delete the
  170. # name. Instead exit immediately.
  171. #
  172. sys.exit(1)
  173. _CheckGitVersion()
  174. try:
  175. if _NeedSetupGnuPG():
  176. can_verify = _SetupGnuPG(opt.quiet)
  177. else:
  178. can_verify = True
  179. dst = os.path.abspath(os.path.join(repodir, S_repo))
  180. _Clone(url, dst, opt.quiet)
  181. if can_verify and not opt.no_repo_verify:
  182. rev = _Verify(dst, branch, opt.quiet)
  183. else:
  184. rev = 'refs/remotes/origin/%s^0' % branch
  185. _Checkout(dst, branch, rev, opt.quiet)
  186. except CloneFailure:
  187. if opt.quiet:
  188. print >>sys.stderr, \
  189. 'fatal: repo init failed; run without --quiet to see why'
  190. raise
  191. def _CheckGitVersion():
  192. cmd = [GIT, '--version']
  193. try:
  194. proc = subprocess.Popen(cmd, stdout=subprocess.PIPE)
  195. except OSError, e:
  196. print >>sys.stderr
  197. print >>sys.stderr, "fatal: '%s' is not available" % GIT
  198. print >>sys.stderr, 'fatal: %s' % e
  199. print >>sys.stderr
  200. print >>sys.stderr, 'Please make sure %s is installed'\
  201. ' and in your path.' % GIT
  202. raise CloneFailure()
  203. ver_str = proc.stdout.read().strip()
  204. proc.stdout.close()
  205. proc.wait()
  206. if not ver_str.startswith('git version '):
  207. print >>sys.stderr, 'error: "%s" unsupported' % ver_str
  208. raise CloneFailure()
  209. ver_str = ver_str[len('git version '):].strip()
  210. ver_act = tuple(map(lambda x: int(x), ver_str.split('.')[0:3]))
  211. if ver_act < MIN_GIT_VERSION:
  212. need = '.'.join(map(lambda x: str(x), MIN_GIT_VERSION))
  213. print >>sys.stderr, 'fatal: git %s or later required' % need
  214. raise CloneFailure()
  215. def _NeedSetupGnuPG():
  216. if not os.path.isdir(home_dot_repo):
  217. return True
  218. kv = os.path.join(home_dot_repo, 'keyring-version')
  219. if not os.path.exists(kv):
  220. return True
  221. kv = open(kv).read()
  222. if not kv:
  223. return True
  224. kv = tuple(map(lambda x: int(x), kv.split('.')))
  225. if kv < KEYRING_VERSION:
  226. return True
  227. return False
  228. def _SetupGnuPG(quiet):
  229. if not os.path.isdir(home_dot_repo):
  230. try:
  231. os.mkdir(home_dot_repo)
  232. except OSError, e:
  233. print >>sys.stderr, \
  234. 'fatal: cannot make %s directory: %s' % (
  235. home_dot_repo, e.strerror)
  236. sys.exit(1)
  237. if not os.path.isdir(gpg_dir):
  238. try:
  239. os.mkdir(gpg_dir, 0700)
  240. except OSError, e:
  241. print >>sys.stderr, \
  242. 'fatal: cannot make %s directory: %s' % (
  243. gpg_dir, e.strerror)
  244. sys.exit(1)
  245. env = os.environ.copy()
  246. env['GNUPGHOME'] = gpg_dir.encode()
  247. cmd = ['gpg', '--import']
  248. try:
  249. proc = subprocess.Popen(cmd,
  250. env = env,
  251. stdin = subprocess.PIPE)
  252. except OSError, e:
  253. if not quiet:
  254. print >>sys.stderr, 'warning: gpg (GnuPG) is not available.'
  255. print >>sys.stderr, 'warning: Installing it is strongly encouraged.'
  256. print >>sys.stderr
  257. return False
  258. proc.stdin.write(MAINTAINER_KEYS)
  259. proc.stdin.close()
  260. if proc.wait() != 0:
  261. print >>sys.stderr, 'fatal: registering repo maintainer keys failed'
  262. sys.exit(1)
  263. print
  264. fd = open(os.path.join(home_dot_repo, 'keyring-version'), 'w')
  265. fd.write('.'.join(map(lambda x: str(x), KEYRING_VERSION)) + '\n')
  266. fd.close()
  267. return True
  268. def _SetConfig(local, name, value):
  269. """Set a git configuration option to the specified value.
  270. """
  271. cmd = [GIT, 'config', name, value]
  272. if subprocess.Popen(cmd, cwd = local).wait() != 0:
  273. raise CloneFailure()
  274. def _InitHttp():
  275. handlers = []
  276. mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
  277. try:
  278. import netrc
  279. n = netrc.netrc()
  280. for host in n.hosts:
  281. p = n.hosts[host]
  282. mgr.add_password(p[1], 'http://%s/' % host, p[0], p[2])
  283. mgr.add_password(p[1], 'https://%s/' % host, p[0], p[2])
  284. except:
  285. pass
  286. handlers.append(urllib2.HTTPBasicAuthHandler(mgr))
  287. handlers.append(urllib2.HTTPDigestAuthHandler(mgr))
  288. if 'http_proxy' in os.environ:
  289. url = os.environ['http_proxy']
  290. handlers.append(urllib2.ProxyHandler({'http': url, 'https': url}))
  291. if 'REPO_CURL_VERBOSE' in os.environ:
  292. handlers.append(urllib2.HTTPHandler(debuglevel=1))
  293. handlers.append(urllib2.HTTPSHandler(debuglevel=1))
  294. urllib2.install_opener(urllib2.build_opener(*handlers))
  295. def _Fetch(url, local, src, quiet):
  296. if not quiet:
  297. print >>sys.stderr, 'Get %s' % url
  298. cmd = [GIT, 'fetch']
  299. if quiet:
  300. cmd.append('--quiet')
  301. err = subprocess.PIPE
  302. else:
  303. err = None
  304. cmd.append(src)
  305. cmd.append('+refs/heads/*:refs/remotes/origin/*')
  306. cmd.append('refs/tags/*:refs/tags/*')
  307. proc = subprocess.Popen(cmd, cwd = local, stderr = err)
  308. if err:
  309. proc.stderr.read()
  310. proc.stderr.close()
  311. if proc.wait() != 0:
  312. raise CloneFailure()
  313. def _DownloadBundle(url, local, quiet):
  314. if not url.endswith('/'):
  315. url += '/'
  316. url += 'clone.bundle'
  317. proc = subprocess.Popen(
  318. [GIT, 'config', '--get-regexp', 'url.*.insteadof'],
  319. cwd = local,
  320. stdout = subprocess.PIPE)
  321. for line in proc.stdout:
  322. m = re.compile(r'^url\.(.*)\.insteadof (.*)$').match(line)
  323. if m:
  324. new_url = m.group(1)
  325. old_url = m.group(2)
  326. if url.startswith(old_url):
  327. url = new_url + url[len(old_url):]
  328. break
  329. proc.stdout.close()
  330. proc.wait()
  331. if not url.startswith('http:') and not url.startswith('https:'):
  332. return False
  333. dest = open(os.path.join(local, '.git', 'clone.bundle'), 'w+b')
  334. try:
  335. try:
  336. r = urllib2.urlopen(url)
  337. except urllib2.HTTPError, e:
  338. if e.code == 404:
  339. return False
  340. print >>sys.stderr, 'fatal: Cannot get %s' % url
  341. print >>sys.stderr, 'fatal: HTTP error %s' % e.code
  342. raise CloneFailure()
  343. except urllib2.URLError, e:
  344. print >>sys.stderr, 'fatal: Cannot get %s' % url
  345. print >>sys.stderr, 'fatal: error %s' % e.reason
  346. raise CloneFailure()
  347. try:
  348. if not quiet:
  349. print >>sys.stderr, 'Get %s' % url
  350. while True:
  351. buf = r.read(8192)
  352. if buf == '':
  353. return True
  354. dest.write(buf)
  355. finally:
  356. r.close()
  357. finally:
  358. dest.close()
  359. def _ImportBundle(local):
  360. path = os.path.join(local, '.git', 'clone.bundle')
  361. try:
  362. _Fetch(local, local, path, True)
  363. finally:
  364. os.remove(path)
  365. def _Clone(url, local, quiet):
  366. """Clones a git repository to a new subdirectory of repodir
  367. """
  368. try:
  369. os.mkdir(local)
  370. except OSError, e:
  371. print >>sys.stderr, \
  372. 'fatal: cannot make %s directory: %s' \
  373. % (local, e.strerror)
  374. raise CloneFailure()
  375. cmd = [GIT, 'init', '--quiet']
  376. try:
  377. proc = subprocess.Popen(cmd, cwd = local)
  378. except OSError, e:
  379. print >>sys.stderr
  380. print >>sys.stderr, "fatal: '%s' is not available" % GIT
  381. print >>sys.stderr, 'fatal: %s' % e
  382. print >>sys.stderr
  383. print >>sys.stderr, 'Please make sure %s is installed'\
  384. ' and in your path.' % GIT
  385. raise CloneFailure()
  386. if proc.wait() != 0:
  387. print >>sys.stderr, 'fatal: could not create %s' % local
  388. raise CloneFailure()
  389. _InitHttp()
  390. _SetConfig(local, 'remote.origin.url', url)
  391. _SetConfig(local, 'remote.origin.fetch',
  392. '+refs/heads/*:refs/remotes/origin/*')
  393. if _DownloadBundle(url, local, quiet):
  394. _ImportBundle(local)
  395. else:
  396. _Fetch(url, local, 'origin', quiet)
  397. def _Verify(cwd, branch, quiet):
  398. """Verify the branch has been signed by a tag.
  399. """
  400. cmd = [GIT, 'describe', 'origin/%s' % branch]
  401. proc = subprocess.Popen(cmd,
  402. stdout=subprocess.PIPE,
  403. stderr=subprocess.PIPE,
  404. cwd = cwd)
  405. cur = proc.stdout.read().strip()
  406. proc.stdout.close()
  407. proc.stderr.read()
  408. proc.stderr.close()
  409. if proc.wait() != 0 or not cur:
  410. print >>sys.stderr
  411. print >>sys.stderr,\
  412. "fatal: branch '%s' has not been signed" \
  413. % branch
  414. raise CloneFailure()
  415. m = re.compile(r'^(.*)-[0-9]{1,}-g[0-9a-f]{1,}$').match(cur)
  416. if m:
  417. cur = m.group(1)
  418. if not quiet:
  419. print >>sys.stderr
  420. print >>sys.stderr, \
  421. "info: Ignoring branch '%s'; using tagged release '%s'" \
  422. % (branch, cur)
  423. print >>sys.stderr
  424. env = os.environ.copy()
  425. env['GNUPGHOME'] = gpg_dir.encode()
  426. cmd = [GIT, 'tag', '-v', cur]
  427. proc = subprocess.Popen(cmd,
  428. stdout = subprocess.PIPE,
  429. stderr = subprocess.PIPE,
  430. cwd = cwd,
  431. env = env)
  432. out = proc.stdout.read()
  433. proc.stdout.close()
  434. err = proc.stderr.read()
  435. proc.stderr.close()
  436. if proc.wait() != 0:
  437. print >>sys.stderr
  438. print >>sys.stderr, out
  439. print >>sys.stderr, err
  440. print >>sys.stderr
  441. raise CloneFailure()
  442. return '%s^0' % cur
  443. def _Checkout(cwd, branch, rev, quiet):
  444. """Checkout an upstream branch into the repository and track it.
  445. """
  446. cmd = [GIT, 'update-ref', 'refs/heads/default', rev]
  447. if subprocess.Popen(cmd, cwd = cwd).wait() != 0:
  448. raise CloneFailure()
  449. _SetConfig(cwd, 'branch.default.remote', 'origin')
  450. _SetConfig(cwd, 'branch.default.merge', 'refs/heads/%s' % branch)
  451. cmd = [GIT, 'symbolic-ref', 'HEAD', 'refs/heads/default']
  452. if subprocess.Popen(cmd, cwd = cwd).wait() != 0:
  453. raise CloneFailure()
  454. cmd = [GIT, 'read-tree', '--reset', '-u']
  455. if not quiet:
  456. cmd.append('-v')
  457. cmd.append('HEAD')
  458. if subprocess.Popen(cmd, cwd = cwd).wait() != 0:
  459. raise CloneFailure()
  460. def _FindRepo():
  461. """Look for a repo installation, starting at the current directory.
  462. """
  463. dir = os.getcwd()
  464. repo = None
  465. olddir = None
  466. while dir != '/' \
  467. and dir != olddir \
  468. and not repo:
  469. repo = os.path.join(dir, repodir, REPO_MAIN)
  470. if not os.path.isfile(repo):
  471. repo = None
  472. olddir = dir
  473. dir = os.path.dirname(dir)
  474. return (repo, os.path.join(dir, repodir))
  475. class _Options:
  476. help = False
  477. def _ParseArguments(args):
  478. cmd = None
  479. opt = _Options()
  480. arg = []
  481. for i in xrange(0, len(args)):
  482. a = args[i]
  483. if a == '-h' or a == '--help':
  484. opt.help = True
  485. elif not a.startswith('-'):
  486. cmd = a
  487. arg = args[i + 1:]
  488. break
  489. return cmd, opt, arg
  490. def _Usage():
  491. print >>sys.stderr,\
  492. """usage: repo COMMAND [ARGS]
  493. repo is not yet installed. Use "repo init" to install it here.
  494. The most commonly used repo commands are:
  495. init Install repo in the current working directory
  496. help Display detailed help on a command
  497. For access to the full online help, install repo ("repo init").
  498. """
  499. sys.exit(1)
  500. def _Help(args):
  501. if args:
  502. if args[0] == 'init':
  503. init_optparse.print_help()
  504. sys.exit(0)
  505. else:
  506. print >>sys.stderr,\
  507. "error: '%s' is not a bootstrap command.\n"\
  508. ' For access to online help, install repo ("repo init").'\
  509. % args[0]
  510. else:
  511. _Usage()
  512. sys.exit(1)
  513. def _NotInstalled():
  514. print >>sys.stderr,\
  515. 'error: repo is not installed. Use "repo init" to install it here.'
  516. sys.exit(1)
  517. def _NoCommands(cmd):
  518. print >>sys.stderr,\
  519. """error: command '%s' requires repo to be installed first.
  520. Use "repo init" to install it here.""" % cmd
  521. sys.exit(1)
  522. def _RunSelf(wrapper_path):
  523. my_dir = os.path.dirname(wrapper_path)
  524. my_main = os.path.join(my_dir, 'main.py')
  525. my_git = os.path.join(my_dir, '.git')
  526. if os.path.isfile(my_main) and os.path.isdir(my_git):
  527. for name in ['git_config.py',
  528. 'project.py',
  529. 'subcmds']:
  530. if not os.path.exists(os.path.join(my_dir, name)):
  531. return None, None
  532. return my_main, my_git
  533. return None, None
  534. def _SetDefaultsTo(gitdir):
  535. global REPO_URL
  536. global REPO_REV
  537. REPO_URL = gitdir
  538. proc = subprocess.Popen([GIT,
  539. '--git-dir=%s' % gitdir,
  540. 'symbolic-ref',
  541. 'HEAD'],
  542. stdout = subprocess.PIPE,
  543. stderr = subprocess.PIPE)
  544. REPO_REV = proc.stdout.read().strip()
  545. proc.stdout.close()
  546. proc.stderr.read()
  547. proc.stderr.close()
  548. if proc.wait() != 0:
  549. print >>sys.stderr, 'fatal: %s has no current branch' % gitdir
  550. sys.exit(1)
  551. def main(orig_args):
  552. main, dir = _FindRepo()
  553. cmd, opt, args = _ParseArguments(orig_args)
  554. wrapper_path = os.path.abspath(__file__)
  555. my_main, my_git = _RunSelf(wrapper_path)
  556. if not main:
  557. if opt.help:
  558. _Usage()
  559. if cmd == 'help':
  560. _Help(args)
  561. if not cmd:
  562. _NotInstalled()
  563. if cmd == 'init':
  564. if my_git:
  565. _SetDefaultsTo(my_git)
  566. try:
  567. _Init(args)
  568. except CloneFailure:
  569. for root, dirs, files in os.walk(repodir, topdown=False):
  570. for name in files:
  571. os.remove(os.path.join(root, name))
  572. for name in dirs:
  573. os.rmdir(os.path.join(root, name))
  574. os.rmdir(repodir)
  575. sys.exit(1)
  576. main, dir = _FindRepo()
  577. else:
  578. _NoCommands(cmd)
  579. if my_main:
  580. main = my_main
  581. ver_str = '.'.join(map(lambda x: str(x), VERSION))
  582. me = [main,
  583. '--repo-dir=%s' % dir,
  584. '--wrapper-version=%s' % ver_str,
  585. '--wrapper-path=%s' % wrapper_path,
  586. '--']
  587. me.extend(orig_args)
  588. me.extend(extra_args)
  589. try:
  590. os.execv(main, me)
  591. except OSError, e:
  592. print >>sys.stderr, "fatal: unable to start %s" % main
  593. print >>sys.stderr, "fatal: %s" % e
  594. sys.exit(148)
  595. if __name__ == '__main__':
  596. main(sys.argv[1:])