import os
from sshtunnel import SSHTunnelForwarder
from dotenv import load_dotenv

load_dotenv()


class config:
    BOT_TOKEN = os.environ.get("TOKEN")
    API_KEY = os.environ.get("API_KEY")
    ADMIN_ID = os.environ.get("ADMIN_ID")
    ENV = os.getenv("ENV", "PROD")

    DB_PORT = int(os.getenv("DB_PORT", 27017))
    if ENV == "DEV":
        server = SSHTunnelForwarder(
            (os.environ.get("SSH_HOST"), 22),
            ssh_private_key=os.getenv("SSH_KEY", None),
            ssh_username=os.environ.get("SSH_USER"),
            ssh_password=os.getenv("SSH_PASSWORD", None),
            remote_bind_address=("localhost", DB_PORT),
        )
        server.start()

        DB_PORT = server.local_bind_port


class BotCommands:
    Start = ["start", "s"]
    Help = ["help", "h"]


class Messages:
    WELCOME_LIST = [
        "Привет! Это бот  Quick dates (Random Coffee). Чтобы зарегистрироваться, пожалуйста, ответь на следующие вопросы."  # несколько сообщений подряд
    ]
    ERROR = "An error occurred"
    ENTER_NAME = "Как тебя зовут?"
    ENTER_AGE = "Сколько тебе лет?"
    ENTER_GENDER = "Выбери пол (женский, мужской)"
    GENDER_WOMAN = "Я девушка"  # кнопки выбора
    GENDER_MAN = "Я парень"  # кнопки выбора
    ENTER_FACULTY = "С какого ты факультета?"
    ENTER_GROUP = "Напиши номер своей группы"
    REGISTRATION_CONFIRM = "Ты ввел следующие данные:\
        \n\nИмя: {}\
        \nВозраст: {}\
        \nПол: {}\
        \nФакультет: {}\
        \nГруппа: {}\
        \n\nВсё верно?"
    ALREADY_REGISTERED = "Ты уже зарегистрирован. Хочешь изменить данные?"