main.py

import os
import sys
from dotenv import load_dotenv

# Load .env BEFORE importing modules that use env vars
load_dotenv()

from src.runners import SyncRunner, StreamRunner, WhatsAppTerminalRunner, WhatsAppAPIRunner

RUNNERS = {
    "sync": SyncRunner,
    "stream": StreamRunner,
    "whatsapp": WhatsAppTerminalRunner,
    "api": WhatsAppAPIRunner,
}


def print_help():
    print("Uso: python main.py [opcoes]")
    print()
    print("Opcoes:")
    print("  (sem args)      Modo normal (sync)")
    print("  -s, --stream    Modo streaming (token por token)")
    print("  -w, --whatsapp  Modo WhatsApp terminal (simula digitando...)")
    print("  -a, --api       Modo WhatsApp API (webhook)")
    print("  -p, --port N    Porta do servidor (default: 8000)")
    print("  -h, --help      Mostra esta ajuda")


def get_port() -> int:
    """Get port from args or env."""
    for i, arg in enumerate(sys.argv):
        if arg in ["--port", "-p"] and i + 1 < len(sys.argv):
            return int(sys.argv[i + 1])
    return int(os.getenv("PORT", "8000"))


def get_runner_mode() -> str:
    if "--help" in sys.argv or "-h" in sys.argv:
        print_help()
        sys.exit(0)

    if "--api" in sys.argv or "-a" in sys.argv:
        return "api"

    if "--whatsapp" in sys.argv or "-w" in sys.argv:
        return "whatsapp"

    if "--stream" in sys.argv or "-s" in sys.argv:
        return "stream"

    return "sync"


def main():
    if not os.getenv("OPENAI_API_KEY"):
        print("Erro: OPENAI_API_KEY nao configurada!")
        print("   Crie um arquivo .env com sua API key ou exporte a variavel de ambiente.")
        print("   Exemplo: export OPENAI_API_KEY='sua-key-aqui'")
        sys.exit(1)

    mode = get_runner_mode()
    port = get_port()

    runner_class = RUNNERS[mode]
    runner = runner_class()

    # API mode accepts port argument
    if mode == "api":
        runner.run(port=port)
    else:
        runner.run()


if __name__ == "__main__":
    main()