Changeset 24 for tabled.py

Timestamp:

02/04/08 15:41:32 (4 years ago)

Author:

mike

Message:

Got web shit working

Files:

• tabled.py (modified) (7 diffs)

tabled.py

@@ -12 +12 @@
 from twisted.protocols.basic import NetstringReceiver
 from twisted.internet import defer
+from twisted.web2 import server, http, resource, channel
 
 tables = {}
-
-class PokerTableProtocol(NetstringReceiver):
+class BasePokerTableHandler:
     def __init__(self):
         self.table = None
         self.id = None
-
+    
     def sendMessage(self, data):
         try:
-            self.sendString( cjson.encode(
-                data) )
+            self.sendMessageViaTransport( cjson.encode(data) )
         except cjson.EncodeError,e:
             print data
@@ -31 +30 @@
         self.table.addDeferred(self.id, d)
 
+    def sendMessageViaTransport(self,data):
+        """This needs to be overidden"""
+        pass
 
-    def stringReceived(self, line):
-        
+    def dataRecieved(self,line):
         try:
             decoded = cjson.decode(line)
             if decoded == None or "method" not in decoded:
-                self.sendString( cjson.encode(
+                self.sendMessageViaTransport( cjson.encode(
                     protocol.MakeError(101,"Messages must contain methods")) )
                 return
@@ -43 +44 @@
             if method == "connect":
                 if self.table or self.id:
-                    self.sendString( cjson.encode(
+                    self.sendMessageViaTransport( cjson.encode(
                         protocol.MakeError(103,"Already connected")) )
                     return
@@ -53 +54 @@
                 self.table = tables[table_name]
                 if self.table.isFull():
-                    self.sendString( cjson.encode(
+                    self.sendMessageViaTransport( cjson.encode(
                         protocol.MakeError(104,"Table is full")))
                     return
@@ -66 +67 @@
                 self.table.action(self.id, decoded["action"],value)
             else:
-                self.sendString( cjson.encode(protocol.MakeError(102,"Unknown Method")) )
+                self.sendMessageViaTransport( cjson.encode(protocol.MakeError(102,"Unknown Method")) )
             print method
         except cjson.DecodeError:
@@ -74 +75 @@
         
         print self.id
+        
+
+class PokerTableProtocol(NetstringReceiver,BasePokerTableHandler):
+    sendMessageViaTransport = NetstringReceiver.sendString
+
+    def stringReceived(self, line):
+        self.dataRecieved(line)
+        
     def connectionLost(self, reason):
         pass
 
 
+def makeToken( id, table ):
+    #TODO make this a bit more secure
+    return id + "$$$" + table
 
+class PokerTableWebHandler(BasePokerTableHandler):
+    def __init__(self):
+        BasePokerTableHandler.__init__(self)
+        self.__waitingRequest=None
+        self.__messageQueue=[]
+
+    def sendMessageViaTransport(self, data):
+        self.__messageQueue.append(data)
+        self.checkWaiting()
+
+    def sendMessage(self, data):
+        #Add your token name to the data
+        data["token"] = makeToken( self.id, self.table.name )
+        BasePokerTableHandler.sendMessage(self,data)
+
+    def checkWaiting(self):
+        if self.__waitingRequest != None and len(self.__messageQueue) > 0:
+            self.__waitingRequest.callback(self.__messageQueue.pop(0))
+            self.__waitingRequest = None
+
+    def doRenderStuff(self,nargs):
+        d = defer.Deferred()
+        #yeah, i know we're going to unparse
+        d.addCallback( lambda resp: http.Response(stream=resp) )
+        self.__waitingRequest = d
+
+        if "method" in nargs and not nargs["method"] == "get_update":
+           self.dataRecieved(cjson.encode(nargs))
+
+        self.checkWaiting()
+
+        return d
+
+
+class PokerTableWebProtocoll(resource.Resource):
+    addSlash = True
+    
+    def __init__(self):
+        resource.Resource.__init__(self)
+        self.__webHandlers = {}
+
+
+       
+
+    def render(self, ctx):
+        nargs = {}
+        for k,v in ctx.args.items(): nargs[k] = v[0]
+        print nargs
+        handler = None
+        if "token" in nargs: handler =  self.__webHandlers[nargs["token"]]
+        elif "method" in nargs and nargs["method"] == "connect":
+            token = makeToken(nargs["client_id"], nargs["table_name"])
+            handler = PokerTableWebHandler()
+            self.__webHandlers[token] = handler
+        else: return http.Response(stream=cjson.encode(protocol.error(200,"Require either token or connect")))
+
+        return handler.doRenderStuff(nargs)
 
 def main():
@@ -89 +158 @@
     factory = Factory()
     factory.protocol = PokerTableProtocol
-
+    
+    site = server.Site(PokerTableWebProtocoll())
+    reactor.listenTCP(8080, channel.HTTPFactory(site))
     # 8007 is the port you want to run under. Choose something >1024
     reactor.listenTCP(port, factory)