from vkbottle.bot import Blueprint
import random
from src.help import for_token
from sqlalchemy import Column, Integer, String, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, scoped_session
from src.for_bd import session, User
from src.dev_mode import dev_mode
from src.keyboards.keyboards import start

bot = Blueprint()
from vkbottle import BaseMiddleware
from vkbottle.bot import Message
class RegistrationMiddleware(BaseMiddleware):
    async def pre(self, message: Message):
        if message.peer_id < 2000000000:
            if session.query(User).filter_by(id=message.from_id).first() == None:
                a = ''
                for i in range(60):
                    a += random.choice(for_token)
                users = 0
                for _ in session.query(User).all():
                    users+=1
                testUser = User(message.from_id, users, 'Игрок', 0, 0, 'start', a, 0, 0, 0)
                session.add(testUser)
                session.commit()
                testUser.token = 'Nope :D'
                await dev_mode(message, testUser)
                testUser.token = a
                await message.answer(f'Вы зарегистрированны, ваш uid {users}')
                await message.answer('Используются обычные кнопки. (можно изменить)')
                await message.answer('Выберите персонажа', keyboard=start)
            else:
                await message.answer('[Debug] Ah, shit, here we go again.')
        else:
            await message.answer('[Debug] Ah, shit, here we go again.')
        return ''
bot.labeler.message_view.register_middleware(RegistrationMiddleware())