from hub.persistence import BaseRepo
from hub.persistence.models import City
from hub.persistence.models import HeatPumpSimulation
from hub.persistence.models import User
from hub.persistence.models import UserRoles
from hub.persistence.models import Application
from hub.persistence.models import UserApplications
from hub.persistence.repositories import UserRepo
from hub.persistence.repositories import ApplicationRepo
from hub.hub_logger import logger


class DBSetup:

  def __init__(self, db_name, app_env, dotenv_path, admin_password, application_uuid):
    """
    Creates database tables a default admin user and a default admin app with the given password and uuid
    :param db_name:
    :param app_env:
    :param dotenv_path:
    """
    repo = BaseRepo(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    User.__table__.create(bind=repo.engine, checkfirst=True)
    City.__table__.create(bind=repo.engine, checkfirst=True)
    Application.__table__.create(bind=repo.engine, checkfirst=True)
    UserApplications.__table__.create(bind=repo.engine, checkfirst=True)
    HeatPumpSimulation.__table__.create(bind=repo.engine, checkfirst=True)
    self._user_repo = UserRepo(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._application_repo = ApplicationRepo(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._create_admin_user(self._user_repo, admin_password)
    self._create_admin_app(self._application_repo, application_uuid)

  @staticmethod
  def _create_admin_app(application_repo, application_uuid):
    name = 'AdminTool'
    description = 'Admin tool to control city persistence and to test the API v1.4'
    print('Creating default admin tool application...')
    application = application_repo.insert(name, description, application_uuid)
    if type(application) is dict:
      logger.info(application)
    else:
      msg = f'Created Admin tool with application_uuid: {application_uuid}'
      print(msg)
      logger.info(msg)

  @staticmethod
  def _create_admin_user(user_repo, admin_password, application_id):
    password = admin_password
    print('Creating default admin user...')
    user = user_repo.insert('Administrator', password, UserRoles.Admin, application_id)
    if type(user) is dict:
      logger.info(user)
    else:
      print(f'Created Admin user')
      logger.info(f'Created Admin user')