#!/usr/bin/env python3

from systemd.journal import JournalHandler
from getpass import getpass
from argparse import ArgumentParser
import random
import threading
import socket
import time
import signal
import logging

from bot import MUCBot
from data import DataStore
from localServer import LocalServer


# Logging
log = logging.getLogger(__name__)
log.addHandler(JournalHandler())
log.setLevel(logging.INFO)

if __name__ == '__main__':

    def cleanExit():
        global localservthread
        localservthread.pleaseStop = True
        # stop tcp server
        sockfd = socket.socket()
        port = 12346
        sockfd.connect(('127.0.0.1', port))
        sockfd.send(b'')
        sockfd.close()

        exit(0)


    def signalHandler(sig, frame):
        log.info("Exiting...")
        cleanExit()

    # Register SIGINT signal handler
    signal.signal(signal.SIGINT, signalHandler)

    # Setup the command line arguments.
    parser = ArgumentParser()

    # Output verbosity options.
    parser.add_argument("-q", "--quiet", help="set logging to ERROR",
                        action="store_const", dest="loglevel",
                        const=logging.ERROR, default=logging.INFO)
    parser.add_argument("-d", "--debug", help="set logging to DEBUG",
                        action="store_const", dest="loglevel",
                        const=logging.DEBUG, default=logging.INFO)

    # JID and password options.
    parser.add_argument("-j", "--jid", dest="jid",
                        help="JID to use")
    parser.add_argument("-p", "--password", dest="password",
                        help="password to use")
    parser.add_argument("-r", "--room", dest="room",
                        help="MUC room to join")
    parser.add_argument("-n", "--nick", dest="nick",
                        help="MUC nickname")

    args = parser.parse_args()

    # Setup logging.
    logging.basicConfig(level=args.loglevel,
                        format='%(levelname)-8s %(message)s')

    # Setup the MUCBot and register plugins. Note that while plugins may
    # have interdependencies, the order in which you register them does
    # not matter.
    xmpp = MUCBot(args.jid, args.password, args.room, args.nick)
    xmpp.register_plugin('xep_0030') # Service Discovery
    xmpp.register_plugin('xep_0045') # Multi-User Chat
    xmpp.register_plugin('xep_0199') # XMPP Ping

    # Create buffer
    store = DataStore(xmpp)
    xmpp.datastore = store

    localservthread = LocalServer(store)
    localservthread.start()

    while True:
        time.sleep(1)

        # Connect to the XMPP server and start processing XMPP stanzas.
        xmpp.connect()

        # Check local server status
        if not localservthread.is_alive():
            exit(1)

        try:
            xmpp.process()
        except KeyboardInterrupt:
            cleanExit()