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
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
23
31
from bzrlib.errors import BzrError
24
from bzrlib.commands import get_cmd_object
32
from bzrlib.workingtree import WorkingTree
37
SHELL_BLACKLIST = set(['rm', 'ls'])
38
COMPLETION_BLACKLIST = set(['shell'])
41
class BlackListedCommand(BzrError):
42
def __init__(self, command):
43
BzrError.__init__(self, "The command %s is blacklisted for shell use" %
47
class CompletionContext(object):
48
def __init__(self, text, command=None, prev_opt=None, arg_pos=None):
50
self.command = command
51
self.prev_opt = prev_opt
54
def get_completions(self):
56
return self.get_completions_or_raise()
61
def get_option_completions(self):
63
command_obj = get_cmd_object(self.command)
66
opts = [o+" " for o in iter_opt_completions(command_obj)]
67
return list(filter_completions(opts, self.text))
69
def get_completions_or_raise(self):
70
if self.command is None:
72
iter = iter_executables(self.text)
74
iter = (c+" " for c in iter_command_names() if
75
c not in COMPLETION_BLACKLIST)
76
return list(filter_completions(iter, self.text))
77
if self.prev_opt is None:
78
completions = self.get_option_completions()
79
if self.command == "cd":
80
iter = iter_dir_completions(self.text)
81
completions.extend(list(filter_completions(iter, self.text)))
83
iter = iter_file_completions(self.text)
84
completions.extend(filter_completions(iter, self.text))
26
88
class PromptCmd(cmd.Cmd):
27
90
def __init__(self):
28
91
cmd.Cmd.__init__(self)
29
92
self.prompt = "bzr> "
31
self.tree = arch.tree_root(".")
94
self.tree = WorkingTree.open_containing('.')[0]
36
99
self.identchars += '-'
37
self.history_file = os.path.expanduser("~/.bazaar/shell-history")
38
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)
39
104
if os.access(self.history_file, os.R_OK) and \
40
105
os.path.isfile(self.history_file):
41
106
readline.read_history_file(self.history_file)
98
161
except Exception, e:
101
self.tree = arch.tree_root(".")
164
self.tree = WorkingTree.open_containing(".")[0]
105
168
def do_help(self, line):
169
self.default("help "+line)
108
171
def default(self, line):
173
args = shlex.split(line)
174
except ValueError, e:
175
print 'Parse error:', e
178
alias_args = get_alias(args[0])
179
if alias_args is not None:
180
args[0] = alias_args.pop(0)
110
182
commandname = args.pop(0)
183
for char in ('|', '<', '>'):
184
commandname = commandname.split(char)[0]
185
if commandname[-1] in ('|', '<', '>'):
186
commandname = commandname[:-1]
188
if commandname in SHELL_BLACKLIST:
189
raise BlackListedCommand(commandname)
112
190
cmd_obj = get_cmd_object(commandname)
191
except (BlackListedCommand, BzrError):
114
192
return os.system(line)
118
return (cmd_obj.run_argv(args) or 0)
195
is_qbzr = cmd_obj.__module__.startswith('bzrlib.plugins.qbzr.')
196
if too_complicated(line) or is_qbzr:
197
return os.system("bzr "+line)
199
return (cmd_obj.run_argv_aliases(args, alias_args) or 0)
119
200
except BzrError, e:
121
202
except KeyboardInterrupt, e:
128
209
def completenames(self, text, line, begidx, endidx):
130
iter = iter_command_names(self.fake_aba)
133
arg = line.split()[-1]
136
iter = cmdutil.iter_munged_completions(iter, arg, text)
210
return CompletionContext(text).get_completions()
141
212
def completedefault(self, text, line, begidx, endidx):
142
213
"""Perform completion for native commands.
144
215
:param text: The text to complete
146
217
:param line: The entire line to complete
150
221
:param endidx: The end of the text in the line
151
222
:type endidx: int
154
(cmd, args, foo) = self.parseline(line)
155
command_obj=find_command(cmd)
156
if command_obj is not None:
157
return command_obj.complete(args.split(), text)
158
elif not self.fake_aba.is_command(cmd) and \
159
cmdutil.is_tla_command(cmd):
160
iter = cmdutil.iter_supported_switches(cmd)
162
arg = args.split()[-1]
165
if arg.startswith("-"):
166
return list(cmdutil.iter_munged_completions(iter, arg,
169
return list(cmdutil.iter_munged_completions(
170
cmdutil.iter_file_completions(arg), arg, text))
175
arg = args.split()[-1]
178
iter = cmdutil.iter_dir_completions(arg)
179
iter = cmdutil.iter_munged_completions(iter, arg, text)
182
arg = args.split()[-1]
183
iter = cmdutil.iter_file_completions(arg)
184
return list(cmdutil.iter_munged_completions(iter, arg, text))
186
return self.completenames(text, line, begidx, endidx)
224
(cmd, args, foo) = self.parseline(line)
227
return CompletionContext(text, command=cmd).get_completions()
195
prompt.write_history()
237
except KeyboardInterrupt:
240
prompt.write_history()
241
except StopIteration:
245
def iter_opt_completions(command_obj):
246
for option_name, option in command_obj.options().items():
247
yield "--" + option_name
248
short_name = option.short_name()
250
yield "-" + short_name
197
253
def iter_file_completions(arg, only_dirs = False):
198
254
"""Generate an iterator that iterates through filename completions.
234
290
return iter_file_completions(arg, True)
293
def iter_command_names(hidden=False):
294
for real_cmd_name in all_command_names():
295
cmd_obj = get_cmd_object(real_cmd_name)
296
if not hidden and cmd_obj.hidden:
298
for name in [real_cmd_name] + cmd_obj.aliases:
299
# Don't complete on aliases that are prefixes of the canonical name
300
if name == real_cmd_name or not real_cmd_name.startswith(name):
304
def iter_executables(path):
305
dirname, partial = os.path.split(path)
306
for filename in os.listdir(dirname):
307
if not filename.startswith(partial):
309
fullpath = os.path.join(dirname, filename)
310
mode=os.lstat(fullpath)[stat.ST_MODE]
311
if stat.S_ISREG(mode) and 0111 & mode:
315
def filter_completions(iter, arg):
316
return (c for c in iter if c.startswith(arg))
319
def iter_munged_completions(iter, arg, text):
320
for completion in iter:
321
completion = str(completion)
322
if completion.startswith(arg):
323
yield completion[len(arg)-len(text):]+" "
326
def too_complicated(line):
added
removed
shell.py
