ClientCLI.py

from ClientUtils import *
from Utils import *
import os
import sys
import getpass


class ClientCLI:
    """
    This class is the command line interface for the client.
    """

    def __init__(self, host, port):
        self.client = ClientUtils(host, port)

    def start(self):
        """
        1. Trigger the login command line interface.
        2. If the user has been authorized, then start the client (which will bind a port and listen to the server).
        3. Start the client CLI.
        """

        try:
            self.__login()
            self.client.invoke()
            self.__interface()
        except KeyboardInterrupt, msg:
            if self.client.authorized:
                self.client.logout()

            print '\nThanks for using the chat room. See you next time!'
            os._exit(0)

    def __login(self):
        """
        This is the CLI for the client to log in.
        """
        while True:
            username = raw_input("Username: ")
            password = getpass.getpass("Password: ")
            if username == '' or password == '':
                print 'Username and password cannot be empty!'
                continue

            if self.client.login(username, password):
                return True

    def __interface(self):
        """
        1. The user can input command here.
        2. This CLI will also display the messages from the server.
        """
        while True:
            encoding = 'utf-8' if sys.stdin.encoding in (None, 'ascii') else sys.stdin.encoding  # This is necessary to support UTF-8 characters.
            line = raw_input('Please enter your command: ').decode(encoding)
            # print('line: ' + line)
            if len(line) == 0:
                continue

            data = line.split(" ", 1)
            command = ''
            params = ''
            if len(data) > 0:
                command = data[0]
                if len(data) > 1:
                    params = data[1]

            if command == 'broadcast':
                if len(params) > 0:
                    message = params
                    self.client.broadcast(message)
                else:
                    print 'The parameters are invalid. Usage: broadcast <message>'

            elif command == 'send':  # TODO: support utf-8 message
                if len(params) > 0:
                    message_to = []
                    message = ''
                    regex_one_user = re.match( r'^([\d\w]+)\s(.+)$', params, re.M|re.I)
                    regex_multi_users = re.match( r'^\((.+)\)\s(.+)$', params, re.M|re.I)
                    if regex_one_user:  # that there is only one user
                        args = params.split(' ', 1)
                        message_to.append(args[0])
                        message = args[1]

                        # message = message.decode('utf-8').encode('utf-8')

                        self.client.send_message(message_to, message)
                    elif regex_multi_users:
                        args = re.match( r'\((.+)\)\s(.+)', params, re.M|re.I)
                        # print 'args.group(1): ' + args.group(1)
                        # print 'args.group(2): ' + args.group(2)
                        user_list = args.group(1)
                        message = args.group(2)
                        user_list = user_list.split(' ')
                        for user in user_list:
                            message_to.append(user)

                        # message = message.deconde('utf-8').encode('utf-8')

                        self.client.send_message(message_to, message)
                    else:
                        print 'The parameters are invalid. Usage: send <user> <message> or send (<user>...<user>) <message>'

                    # print message_to

                    # self.client.send_message(message_to, message)
                else:
                    print 'The parameters are invalid. Usage: send <user> <message> or send (<user>...<user>) <message>'

            elif command == 'logout':
                self.client.logout()
                print 'Thanks for using the chat room. See you next time!'
                # sys.exit(0)
                os._exit(0)

            elif command == 'who':
                self.client.who()

            elif command == 'last':
                # print self.__is_number(params)
                if len(params) > 0 and Utils.is_number(params) and int(params) > 0 and int(params) <= 60:
                    number = int(params)
                    self.client.last(number)
                else:
                    print 'The parameters are invalid. Usage: last <number>, where 0 < number <= 60'

            elif command == 'check':  # check user's status. mainly for debugging use
                target = ''
                if len(params) > 0:
                    target = params
                self.client.check(target)

            # when a user B is blacklisted by user A, user B cannot send messages to user A. QUESTION: should B be able to broadcast messages to A?
            # the answer for the "QUESTION" now is yes.
            elif command == 'blacklist' or command == 'whitelist':
                if len(params) > 0:
                    targets = []
                    regex_one_user = re.match( r'^([\d\w]+)$', params, re.M|re.I)
                    regex_multi_users = re.match( r'^\((.+)\)$', params, re.M|re.I)
                    if regex_one_user:
                        targets.append(params)

                        self.client.update_blacklist(targets, command)
                    elif regex_multi_users:
                        user_list = regex_multi_users.group(1)
                        user_list = user_list.split(' ')
                        for user in user_list:
                            targets.append(user)

                        self.client.update_blacklist(targets, command)
                    else:
                        print 'The parameters are invalid. Usage: %s <user> or %s (<user>...<user>)' % (command, command)

            # to check what users are active within the last specific minutes
            elif command == 'active':
                if len(params) > 0 and Utils.is_number(params) and int(params) > 0 and int(params) <= 60:
                    number = int(params)
                    self.client.active(number)
                else:
                    print 'The parameters are invalid. Usage: active <number>, where 0 < number <= 60'

            else:
                print command + ' - is an invalid command.'