blob: cd5e3c3e3c5005035b7ec23b48a0febdcae84cf3 [file] [log] [blame]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001#
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
16import os
17import optparse
Conley Owensd21720d2012-04-16 11:02:21 -070018import platform
Colin Cross5acde752012-03-28 20:15:45 -070019import re
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070020import sys
21
22from error import NoSuchProjectError
Colin Cross5acde752012-03-28 20:15:45 -070023from error import InvalidProjectGroupsError
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070024
David Pursehouseb148ac92012-11-16 09:33:39 +090025
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070026class Command(object):
27 """Base class for any command line action in repo.
28 """
29
30 common = False
31 manifest = None
32 _optparse = None
33
Shawn O. Pearcedb45da12009-04-18 13:49:13 -070034 def WantPager(self, opt):
35 return False
36
David Pursehouseb148ac92012-11-16 09:33:39 +090037 def ReadEnvironmentOptions(self, opts):
38 """ Set options from environment variables. """
39
40 env_options = self._RegisteredEnvironmentOptions()
41
42 for env_key, opt_key in env_options.items():
43 # Get the user-set option value if any
44 opt_value = getattr(opts, opt_key)
45
46 # If the value is set, it means the user has passed it as a command
47 # line option, and we should use that. Otherwise we can try to set it
48 # with the value from the corresponding environment variable.
49 if opt_value is not None:
50 continue
51
52 env_value = os.environ.get(env_key)
53 if env_value is not None:
54 setattr(opts, opt_key, env_value)
55
56 return opts
57
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070058 @property
59 def OptionParser(self):
60 if self._optparse is None:
61 try:
62 me = 'repo %s' % self.NAME
63 usage = self.helpUsage.strip().replace('%prog', me)
64 except AttributeError:
65 usage = 'repo %s' % self.NAME
66 self._optparse = optparse.OptionParser(usage = usage)
67 self._Options(self._optparse)
68 return self._optparse
69
70 def _Options(self, p):
71 """Initialize the option parser.
72 """
73
David Pursehouseb148ac92012-11-16 09:33:39 +090074 def _RegisteredEnvironmentOptions(self):
75 """Get options that can be set from environment variables.
76
77 Return a dictionary mapping environment variable name
78 to option key name that it can override.
79
80 Example: {'REPO_MY_OPTION': 'my_option'}
81
82 Will allow the option with key value 'my_option' to be set
83 from the value in the environment variable named 'REPO_MY_OPTION'.
84
85 Note: This does not work properly for options that are explicitly
86 set to None by the user, or options that are defined with a
87 default value other than None.
88
89 """
90 return {}
91
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070092 def Usage(self):
93 """Display usage and terminate.
94 """
95 self.OptionParser.print_usage()
96 sys.exit(1)
97
98 def Execute(self, opt, args):
99 """Perform the action, after option parsing is complete.
100 """
101 raise NotImplementedError
Conley Owens971de8e2012-04-16 10:36:08 -0700102
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800103 def _ResetPathToProjectMap(self, projects):
104 self._by_path = dict((p.worktree, p) for p in projects)
105
106 def _UpdatePathToProjectMap(self, project):
107 self._by_path[project.worktree] = project
108
Simran Basib9a1b732015-08-20 12:19:28 -0700109 def _GetProjectByPath(self, manifest, path):
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800110 project = None
111 if os.path.exists(path):
112 oldpath = None
113 while path \
114 and path != oldpath \
Simran Basib9a1b732015-08-20 12:19:28 -0700115 and path != manifest.topdir:
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800116 try:
117 project = self._by_path[path]
118 break
119 except KeyError:
120 oldpath = path
121 path = os.path.dirname(path)
122 else:
123 try:
124 project = self._by_path[path]
125 except KeyError:
126 pass
127 return project
128
Simran Basib9a1b732015-08-20 12:19:28 -0700129 def GetProjects(self, args, manifest=None, groups='', missing_ok=False,
130 submodules_ok=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700131 """A list of projects that match the arguments.
132 """
Simran Basib9a1b732015-08-20 12:19:28 -0700133 if not manifest:
134 manifest = self.manifest
135 all_projects_list = manifest.projects
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700136 result = []
137
Simran Basib9a1b732015-08-20 12:19:28 -0700138 mp = manifest.manifestProject
Colin Cross5acde752012-03-28 20:15:45 -0700139
Graham Christensen0369a062015-07-29 17:02:54 -0500140 if not groups:
141 groups = mp.config.GetString('manifest.groups')
Colin Crossc39864f2012-04-23 13:41:58 -0700142 if not groups:
David Holmer0a1c6a12012-11-14 19:19:00 -0500143 groups = 'default,platform-' + platform.system().lower()
David Pursehouse1d947b32012-10-25 12:23:11 +0900144 groups = [x for x in re.split(r'[,\s]+', groups) if x]
Colin Cross5acde752012-03-28 20:15:45 -0700145
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700146 if not args:
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800147 derived_projects = {}
148 for project in all_projects_list:
149 if submodules_ok or project.sync_s:
150 derived_projects.update((p.name, p)
151 for p in project.GetDerivedSubprojects())
152 all_projects_list.extend(derived_projects.values())
153 for project in all_projects_list:
Colin Cross5acde752012-03-28 20:15:45 -0700154 if ((missing_ok or project.Exists) and
155 project.MatchesGroups(groups)):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700156 result.append(project)
157 else:
David James8d201162013-10-11 17:03:19 -0700158 self._ResetPathToProjectMap(all_projects_list)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700159
160 for arg in args:
Simran Basib9a1b732015-08-20 12:19:28 -0700161 projects = manifest.GetProjectsWithName(arg)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700162
David James8d201162013-10-11 17:03:19 -0700163 if not projects:
Anthony Newnamdf14a702011-01-09 17:31:57 -0800164 path = os.path.abspath(arg).replace('\\', '/')
Simran Basib9a1b732015-08-20 12:19:28 -0700165 project = self._GetProjectByPath(manifest, path)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700166
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800167 # If it's not a derived project, update path->project mapping and
168 # search again, as arg might actually point to a derived subproject.
169 if (project and not project.Derived and
170 (submodules_ok or project.sync_s)):
171 search_again = False
172 for subproject in project.GetDerivedSubprojects():
173 self._UpdatePathToProjectMap(subproject)
174 search_again = True
175 if search_again:
Simran Basib9a1b732015-08-20 12:19:28 -0700176 project = self._GetProjectByPath(manifest, path) or project
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700177
David James8d201162013-10-11 17:03:19 -0700178 if project:
179 projects = [project]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700180
David James8d201162013-10-11 17:03:19 -0700181 if not projects:
182 raise NoSuchProjectError(arg)
183
184 for project in projects:
185 if not missing_ok and not project.Exists:
186 raise NoSuchProjectError(arg)
187 if not project.MatchesGroups(groups):
188 raise InvalidProjectGroupsError(arg)
189
190 result.extend(projects)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700191
192 def _getpath(x):
193 return x.relpath
194 result.sort(key=_getpath)
195 return result
196
Zhiguang Lia8864fb2013-03-15 10:32:10 +0800197 def FindProjects(self, args):
198 result = []
David Pursehouse84c4d3c2013-04-30 10:57:37 +0900199 patterns = [re.compile(r'%s' % a, re.IGNORECASE) for a in args]
Zhiguang Lia8864fb2013-03-15 10:32:10 +0800200 for project in self.GetProjects(''):
David Pursehouse84c4d3c2013-04-30 10:57:37 +0900201 for pattern in patterns:
Zhiguang Lia8864fb2013-03-15 10:32:10 +0800202 if pattern.search(project.name) or pattern.search(project.relpath):
203 result.append(project)
204 break
205 result.sort(key=lambda project: project.relpath)
206 return result
207
David Pursehouse4f7bdea2012-10-22 12:50:15 +0900208# pylint: disable=W0223
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900209# Pylint warns that the `InteractiveCommand` and `PagedCommand` classes do not
210# override method `Execute` which is abstract in `Command`. Since that method
211# is always implemented in classes derived from `InteractiveCommand` and
212# `PagedCommand`, this warning can be suppressed.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700213class InteractiveCommand(Command):
214 """Command which requires user interaction on the tty and
215 must not run within a pager, even if the user asks to.
216 """
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700217 def WantPager(self, opt):
218 return False
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700219
220class PagedCommand(Command):
221 """Command which defaults to output in a pager, as its
222 display tends to be larger than one screen full.
223 """
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700224 def WantPager(self, opt):
225 return True
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800226
David Pursehouse4f7bdea2012-10-22 12:50:15 +0900227# pylint: enable=W0223
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900228
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800229class MirrorSafeCommand(object):
230 """Command permits itself to run within a mirror,
231 and does not require a working directory.
232 """
Dan Willemsen9ff2ece2015-08-31 15:45:06 -0700233
Dan Willemsen79360642015-08-31 15:45:06 -0700234class GitcAvailableCommand(object):
Dan Willemsen9ff2ece2015-08-31 15:45:06 -0700235 """Command that requires GITC to be available, but does
236 not require the local client to be a GITC client.
237 """
Dan Willemsen79360642015-08-31 15:45:06 -0700238
239class GitcClientCommand(object):
240 """Command that requires the local client to be a GITC
241 client.
242 """