import gradio as gr
import os
import asyncio
import socket
import aiodns
import aiohttp
from aiogram import Bot, Dispatcher
from aiogram import Router
from aiogram.fsm.storage.memory import MemoryStorage
from aiogram.types import Message
from aiogram.filters.command import Command
from dotenv import load_dotenv
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
import random
from aiohttp.resolver import AsyncResolver
from aiogram.client.session.aiohttp import AiohttpSession


load_dotenv()
# resolver = aiodns.DNSResolver()
# connector = aiohttp.TCPConnector(resolver=resolver)

dp = Dispatcher(storage=MemoryStorage())
bot_router = Router()


class Step(StatesGroup):
    guesses = State()
    finish = State()


@dp.message(Command("start"))
async def cmd_start(message: Message, state: FSMContext):
    await message.answer("Начнём игру. Я загадал число от 1 до 100 (1 тоже может быть, и 100 тоже может быть). Число целое. Попробуй угадать какое за наименьшее количество попыток.")
    await state.set_state(Step.guesses)
    the_number = random.randint(1, 100)
    await state.update_data(the_number=the_number)
    await state.update_data(attempt=0)


@dp.message(Step.guesses)
async def ask(message: Message, state: FSMContext):
    storage_data = await state.get_data()
    attempt = int(storage_data.get('attempt', 0))
    the_number = int(storage_data.get('the_number'))
    attempt = attempt + 1

    await state.update_data(attempt=attempt)
    try:
        a_number = int(message.text)
        if a_number == the_number:
            await message.answer('Молодец, ты отгадал!')
            await message.answer(f'За {attempt} попыток')
        else:
            # не отгадал. дадим подсказку
            if the_number > a_number:
                await message.answer('Загаданное число больше')
            else:
                await message.answer('Загаданное число меньше')
    except Exception as e:
        await message.answer('Это не число')


async def main():
    # print(await aiodns.getaddrinfo('api.telegram.org', port, family=AF_INET, type=0, proto=0, flags=0))
    nameservers = ["8.8.8.8", "8.8.4.4"]
    resolver = AsyncResolver(nameservers=nameservers)
    
    # Create a TCPConnector with the custom resolver
    connector = aiohttp.TCPConnector(resolver=resolver)
    # resolver = aiohttp.AsyncResolver()
    # connector = aiohttp.TCPConnector(resolver=resolver, family=socket.AF_INET)
    
    session = aiohttp.ClientSession(connector=connector)
    # async with aiohttp.ClientSession(connector=connector) as session:
    # session = AiohttpSession()
    bot = Bot(token=os.getenv('BOT_TOKEN'), session=session)
    print(session)
    await dp.start_polling(bot)
    demo = await gr.Interface(fn=greet, inputs="text", outputs="text")
    await demo.launch()

def greet(name):
    return "Hello " + name + "!!"


# async def resolve_hostname(hostname, nameservers):
#     resolver = aiodns.DNSResolver(nameservers=nameservers)
#     result = await resolver.gethostbyname(hostname, socket.AF_UNSPEC)
#     return result

# # Example usage with custom DNS servers
# custom_dns_servers = ['1.1.1.1', '8.8.8.8']  # Cloudflare and Google DNS
# hostname = 'api.telegram.org'

# addresses = asyncio.run(resolve_hostname(hostname, custom_dns_servers))
# print(addresses.addresses)

if __name__ == "__main__":
    # aiodns.servers = set(["8.8.8.8", "1.1.1.1"])
    
    # loop = asyncio.get_event_loop()
    # resolver = aiodns.DNSResolver(loop=loop)
    asyncio.run(main())