#!/usr/bin/python
# First the server



"""
A basic, multiclient 'chat server' using Python's select module
with interrupt handling.

Entering any line of input at the terminal will exit the server.
"""

import select
import socket
import sys
import signal
from communication import send, receive
import threading
import GlobalVariables
import time
BUFSIZ = 1024
class Sender(threading.Thread):
    """ Sends the messages """
    outputs = []
    oldmessage=''
     
    def run(self):
        running = 1
        while running:
            time.sleep(0.001)
            if(self.oldmessage!=GlobalVariables.ServerMessage):
                self.oldmessage=GlobalVariables.ServerMessage
             
                for o in self.outputs:
                    # o.send(msg)
                    #print "sending (glob): ",GlobalVariables.ServerMessage,"\n"
                    send(o,GlobalVariables.ServerMessage )
                if(GlobalVariables.UserInput[1:]=='exit' or GlobalVariables.UserInput[1:]=='EXIT'):
                    running=0
              


         
class CommServer(threading.Thread):
    """ Simple  server using select """
    oldmessage=''
    sender=Sender()
    def __init__(self, port=3490, backlog=5):
        self.clients = 0
        # Client map
        self.clientmap = {}
        # Output socket list
        self.outputs = []
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.server.bind(('',port))
        print 'SERVER: Listening to port',port,'...'
        self.server.listen(backlog)
        self.sender.start()
        # Trap keyboard interrupts
        #signal.signal(signal.SIGINT, self.sighandler)
        threading.Thread.__init__(self)
    #def sighandler(self, signum, frame):
    #    if(GlobalVariables.UserInput[1:]=='exit' or GlobalVariables.UserInput[1:]=='EXIT'):
    #        running=0
    #        for o in self.outputs:
    #            o.close()
    #        
    #        self.server.close()
     #   print "TEST"
        # Close the server
    #    print 'SERVER: Shutting down server...'
        # Close existing client sockets
     #   for o in self.outputs:
      #      o.close()
            
     #   self.server.close()

    def getname(self, client):

        # Return the printable name of the
        # client, given its socket...
        info = self.clientmap[client]
        host, name = info[0][0], info[1]
        return '@'.join((name, host))
        
    def run(self):
        
        inputs = [self.server,sys.stdin]
        self.outputs = []

        running = 1

        while running:
            time.sleep(0.001)
            #if(self.oldmessage!=GlobalVariables.ServerMessage):
            #    self.oldmessage=GlobalVariables.ServerMessage
            #    for o in self.outputs:
            #        # o.send(msg)
            #        #print "sending (glob): ",GlobalVariables.ServerMessage,"\n"
            #        send(o,GlobalVariables.ServerMessage )
            if(GlobalVariables.UserInput[1:]=='exit' or GlobalVariables.UserInput[1:]=='EXIT'):
                running=0
            
            try:
                inputready,outputready,exceptready = select.select(inputs, self.outputs, [])
            except select.error, e:
                break
            except socket.error, e:
                break

            for s in inputready:

                if s == self.server:
                    # handle the server socket
                    client, address = self.server.accept()
                    #print 'SERVER: got connection %d from %s' % (client.fileno(), address)
                    GlobalVariables.ServerMessage = 'SERVER: got connection %d from %s' %( client.fileno(), address)+"\n"
                    # Read the login name
                    cname="TESTUSER"
                    #cname = receive(client).split('NAME: ')[1]
                    #print cname
                    # Compute client name and send back
                    self.clients += 1
                    #send(client, 'FROM SERVER: CLIENT: ' + str(address[0]))
                    #send(client,"test");
                    #send(client,"test2");
                    
                    inputs.append(client)

                    self.clientmap[client] = (address, cname)
                    # Send joining information to other clients
                    msg = '\nFROM SERVER: (Connected: New client (%d) from %s)' % (self.clients, self.getname(client))
                    GlobalVariables.ServerMessage=msg;
                    time.sleep(0.001)
                    #print "appending ", client
                    self.outputs.append(client)
                    self.sender.outputs=self.outputs
                #elif s == sys.stdin:
                    # handle standard input
                 #   junk = sys.stdin.readline()
                 #   if (junk[:-1]=='exit' or junk[:-1]=='EXIT' ):
                 #       running = 0
                        # Close the server
                 #       print 'SERVER: Shutting down server...'
                 #       # Close existing client sockets
                 #       for o in self.outputs:
                 #           o.close()
                        
                elif s!=sys.stdin:
                    # handle all other sockets
                    try:
                        data = s.recv(BUFSIZ)
                        
                        #data = receive(s)
                        #print data
                        if data:
                            GlobalVariables.counter += 1
                            GlobalVariables.counter %= 10
                            data = str(GlobalVariables.counter)+data
                            # Send as new client's message...
                            msg = 'SERVER: You sent: >> ' + data
                            msg+="\n"
                            GlobalVariables.UserInput=data
                            #print "SERVER: recv data: "+data
                            #print "msg is: ",msg

                            #for o in self.outputs:
                                #msg="TSTSTS"
                             #   print "sending ",msg," to ",o
                                
                              #  send(o, msg)
                            if(GlobalVariables.UserInput[1:]=='exit' or GlobalVariables.UserInput[1:]=='EXIT'):
                                running=0
                                # Send data to all except ourselves
                                                            
                        else:
                            GlobalVariables.ServerMessage ='SERVER: %d hung up' % s.fileno()
                            self.clients -= 1
                            s.close()
                            inputs.remove(s)
                            self.outputs.remove(s)
                            self.sender.outputs=self.outputs
                            # Send client leaving information to others
                            msg = 'SERVER: \n(Hung up: Client from %s)' % self.getname(s)
                            #for o in self.outputs:
                                # o.send(msg)
                                # send(o, msg)
                                
                    except socket.error, e:
                        # Remove
                        inputs.remove(s)
                        self.outputs.remove(s)
                        

        GlobalVariables.ServerMessage ='SERVER: Shutting down server...'
        
        # Close existing client sockets
        for o in self.outputs:
            o.close()
        self.server.close()
        #def run(self):
        #    self.serve()
            
if __name__ == "__main__":
    commServer=CommServer(3491,5)
    commServer.start()