blob: df1818356dbcb6be7c60790cd8ce9471168eceb9 [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. Pearce0ed2bd12009-03-09 18:26:31 -070030import git_command
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080031from command import InteractiveCommand
32from command import MirrorSafeCommand
33from command import PagedCommand
Shawn O. Pearce7965f9f2008-10-29 15:20:02 -070034from editor import Editor
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
38from manifest import Manifest
39from pager import RunPager
40
41from subcmds import all as all_commands
42
43global_options = optparse.OptionParser(
44 usage="repo [-p|--paginate|--no-pager] COMMAND [ARGS]"
45 )
46global_options.add_option('-p', '--paginate',
47 dest='pager', action='store_true',
48 help='display command output in the pager')
49global_options.add_option('--no-pager',
50 dest='no_pager', action='store_true',
51 help='disable the pager')
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -070052global_options.add_option('--trace',
53 dest='trace', action='store_true',
54 help='trace git command execution')
Shawn O. Pearce47c1a632009-03-02 18:24:23 -080055global_options.add_option('--version',
56 dest='show_version', action='store_true',
57 help='display this version of repo')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070058
59class _Repo(object):
60 def __init__(self, repodir):
61 self.repodir = repodir
62 self.commands = all_commands
63
64 def _Run(self, argv):
65 name = None
66 glob = []
67
68 for i in xrange(0, len(argv)):
69 if not argv[i].startswith('-'):
70 name = argv[i]
71 if i > 0:
72 glob = argv[:i]
73 argv = argv[i + 1:]
74 break
75 if not name:
76 glob = argv
77 name = 'help'
78 argv = []
79 gopts, gargs = global_options.parse_args(glob)
80
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -070081 if gopts.trace:
82 git_command.TRACE = True
Shawn O. Pearce47c1a632009-03-02 18:24:23 -080083 if gopts.show_version:
84 if name == 'help':
85 name = 'version'
86 else:
87 print >>sys.stderr, 'fatal: invalid usage of --version'
88 sys.exit(1)
89
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070090 try:
91 cmd = self.commands[name]
92 except KeyError:
93 print >>sys.stderr,\
94 "repo: '%s' is not a repo command. See 'repo help'."\
95 % name
96 sys.exit(1)
97
98 cmd.repodir = self.repodir
99 cmd.manifest = Manifest(cmd.repodir)
Shawn O. Pearce7965f9f2008-10-29 15:20:02 -0700100 Editor.globalConfig = cmd.manifest.globalConfig
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700101
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800102 if not isinstance(cmd, MirrorSafeCommand) and cmd.manifest.IsMirror:
103 print >>sys.stderr, \
104 "fatal: '%s' requires a working directory"\
105 % name
106 sys.exit(1)
107
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700108 if not gopts.no_pager and not isinstance(cmd, InteractiveCommand):
109 config = cmd.manifest.globalConfig
110 if gopts.pager:
111 use_pager = True
112 else:
113 use_pager = config.GetBoolean('pager.%s' % name)
114 if use_pager is None:
115 use_pager = isinstance(cmd, PagedCommand)
116 if use_pager:
117 RunPager(config)
118
119 copts, cargs = cmd.OptionParser.parse_args(argv)
120 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:
214 repo._Run(argv)
215 except KeyboardInterrupt:
216 sys.exit(1)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800217 except RepoChangedException, rce:
218 # If repo changed, re-exec ourselves.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700219 #
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800220 argv = list(sys.argv)
221 argv.extend(rce.extra_args)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700222 try:
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800223 os.execv(__file__, argv)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700224 except OSError, e:
225 print >>sys.stderr, 'fatal: cannot restart repo after upgrade'
226 print >>sys.stderr, 'fatal: %s' % e
227 sys.exit(128)
228
229if __name__ == '__main__':
230 _Main(sys.argv[1:])