#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sun Jan 26 17:50:25 2020

@author: suwako
"""

import sys, multiprocessing, socket, time
if sys.platform == 'win32':
    import multiprocessing.reduction
from colorama import Fore, Style
import numpy as np


def debug(*s): print("[D  ] : " +
                     str(', '.join(map(str, s))) +
                     Style.RESET_ALL)


def info(*s): print(Fore.GREEN +
                    Fore.LIGHTGREEN_EX +
                    "[I -] : " +
                    Style.RESET_ALL +
                    str(', '.join(map(str, s))) +
                    Style.RESET_ALL)


def warn(*s): print(Fore.MAGENTA +
                    "[W ~] : " +
                    Style.RESET_ALL +
                    str(', '.join(map(str, s))) +
                    Style.RESET_ALL)


def error(*s): print(Fore.RED +
                     Fore.LIGHTRED_EX +
                     "[E !] : " +
                     Style.RESET_ALL +
                     str(', '.join(map(str, s))) +
                     Style.RESET_ALL)

level = 1
for i in range(level):
    exec("debug info warn error".split()[i] + "= lambda *x: None")


def handle(connection, address):
    try:
        info("Connected at "+ ':'.join(map(str,address)))
        while True:
            data = connection.recv(1024)
            if data == b"":
                info("Socket closed remotely at " + ':'.join(map(str,address)))
                break
            debug("Received data %r" % data)
            connection.sendall(data)
            debug("Sent data")
    except:
        warn("Problem handling request")
        connection.close()
        raise
    finally:
        debug("Closing socket with " + ':'.join(map(str,address)))
        connection.close()

class Server(object):
    def __init__(self, hostname, port):
        self.hostname = hostname
        self.port = port

    def start(self):
        debug("listening")
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind((self.hostname, self.port))
        self.socket.listen(1)

        while True:
            conn, address = self.socket.accept()
            process = multiprocessing.Process(target=handle,
                                              args=(conn, address))
            process.daemon = True
            process.start()


if __name__ == "__main__":
    server = Server("0.0.0.0", 9000)
    try:
        info("Listening")
        server.start()
    except:
        error("Unexpected exception")
        for process in multiprocessing.active_children():
            info("Shutting down process %r", process)
            process.terminate()
            process.join()
        raise
    finally:
        info("Shutting down")
        for process in multiprocessing.active_children():
            info("Shutting down process %r", process)
            process.terminate()
            process.join()
    info("All done")