gnuradionetwork/frn/protocol/client.py

# -*- coding: utf-8 -*-

from twisted.internet.protocol import ClientFactory
from twisted.protocols.basic import LineReceiver
from frn.utility import *


class FRNClient(LineReceiver):

    client_version = 2010002

    def connectionMade(self):
        self.login()

    def ready(self):
        self.status = 'READY'
        self.msgbuffer = []
        self.phase = 0
        self.setRawMode()

    def startMultiLineMessage(self, msgtype, rest=''):
        self.status = msgtype
        self.phase = None
        self.setLineMode(rest)

    def collectMultiLineMessage(self, line):
        if self.phase is None:
            while line[0] not in '0123456789': # needed for client list
                line = line[1:]
            self.expected_lines = int(line.strip())
            self.msgbuffer = []
            self.phase = 0
        else:
            self.msgbuffer.append(line)
            self.phase += 1
            if self.phase >= self.expected_lines:
                handler = getattr(self, 'on_'+self.status, self.on_unimplemented)
                message = self.msgbuffer
                self.ready()
                handler(message)

    def startAudioMessage(self, rest=''):
        self.status = 'AUDIO'
        self.phase = None
        self.msgbuffer = ''
        if len(rest) > 0:
            self.collectAudioMessage(rest)

    def collectAudioMessage(self, data):
        needed = min(327, max(0, 327-len(self.msgbuffer)))
        if len(data) > 0:
            self.msgbuffer += data[:needed]
        if len(self.msgbuffer) >= 327:
            audio_data = self.msgbuffer
            source = ord(audio_data[0])*256+ord(audio_data[1])
            self.ready()
            self.on_AUDIO(source, audio_data[2:])
        if len(data) > needed:
            self.dataReceived(data[needed:])

    def lineReceived(self, line):
        if self.status == 'AUTH':
            if self.phase == 0:
                self.latest_client_version = int(line.strip())
                self.phase = 1
            else:
                self.serverdata = parseSimpleXML(line.strip())
                print self.serverdata
                if int(self.serverdata['sv']) > 2009004:
                    self.sendLine(makeAuthKey(self.serverdata['kp']))
                self.ready()
        else:
            self.collectMultiLineMessage(line)

    def rawDataReceived(self, data):
        if self.status == 'READY':
            packet_type = ord(data[0])
            if packet_type == 0: # Keepalive
                self.sendLine('P')
            elif packet_type == 1: # TX ack
                self.status == 'TX'
                self.phase = 0
                if len(data) > 1:
                    self.dataReceived(data[1:])
            elif packet_type == 2: # Audio
                self.startAudioMessage(data[1:])
            elif packet_type == 3: # Client list
                self.startMultiLineMessage('CLIENTS', data[1:])
            elif packet_type == 4: # SMS
                self.startMultiLineMessage('SMS', data[1:])
            elif packet_type == 5: # Channel list
                self.startMultiLineMessage('CHANNELS', data[1:])
            else:
                print "Unknown packet type %d" % packet_type
        elif self.status == 'AUDIO':
            self.collectAudioMessage(data)
        elif self.status == 'TX':
            if self.phase == 0:
                self.phase = 1
                self.on_TX(ord(data[0])*256+ord(data[1]))

    def login(self):
        d = self.factory.client_id
        fields = [
            ('VX', self.client_version),
            ('EA', d['email']),
            ('PW', d['password']),
            ('ON', d['operator']),
            ('BC', d['transmission']),
            ('DS', d['description']),
            ('NN', d['country']),
            ('CT', d['city']),
            ('NT', d['network'])
        ]
        ap = "CT:"+formatSimpleXML(fields)
        self.status = 'AUTH'
        self.phase = 0
        self.sendLine(ap)
        #self.request_rx()

    def set_status(self, status):
        self.sendLine('ST:%s' % str(status))

    def request_rx(self):
        self.sendLine('RX0')

    def request_tx(self):
        self.sendLine('TX0')

    def send_audio(self, frame):
        self.sendLine('TX1')
        self.transport.write(frame)

    def send_SMS(self, dest, text):
        self.sendLine('TM:'+formatSimpleXML(dict(ID=dest, MS=text)))

    def on_unimplemented(self, msg):
        print msg

    def on_AUDIO(self, from_id, frames):
        pass

    def on_TX(self, my_id):
        pass


class FRNClientFactory(ClientFactory):

    protocol = FRNClient

    def __init__(self, **kw):
        self.client_id = kw

    def startedConnecting(self, connector):
        print 'Started to connect.'

    def buildProtocol(self, addr):
		print 'Connected.'
		return ClientFactory.buildProtocol(self, addr)

    def clientConnectionLost(self, connector, reason):
        print 'Lost connection.  Reason:', reason

    def clientConnectionFailed(self, connector, reason):
        print 'Connection failed. Reason:', reason

# vim: set et ai sw=4 ts=4 sts=4: