import os
import sys

# Добавляем путь к корневой директории проекта
sys.path.append(os.path.dirname(os.path.dirname(__file__)))

from database.db import BotDB

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

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

BotDB = BotDB(parent_dir, 'tg-bot-database.db')


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 < 3:
        # Скрипт миграции для создания таблицы our_users
        create_table_sql = """
            CREATE TABLE IF NOT EXISTS "our_users" (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                user_id INTEGER NOT NULL UNIQUE,
                first_name TEXT,
                full_name TEXT,
                username TEXT,
                is_bot BOOLEAN DEFAULT 0,
                language_code TEXT,
                date_added TEXT,
                date_changed TEXT,
                has_stickers BOOLEAN DEFAULT 0
            );
        """
        
        # Применение миграции
        BotDB.create_table(create_table_sql)
        filename = get_filename()

        BotDB.update_version(3, filename)


if __name__ == "__main__":
    main()