summaryrefslogtreecommitdiff
path: root/plugins/alias.py
blob: 5a35d1c63fc9f934491ea1f89fe3c1751319fef7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
"""
Alias plugin.

Allows the creation and the removal of personal aliases.
"""

from plugin import BasePlugin
import common
from common import parse_command_args_to_alias as parse

class Plugin(BasePlugin):
    def init(self):
        self.add_command('alias', self.command_alias, '/alias <alias> <command> <args>\nAlias: create an alias command')
        self.add_command('unalias', self.command_unalias, '/unalias <alias>\nUnalias: remove a previously created alias')
        self.commands = {}

    def command_alias(self, line):
        """
        /alias <alias> <command> [args]
        """
        arg = common.shell_split(line)
        if len(arg) < 2:
            self.core.information('Alias: Not enough parameters', 'Error')
            return
        alias = arg[0]
        command = arg[1]
        tmp_args = arg[2] if len(arg) > 2 else ''

        if alias in self.core.commands or alias in self.commands:
            self.core.information('Alias: command already exists', 'Error')
            return
        self.commands[alias] = lambda arg: self.get_command(command)(parse(arg, tmp_args))
        self.add_command(alias, self.commands[alias], 'This command is an alias for /%s %s' %( command, tmp_args))
        self.core.information('Alias /%s successfuly created' % alias, 'Info')

    def command_unalias(self, alias):
        """
        /unalias <existing alias>
        """
        if alias in self.commands:
            del self.commands[alias]
            self.del_command(alias)
            self.core.information('Alias /%s successfuly deleted' % alias, 'Info')

    def get_command(self, name):
        """Returns the function associated with a command"""
        def dummy(args):
            """Dummy function called if the command doesn’t exist"""
            pass
        if name in self.core.commands:
            return self.core.commands[name][0]
        elif name in self.core.current_tab().commands:
            return self.core.current_tab().commands[name][0]
        return dummy