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
32
from bzrlib import osutils, trace
33
from bzrlib.branch import Branch
34
from bzrlib.config import config_dir, ensure_config_dir_exists
35
from bzrlib.commands import get_cmd_object, all_command_names, get_alias
24
36
from bzrlib.errors import BzrError
25
from bzrlib.commands import get_cmd_object, get_all_cmds
37
from bzrlib.workingtree import WorkingTree
27
42
SHELL_BLACKLIST = set(['rm', 'ls'])
28
43
COMPLETION_BLACKLIST = set(['shell'])
30
46
class BlackListedCommand(BzrError):
31
47
def __init__(self, command):
32
48
BzrError.__init__(self, "The command %s is blacklisted for shell use" %
35
52
class CompletionContext(object):
36
53
def __init__(self, text, command=None, prev_opt=None, arg_pos=None):
40
57
self.arg_pos = None
42
59
def get_completions(self):
44
iter = (c for c in iter_command_names() if
45
c not in COMPLETION_BLACKLIST)
47
iter = list(filter_completions(iter, text))
61
return self.get_completions_or_raise()
66
def get_option_completions(self):
68
command_obj = get_cmd_object(self.command)
71
opts = [o+" " for o in iter_opt_completions(command_obj)]
72
return list(filter_completions(opts, self.text))
74
def get_completions_or_raise(self):
75
if self.command is None:
77
iter = iter_executables(self.text)
79
iter = (c+" " for c in iter_command_names() if
80
c not in COMPLETION_BLACKLIST)
81
return list(filter_completions(iter, self.text))
82
if self.prev_opt is None:
83
completions = self.get_option_completions()
84
if self.command == "cd":
85
iter = iter_dir_completions(self.text)
86
completions.extend(list(filter_completions(iter, self.text)))
88
iter = iter_file_completions(self.text)
89
completions.extend(filter_completions(iter, self.text))
53
93
class PromptCmd(cmd.Cmd):
54
95
def __init__(self):
55
96
cmd.Cmd.__init__(self)
56
97
self.prompt = "bzr> "
58
self.tree = arch.tree_root(".")
99
self.tree = WorkingTree.open_containing('.')[0]
63
104
self.identchars += '-'
64
self.history_file = os.path.expanduser("~/.bazaar/shell-history")
65
readline.set_completer_delims(string.whitespace)
66
if os.access(self.history_file, os.R_OK) and \
67
os.path.isfile(self.history_file):
68
readline.read_history_file(self.history_file)
105
ensure_config_dir_exists()
106
self.history_file = osutils.pathjoin(config_dir(), 'shell-history')
107
whitespace = ''.join(c for c in string.whitespace if c < chr(127))
109
readline.set_completer_delims(whitespace)
110
if os.access(self.history_file, os.R_OK) and \
111
os.path.isfile(self.history_file):
112
readline.read_history_file(self.history_file)
69
113
self.cwd = os.getcwd()
71
115
def write_history(self):
72
readline.write_history_file(self.history_file)
117
readline.write_history_file(self.history_file)
74
119
def do_quit(self, args):
75
120
self.write_history()
89
134
def set_prompt(self):
90
135
if self.tree is not None:
92
prompt = pylon.alias_or_version(self.tree.tree_version,
95
if prompt is not None:
96
prompt = " " + prompt +":"+ pylon.tree_cwd(self.tree)
137
prompt_data = (self.tree.branch.nick, self.tree.branch.revno(),
138
self.tree.relpath('.'))
139
prompt = " %s:%d/%s" % prompt_data
103
146
def set_title(self, command=None):
105
version = pylon.alias_or_version(self.tree.tree_version, self.tree,
148
b = Branch.open_containing('.')[0]
149
version = "%s:%d" % (b.nick, b.revno())
108
151
version = "[no version]"
109
152
if command is None:
133
176
self.default("help "+line)
135
178
def default(self, line):
180
args = shlex.split(line)
181
except ValueError, e:
182
print 'Parse error:', e
185
alias_args = get_alias(args[0])
186
if alias_args is not None:
187
args[0] = alias_args.pop(0)
137
189
commandname = args.pop(0)
138
190
for char in ('|', '<', '>'):
139
191
commandname = commandname.split(char)[0]
147
199
return os.system(line)
150
if too_complicated(line):
202
is_qbzr = cmd_obj.__module__.startswith('bzrlib.plugins.qbzr.')
203
if too_complicated(line) or is_qbzr:
151
204
return os.system("bzr "+line)
153
return (cmd_obj.run_argv(args) or 0)
206
return (cmd_obj.run_argv_aliases(args, alias_args) or 0)
154
207
except BzrError, e:
208
trace.log_exception_quietly()
156
210
except KeyboardInterrupt, e:
157
211
print "Interrupted"
158
212
except Exception, e:
159
# print "Unhandled error:\n%s" % errors.exception_str(e)
213
trace.log_exception_quietly()
160
214
print "Unhandled error:\n%s" % (e)
163
217
def completenames(self, text, line, begidx, endidx):
164
from bzrlib.trace import mutter
166
iter = (c for c in iter_command_names() if
167
c not in COMPLETION_BLACKLIST)
170
arg = line.split()[-1]
173
iter = filter_completions(iter, arg)
218
return CompletionContext(text).get_completions()
178
220
def completedefault(self, text, line, begidx, endidx):
179
221
"""Perform completion for native commands.
181
223
:param text: The text to complete
183
225
:param line: The entire line to complete
190
232
(cmd, args, foo) = self.parseline(line)
193
return self.completenames(text, line, begidx, endidx)
197
command_obj = get_cmd_object(cmd)
201
if command_obj is not None:
202
opts = list(iter_opt_completions(command_obj))
203
files = list(iter_file_completions(text))
204
return list(filter_completions(opts+files, text))
207
arg = args.split()[-1]
210
iter = iter_dir_completions(text)
211
iter = filter_completions(iter, text)
214
arg = args.split()[-1]
215
iter = iter_file_completions(arg)
216
return list(iter_munged_completions(iter, arg, text))
218
return self.completenames(text, line, begidx, endidx)
235
return CompletionContext(text, command=cmd).get_completions()
238
def run_shell(directory=None):
240
if not directory is None:
224
242
prompt = PromptCmd()
228
prompt.write_history()
247
except KeyboardInterrupt:
250
prompt.write_history()
229
251
except StopIteration:
232
255
def iter_opt_completions(command_obj):
233
256
for option_name, option in command_obj.options().items():
234
257
yield "--" + option_name
276
300
return iter_file_completions(arg, True)
278
303
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:
304
for real_cmd_name in all_command_names():
305
cmd_obj = get_cmd_object(real_cmd_name)
306
if not hidden and cmd_obj.hidden:
282
for name in [real_cmd_name] + cmd_class.aliases:
308
for name in [real_cmd_name] + cmd_obj.aliases:
283
309
# Don't complete on aliases that are prefixes of the canonical name
284
310
if name == real_cmd_name or not real_cmd_name.startswith(name):
314
def iter_executables(path):
315
dirname, partial = os.path.split(path)
316
for filename in os.listdir(dirname):
317
if not filename.startswith(partial):
319
fullpath = os.path.join(dirname, filename)
320
mode=os.lstat(fullpath)[stat.ST_MODE]
321
if stat.S_ISREG(mode) and 0111 & mode:
287
325
def filter_completions(iter, arg):
288
326
return (c for c in iter if c.startswith(arg))
290
329
def iter_munged_completions(iter, arg, text):
291
330
for completion in iter:
292
331
completion = str(completion)
293
332
if completion.startswith(arg):
294
333
yield completion[len(arg)-len(text):]+" "
296
336
def too_complicated(line):
297
for char in '|<>"\"*?':
added
removed
shell.py
