import time
import mysql.connector
from flask import Flask
from main import app
from main import db_init
import main
from gunicorn.app.base import BaseApplication
import os
import dotenv
from apscheduler.schedulers.background import BackgroundScheduler
import hvac
import sys
import json

dotenv.load_dotenv()
db_config = {
    'host': os.getenv('DB_HOST'),
    'database': os.getenv('DB_NAME'),
    'user': os.getenv('DB_USER'),
    'password': os.getenv('DB_PASSWORD')
}

class GunicornApp(BaseApplication):
    def __init__(self, app, options=None):
        self.options = options or {}
        self.application = app
        super().__init__()

    def load_config(self):
        for key, value in self.options.items():
            if key in self.cfg.settings and value is not None:
                self.cfg.set(key.lower(), value)

    def load(self):
        return self.application

def establish_database_connection():
    while True:
        try:
            conn = mysql.connector.connect(**db_config)
            if conn.is_connected():
                print('Connected to database')
                break
        except mysql.connector.Error as e:
            print('Connecting to database...', e)
        time.sleep(1)
    conn.close()



if __name__ == '__main__':
    # scheduler = BackgroundScheduler()
    # Run main.update_events every hour
    # scheduler.add_job(update_events, 'cron', hour='*')
    # Run main.reset_worship_chat every monday at 12am
    # scheduler.add_job(reset_worship_chat, 'cron', day_of_week='mon', hour=0, minute=0)
    # Run elvanto.refresh_all_tokens every 6 hours
    # scheduler.add_job(refresh_oauth_tokens, 'interval', hours=6)
    # scheduler.start()

    establish_database_connection()
    db_init()
    workers = os.getenv('WORKERS')
    threads = os.getenv('THREADS')
    if workers is None:
        workers = 1
    if threads is None:
        threads = 2
    workers = int(workers)
    threads = int(threads)
    options = {
        'bind': '0.0.0.0:5000',
        'workers': workers,
        'threads': threads,
    }
    gunicorn_app = GunicornApp(app, options)
    print('Starting server with ' + str(workers) + ' workers and ' + str(threads) + ' threads', flush=True)
    gunicorn_app.run()