1
1
# Copyright (C) 2004, 2005 Aaron Bentley
2
# <aaron.bentley@utoronto.ca>
2
# <aaron@aaronbentley.com>
4
4
# This program is free software; you can redistribute it and/or modify
5
5
# it under the terms of the GNU General Public License as published by
14
14
# You should have received a copy of the GNU General Public License
15
15
# along with this program; if not, write to the Free Software
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
from itertools import chain
23
from itertools import chain
27
from bzrlib import osutils
28
from bzrlib.branch import Branch
29
from bzrlib.config import config_dir, ensure_config_dir_exists
30
from bzrlib.commands import get_cmd_object, all_command_names, get_alias
24
31
from bzrlib.errors import BzrError
25
from bzrlib.commands import get_cmd_object, get_all_cmds, get_alias
26
from bzrlib.branch import Branch
32
from bzrlib.workingtree import WorkingTree
28
37
SHELL_BLACKLIST = set(['rm', 'ls'])
29
38
COMPLETION_BLACKLIST = set(['shell'])
31
41
class BlackListedCommand(BzrError):
32
42
def __init__(self, command):
33
43
BzrError.__init__(self, "The command %s is blacklisted for shell use" %
36
47
class CompletionContext(object):
37
48
def __init__(self, text, command=None, prev_opt=None, arg_pos=None):
58
69
def get_completions_or_raise(self):
59
70
if self.command is None:
60
iter = (c+" " for c in iter_command_names() if
61
c not in COMPLETION_BLACKLIST)
72
iter = iter_executables(self.text)
74
iter = (c+" " for c in iter_command_names() if
75
c not in COMPLETION_BLACKLIST)
62
76
return list(filter_completions(iter, self.text))
63
77
if self.prev_opt is None:
64
78
completions = self.get_option_completions()
67
81
completions.extend(list(filter_completions(iter, self.text)))
69
83
iter = iter_file_completions(self.text)
70
completions.extend([f+" " for f in
71
filter_completions(iter, self.text)])
84
completions.extend(filter_completions(iter, self.text))
75
88
class PromptCmd(cmd.Cmd):
76
90
def __init__(self):
77
91
cmd.Cmd.__init__(self)
78
92
self.prompt = "bzr> "
85
99
self.identchars += '-'
86
self.history_file = os.path.expanduser("~/.bazaar/shell-history")
87
readline.set_completer_delims(string.whitespace)
100
ensure_config_dir_exists()
101
self.history_file = osutils.pathjoin(config_dir(), 'shell-history')
102
whitespace = ''.join(c for c in string.whitespace if c < chr(127))
103
readline.set_completer_delims(whitespace)
88
104
if os.access(self.history_file, os.R_OK) and \
89
105
os.path.isfile(self.history_file):
90
106
readline.read_history_file(self.history_file)
111
127
def set_prompt(self):
112
128
if self.tree is not None:
114
prompt_data = (self.tree.branch.nick, self.tree.branch.revno(),
115
self.tree.branch.relpath('.'))
130
prompt_data = (self.tree.branch.nick, self.tree.branch.revno(),
131
self.tree.relpath('.'))
116
132
prompt = " %s:%d/%s" % prompt_data
153
169
self.default("help "+line)
155
171
def default(self, line):
173
args = shlex.split(line)
174
except ValueError, e:
175
print 'Parse error:', e
157
178
alias_args = get_alias(args[0])
158
179
if alias_args is not None:
159
180
args[0] = alias_args.pop(0)
161
182
commandname = args.pop(0)
162
183
for char in ('|', '<', '>'):
163
184
commandname = commandname.split(char)[0]
190
211
def completedefault(self, text, line, begidx, endidx):
191
212
"""Perform completion for native commands.
193
214
:param text: The text to complete
195
216
:param line: The entire line to complete
205
226
return CompletionContext(text, command=cmd).get_completions()
209
231
prompt = PromptCmd()
213
prompt.write_history()
236
except KeyboardInterrupt:
239
prompt.write_history()
214
240
except StopIteration:
217
244
def iter_opt_completions(command_obj):
218
245
for option_name, option in command_obj.options().items():
219
246
yield "--" + option_name
261
289
return iter_file_completions(arg, True)
263
292
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:
293
for real_cmd_name in all_command_names():
294
cmd_obj = get_cmd_object(real_cmd_name)
295
if not hidden and cmd_obj.hidden:
267
for name in [real_cmd_name] + cmd_class.aliases:
297
for name in [real_cmd_name] + cmd_obj.aliases:
268
298
# Don't complete on aliases that are prefixes of the canonical name
269
299
if name == real_cmd_name or not real_cmd_name.startswith(name):
303
def iter_executables(path):
304
dirname, partial = os.path.split(path)
305
for filename in os.listdir(dirname):
306
if not filename.startswith(partial):
308
fullpath = os.path.join(dirname, filename)
309
mode=os.lstat(fullpath)[stat.ST_MODE]
310
if stat.S_ISREG(mode) and 0111 & mode:
272
314
def filter_completions(iter, arg):
273
315
return (c for c in iter if c.startswith(arg))
275
318
def iter_munged_completions(iter, arg, text):
276
319
for completion in iter:
277
320
completion = str(completion)
278
321
if completion.startswith(arg):
279
322
yield completion[len(arg)-len(text):]+" "
281
325
def too_complicated(line):
282
for char in '|<>"\"*?':
added
removed
shell.py
