stream.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # Copyright 2010 Maurizio Porrato <maurizio.porrato@gmail.com>
   5 # See LICENSE.txt for copyright info
   6
   7 import subprocess
   8 from frn.protocol.client import FRNClient, FRNClientFactory
   9 from frn.user import FRNUser
  10 from twisted.internet import reactor, task
  11 from twisted.internet.defer import DeferredList
  12 from twisted.python import log
  13 import os, string
  14 from Queue import Queue
  15 from twisted.internet.task import LoopingCall
  16 import fcntl
  17
  18 from twisted.internet.protocol import ProcessProtocol
  19
  20 class StreamingProtocol(ProcessProtocol):
  21
  22     _silenceFrame = open("sounds/silence-gsm.gsm", "rb").read()
  23
  24     def __init__(self):
  25         self._frames = Queue()
  26         self._streamTimer = LoopingCall.withCount(self.sendStreamFrame)
  27
  28     def connectionMade(self):
  29         log.msg("Streaming process started")
  30         self._streamTimer.start(0.20)
  31
  32     def inConnectionLost(self):
  33         log.msg("connection lost")
  34         self._streamTimer.stop()
  35
  36     def processExited(self, status):
  37         log.msg("Streaming process exited (%s)" % str(status))
  38
  39     def sendStreamFrame(self, count):
  40         for i in range(count):
  41             if self._frames.empty():
  42                 frames = self._silenceFrame
  43             else:
  44                 frames = self._frames.get_nowait()
  45             self.transport.write(frames)
  46
  47     def feed(self, frame):
  48         self._frames.put_nowait(frame)
  49
  50     def eof(self):
  51         self.transport.closeStdin()
  52
  53
  54 STREAM_CMD="""tools/gsmstream.sh"""
  55
  56 class FRNStream(FRNClient):
  57
  58     def __init__(self):
  59         self._stream = StreamingProtocol()
  60
  61     def connectionMade(self):
  62         FRNClient.connectionMade(self)
  63         reactor.spawnProcess(self._stream, "bash", ["bash", "-c", STREAM_CMD], os.environ)
  64
  65     def connectionLost(self, reason):
  66         FRNClient.connectionLost(self, reason)
  67         self._stream.eof()
  68
  69     def getClientName(self, client_id):
  70         if self.clientsById.has_key(client_id):
  71             return self.clientsById[client_id]['ON']
  72         else:
  73             return client_id
  74
  75     def audioFrameReceived(self, from_id, frames):
  76         self._stream.feed(frames)
  77         self.pong()
  78
  79     def loginResponse(self, info):
  80         log.msg("Login: %s" % info['AL'])
  81
  82     def clientsListUpdated(self, clients):
  83         self.clients = clients
  84         self.clientsById = dict([(i['ID'], i) for i in clients])
  85
  86
  87 class FRNStreamFactory(FRNClientFactory):
  88     protocol = FRNStream
  89     reactor = reactor
  90
  91
  92 if __name__ == '__main__':
  93     import sys
  94     from os.path import dirname, join as pjoin
  95     from ConfigParser import ConfigParser
  96
  97     log.startLogging(sys.stderr)
  98
  99     basedir = dirname(__file__)
 100
 101     acfg = ConfigParser()
 102     acfg.read(['/etc/grn/accounts.conf',
 103         pjoin(basedir,'accounts.conf'), 'accounts.conf'])
 104
 105     scfg = ConfigParser()
 106     scfg.read(['/etc/grn/servers.conf',
 107         pjoin(basedir,'servers.conf'), 'servers.conf'])
 108
 109     argc = len(sys.argv)
 110     if argc >= 3:
 111         server_name, network_name = sys.argv[2].split(':',1)
 112         account_cfg = acfg.items(sys.argv[1])+[('network', network_name)]
 113         server_cfg = scfg.items(server_name)
 114         server = scfg.get(server_name, 'server')
 115         port = scfg.getint(server_name, 'port')
 116
 117         d = dict(account_cfg)
 118         user = FRNUser(
 119             EA=d['email'],
 120             PW=d['password'], ON=d['operator'],
 121             BC=d['transmission'], DS=d['description'],
 122             NN=d['country'], CT=d['city'], NT=d['network'])
 123         reactor.connectTCP(server, port, FRNStreamFactory(user))
 124         reactor.run()
 125
 126
 127 # vim: set et ai sw=4 ts=4 sts=4: