1
# Copyright (C) 2004, 2005 Aaron Bentley
2
# <aaron.bentley@utoronto.ca>
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23
from itertools import chain
24
from bzrlib.errors import BzrError
25
from bzrlib.commands import get_cmd_object, get_all_cmds, get_alias
26
from bzrlib.branch import Branch
28
SHELL_BLACKLIST = set(['rm', 'ls'])
29
COMPLETION_BLACKLIST = set(['shell'])
31
class BlackListedCommand(BzrError):
32
def __init__(self, command):
33
BzrError.__init__(self, "The command %s is blacklisted for shell use" %
36
class CompletionContext(object):
37
def __init__(self, text, command=None, prev_opt=None, arg_pos=None):
39
self.command = command
40
self.prev_opt = prev_opt
43
def get_completions(self):
45
return self.get_completions_or_raise()
50
def get_option_completions(self):
52
command_obj = get_cmd_object(self.command)
55
opts = [o+" " for o in iter_opt_completions(command_obj)]
56
return list(filter_completions(opts, self.text))
58
def get_completions_or_raise(self):
59
if self.command is None:
60
iter = (c+" " for c in iter_command_names() if
61
c not in COMPLETION_BLACKLIST)
62
return list(filter_completions(iter, self.text))
63
if self.prev_opt is None:
64
completions = self.get_option_completions()
65
if self.command == "cd":
66
iter = iter_dir_completions(self.text)
67
completions.extend(list(filter_completions(iter, self.text)))
69
iter = iter_file_completions(self.text)
70
completions.extend([f+" " for f in
71
filter_completions(iter, self.text)])
75
class PromptCmd(cmd.Cmd):
77
cmd.Cmd.__init__(self)
80
self.tree = WorkingTree.open_containing('.')[0]
85
self.identchars += '-'
86
self.history_file = os.path.expanduser("~/.bazaar/shell-history")
87
readline.set_completer_delims(string.whitespace)
88
if os.access(self.history_file, os.R_OK) and \
89
os.path.isfile(self.history_file):
90
readline.read_history_file(self.history_file)
91
self.cwd = os.getcwd()
93
def write_history(self):
94
readline.write_history_file(self.history_file)
96
def do_quit(self, args):
100
def do_exit(self, args):
103
def do_EOF(self, args):
107
def postcmd(self, line, bar):
111
def set_prompt(self):
112
if self.tree is not None:
114
prompt_data = (self.tree.branch.nick, self.tree.branch.revno(),
115
self.tree.branch.relpath('.'))
116
prompt = " %s:%d/%s" % prompt_data
121
self.prompt = "bzr%s> " % prompt
123
def set_title(self, command=None):
125
b = Branch.open_containing('.')[0]
126
version = "%s:%d" % (b.nick, b.revno())
128
version = "[no version]"
131
sys.stdout.write(terminal.term_title("bzr %s %s" % (command, version)))
133
def do_cd(self, line):
136
line = os.path.expanduser(line)
137
if os.path.isabs(line):
140
newcwd = self.cwd+'/'+line
141
newcwd = os.path.normpath(newcwd)
148
self.tree = WorkingTree.open_containing(".")[0]
152
def do_help(self, line):
153
self.default("help "+line)
155
def default(self, line):
157
alias_args = get_alias(args[0])
158
if alias_args is not None:
159
args[0] = alias_args.pop(0)
161
commandname = args.pop(0)
162
for char in ('|', '<', '>'):
163
commandname = commandname.split(char)[0]
164
if commandname[-1] in ('|', '<', '>'):
165
commandname = commandname[:-1]
167
if commandname in SHELL_BLACKLIST:
168
raise BlackListedCommand(commandname)
169
cmd_obj = get_cmd_object(commandname)
170
except (BlackListedCommand, BzrError):
171
return os.system(line)
174
if too_complicated(line):
175
return os.system("bzr "+line)
177
return (cmd_obj.run_argv_aliases(args, alias_args) or 0)
180
except KeyboardInterrupt, e:
183
# print "Unhandled error:\n%s" % errors.exception_str(e)
184
print "Unhandled error:\n%s" % (e)
187
def completenames(self, text, line, begidx, endidx):
188
return CompletionContext(text).get_completions()
190
def completedefault(self, text, line, begidx, endidx):
191
"""Perform completion for native commands.
193
:param text: The text to complete
195
:param line: The entire line to complete
197
:param begidx: The start of the text in the line
199
:param endidx: The end of the text in the line
202
(cmd, args, foo) = self.parseline(line)
205
return CompletionContext(text, command=cmd).get_completions()
213
prompt.write_history()
214
except StopIteration:
217
def iter_opt_completions(command_obj):
218
for option_name, option in command_obj.options().items():
219
yield "--" + option_name
220
short_name = option.short_name()
222
yield "-" + short_name
224
def iter_file_completions(arg, only_dirs = False):
225
"""Generate an iterator that iterates through filename completions.
227
:param arg: The filename fragment to match
229
:param only_dirs: If true, match only directories
230
:type only_dirs: bool
237
(dir, file) = os.path.split(arg)
239
listingdir = os.path.expanduser(dir)
242
for file in chain(os.listdir(listingdir), extras):
244
userfile = dir+'/'+file
247
if userfile.startswith(arg):
248
if os.path.isdir(listingdir+'/'+file):
255
def iter_dir_completions(arg):
256
"""Generate an iterator that iterates through directory name completions.
258
:param arg: The directory name fragment to match
261
return iter_file_completions(arg, True)
263
def iter_command_names(hidden=False):
264
for real_cmd_name, cmd_class in get_all_cmds():
265
if not hidden and cmd_class.hidden:
267
for name in [real_cmd_name] + cmd_class.aliases:
268
# Don't complete on aliases that are prefixes of the canonical name
269
if name == real_cmd_name or not real_cmd_name.startswith(name):
272
def filter_completions(iter, arg):
273
return (c for c in iter if c.startswith(arg))
275
def iter_munged_completions(iter, arg, text):
276
for completion in iter:
277
completion = str(completion)
278
if completion.startswith(arg):
279
yield completion[len(arg)-len(text):]+" "
281
def too_complicated(line):
282
for char in '|<>"\"*?':
added
removed
shell.py
