main.py

import asyncio
import os
import threading
import time
import pygame

import paho.mqtt.client as mqtt
from bot.telegram import TelegramBot

# Mosquitto MQTT Config
MQTT_BROKER = "localhost"
MQTT_PORT = 1883
CLIENT_ID = "raspberry_monyet"
TOPIC = "esp8266/pub"

# Try to initialize pygame.mixer with retries
def init_audio_with_retry(retries=5, delay=5):
    for i in range(retries):
        try:
            pygame.mixer.init()
            print("🔊 Audio initialized successfully")
            return True
        except pygame.error as e:
            print(f"⚠️ Audio init failed (attempt {i+1}/{retries}): {e}")
            time.sleep(delay)
    print("❌ Audio init failed after all retries")
    return False

audio_ready = init_audio_with_retry()

# Telegram Bot
bot = TelegramBot()

# Play sound one time
def play_sound_once():
    if not audio_ready:
        print("🔇 Skipping sound playback, audio device not ready")
        return

    try:
        # Check for various alarm file formats
        alarm_files = ["alarm/alarm.wav", "alarm/alarm.mp3", "alarm/alarm.ogg", "alarm/alarm.flac"]
        sound_file = "example.mp3"  # Default fallback

        for file in alarm_files:
            if os.path.exists(file):
                sound_file = file
                break

        pygame.mixer.music.load(sound_file)
        pygame.mixer.music.play()
        print("🔊 Alarm playing once")
        while pygame.mixer.music.get_busy():
            time.sleep(0.1)
        print("🔇 Alarm finished")
    except Exception as e:
        print(f"⚠️ Error playing sound: {e}")

# MQTT callbacks
def on_connect(client, userdata, flags, rc):
    if rc == 0:
        print("✅ Connected to MQTT broker!")
        client.subscribe(TOPIC)
        print(f"📡 Subscribed to topic '{TOPIC}'")
    else:
        print(f"❌ Failed to connect, return code {rc}")

def on_message(client, userdata, msg):
    message = msg.payload.decode()
    print(f"📩 Received MQTT message from '{msg.topic}': {message}")

    threading.Thread(target=play_sound_once, daemon=True).start()
    asyncio.run(bot.send_message("🐒 MQTT message received", sensor_active=True))

# MQTT Client Setup
client = mqtt.Client(CLIENT_ID)
client.on_connect = on_connect
client.on_message = on_message

# Async main
async def main():
    print(f"🔌 Connecting to Mosquitto broker at {MQTT_BROKER}:{MQTT_PORT}...")
    client.connect(MQTT_BROKER, MQTT_PORT, 60)
    client.loop_start()

    try:
        while True:
            await asyncio.sleep(1)
    except KeyboardInterrupt:
        print("🔌 Disconnecting...")
        client.loop_stop()
        client.disconnect()
        print("🚫 Disconnected.")

if __name__ == "__main__":
    #try:
     #   asyncio.run(main())
    #except KeyboardInterrupt:
        #print("🔌 Disconnecting...")
        #client.loop_stop()
        #client.disconnect()
        #print("🚫 Disconnected.")
    bot.run()