bitty.py

#!/usr/bin/env python
from kivy.app import App
from kivy.uix.floatlayout import FloatLayout
from kivy.factory import Factory
from kivy.properties import ObjectProperty
from kivy.uix.tabbedpanel import TabbedPanel
from kivy.uix.textinput import TextInput
from kivy.uix.popup import Popup

from kivy.support import install_twisted_reactor
install_twisted_reactor()

from twisted.internet import ssl, reactor
from twisted.internet.protocol import ClientFactory, Protocol

import json
import sys

class TenthbitClient(Protocol):
    def connectionMade(self):
        self.factory.root.on_connection(self.transport)

    def dataReceived(self, data):
        self.factory.root.handle_payload(data)

class TenthbitClientFactory(ClientFactory):
    protocol = TenthbitClient

    def __init__(self, root):
        self.root = root

    def clientConnectionFailed(self, connector, reason):
        print "Connection failed - goodbye!"
        reactor.stop()

    def clientConnectionLost(self, connector, reason):
        print reason
        print "Connection lost - goodbye!"
        reactor.stop()

class ConnectDialog(FloatLayout):
    connect = ObjectProperty(None)
    cancel = ObjectProperty(None)

class Root(TabbedPanel):
    connection = None
    chat = ObjectProperty(None)
    scrollback = ObjectProperty(None)

    username = None
    password = None

    def send_payload(self, payload, dumps=True):
        '''Take a JSON string and send it down the wire.'''
        if dumps:
            encoded = json.dumps(payload)
        else:
            encoded = payload
        print '<<< ' + encoded + '\n'
        self.connection.write(encoded + '\n')

    def send_auth(self):
        auth = {
            'op': 'auth',
            'ex': {
                'username': self.username,
                'password': self.password,
            }
        }
        self.scrollback.text += '*Authenticating*\n'
        self.send_payload(auth)

    def dismiss_popup(self):
        self._popup.dismiss()

    def send_from_inputbox(self):
        allowed_commands_without_connection = [
            '/connect',
        ]
        command, space, arguments = self.chat.text.partition(' ')
        if self.connection or command in allowed_commands_without_connection and command.startswith('/'):
            if command.startswith('/'):
                command = command[1:]
                if command == 'raw':
                    # Send a raw JSON packet down the wire.
                    self.send_payload(arguments, False)
                elif command == 'quit' or command == 'disconnect':
                    if arguments:
                        self.send_payload(
                            {
                                'op': 'disconnect',
                                'ex': {
                                    'message': arguments,
                                }
                            })
                    else:
                        self.send_payload({'op': 'disconnect'})
                    sys.exit(0)
                elif command == 'connect':
                    # Connect to a server without the connection dialog.
                    # Format should be user:pass server[:port]
                    split = arguments.split(' ', 1)
                    if ':' in split[0] and len(split) == 2:
                        username, password = split[0].split(':')
                        if ':' in split[1]:
                            server, port = split[1].split(':')
                        else:
                            server, port = split[1], 10817
                        self.connect(server, port, username, password)
                    else:
                        self.add_to_scrollback('Format: /connect username:password server[:port]')

                elif command == 'me':
                    # Perform a third-person action.
                    msg = {
                        'op': 'act',
                        'rm': '48557f95', # TODO: Unhardcode
                        'ex': {
                            'message': arguments,
                            'isaction': True
                        }
                    }
                    self.add_to_scrollback('>> ' + arguments)
                    self.send_payload(msg)
            else:
                msg = {
                    'op': 'act',
                    'rm': '48557f95', # TODO: Unhardcode
                    'ex': {
                        'message': self.chat.text
                    }
                }
                self.add_to_scrollback('>> ' + self.chat.text)
                self.send_payload(msg)
        else:
            self.add_to_scrollback(':-( :: Not connected.')

        self.chat.text = ''
        self.chat.focus = True
        self.chat.select_all()

    def add_to_scrollback(self, data):
        self.scrollback.text += data + '\n'

    def handle_payload(self, data):
        print "*** " + data
        parsed = json.loads(data)

        if not parsed.get('op'):
            pass

        if parsed.get('ex') and parsed['ex'].get('isack') == True:
            return

        # TODO: Use fn.py's Option monad.
        if parsed['op'] == 'welcome' and self.username != None and self.password != None:
            self.send_auth()

        if parsed['op'] == 'act':
            if parsed.get('sr') and parsed['ex'].get('message'):
                if parsed['ex'].get('isaction'):
                    self.add_to_scrollback('*** %s %s' % (parsed['sr'], parsed['ex']['message']))
                else:
                    self.add_to_scrollback('<%s> %s' % (parsed['sr'], parsed['ex']['message']))

        if parsed['op'] == 'join':
            self.add_to_scrollback('[JOINED] %s' % (parsed['sr']))

        if parsed['op'] == 'leave' or parsed['op'] == 'disconnect':
            bracketed = 'LEFT' if parsed['op'] == 'leave' else 'QUIT'
            if parsed.get('ex') and parsed['ex']['message']:
                msg = '[%s] %s (%s)' % (bracketed, parsed['sr'], parsed['ex']['message'])
            else:
                msg = '[%s] %s' % (bracketed, parsed['sr'])

            self.add_to_scrollback(msg)

    def on_connection(self, connection):
        print "/!\ Connected successfully!"
        self.connection = connection

    def show_connect(self):
        content = ConnectDialog(connect=self.connect, cancel=self.dismiss_popup)
        self._popup = Popup(title="Connect to Network", content=content, size_hint=(0.9, 0.9))
        self._popup.open()

    def connect(self, server, port, username, password):
        self.username = username
        self.password = password
        factory = TenthbitClientFactory(self)
        reactor.connectSSL(server, int(port), factory, ssl.ClientContextFactory())

class Bitty(App):
    pass

Factory.register('Root', cls=Root)
Factory.register('ConnectDialog', cls=ConnectDialog)

if __name__ == '__main__':
    Bitty().run()