path: root/cmd2/parsing.py

#
# -*- coding: utf-8 -*-
"""Statement parsing classes for cmd2"""

import re
import shlex
from typing import List, Tuple

from . import constants

BLANK_LINE = '\n'

def _comment_replacer(match):
    matched_string = match.group(0)
    if matched_string.startswith('/'):
        # the matched string was a comment, so remove it
        return ''
    # the matched string was a quoted string, return the match
    return matched_string

class Statement(str):
    """String subclass with additional attributes to store the results of parsing.

    The cmd module in the standard library passes commands around as a
    string. To retain backwards compatibility, cmd2 does the same. However, we
    need a place to capture the additional output of the command parsing, so we add
    our own attributes to this subclass.

    The string portion of the class contains the arguments, but not the command, nor
    the output redirection clauses.
    """
    def __init__(self, obj):
        super().__init__()
        self.raw = str(obj)
        self.command = None
        self.multiline_command = None
        # has to be an empty string for compatibility with standard library cmd
        self.args = ''
        self.terminator = None
        self.suffix = None
        self.pipe_to = None
        self.output = None
        self.output_to = None

    @property
    def command_and_args(self):
        """Combine command and args with a space separating them"""
        return '{} {}'.format('' if self.command is None else self.command, self.args).strip()

class StatementParser():
    """Parse raw text into command components.

    Shortcuts is a list of tuples with each tuple containing the shortcut and the expansion.
    """
    def __init__(
            self,
            allow_redirection=True,
            terminators=None,
            multiline_commands=None,
            aliases=None,
            shortcuts=[],
        ):
        self.allow_redirection = allow_redirection
        if terminators is None:
            self.terminators = [';']
        else:
            self.terminators = terminators
        if multiline_commands is None:
            self.multiline_commands = []
        else:
            self.multiline_commands = multiline_commands
        if aliases is None:
            self.aliases = {}
        else:
            self.aliases = aliases
        if shortcuts is None:
            self.shortcuts = []
        else:
            self.shortcuts = shortcuts

        # this regular expression matches C-style comments and quoted
        # strings, i.e. stuff between single or double quote marks
        # it's used with _comment_replacer() to strip out the C-style
        # comments, while leaving C-style comments that are inside either
        # double or single quotes.
        #
        # this big regular expression can be broken down into 3 regular
        # expressions that are OR'ed together.
        #
        # /\*.*?(\*/|$)          matches C-style comments, with an optional
        #                        closing '*/'. The optional closing '*/' is
        #                        there to retain backward compatibility with
        #                        the pyparsing implementation of cmd2 < 0.9.0
        # \'(?:\\.|[^\\\'])*\'   matches a single quoted string, allowing
        #                        for embedded backslash escaped single quote
        #                        marks
        # "(?:\\.|[^\\"])*"      matches a double quoted string, allowing
        #                        for embedded backslash escaped double quote
        #                        marks
        #
        # by way of reminder the (?:...) regular expression syntax is just
        # a non-capturing version of regular parenthesis. We need the non-
        # capturing syntax because _comment_replacer() looks at match
        # groups
        self.comment_pattern = re.compile(
            r'/\*.*?(\*/|$)|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',
            re.DOTALL | re.MULTILINE
        )

    def parse(self, rawinput: str) -> Statement:
        """Parse input into a Statement object, stripping comments, expanding
        aliases and shortcuts, and extracting output redirection directives.
        """
        # strip C-style comments
        # shlex will handle the python/shell style comments for us
        # save rawinput for later
        rawinput = re.sub(self.comment_pattern, _comment_replacer, rawinput)
        # we are going to modify line, so create a copy of the raw input
        line = rawinput
        command = None
        args = ''

        # expand shortcuts, have to do this first because
        # a shortcut can expand into multiple tokens, ie '!ls' becomes
        # 'shell ls'
        line = self.expand_shortcuts(line)

        # handle the special case/hardcoded terminator of a blank line
        # we have to do this before we shlex on whitespace because it
        # destroys all unquoted whitespace in the input
        terminator = None
        if line[-1:] == BLANK_LINE:
            terminator = BLANK_LINE

        # split the input on whitespace
        lexer = shlex.shlex(line, posix=False)
        lexer.whitespace_split = True
        tokens = self.split_on_punctuation(list(lexer))

        # expand aliases
        if tokens:
            command_to_expand = tokens[0]
            tokens[0] = self.expand_aliases(command_to_expand)

        # of the valid terminators, find the first one to occur in the input
        terminator_pos = len(tokens)+1
        for test_terminator in self.terminators:
            try:
                pos = tokens.index(test_terminator)
                if pos < terminator_pos:
                    terminator_pos = pos
                    terminator = test_terminator
                    break
            except ValueError:
                # the terminator is not in the tokens
                pass

        if terminator:
            if terminator == BLANK_LINE:
                terminator_pos = len(tokens)+1
            else:
                terminator_pos = tokens.index(terminator)
            # everything before the first terminator is the command and the args
            (command, args) = self._command_and_args(tokens[:terminator_pos])
            # we will set the suffix later
            # remove all the tokens before and including the terminator
            tokens = tokens[terminator_pos+1:]
        else:
            (testcommand, testargs) = self._command_and_args(tokens)
            if testcommand in self.multiline_commands:
                # no terminator on this line but we have a multiline command
                # everything else on the line is part of the args
                # because redirectors can only be after a terminator
                command = testcommand
                args = testargs
                tokens = []

        # check for output redirect
        output = None
        output_to = None
        try:
            output_pos = tokens.index('>')
            output = '>'
            output_to = ' '.join(tokens[output_pos+1:])
            # remove all the tokens after the output redirect
            tokens = tokens[:output_pos]
        except ValueError:
            pass

        try:
            output_pos = tokens.index('>>')
            output = '>>'
            output_to = ' '.join(tokens[output_pos+1:])
            # remove all tokens after the output redirect
            tokens = tokens[:output_pos]
        except ValueError:
            pass

        # check for pipes
        try:
            # find the first pipe if it exists
            pipe_pos = tokens.index('|')
            # set everything after the first pipe to result.pipe_to
            pipe_to = ' '.join(tokens[pipe_pos+1:])
            # remove all the tokens after the pipe
            tokens = tokens[:pipe_pos]
        except ValueError:
            # no pipe in the tokens
            pipe_to = None

        if terminator:
            # whatever is left is the suffix
            suffix = ' '.join(tokens)
        else:
            # no terminator, so whatever is left is the command and the args
            suffix = None
            if not command:
                # command could already have been set, if so, don't set it again
                (command, args) = self._command_and_args(tokens)

        # set multiline
        if command in self.multiline_commands:
            multiline_command = command
        else:
            multiline_command = None

        # build Statement object
        result = Statement(args)
        result.raw = rawinput
        result.command = command
        result.args = args
        result.terminator = terminator
        result.output = output
        result.output_to = output_to
        result.pipe_to = pipe_to
        result.suffix = suffix
        result.multiline_command = multiline_command
        return result

    def parse_command_only(self, rawinput: str) -> Statement:
        """Partially parse input into a Statement object. The command is
        identified, and shortcuts and aliases are expanded.
        Terminators, multiline commands, and output redirection are not
        parsed.
        """
        # strip C-style comments
        # shlex will handle the python/shell style comments for us
        # save rawinput for later
        rawinput = re.sub(self.comment_pattern, _comment_replacer, rawinput)
        # we are going to modify line, so create a copy of the raw input
        line = rawinput
        command = None
        args = ''

        # expand shortcuts, have to do this first because
        # a shortcut can expand into multiple tokens, ie '!ls' becomes
        # 'shell ls'
        line = self.expand_shortcuts(line)

        # split the input on whitespace
        lexer = shlex.shlex(line, posix=False)
        lexer.whitespace_split = True
        tokens = self.split_on_punctuation(list(lexer))

        # expand aliases
        if tokens:
            command_to_expand = tokens[0]
            tokens[0] = self.expand_aliases(command_to_expand)

        (command, args) = self._command_and_args(tokens)

        # build Statement object
        result = Statement(args)
        result.raw = rawinput
        result.command = command
        result.args = args
        return result

    def expand_shortcuts(self, line: str) -> str:
        """Expand shortcuts at the beginning of input."""
        for (shortcut, expansion) in self.shortcuts:
            if  line.startswith(shortcut):
                # If the next character after the shortcut isn't a space, then insert one
                shortcut_len = len(shortcut)
                if len(line) == shortcut_len or line[shortcut_len] != ' ':
                    expansion += ' '

                # Expand the shortcut
                line = line.replace(shortcut, expansion, 1)
                break
        return line

    def expand_aliases(self, command: str) -> str:
        """Given a command, expand any aliases for the command"""
        # make a copy of aliases so we can edit it
        tmp_aliases = list(self.aliases.keys())
        keep_expanding = bool(tmp_aliases)

        while keep_expanding:
            for cur_alias in tmp_aliases:
                keep_expanding = False
                if command == cur_alias:
                    command = self.aliases[cur_alias]
                    tmp_aliases.remove(cur_alias)
                    keep_expanding = bool(tmp_aliases)
                    break
        return command

    @staticmethod
    def _command_and_args(tokens: List[str]) -> Tuple[str, str]:
        """given a list of tokens, and return a tuple of the command
        and the args as a string.
        """
        command = None
        args = ''

        if tokens:
            command = tokens[0]

        if len(tokens) > 1:
            args = ' '.join(tokens[1:])

        return (command, args)

    def split_on_punctuation(self, tokens: List[str]) -> List[str]:
        """
        # Further splits tokens from a command line using punctuation characters
        # as word breaks when they are in unquoted strings. Each run of punctuation
        # characters is treated as a single token.

        :param initial_tokens: the tokens as parsed by shlex
        :return: the punctuated tokens
        """
        punctuation = []
        punctuation.extend(self.terminators)
        if self.allow_redirection:
            punctuation.extend(constants.REDIRECTION_CHARS)

        punctuated_tokens = []

        for cur_initial_token in tokens:

            # Save tokens up to 1 character in length or quoted tokens. No need to parse these.
            if len(cur_initial_token) <= 1 or cur_initial_token[0] in constants.QUOTES:
                punctuated_tokens.append(cur_initial_token)
                continue

            # Iterate over each character in this token
            cur_index = 0
            cur_char = cur_initial_token[cur_index]

            # Keep track of the token we are building
            new_token = ''

            while True:
                if cur_char not in punctuation:

                    # Keep appending to new_token until we hit a punctuation char
                    while cur_char not in punctuation:
                        new_token += cur_char
                        cur_index += 1
                        if cur_index < len(cur_initial_token):
                            cur_char = cur_initial_token[cur_index]
                        else:
                            break

                else:
                    cur_punc = cur_char

                    # Keep appending to new_token until we hit something other than cur_punc
                    while cur_char == cur_punc:
                        new_token += cur_char
                        cur_index += 1
                        if cur_index < len(cur_initial_token):
                            cur_char = cur_initial_token[cur_index]
                        else:
                            break

                # Save the new token
                punctuated_tokens.append(new_token)
                new_token = ''

                # Check if we've viewed all characters
                if cur_index >= len(cur_initial_token):
                    break

        return punctuated_tokens