import os

from database.db import BotDB

# Получаем текущую директорию
current_dir = os.path.dirname(__file__)

# Переходим на уровень выше
parent_dir = os.path.dirname(current_dir)

BotDB = BotDB(parent_dir, 'database/tg-bot-database')


def get_filename():
    """Возвращает имя файла без расширения."""
    filename = os.path.basename(__file__)
    filename = os.path.splitext(filename)[0]
    return filename


def main():
    # Проверка версии миграций
    current_version = BotDB.get_current_version()  # Добавьте функцию для получения версии

    # Выполнение миграций и проверка последней версии
    if current_version < 2:
        # Скрипты миграции
        create_table_sql_1 = """
            CREATE TABLE IF NOT EXISTS "post_from_telegram_suggest"
                (
                    message_id             INTEGER not null,
                    text                   TEXT,
                    helper_text_message_id INTEGER,
                    author_id              INTEGER,
                    created_at             TEXT
                );
            """
        create_table_sql_2 = """
                    CREATE TABLE IF NOT EXISTS message_link_to_content (
                    post_id INTEGER NOT NULL,
                    message_id INTEGER NOT NULL
                );
            """
        create_table_sql_3 = """
            CREATE TABLE IF NOT EXISTS content_post_from_telegram (
                    message_id INTEGER NOT NULL,
                    content_name TEXT NOT NULL,
                    content_type TEXT
                );
        """
        # Применение миграции
        BotDB.create_table(create_table_sql_1)
        BotDB.create_table(create_table_sql_2)
        BotDB.create_table(create_table_sql_3)
        filename = get_filename()

        BotDB.update_version(2, filename)


if __name__ == "__main__":
    main()