from telegram import Update
from telegram.ext import Updater, CommandHandler, CallbackContext, MessageHandler, Filters
import requests
import os

updater = Updater(os.environ['BOT_TOKEN'])


@bot.message_handler(commands=['start', 'hello'])
def send_welcome(message):
    bot.reply_to(message, "Posso gerar números aleatórios verdadeiros")
    
@bot.message_handler(commands=['aleatorio'])

def sign_handler(message):
    text = "Por favor escolha um tipo de dado:\nuint8 (Inteiros 0–255), uint16 (Inteiros 0–65535)"
    bot.reply_to(message, text)

    # Agora, vamos esperar a resposta do usuário
    bot.register_next_step_handler(message, process_data_type_choice)

def process_data_type_choice(message):
    # A resposta do usuário estará em message.text
    data_type = message.text

    # Pergunte ao usuário sobre o comprimento da matriz
    bot.reply_to(message, "Por favor, escolha o comprimento da matriz. O valor deve variar de 1 a 1024.")

    # Agora, vamos esperar a resposta do usuário sobre o comprimento da matriz
    bot.register_next_step_handler(message, lambda msg: process_array_length_choice(msg, data_type))

def process_array_length_choice(message, data_type):
    # A resposta do usuário sobre o comprimento da matriz estará em message.text
    array_length = message.text

    # Faça a solicitação à API QRNG@ANU
    api_url = f"https://qrng.anu.edu.au/API/jsonI.php?length={array_length}&type={data_type}"

    # Tente obter os dados da API
    try:
        response = requests.get(api_url)
        response.raise_for_status()  # Verifica se a solicitação foi bem-sucedida

        # Converte a resposta para JSON
        result = response.json()

        # Obtém a matriz de dados da resposta
        random_data = result.get('data', [])

        # Informa ao usuário
        bot.reply_to(message, f"Números aleatórios verdadeiros: {random_data}")

    except requests.exceptions.RequestException as e:
        # Se ocorrer um erro na solicitação à API, informe ao usuário
        bot.reply_to(message, f"Aguarde 1 min depois da ultima solicitação")

# Adicione a seguinte linha para iniciar o polling
bot.infinity_polling()