import lib

import time

import socket

import logging as log

class Server(object):

    def __init__(self, config):

        self.host = '0.0.0.0'

        self.port = int(config.get("Comm", "port"))

        self.finish = 0

        addr = (self.host, self.port)

        log.debug("Setup TCP server at %s:%s" % addr)

        self.tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        self.tcp.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

        self.tcp.bind(addr)

        self.tcp.listen(1)

        log.info("TCP server listening at %s:%s (sock=%d)" %

                 (self.host, self.port, self.tcp.fileno()))

    def getMsg(self):

        bytes = []

        try:

            while 1:

                c = self.con.recv(1)

                bytes.append(c)

                if not c or c == "\n":

                    break

        except Exception, e:

            log.error("Error reading message from client: %s" % e)

            return None

        if not bytes or bytes[-1] != "\n":

            msg = "".join(bytes)

            log.error("Invalid message from client: %r" % msg)

            return None

        # remove \n and \r

        bytes.pop()

        if bytes[-1] == "\r":

            bytes.pop()

        msg = "".join(bytes)

        log.debug("RECV: %r" % msg)

        return msg

    def sendMsg(self, msg):

        log.debug("SEND: %r" % msg)

        self.con.send(msg + "\n")

    def sendOk(self, payload=None):

        self.sendMsg("OK %d" % bool(payload is not None))

        if payload is not None:

            if not isinstance(payload, (tuple, list)):

                payload = (payload,)

            for e in payload:

                self.sendMsg("+%s" % e)

            self.sendMsg(".")

    def sendNotOk(self, reason=""):

        self.sendMsg("NOTOK %r" % reason)

    def getRequest(self):

        log.debug("Wait for client request at %s:%s (sock=%d)" %

                  (self.host, self.port, self.tcp.fileno()))

        self.con, self.client = self.tcp.accept()

        log.info("Incoming request from %s (con=%s)" %

                 (self.client, self.con.fileno()))

        return (self.con, self.client, self.port)

    def disconnect_client(self, connection):

        log.info("Closed request from %s (con=%s)" %

                 (self.client, self.con.fileno()))

        connection.close()

    def stop(self):

        log.debug("Stop")

        self.tcp.close()