blob: 07b26ef7a01c75ef272e6cc51018e0da6340ec8a [file] [log] [blame]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001#!/bin/sh
2#
3# Copyright (C) 2008 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17magic='--calling-python-from-/bin/sh--'
Shawn O. Pearce7542d662008-10-21 07:11:36 -070018"""exec" python -E "$0" "$@" """#$magic"
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070019if __name__ == '__main__':
20 import sys
21 if sys.argv[-1] == '#%s' % magic:
22 del sys.argv[-1]
23del magic
24
25import optparse
26import os
27import re
28import sys
29
Shawn O. Pearcead3193a2009-04-18 09:54:51 -070030from trace import SetTrace
Doug Anderson0048b692010-12-21 13:39:23 -080031from git_config import init_ssh, close_ssh
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080032from command import InteractiveCommand
33from command import MirrorSafeCommand
34from command import PagedCommand
Shawn O. Pearce559b8462009-03-02 12:56:08 -080035from error import ManifestInvalidRevisionError
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070036from error import NoSuchProjectError
37from error import RepoChangedException
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070038from pager import RunPager
39
40from subcmds import all as all_commands
41
42global_options = optparse.OptionParser(
43 usage="repo [-p|--paginate|--no-pager] COMMAND [ARGS]"
44 )
45global_options.add_option('-p', '--paginate',
46 dest='pager', action='store_true',
47 help='display command output in the pager')
48global_options.add_option('--no-pager',
49 dest='no_pager', action='store_true',
50 help='disable the pager')
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -070051global_options.add_option('--trace',
52 dest='trace', action='store_true',
53 help='trace git command execution')
Shawn O. Pearce47c1a632009-03-02 18:24:23 -080054global_options.add_option('--version',
55 dest='show_version', action='store_true',
56 help='display this version of repo')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070057
58class _Repo(object):
59 def __init__(self, repodir):
60 self.repodir = repodir
61 self.commands = all_commands
Mike Lockwood33f0e782009-07-14 15:23:39 -040062 # add 'branch' as an alias for 'branches'
63 all_commands['branch'] = all_commands['branches']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070064
65 def _Run(self, argv):
66 name = None
67 glob = []
68
69 for i in xrange(0, len(argv)):
70 if not argv[i].startswith('-'):
71 name = argv[i]
72 if i > 0:
73 glob = argv[:i]
74 argv = argv[i + 1:]
75 break
76 if not name:
77 glob = argv
78 name = 'help'
79 argv = []
80 gopts, gargs = global_options.parse_args(glob)
81
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -070082 if gopts.trace:
Shawn O. Pearcead3193a2009-04-18 09:54:51 -070083 SetTrace()
Shawn O. Pearce47c1a632009-03-02 18:24:23 -080084 if gopts.show_version:
85 if name == 'help':
86 name = 'version'
87 else:
88 print >>sys.stderr, 'fatal: invalid usage of --version'
89 sys.exit(1)
90
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070091 try:
92 cmd = self.commands[name]
93 except KeyError:
94 print >>sys.stderr,\
95 "repo: '%s' is not a repo command. See 'repo help'."\
96 % name
97 sys.exit(1)
98
99 cmd.repodir = self.repodir
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700100
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800101 if not isinstance(cmd, MirrorSafeCommand) and cmd.manifest.IsMirror:
102 print >>sys.stderr, \
103 "fatal: '%s' requires a working directory"\
104 % name
105 sys.exit(1)
106
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700107 copts, cargs = cmd.OptionParser.parse_args(argv)
108
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700109 if not gopts.no_pager and not isinstance(cmd, InteractiveCommand):
110 config = cmd.manifest.globalConfig
111 if gopts.pager:
112 use_pager = True
113 else:
114 use_pager = config.GetBoolean('pager.%s' % name)
115 if use_pager is None:
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700116 use_pager = cmd.WantPager(copts)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700117 if use_pager:
118 RunPager(config)
119
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700120 try:
121 cmd.Execute(copts, cargs)
Shawn O. Pearce559b8462009-03-02 12:56:08 -0800122 except ManifestInvalidRevisionError, e:
123 print >>sys.stderr, 'error: %s' % str(e)
124 sys.exit(1)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700125 except NoSuchProjectError, e:
126 if e.name:
127 print >>sys.stderr, 'error: project %s not found' % e.name
128 else:
129 print >>sys.stderr, 'error: no project in current directory'
130 sys.exit(1)
131
132def _MyWrapperPath():
133 return os.path.join(os.path.dirname(__file__), 'repo')
134
135def _CurrentWrapperVersion():
136 VERSION = None
137 pat = re.compile(r'^VERSION *=')
138 fd = open(_MyWrapperPath())
139 for line in fd:
140 if pat.match(line):
141 fd.close()
142 exec line
143 return VERSION
144 raise NameError, 'No VERSION in repo script'
145
146def _CheckWrapperVersion(ver, repo_path):
147 if not repo_path:
148 repo_path = '~/bin/repo'
149
150 if not ver:
151 print >>sys.stderr, 'no --wrapper-version argument'
152 sys.exit(1)
153
154 exp = _CurrentWrapperVersion()
155 ver = tuple(map(lambda x: int(x), ver.split('.')))
156 if len(ver) == 1:
157 ver = (0, ver[0])
158
159 if exp[0] > ver[0] or ver < (0, 4):
160 exp_str = '.'.join(map(lambda x: str(x), exp))
161 print >>sys.stderr, """
162!!! A new repo command (%5s) is available. !!!
163!!! You must upgrade before you can continue: !!!
164
165 cp %s %s
166""" % (exp_str, _MyWrapperPath(), repo_path)
167 sys.exit(1)
168
169 if exp > ver:
170 exp_str = '.'.join(map(lambda x: str(x), exp))
171 print >>sys.stderr, """
172... A new repo command (%5s) is available.
173... You should upgrade soon:
174
175 cp %s %s
176""" % (exp_str, _MyWrapperPath(), repo_path)
177
178def _CheckRepoDir(dir):
179 if not dir:
180 print >>sys.stderr, 'no --repo-dir argument'
181 sys.exit(1)
182
183def _PruneOptions(argv, opt):
184 i = 0
185 while i < len(argv):
186 a = argv[i]
187 if a == '--':
188 break
189 if a.startswith('--'):
190 eq = a.find('=')
191 if eq > 0:
192 a = a[0:eq]
193 if not opt.has_option(a):
194 del argv[i]
195 continue
196 i += 1
197
198def _Main(argv):
199 opt = optparse.OptionParser(usage="repo wrapperinfo -- ...")
200 opt.add_option("--repo-dir", dest="repodir",
201 help="path to .repo/")
202 opt.add_option("--wrapper-version", dest="wrapper_version",
203 help="version of the wrapper script")
204 opt.add_option("--wrapper-path", dest="wrapper_path",
205 help="location of the wrapper script")
206 _PruneOptions(argv, opt)
207 opt, argv = opt.parse_args(argv)
208
209 _CheckWrapperVersion(opt.wrapper_version, opt.wrapper_path)
210 _CheckRepoDir(opt.repodir)
211
212 repo = _Repo(opt.repodir)
213 try:
Shawn O. Pearcefb231612009-04-10 18:53:46 -0700214 try:
Doug Anderson0048b692010-12-21 13:39:23 -0800215 init_ssh()
Shawn O. Pearcefb231612009-04-10 18:53:46 -0700216 repo._Run(argv)
217 finally:
218 close_ssh()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700219 except KeyboardInterrupt:
220 sys.exit(1)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800221 except RepoChangedException, rce:
222 # If repo changed, re-exec ourselves.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700223 #
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800224 argv = list(sys.argv)
225 argv.extend(rce.extra_args)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700226 try:
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800227 os.execv(__file__, argv)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700228 except OSError, e:
229 print >>sys.stderr, 'fatal: cannot restart repo after upgrade'
230 print >>sys.stderr, 'fatal: %s' % e
231 sys.exit(128)
232
233if __name__ == '__main__':
234 _Main(sys.argv[1:])