import os
import sqlalchemy

from config import Config
from sqlalchemy.orm import (sessionmaker, scoped_session)

MM_ENV = os.environ.get('MM_ENV', 'PRODUCTION')
testing = False

if MM_ENV == 'PRODUCTION':
    dbname = os.environ.get('MM_PRODUCTION_DBNAME', 'musicmuster_prod')
    dbuser = os.environ.get('MM_PRODUCTION_DBUSER', 'musicmuster')
    dbpw = os.environ.get('MM_PRODUCTION_DBPW', 'musicmuster')
    dbhost = os.environ.get('MM_PRODUCTION_DBHOST', 'localhost')
elif MM_ENV == 'TESTING':
    dbname = os.environ.get('MM_TESTING_DBNAME', 'musicmuster_testing')
    dbuser = os.environ.get('MM_TESTING_DBUSER', 'musicmuster_testing')
    dbpw = os.environ.get('MM_TESTING_DBPW', 'musicmuster_testing')
    dbhost = os.environ.get('MM_TESTING_DBHOST', 'localhost')
    testing = True
elif MM_ENV == 'DEVELOPMENT':
    dbname = os.environ.get('MM_DEVELOPMENT_DBNAME', 'musicmuster_dev')
    dbuser = os.environ.get('MM_DEVELOPMENT_DBUSER', 'musicmuster')
    dbpw = os.environ.get('MM_DEVELOPMENT_DBPW', 'musicmuster')
    dbhost = os.environ.get('MM_DEVELOPMENT_DBHOST', 'localhost')
else:
    raise ValueError(f"Unknown MusicMuster environment: {MM_ENV=}")

MYSQL_CONNECT = f"mysql+mysqldb://{dbuser}:{dbpw}@{dbhost}/{dbname}"

engine = sqlalchemy.create_engine(
    MYSQL_CONNECT,
    encoding='utf-8',
    echo=Config.DISPLAY_SQL,
    pool_pre_ping=True
)

Session = scoped_session(sessionmaker(bind=engine))