diff options
Diffstat (limited to 'main.py')
-rwxr-xr-x | main.py | 27 |
1 files changed, 26 insertions, 1 deletions
@@ -26,6 +26,7 @@ import getpass | |||
26 | import netrc | 26 | import netrc |
27 | import optparse | 27 | import optparse |
28 | import os | 28 | import os |
29 | import shlex | ||
29 | import sys | 30 | import sys |
30 | import textwrap | 31 | import textwrap |
31 | import time | 32 | import time |
@@ -48,7 +49,7 @@ from color import SetDefaultColoring | |||
48 | import event_log | 49 | import event_log |
49 | from repo_trace import SetTrace | 50 | from repo_trace import SetTrace |
50 | from git_command import user_agent | 51 | from git_command import user_agent |
51 | from git_config import init_ssh, close_ssh | 52 | from git_config import init_ssh, close_ssh, RepoConfig |
52 | from command import InteractiveCommand | 53 | from command import InteractiveCommand |
53 | from command import MirrorSafeCommand | 54 | from command import MirrorSafeCommand |
54 | from command import GitcAvailableCommand, GitcClientCommand | 55 | from command import GitcAvailableCommand, GitcClientCommand |
@@ -155,6 +156,9 @@ class _Repo(object): | |||
155 | argv = [] | 156 | argv = [] |
156 | gopts, _gargs = global_options.parse_args(glob) | 157 | gopts, _gargs = global_options.parse_args(glob) |
157 | 158 | ||
159 | name, alias_args = self._ExpandAlias(name) | ||
160 | argv = alias_args + argv | ||
161 | |||
158 | if gopts.help: | 162 | if gopts.help: |
159 | global_options.print_help() | 163 | global_options.print_help() |
160 | commands = ' '.join(sorted(self.commands)) | 164 | commands = ' '.join(sorted(self.commands)) |
@@ -165,6 +169,27 @@ class _Repo(object): | |||
165 | 169 | ||
166 | return (name, gopts, argv) | 170 | return (name, gopts, argv) |
167 | 171 | ||
172 | def _ExpandAlias(self, name): | ||
173 | """Look up user registered aliases.""" | ||
174 | # We don't resolve aliases for existing subcommands. This matches git. | ||
175 | if name in self.commands: | ||
176 | return name, [] | ||
177 | |||
178 | key = 'alias.%s' % (name,) | ||
179 | alias = RepoConfig.ForRepository(self.repodir).GetString(key) | ||
180 | if alias is None: | ||
181 | alias = RepoConfig.ForUser().GetString(key) | ||
182 | if alias is None: | ||
183 | return name, [] | ||
184 | |||
185 | args = alias.strip().split(' ', 1) | ||
186 | name = args[0] | ||
187 | if len(args) == 2: | ||
188 | args = shlex.split(args[1]) | ||
189 | else: | ||
190 | args = [] | ||
191 | return name, args | ||
192 | |||
168 | def _Run(self, name, gopts, argv): | 193 | def _Run(self, name, gopts, argv): |
169 | """Execute the requested subcommand.""" | 194 | """Execute the requested subcommand.""" |
170 | result = 0 | 195 | result = 0 |