mirror of
https://github.com/nvbn/thefuck.git
synced 2025-03-25 03:59:08 +00:00
2d81166213
This way it's ensured that whatever is used as cache key is always ordered. Sets are unordered collections.
129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
from subprocess import Popen, PIPE
|
|
from time import time
|
|
import os
|
|
import sys
|
|
import six
|
|
from .. import logs
|
|
from ..conf import settings
|
|
from ..utils import DEVNULL, cache
|
|
from .generic import Generic
|
|
|
|
|
|
@cache('~/.config/fish/config.fish', '~/.config/fish/functions')
|
|
def _get_functions(overridden):
|
|
proc = Popen(['fish', '-ic', 'functions'], stdout=PIPE, stderr=DEVNULL)
|
|
functions = proc.stdout.read().decode('utf-8').strip().split('\n')
|
|
return {func: func for func in functions if func not in overridden}
|
|
|
|
|
|
@cache('~/.config/fish/config.fish')
|
|
def _get_aliases(overridden):
|
|
aliases = {}
|
|
proc = Popen(['fish', '-ic', 'alias'], stdout=PIPE, stderr=DEVNULL)
|
|
alias_out = proc.stdout.read().decode('utf-8').strip()
|
|
if not alias_out:
|
|
return aliases
|
|
for alias in alias_out.split('\n'):
|
|
for separator in (' ', '='):
|
|
split_alias = alias.replace('alias ', '', 1).split(separator, 1)
|
|
if len(split_alias) == 2:
|
|
name, value = split_alias
|
|
break
|
|
else:
|
|
continue
|
|
if name not in overridden:
|
|
aliases[name] = value
|
|
return aliases
|
|
|
|
|
|
class Fish(Generic):
|
|
def _get_overridden_aliases(self):
|
|
overridden = os.environ.get('THEFUCK_OVERRIDDEN_ALIASES',
|
|
os.environ.get('TF_OVERRIDDEN_ALIASES', ''))
|
|
default = {'cd', 'grep', 'ls', 'man', 'open'}
|
|
for alias in overridden.split(','):
|
|
default.add(alias.strip())
|
|
return sorted(default)
|
|
|
|
def app_alias(self, alias_name):
|
|
if settings.alter_history:
|
|
alter_history = (' builtin history delete --exact'
|
|
' --case-sensitive -- $fucked_up_command\n'
|
|
' builtin history merge ^ /dev/null\n')
|
|
else:
|
|
alter_history = ''
|
|
# It is VERY important to have the variables declared WITHIN the alias
|
|
return ('function {0} -d "Correct your previous console command"\n'
|
|
' set -l fucked_up_command $history[1]\n'
|
|
' env TF_SHELL=fish TF_ALIAS={0} PYTHONIOENCODING=utf-8'
|
|
' thefuck $fucked_up_command | read -l unfucked_command\n'
|
|
' if [ "$unfucked_command" != "" ]\n'
|
|
' eval $unfucked_command\n{1}'
|
|
' end\n'
|
|
'end').format(alias_name, alter_history)
|
|
|
|
def get_aliases(self):
|
|
overridden = self._get_overridden_aliases()
|
|
functions = _get_functions(overridden)
|
|
raw_aliases = _get_aliases(overridden)
|
|
functions.update(raw_aliases)
|
|
return functions
|
|
|
|
def _expand_aliases(self, command_script):
|
|
aliases = self.get_aliases()
|
|
binary = command_script.split(' ')[0]
|
|
if binary in aliases and aliases[binary] != binary:
|
|
return command_script.replace(binary, aliases[binary], 1)
|
|
elif binary in aliases:
|
|
return u'fish -ic "{}"'.format(command_script.replace('"', r'\"'))
|
|
else:
|
|
return command_script
|
|
|
|
def _get_history_file_name(self):
|
|
return os.path.expanduser('~/.config/fish/fish_history')
|
|
|
|
def _get_history_line(self, command_script):
|
|
return u'- cmd: {}\n when: {}\n'.format(command_script, int(time()))
|
|
|
|
def _script_from_history(self, line):
|
|
if '- cmd: ' in line:
|
|
return line.split('- cmd: ', 1)[1]
|
|
else:
|
|
return ''
|
|
|
|
def and_(self, *commands):
|
|
return u'; and '.join(commands)
|
|
|
|
def or_(self, *commands):
|
|
return u'; or '.join(commands)
|
|
|
|
def how_to_configure(self):
|
|
return self._create_shell_configuration(
|
|
content=u"thefuck --alias | source",
|
|
path='~/.config/fish/config.fish',
|
|
reload='fish')
|
|
|
|
def info(self):
|
|
"""Returns the name and version of the current shell"""
|
|
proc = Popen(['fish', '--version'],
|
|
stdout=PIPE, stderr=DEVNULL)
|
|
version = proc.stdout.read().decode('utf-8').split()[-1]
|
|
return u'Fish Shell {}'.format(version)
|
|
|
|
def put_to_history(self, command):
|
|
try:
|
|
return self._put_to_history(command)
|
|
except IOError:
|
|
logs.exception("Can't update history", sys.exc_info())
|
|
|
|
def _put_to_history(self, command_script):
|
|
"""Puts command script to shell history."""
|
|
history_file_name = self._get_history_file_name()
|
|
if os.path.isfile(history_file_name):
|
|
with open(history_file_name, 'a') as history:
|
|
entry = self._get_history_line(command_script)
|
|
if six.PY2:
|
|
history.write(entry.encode('utf-8'))
|
|
else:
|
|
history.write(entry)
|