from modules.data.database import create as db_create
from modules.data.database import encryption
from modules.data.config import read as config_read
from modules.track.logging import log
import os

def main(session):
    create_directories()
    log("INFO", "Ensuring server directories")

    from modules.data.config import create as config_create
    log("INFO", "Ensuring config file")
    config_create()

    log("INFO", "Ensuring database")
    db_create().tables()

    if session.db_encrypted:
        log("INFO", "Checking encryption")
        encryption(session).mode()

def final_startup(session):
    from modules.data.datetime import timestamp as datetime_timestamp
    datetime_timestamp().generate_slot()

def create_directories():
    paths = ["data", "data/images"]
    if config_read("database", "ShamirSecretSharing"):
        paths = ["data", "data/images", "data/shares/"]
    for path in paths:
        if not os.path.exists(path):
            os.mkdir(path)
            log("INFO", f"Created new directory: {path}")

if __name__ == "__main__":
    main()