def main():

    import socket
    import os
    import config
    import datetime
    import signal
    import threading
    from log import logsManager
    import sys
    import select

    socketFile = '/etc/monutil/ip.socket'

    if len(sys.argv) == 4:

        hostname = socket.gethostname()
        ipAddress = socket.gethostbyname(hostname)
        blockedIPAddress = sys.argv[1]
        jail = sys.argv[2]
        live = int(sys.argv[3])
        logTime = datetime.datetime.now()

        clientSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        clientSocket.connect(socketFile)
        data = f"{hostname}|{ipAddress}|{logTime}|ipBlock|{blockedIPAddress}|{jail}|{live}"
        clientSocket.send(data.encode('utf-8'))
        clientSocket.close()
        sys.exit(0)

    else:
        
        if os.path.exists(socketFile):
            os.remove(socketFile)

        stop_event = threading.Event()
        dataBuffer = []

        def publishData(stop_event):

            if config.loggingMode == 'rabbitmq':

                import rabbitmq
                rabbitmq = rabbitmq.rabbitMQClient(config.rabbitmqca,config.rabbitmqcacert,config.rabbitmqcakey,config.rabbitmqHost,config.rabbitmqPort,config.rabbitmqRoutingKey)
            
            while not (stop_event.is_set()):
                if dataBuffer:
                    data = dataBuffer.pop(0)
                    
                    if config.loggingMode == 'rabbitmq':
                        rabbitmq.publish(f"{data}")
                    else:
                        print("Not yet implemented")

        def cleanup(signum, frame):
            
            print("Signal received, shutting down...")
            stop_event.set() 

        signal.signal(signal.SIGTERM, cleanup)
        signal.signal(signal.SIGINT, cleanup)

        def server(stop_event):

            serverSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            serverSocket.bind(socketFile)
            serverSocket.listen(50)

            while not (stop_event.is_set()):
            
                readable, _, _ = select.select([serverSocket], [], [], 1.0)
                
                if readable:
                    clientSocket, _ = serverSocket.accept()
                    data = clientSocket.recv(1024).decode('utf-8')
                    
                    if data:
                
                        dataBuffer.append(data)
                        print(data)
            
                    clientSocket.close()

        publishThread = threading.Thread(target=publishData, args=(stop_event,))
        serverThread = threading.Thread(target=server, args=(stop_event,))
        
        publishThread.start()
        serverThread.start()
        
        publishThread.join()
        serverThread.join()

if __name__ == "__main__":
    main()