The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 1 | # |
| 2 | # Copyright (C) 2008 The Android Open Source Project |
| 3 | # |
| 4 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | # you may not use this file except in compliance with the License. |
| 6 | # You may obtain a copy of the License at |
| 7 | # |
| 8 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | # |
| 10 | # Unless required by applicable law or agreed to in writing, software |
| 11 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | # See the License for the specific language governing permissions and |
| 14 | # limitations under the License. |
| 15 | |
| 16 | import os |
Doug Anderson | 2630dd9 | 2011-04-07 13:36:30 -0700 | [diff] [blame] | 17 | import shutil |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 18 | import sys |
| 19 | |
| 20 | from color import Coloring |
Shawn O. Pearce | c95583b | 2009-03-03 17:47:06 -0800 | [diff] [blame] | 21 | from command import InteractiveCommand, MirrorSafeCommand |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 22 | from error import ManifestParseError |
Shawn O. Pearce | 350cde4 | 2009-04-16 11:21:18 -0700 | [diff] [blame] | 23 | from project import SyncBuffer |
Shawn O. Pearce | f322b9a | 2011-09-19 14:50:58 -0700 | [diff] [blame] | 24 | from git_config import GitConfig |
Shawn O. Pearce | 2ec00b9 | 2009-06-12 09:32:50 -0700 | [diff] [blame] | 25 | from git_command import git_require, MIN_GIT_VERSION |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 26 | from git_config import GitConfig |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 27 | |
Shawn O. Pearce | c95583b | 2009-03-03 17:47:06 -0800 | [diff] [blame] | 28 | class Init(InteractiveCommand, MirrorSafeCommand): |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 29 | common = True |
| 30 | helpSummary = "Initialize repo in the current directory" |
| 31 | helpUsage = """ |
| 32 | %prog [options] |
| 33 | """ |
| 34 | helpDescription = """ |
| 35 | The '%prog' command is run once to install and initialize repo. |
| 36 | The latest repo source code and manifest collection is downloaded |
| 37 | from the server and is installed in the .repo/ directory in the |
| 38 | current working directory. |
| 39 | |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 40 | The optional -u argument can be used to specify a URL to the |
| 41 | manifest project. It is also possible to have a git configuration |
| 42 | section as below to use 'identifier' as argument to -u: |
| 43 | |
| 44 | [repo-manifest "identifier"] |
| 45 | url = ... |
| 46 | server = ... |
| 47 | reference = ... |
| 48 | |
| 49 | Only the url is required - the others are optional. |
| 50 | |
| 51 | If no -u argument is specified, the 'repo-manifest' section named |
| 52 | 'default' will be used if present. |
| 53 | |
Shawn O. Pearce | 77bb4af | 2009-04-18 11:33:32 -0700 | [diff] [blame] | 54 | The optional -b argument can be used to select the manifest branch |
| 55 | to checkout and use. If no branch is specified, master is assumed. |
| 56 | |
| 57 | The optional -m argument can be used to specify an alternate manifest |
| 58 | to be used. If no manifest is specified, the manifest default.xml |
| 59 | will be used. |
| 60 | |
Shawn O. Pearce | 8844338 | 2010-10-08 10:02:09 +0200 | [diff] [blame] | 61 | The --reference option can be used to point to a directory that |
| 62 | has the content of a --mirror sync. This will make the working |
| 63 | directory use as much data as possible from the local reference |
| 64 | directory when fetching from the server. This will make the sync |
| 65 | go a lot faster by reducing data traffic on the network. |
| 66 | |
| 67 | |
Shawn O. Pearce | 77bb4af | 2009-04-18 11:33:32 -0700 | [diff] [blame] | 68 | Switching Manifest Branches |
| 69 | --------------------------- |
| 70 | |
| 71 | To switch to another manifest branch, `repo init -b otherbranch` |
| 72 | may be used in an existing client. However, as this only updates the |
| 73 | manifest, a subsequent `repo sync` (or `repo sync -d`) is necessary |
| 74 | to update the working directory files. |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 75 | """ |
| 76 | |
| 77 | def _Options(self, p): |
| 78 | # Logging |
| 79 | g = p.add_option_group('Logging options') |
| 80 | g.add_option('-q', '--quiet', |
| 81 | dest="quiet", action="store_true", default=False, |
| 82 | help="be quiet") |
| 83 | |
| 84 | # Manifest |
| 85 | g = p.add_option_group('Manifest options') |
| 86 | g.add_option('-u', '--manifest-url', |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 87 | dest='manifest_url', default='default', |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 88 | help='manifest repository location', metavar='URL') |
| 89 | g.add_option('-b', '--manifest-branch', |
| 90 | dest='manifest_branch', |
| 91 | help='manifest branch or revision', metavar='REVISION') |
| 92 | g.add_option('-m', '--manifest-name', |
| 93 | dest='manifest_name', default='default.xml', |
| 94 | help='initial manifest file', metavar='NAME.xml') |
Shawn O. Pearce | e284ad1 | 2008-11-04 07:37:10 -0800 | [diff] [blame] | 95 | g.add_option('--mirror', |
| 96 | dest='mirror', action='store_true', |
| 97 | help='mirror the forrest') |
Shawn O. Pearce | 8844338 | 2010-10-08 10:02:09 +0200 | [diff] [blame] | 98 | g.add_option('--reference', |
| 99 | dest='reference', |
| 100 | help='location of mirror directory', metavar='DIR') |
Doug Anderson | 30d4529 | 2011-05-04 15:01:04 -0700 | [diff] [blame] | 101 | g.add_option('--depth', type='int', default=None, |
| 102 | dest='depth', |
| 103 | help='create a shallow clone with given depth; see git clone') |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 104 | |
| 105 | # Tool |
Shawn O. Pearce | fd89b67 | 2009-04-18 11:28:57 -0700 | [diff] [blame] | 106 | g = p.add_option_group('repo Version options') |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 107 | g.add_option('--repo-url', |
| 108 | dest='repo_url', |
| 109 | help='repo repository location', metavar='URL') |
| 110 | g.add_option('--repo-branch', |
| 111 | dest='repo_branch', |
| 112 | help='repo branch or revision', metavar='REVISION') |
| 113 | g.add_option('--no-repo-verify', |
| 114 | dest='no_repo_verify', action='store_true', |
| 115 | help='do not verify repo source code') |
| 116 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 117 | def _SyncManifest(self, opt): |
| 118 | m = self.manifest.manifestProject |
Shawn O. Pearce | 5470df6 | 2009-03-09 18:51:58 -0700 | [diff] [blame] | 119 | is_new = not m.Exists |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 120 | manifest_server = None |
| 121 | |
| 122 | # The manifest url may point out a manifest section in the config |
| 123 | key = 'repo-manifest.%s.' % opt.manifest_url |
| 124 | if GitConfig.ForUser().GetString(key + 'url'): |
| 125 | opt.manifest_url = GitConfig.ForUser().GetString(key + 'url') |
| 126 | |
| 127 | # Also copy other options to the manifest config if not specified already. |
| 128 | if not opt.reference: |
| 129 | opt.reference = GitConfig.ForUser().GetString(key + 'reference') |
| 130 | manifest_server = GitConfig.ForUser().GetString(key + 'server') |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 131 | |
Shawn O. Pearce | 5470df6 | 2009-03-09 18:51:58 -0700 | [diff] [blame] | 132 | if is_new: |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 133 | if not opt.manifest_url or opt.manifest_url == 'default': |
| 134 | print >>sys.stderr, """\ |
| 135 | fatal: missing manifest url (-u) and no default found. |
| 136 | |
| 137 | tip: The global git configuration key 'repo-manifest.default.url' can |
| 138 | be used to specify a default url.""" |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 139 | sys.exit(1) |
| 140 | |
| 141 | if not opt.quiet: |
Shawn O. Pearce | f322b9a | 2011-09-19 14:50:58 -0700 | [diff] [blame] | 142 | print >>sys.stderr, 'Get %s' \ |
| 143 | % GitConfig.ForUser().UrlInsteadOf(opt.manifest_url) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 144 | m._InitGitDir() |
| 145 | |
| 146 | if opt.manifest_branch: |
Shawn O. Pearce | 3c8dea1 | 2009-05-29 18:38:17 -0700 | [diff] [blame] | 147 | m.revisionExpr = opt.manifest_branch |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 148 | else: |
Shawn O. Pearce | 3c8dea1 | 2009-05-29 18:38:17 -0700 | [diff] [blame] | 149 | m.revisionExpr = 'refs/heads/master' |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 150 | else: |
| 151 | if opt.manifest_branch: |
Shawn O. Pearce | 3c8dea1 | 2009-05-29 18:38:17 -0700 | [diff] [blame] | 152 | m.revisionExpr = opt.manifest_branch |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 153 | else: |
| 154 | m.PreSync() |
| 155 | |
| 156 | if opt.manifest_url: |
| 157 | r = m.GetRemote(m.remote.name) |
| 158 | r.url = opt.manifest_url |
| 159 | r.ResetFetch() |
| 160 | r.Save() |
| 161 | |
Victor Boivie | ee1c2f5 | 2011-04-19 10:50:12 +0200 | [diff] [blame^] | 162 | if manifest_server: |
| 163 | m.config.SetString('repo.manifest-server', manifest_server) |
| 164 | |
Shawn O. Pearce | 8844338 | 2010-10-08 10:02:09 +0200 | [diff] [blame] | 165 | if opt.reference: |
| 166 | m.config.SetString('repo.reference', opt.reference) |
| 167 | |
Shawn O. Pearce | e284ad1 | 2008-11-04 07:37:10 -0800 | [diff] [blame] | 168 | if opt.mirror: |
Shawn O. Pearce | 5470df6 | 2009-03-09 18:51:58 -0700 | [diff] [blame] | 169 | if is_new: |
| 170 | m.config.SetString('repo.mirror', 'true') |
| 171 | else: |
| 172 | print >>sys.stderr, 'fatal: --mirror not supported on existing client' |
| 173 | sys.exit(1) |
Shawn O. Pearce | e284ad1 | 2008-11-04 07:37:10 -0800 | [diff] [blame] | 174 | |
Shawn O. Pearce | f322b9a | 2011-09-19 14:50:58 -0700 | [diff] [blame] | 175 | if not m.Sync_NetworkHalf(is_new=is_new): |
Shawn O. Pearce | 1fc99f4 | 2009-03-17 08:06:18 -0700 | [diff] [blame] | 176 | r = m.GetRemote(m.remote.name) |
| 177 | print >>sys.stderr, 'fatal: cannot obtain manifest %s' % r.url |
Doug Anderson | 2630dd9 | 2011-04-07 13:36:30 -0700 | [diff] [blame] | 178 | |
| 179 | # Better delete the manifest git dir if we created it; otherwise next |
| 180 | # time (when user fixes problems) we won't go through the "is_new" logic. |
| 181 | if is_new: |
| 182 | shutil.rmtree(m.gitdir) |
Shawn O. Pearce | 1fc99f4 | 2009-03-17 08:06:18 -0700 | [diff] [blame] | 183 | sys.exit(1) |
| 184 | |
Shawn O. Pearce | 350cde4 | 2009-04-16 11:21:18 -0700 | [diff] [blame] | 185 | syncbuf = SyncBuffer(m.config) |
| 186 | m.Sync_LocalHalf(syncbuf) |
| 187 | syncbuf.Finish() |
| 188 | |
Shawn O. Pearce | df01883 | 2009-03-17 08:15:27 -0700 | [diff] [blame] | 189 | if is_new or m.CurrentBranch is None: |
Shawn O. Pearce | 0a389e9 | 2009-04-10 16:21:18 -0700 | [diff] [blame] | 190 | if not m.StartBranch('default'): |
| 191 | print >>sys.stderr, 'fatal: cannot create default in manifest' |
| 192 | sys.exit(1) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 193 | |
| 194 | def _LinkManifest(self, name): |
| 195 | if not name: |
| 196 | print >>sys.stderr, 'fatal: manifest name (-m) is required.' |
| 197 | sys.exit(1) |
| 198 | |
| 199 | try: |
| 200 | self.manifest.Link(name) |
| 201 | except ManifestParseError, e: |
| 202 | print >>sys.stderr, "fatal: manifest '%s' not available" % name |
| 203 | print >>sys.stderr, 'fatal: %s' % str(e) |
| 204 | sys.exit(1) |
| 205 | |
Shawn O. Pearce | 37dbf2b | 2009-07-02 10:53:04 -0700 | [diff] [blame] | 206 | def _Prompt(self, prompt, value): |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 207 | mp = self.manifest.manifestProject |
| 208 | |
| 209 | sys.stdout.write('%-10s [%s]: ' % (prompt, value)) |
| 210 | a = sys.stdin.readline().strip() |
Shawn O. Pearce | 37dbf2b | 2009-07-02 10:53:04 -0700 | [diff] [blame] | 211 | if a == '': |
| 212 | return value |
| 213 | return a |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 214 | |
| 215 | def _ConfigureUser(self): |
| 216 | mp = self.manifest.manifestProject |
| 217 | |
Shawn O. Pearce | 37dbf2b | 2009-07-02 10:53:04 -0700 | [diff] [blame] | 218 | while True: |
| 219 | print '' |
| 220 | name = self._Prompt('Your Name', mp.UserName) |
| 221 | email = self._Prompt('Your Email', mp.UserEmail) |
| 222 | |
| 223 | print '' |
| 224 | print 'Your identity is: %s <%s>' % (name, email) |
Mike Frysinger | e931127 | 2011-08-11 15:46:43 -0400 | [diff] [blame] | 225 | sys.stdout.write('is this correct [y/N]? ') |
Nico Sallembien | 6d7508b | 2010-04-01 11:03:53 -0700 | [diff] [blame] | 226 | a = sys.stdin.readline().strip() |
| 227 | if a in ('yes', 'y', 't', 'true'): |
Shawn O. Pearce | 37dbf2b | 2009-07-02 10:53:04 -0700 | [diff] [blame] | 228 | break |
| 229 | |
| 230 | if name != mp.UserName: |
| 231 | mp.config.SetString('user.name', name) |
| 232 | if email != mp.UserEmail: |
| 233 | mp.config.SetString('user.email', email) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 234 | |
| 235 | def _HasColorSet(self, gc): |
| 236 | for n in ['ui', 'diff', 'status']: |
| 237 | if gc.Has('color.%s' % n): |
| 238 | return True |
| 239 | return False |
| 240 | |
| 241 | def _ConfigureColor(self): |
| 242 | gc = self.manifest.globalConfig |
| 243 | if self._HasColorSet(gc): |
| 244 | return |
| 245 | |
| 246 | class _Test(Coloring): |
| 247 | def __init__(self): |
| 248 | Coloring.__init__(self, gc, 'test color display') |
| 249 | self._on = True |
| 250 | out = _Test() |
| 251 | |
| 252 | print '' |
| 253 | print "Testing colorized output (for 'repo diff', 'repo status'):" |
| 254 | |
| 255 | for c in ['black','red','green','yellow','blue','magenta','cyan']: |
| 256 | out.write(' ') |
| 257 | out.printer(fg=c)(' %-6s ', c) |
| 258 | out.write(' ') |
| 259 | out.printer(fg='white', bg='black')(' %s ' % 'white') |
| 260 | out.nl() |
| 261 | |
| 262 | for c in ['bold','dim','ul','reverse']: |
| 263 | out.write(' ') |
| 264 | out.printer(fg='black', attr=c)(' %-6s ', c) |
| 265 | out.nl() |
| 266 | |
Mike Frysinger | e931127 | 2011-08-11 15:46:43 -0400 | [diff] [blame] | 267 | sys.stdout.write('Enable color display in this user account (y/N)? ') |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 268 | a = sys.stdin.readline().strip().lower() |
| 269 | if a in ('y', 'yes', 't', 'true', 'on'): |
| 270 | gc.SetString('color.ui', 'auto') |
| 271 | |
Doug Anderson | 30d4529 | 2011-05-04 15:01:04 -0700 | [diff] [blame] | 272 | def _ConfigureDepth(self, opt): |
| 273 | """Configure the depth we'll sync down. |
| 274 | |
| 275 | Args: |
| 276 | opt: Options from optparse. We care about opt.depth. |
| 277 | """ |
| 278 | # Opt.depth will be non-None if user actually passed --depth to repo init. |
| 279 | if opt.depth is not None: |
| 280 | if opt.depth > 0: |
| 281 | # Positive values will set the depth. |
| 282 | depth = str(opt.depth) |
| 283 | else: |
| 284 | # Negative numbers will clear the depth; passing None to SetString |
| 285 | # will do that. |
| 286 | depth = None |
| 287 | |
| 288 | # We store the depth in the main manifest project. |
| 289 | self.manifest.manifestProject.config.SetString('repo.depth', depth) |
| 290 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 291 | def Execute(self, opt, args): |
Shawn O. Pearce | 2ec00b9 | 2009-06-12 09:32:50 -0700 | [diff] [blame] | 292 | git_require(MIN_GIT_VERSION, fail=True) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 293 | self._SyncManifest(opt) |
| 294 | self._LinkManifest(opt.manifest_name) |
| 295 | |
Shawn O. Pearce | 8630f39 | 2009-03-19 10:17:12 -0700 | [diff] [blame] | 296 | if os.isatty(0) and os.isatty(1) and not self.manifest.IsMirror: |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 297 | self._ConfigureUser() |
| 298 | self._ConfigureColor() |
| 299 | |
Doug Anderson | 30d4529 | 2011-05-04 15:01:04 -0700 | [diff] [blame] | 300 | self._ConfigureDepth(opt) |
| 301 | |
Shawn O. Pearce | 8630f39 | 2009-03-19 10:17:12 -0700 | [diff] [blame] | 302 | if self.manifest.IsMirror: |
Shawn O. Pearce | e284ad1 | 2008-11-04 07:37:10 -0800 | [diff] [blame] | 303 | type = 'mirror ' |
| 304 | else: |
| 305 | type = '' |
| 306 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 307 | print '' |
Shawn O. Pearce | e284ad1 | 2008-11-04 07:37:10 -0800 | [diff] [blame] | 308 | print 'repo %sinitialized in %s' % (type, self.manifest.topdir) |