#!/usr/bin/env python3
import logging
import yaml
import threading
import sys
import os
import shelve
import getopt
from pickle import Unpickler

from . import bot

def start_new_thread(func, join=False, args=(), kwargs=None):
	t = threading.Thread(target=func, args=args, kwargs=kwargs)
	if not join:
		t.daemon = True
	t.start()
	if join:
		t.join()

def usage():
	print("Usage: %s [-q|-d] [-c file]" % sys.argv[0])
	print("Options:")
	print("  -h    Display this text")
	print("  -q    Quiet, set log level to WARNING")
	print("  -c    Location of config file (default: ./config.yaml)")

# well this is just dumb
class RenamingUnpickler(Unpickler):
	def find_class(self, module, name):
		if module == "src.core":
			module = "supportbot.bot"
		return super().find_class(module, name)

def main(configpath, loglevel=logging.INFO):
	with open(configpath, "r") as f:
		config = yaml.safe_load(f)

	logging.basicConfig(format="%(levelname)-7s [%(asctime)s] %(message)s", datefmt="%Y-%m-%d %H:%M:%S", level=loglevel)

	shelve.Unpickler = RenamingUnpickler
	db = shelve.open(config["database"])

	bot.init(config, db)

	try:
		start_new_thread(bot.run, join=True)
	except KeyboardInterrupt:
		logging.info("Interrupted, exiting")
		db.close()
		os._exit(1)

if __name__ == "__main__":
	try:
		opts, args = getopt.getopt(sys.argv[1:], "hqc:", ["help"])
	except getopt.GetoptError as e:
		print(str(e))
		exit(1)

	# Process command line args
	def readopt(name):
		for e in opts:
			if e[0] == name:
				return e[1]
	if readopt("-h") is not None or readopt("--help") is not None:
		usage()
		exit(0)
	loglevel = logging.INFO
	if readopt("-q") is not None:
		loglevel = logging.WARNING
	configpath = "./config.yaml"
	if readopt("-c") is not None:
		configpath = readopt("-c")

	# Run the actual program
	main(configpath, loglevel)