113 lines
3.1 KiB
Python
Executable File
113 lines
3.1 KiB
Python
Executable File
#!/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
|
|
|
|
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", "--rooms", dest="rooms",
|
|
help="MUC rooms to join separated by comma")
|
|
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')
|
|
log = logging.getLogger(__name__)
|
|
log.addHandler(JournalHandler())
|
|
|
|
|
|
# Setup the MUCBot and register plugins. Note that while plugins may
|
|
# have interdependencies, the order in which you register them does
|
|
# not matter.
|
|
|
|
jid = args.jid
|
|
password = args.password
|
|
rooms = args.rooms
|
|
nick = args.nick
|
|
|
|
if not jid:
|
|
jid = "isengard@a-lec.org"
|
|
if not password:
|
|
password = getpass()
|
|
if not rooms:
|
|
rooms = "tests-isengard@salons.a-lec.org"
|
|
if not nick:
|
|
nick = "Isengard"
|
|
|
|
xmpp = MUCBot(jid, password, rooms.split(","), 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()
|