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
19
from itertools import chain
27
from bzrlib.branch import Branch
28
from bzrlib.commands import get_cmd_object, get_all_cmds, get_alias
29
from bzrlib.errors import BzrError
30
from bzrlib.workingtree import WorkingTree
35
SHELL_BLACKLIST = set(['rm', 'ls'])
36
COMPLETION_BLACKLIST = set(['shell'])
39
class BlackListedCommand(BzrError):
40
def __init__(self, command):
41
BzrError.__init__(self, "The command %s is blacklisted for shell use" %
45
class CompletionContext(object):
46
def __init__(self, text, command=None, prev_opt=None, arg_pos=None):
48
self.command = command
49
self.prev_opt = prev_opt
52
def get_completions(self):
54
return self.get_completions_or_raise()
59
def get_option_completions(self):
61
command_obj = get_cmd_object(self.command)
64
opts = [o+" " for o in iter_opt_completions(command_obj)]
65
return list(filter_completions(opts, self.text))
67
def get_completions_or_raise(self):
68
if self.command is None:
70
iter = iter_executables(self.text)
72
iter = (c+" " for c in iter_command_names() if
73
c not in COMPLETION_BLACKLIST)
74
return list(filter_completions(iter, self.text))
75
if self.prev_opt is None:
76
completions = self.get_option_completions()
77
if self.command == "cd":
78
iter = iter_dir_completions(self.text)
79
completions.extend(list(filter_completions(iter, self.text)))
81
iter = iter_file_completions(self.text)
82
completions.extend(filter_completions(iter, self.text))
86
class PromptCmd(cmd.Cmd):
88
cmd.Cmd.__init__(self)
91
self.tree = WorkingTree.open_containing('.')[0]
96
self.identchars += '-'
97
self.history_file = os.path.expanduser("~/.bazaar/shell-history")
98
readline.set_completer_delims(string.whitespace)
99
if os.access(self.history_file, os.R_OK) and \
100
os.path.isfile(self.history_file):
101
readline.read_history_file(self.history_file)
102
self.cwd = os.getcwd()
104
def write_history(self):
105
readline.write_history_file(self.history_file)
107
def do_quit(self, args):
111
def do_exit(self, args):
114
def do_EOF(self, args):
118
def postcmd(self, line, bar):
122
def set_prompt(self):
123
if self.tree is not None:
125
prompt_data = (self.tree.branch.nick, self.tree.branch.revno(),
126
self.tree.relpath('.'))
127
prompt = " %s:%d/%s" % prompt_data
132
self.prompt = "bzr%s> " % prompt
134
def set_title(self, command=None):
136
b = Branch.open_containing('.')[0]
137
version = "%s:%d" % (b.nick, b.revno())
139
version = "[no version]"
142
sys.stdout.write(terminal.term_title("bzr %s %s" % (command, version)))
144
def do_cd(self, line):
147
line = os.path.expanduser(line)
148
if os.path.isabs(line):
151
newcwd = self.cwd+'/'+line
152
newcwd = os.path.normpath(newcwd)
159
self.tree = WorkingTree.open_containing(".")[0]
163
def do_help(self, line):
164
self.default("help "+line)
166
def default(self, line):
167
args = shlex.split(line)
168
alias_args = get_alias(args[0])
169
if alias_args is not None:
170
args[0] = alias_args.pop(0)
172
commandname = args.pop(0)
173
for char in ('|', '<', '>'):
174
commandname = commandname.split(char)[0]
175
if commandname[-1] in ('|', '<', '>'):
176
commandname = commandname[:-1]
178
if commandname in SHELL_BLACKLIST:
179
raise BlackListedCommand(commandname)
180
cmd_obj = get_cmd_object(commandname)
181
except (BlackListedCommand, BzrError):
182
return os.system(line)
185
if too_complicated(line):
186
return os.system("bzr "+line)
188
return (cmd_obj.run_argv_aliases(args, alias_args) or 0)
191
except KeyboardInterrupt, e:
194
# print "Unhandled error:\n%s" % errors.exception_str(e)
195
print "Unhandled error:\n%s" % (e)
198
def completenames(self, text, line, begidx, endidx):
199
return CompletionContext(text).get_completions()
201
def completedefault(self, text, line, begidx, endidx):
202
"""Perform completion for native commands.
204
:param text: The text to complete
206
:param line: The entire line to complete
208
:param begidx: The start of the text in the line
210
:param endidx: The end of the text in the line
213
(cmd, args, foo) = self.parseline(line)
216
return CompletionContext(text, command=cmd).get_completions()
225
prompt.write_history()
226
except StopIteration:
230
def iter_opt_completions(command_obj):
231
for option_name, option in command_obj.options().items():
232
yield "--" + option_name
233
short_name = option.short_name()
235
yield "-" + short_name
238
def iter_file_completions(arg, only_dirs = False):
239
"""Generate an iterator that iterates through filename completions.
241
:param arg: The filename fragment to match
243
:param only_dirs: If true, match only directories
244
:type only_dirs: bool
251
(dir, file) = os.path.split(arg)
253
listingdir = os.path.expanduser(dir)
256
for file in chain(os.listdir(listingdir), extras):
258
userfile = dir+'/'+file
261
if userfile.startswith(arg):
262
if os.path.isdir(listingdir+'/'+file):
269
def iter_dir_completions(arg):
270
"""Generate an iterator that iterates through directory name completions.
272
:param arg: The directory name fragment to match
275
return iter_file_completions(arg, True)
278
def iter_command_names(hidden=False):
279
for real_cmd_name, cmd_class in get_all_cmds():
280
if not hidden and cmd_class.hidden:
282
for name in [real_cmd_name] + cmd_class.aliases:
283
# Don't complete on aliases that are prefixes of the canonical name
284
if name == real_cmd_name or not real_cmd_name.startswith(name):
288
def iter_executables(path):
289
dirname, partial = os.path.split(path)
290
for filename in os.listdir(dirname):
291
if not filename.startswith(partial):
293
fullpath = os.path.join(dirname, filename)
294
mode=os.lstat(fullpath)[stat.ST_MODE]
295
if stat.S_ISREG(mode) and 0111 & mode:
299
def filter_completions(iter, arg):
300
return (c for c in iter if c.startswith(arg))
303
def iter_munged_completions(iter, arg, text):
304
for completion in iter:
305
completion = str(completion)
306
if completion.startswith(arg):
307
yield completion[len(arg)-len(text):]+" "
310
def too_complicated(line):
added
removed
shell.py
