#include <WiFi.h>
#include <WiFiClientSecure.h>
#include <UniversalTelegramBot.h>
#include <ArduinoJson.h>
#include <string.h> 
#include <HTTPClient.h>


//@TradutorMorseBot

// --- Configurações WiFi ---
const char* ssid = "Wokwi-GUEST";
const char* password = "";

// --- Configurações Telegram ---
#define BOT_TOKEN "8415451527:AAH_PdB19DBFTxVX7Bb-AalmIp-jz1CxR3s"   

WiFiClientSecure client;
UniversalTelegramBot bot(BOT_TOKEN, client);

unsigned long lastTimeCheck = 0;
const unsigned long checkInterval = 2000; 

// --- Configurações de Hardware ---
#define BOTAO  14
#define BUZZER 15
#define LED    2

// --- Configurações Morse ---
#define MAX_CARACTERES 50
#define TEMPO_SEGURO 50  
#define UNIDADE_TEMPO 500 

// --- Variáveis de Estado do Botão ---
unsigned long tempoPressionado = 0;
unsigned long tempoSolto = 0;

bool estadoAnterior = HIGH;
bool botaoPressionado = false;
bool pausaProcessada = false;

int i = 0;
char letras_aleatorias[MAX_CARACTERES + 1];

const char *letras = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
const char *morse[] = {
    ".-", "-...", "-.-.", "-..", ".", "..-.", "--.", "....", "..", ".---",
    "-.-", ".-..", "--", "-.", "---", ".--.", "--.-", ".-.", "...", "-",
    "..-", "...-", ".--", "-..-", "-.--", "--..",
    "-----", ".----", "..---", "...--", "....-", ".....",
    "-....", "--...", "---..", "----."
};

void connectWiFi();
void sendMenu(String chat_id, String nome);
void pegarMensagemTelegram(int numNewMessages);
void pegarMensagemMorse();
void texto_para_morse(const char *texto); 
String morse_para_texto(const char *codigo); 

void sendMenu(String chat_id, String nome) {
  String keyboardJson = F(
    "{\"keyboard\":["
    " [{\"text\":\"1️⃣ Bom dia\"},{\"text\":\"2️⃣ SOS\"}],"
    " [{\"text\":\"3️⃣ Estou perdido\"}]"
    "],\"resize_keyboard\":true}"
  );

  String mensagem =
    "👋 *Bem-vindo, " + nome + "!*\n\n"
    "👉 Para *traduzir de Morse → Texto*, digite o código no protótipo.\n"
    "👉 Para *traduzir de Texto → Morse*, escolha uma das frases abaixo ou digite uma nova.\n\n"
    "📋 *Escolha uma das opções:*\n"
    "1️⃣ Bom dia\n"
    "2️⃣ SOS\n"
    "3️⃣ Estou perdido\n\n"
    "Toque no botão abaixo";

  Serial.println("📤 Enviando menu...");

  String url = "https://api.telegram.org/bot" + String(BOT_TOKEN) + "/sendMessage";

  String payload = 
    "chat_id=" + chat_id + 
    "&text=" + mensagem + 
    "&parse_mode=Markdown" +
    "&reply_markup=" + keyboardJson;

  HTTPClient https;
  https.begin(url);
  https.addHeader("Content-Type", "application/x-www-form-urlencoded");

  int httpCode = https.POST(payload);
  String response = https.getString();

  if (httpCode == 200)
    Serial.println("Menu enviado com sucesso!");
  else {
    Serial.print("Erro ao enviar menu. Código: ");
    Serial.println(httpCode);
    Serial.println(response);
  }

  https.end();
}

void setup() {
  Serial.begin(115200);
  connectWiFi();
  WiFi.mode(WIFI_STA);


  pinMode(BOTAO, INPUT_PULLUP);
  pinMode(BUZZER, OUTPUT);
  pinMode(LED, OUTPUT);
  letras_aleatorias[0] = '\0';
}

void loop() {
  pegarMensagemMorse();

  if (millis() - lastTimeCheck > checkInterval) {
    int numNewMessages = bot.getUpdates(bot.last_message_received + 1);

    if (numNewMessages) {
      pegarMensagemTelegram(numNewMessages); 
    }
    lastTimeCheck = millis();
  }
}

void pegarMensagemTelegram(int numNewMessages){ 
  Serial.printf("📩 Novas mensagens: %d\n", numNewMessages);

      for (int i = 0; i < numNewMessages; i++) {
        String chat_id = bot.messages[i].chat_id.c_str();
        String msgText = bot.messages[i].text;
        String fromName = bot.messages[i].from_name;
        if (fromName == "") fromName = "usuário";

        Serial.printf("Mensagem de %s: %s\n", fromName.c_str(), msgText);

        if (msgText == "/start" || msgText == "Iniciar" || msgText == "oi" || msgText == "olá") {
          sendMenu(chat_id, fromName);
        }
        else if (msgText.startsWith("1")) {
          texto_para_morse("Bom dia"); 
        }
        else if (msgText.startsWith("2")) {
          texto_para_morse("SOS"); 
        }
        else if (msgText.startsWith("3")) {
          texto_para_morse("Estou perdida"); 
        }
        else {
          texto_para_morse(msgText.c_str()); 
        }
      }
}

void pegarMensagemMorse(){
  bool estadoAtual = digitalRead(BOTAO);

  if (estadoAnterior == HIGH && estadoAtual == LOW) {
    tempoPressionado = millis();
    botaoPressionado = true;
    pausaProcessada = false;

    tone(BUZZER, 600);
    digitalWrite(LED, HIGH);
  }

  if (botaoPressionado && estadoAtual == LOW) {
  }

  if (estadoAnterior == LOW && estadoAtual == HIGH && botaoPressionado) {
    unsigned long duracao = millis() - tempoPressionado;
    noTone(BUZZER);
    digitalWrite(LED, LOW);

    if (i < MAX_CARACTERES) {
      if (duracao > 2 && duracao <= TEMPO_SEGURO) {
        letras_aleatorias[i] = '.';
        i++;
      } else if (duracao > TEMPO_SEGURO)  {
        letras_aleatorias[i] = '-';
        i++;
      }
      letras_aleatorias[i] = '\0';
    }

    botaoPressionado = false;
    tempoSolto = millis();
  }

  
  if (!botaoPressionado && millis() - tempoSolto > 4000 && !pausaProcessada) {
    String mensagem_traduzida = morse_para_texto(letras_aleatorias);
    
    // O chat_id não está disponível globalmente, é necessário obter o último chat_id conhecido
    // ou assumir que o usuário só interage com o bot após um /start.
    // Para simplificar, vamos assumir que o chat_id é o do último usuário que interagiu.
    // **NOTA: Isso é uma simplificação e pode falhar em um ambiente multi-usuário.**
    // Uma solução robusta exigiria armazenar o chat_id em uma variável global ou EEPROM.
    // Como o código original não tinha essa lógica, vou deixar um placeholder e uma nota.
    String last_chat_id = "SEU_ULTIMO_CHAT_ID_CONHECIDO"; 
    
    if (mensagem_traduzida.length() > 0) {
        bot.sendMessage(last_chat_id, "Tradução Morse: " + mensagem_traduzida);
    }

    // Resetar o buffer
    i = 0;
    letras_aleatorias[0] = '\0';
    pausaProcessada = true; 
  }

  estadoAnterior = estadoAtual;
}

// --- Converte texto para Morse ---
void texto_para_morse(const char *texto) { 
    for (int i = 0; texto[i] != '\0'; i++) {
    char c = toupper(texto[i]);

    if (c == ' ') {
      delay(UNIDADE_TEMPO * 7);  
      continue;
    }

    const char *ptr = strchr(letras, c);
    if (ptr) {
      int idx = ptr - letras;
      const char *codigo = morse[idx];

      for (int j = 0; codigo[j] != '\0'; j++) {
        if (codigo[j] == '.') { 
          tone(BUZZER, 300);
          digitalWrite(LED, HIGH);
          delay(UNIDADE_TEMPO); 
          noTone(BUZZER);
          digitalWrite(LED, LOW);
          delay(UNIDADE_TEMPO); 
        }
        else if (codigo[j] == '-') {
          tone(BUZZER, 600);
          digitalWrite(LED, HIGH);
          delay(UNIDADE_TEMPO * 3);
          noTone(BUZZER);
          digitalWrite(LED, LOW);
          delay(UNIDADE_TEMPO); 
        }
      }

      delay(UNIDADE_TEMPO * 3);  
    }
  }
}

// --- Converte Morse para texto ---
String morse_para_texto(const char *codigo) { 
    String saida = ""; 
    char buffer[10];
    int pos = 0;

    for (int i = 0; ; i++) {
        char c = codigo[i];
        if (c != ' ' && c != '\0') {
            buffer[pos++] = c;
        } else {
            buffer[pos] = '\0';
            if (pos > 0) {
                for (int j = 0; j < strlen(letras); j++) {
                    if (strcmp(buffer, morse[j]) == 0) {
                        saida += letras[j]; 
                        break;
                    }
                }
                pos = 0;
            }

            
            if (c == ' ') {
                if (codigo[i+1] == ' ') {
                    saida += " "; 
                    i++; 
                }
            }

            if (c == '\0') break;
        }
    }
    return saida;
}

void connectWiFi() {
  Serial.print("Conectando ao WiFi ");
  Serial.print(ssid);
  Serial.println(" ...");

  WiFi.mode(WIFI_STA);
  WiFi.begin(ssid, password);
  client.setInsecure(); 

  // unsigned long start = millis();
  // while (WiFi.status() != WL_CONNECTED && millis() - start < 20000) {
  while (WiFi.status() != WL_CONNECTED) {
    Serial.print(".");
    delay(500);
  }
  Serial.println();
  if (WiFi.status() == WL_CONNECTED) {
    Serial.print("Conectado! IP: ");
    Serial.println(WiFi.localIP());
  } else {
    Serial.println("Falha ao conectar WiFi");
  }
}