#!/usr/bin/env python
from main import app
from app.extensions import db
from app.tools.data import save
from app.tools.logs import write
from sqlalchemy_utils import create_database, database_exists
from cryptography.fernet import Fernet
from os import mkdir, path
from pathlib import Path

database_uri = app.config.get('SQLALCHEMY_DATABASE_URI')

with app.app_context():
    data = Path(app.config.get('DATA'))
    if not path.isdir(f'./{data}'): mkdir(f'./{data}')
    if not path.isdir(f'./{data}/questions'): mkdir(f'./{data}/questions')
    if not path.isfile(f'./{data}/.gitignore'):
        with open(f'./{data}/.gitignore', 'w') as file: file.write(f'*')
    if not path.isfile(f'./{data}/config.json'): save({}, 'config.json')
    if not path.isdir(f'./{data}/logs'): mkdir(f'./{data}/logs')
    if not path.isfile(f'./{data}/logs/users.log'): write('users.log', 'Log file created.')
    if not path.isfile(f'./{data}/logs/system.log'): write('system.log', 'Log file created.')
    if not path.isfile(f'./{data}/logs/tests.log'): write('tests.log', 'Log file created.')
    if not database_exists(database_uri):
        create_database(database_uri)
        write('system.log', 'No database found. Creating a new database.')
        from app.models import *
        db.create_all()
        write('system.log', 'Creating database schema.')
    if not path.isfile(f'./{data}/.encryption.key'):
        write('system.log', 'No encryption key found. Generating new encryption key.')
        with open(f'./{data}/.encryption.key', 'wb') as key_file:
            key = Fernet.generate_key()
            key_file.write(key)