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

import src.core as core

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

def readopt(name):
	global opts
	for e in opts:
		if e[0] == name:
			return e[1]
	return None

def usage():
	print("Usage: %s [-q|-d] [-c config.yaml]" % 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)")

def open_db(config):
	return shelve.open(config["database"])

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)

	db = open_db(config)

	core.init(config, db)

	try:
		start_new_thread(core.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
	if readopt("-h") or readopt("--help"):
		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)