mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 08:24:40 +01:00
149 lines
3.8 KiB
Python
149 lines
3.8 KiB
Python
"""Bash-like interface extensions for xonsh."""
|
|
import shlex
|
|
import sys
|
|
import re
|
|
import builtins
|
|
|
|
|
|
__all__ = ()
|
|
|
|
|
|
@events.on_transform_command
|
|
def bash_preproc(cmd, **kw):
|
|
bang_previous = {
|
|
"!": lambda x: x,
|
|
"$": lambda x: shlex.split(x)[-1],
|
|
"^": lambda x: shlex.split(x)[0],
|
|
"*": lambda x: " ".join(shlex.split(x)[1:]),
|
|
}
|
|
|
|
def replace_bang(m):
|
|
arg = m.group(1)
|
|
inputs = __xonsh__.history.inps
|
|
|
|
# Dissect the previous command.
|
|
if arg in bang_previous:
|
|
try:
|
|
return bang_previous[arg](inputs[-1])
|
|
except IndexError:
|
|
print("xonsh: no history for '!{}'".format(arg))
|
|
return ""
|
|
|
|
# Look back in history for a matching command.
|
|
else:
|
|
try:
|
|
return next((x for x in reversed(inputs) if x.startswith(arg)))
|
|
except StopIteration:
|
|
print("xonsh: no previous commands match '!{}'".format(arg))
|
|
return ""
|
|
|
|
return re.sub(r"!([!$^*]|[\w]+)", replace_bang, cmd)
|
|
|
|
|
|
def alias(args, stdin=None):
|
|
ret = 0
|
|
|
|
if args:
|
|
for arg in args:
|
|
if "=" in arg:
|
|
# shlex.split to remove quotes, e.g. "foo='echo hey'" into
|
|
# "foo=echo hey"
|
|
name, cmd = shlex.split(arg)[0].split("=", 1)
|
|
aliases[name] = shlex.split(cmd)
|
|
elif arg in aliases:
|
|
print("{}={}".format(arg, aliases[arg]))
|
|
else:
|
|
print("alias: {}: not found".format(arg), file=sys.stderr)
|
|
ret = 1
|
|
else:
|
|
for alias, cmd in aliases.items():
|
|
print("{}={}".format(alias, cmd))
|
|
|
|
return ret
|
|
|
|
|
|
aliases["alias"] = alias
|
|
builtins.__xonsh__.env["THREAD_SUBPROCS"] = False
|
|
|
|
|
|
def _unset(args):
|
|
if not args:
|
|
print("Usage: unset ENV_VARIABLE", file=sys.stderr)
|
|
|
|
for v in args:
|
|
try:
|
|
__xonsh__.env.pop(v)
|
|
except KeyError:
|
|
print(f"{v} not found", file=sys.stderr)
|
|
|
|
|
|
aliases["unset"] = _unset
|
|
|
|
|
|
def _export(args):
|
|
if not args:
|
|
print("Usage: export ENV_VARIABLE=VALUE", file=sys.stderr)
|
|
|
|
for eq in args:
|
|
if "=" in eq:
|
|
name, val = shlex.split(eq)[0].split("=", 1)
|
|
__xonsh__.env[name] = val
|
|
else:
|
|
print(f"{eq} equal sign not found", file=sys.stderr)
|
|
|
|
|
|
aliases["export"] = _export
|
|
|
|
|
|
def _set(args):
|
|
arg = args[0]
|
|
if arg == "-e":
|
|
__xonsh__.env["RAISE_SUBPROC_ERROR"] = True
|
|
elif arg == "+e":
|
|
__xonsh__.env["RAISE_SUBPROC_ERROR"] = False
|
|
elif arg == "-x":
|
|
__xonsh__.env["XONSH_TRACE_SUBPROC"] = True
|
|
elif arg == "+x":
|
|
__xonsh__.env["XONSH_TRACE_SUBPROC"] = False
|
|
else:
|
|
print(
|
|
"Not supported in xontrib bashisms.\nPRs are welcome - https://github.com/xonsh/xonsh/blob/master/xontrib/bashisms.py",
|
|
file=sys.stderr,
|
|
)
|
|
|
|
|
|
aliases["set"] = _set
|
|
|
|
|
|
def _shopt(args):
|
|
|
|
supported_shopt = ["DOTGLOB"]
|
|
|
|
args_len = len(args)
|
|
if args_len == 0:
|
|
for so in supported_shopt:
|
|
onoff = "on" if so in __xonsh__.env and __xonsh__.env[so] else "off"
|
|
print(f"dotglob\t{onoff}")
|
|
return
|
|
elif args_len < 2 or args[0] in ["-h", "--help"]:
|
|
print(f'Usage: shopt <-s|-u> <{"|".join(supported_shopt).lower()}>')
|
|
return
|
|
|
|
opt = args[0]
|
|
optname = args[1]
|
|
|
|
if opt == "-s" and optname == "dotglob":
|
|
__xonsh__.env["DOTGLOB"] = True
|
|
elif opt == "-u" and optname == "dotglob":
|
|
__xonsh__.env["DOTGLOB"] = False
|
|
else:
|
|
print(
|
|
"Not supported in xontrib bashisms.\nPRs are welcome - https://github.com/xonsh/xonsh/blob/master/xontrib/bashisms.py",
|
|
file=sys.stderr,
|
|
)
|
|
|
|
|
|
aliases["shopt"] = _shopt
|
|
|
|
|
|
aliases["complete"] = "completer list"
|