import argparse
import json
import os

import pika


def main():
    parser = argparse.ArgumentParser(
        description="Send a test notification to RabbitMQ."
    )
    parser.add_argument(
        "--title", default="Test Notification", help="Notification title"
    )
    parser.add_argument(
        "--message",
        default="Hello from the producer script!",
        help="Notification message",
    )
    parser.add_argument(
        "--channels",
        default="ntfy",
        help="Comma-separated list of channels (e.g. ntfy,discord,loki)",
    )
    parser.add_argument(
        "--ntfy_topic",
        help="Override the default Ntfy topic",
    )
    parser.add_argument(
        "--mattermost_channel",
        help="Override the default Mattermost channel",
    )
    parser.add_argument(
        "--priority",
        type=str,
        help="Override the default priority for Gotify notifications",
    )
    parser.add_argument(
        "--gotify_app",
        help="Override the default Gotify app token",
    )
    parser.add_argument(
        "--pushover_device",
        help="Override the default Pushover device",
    )
    parser.add_argument(
        "--pushover_app",
        help="Specify which Pushover application to use (e.g., infra, cicd, monitoring)",
    )
    args = parser.parse_args()

    rabbitmq_host = os.environ.get("RABBITMQ_HOST", "localhost")
    rabbitmq_port = int(os.environ.get("RABBITMQ_PORT", 5672))
    rabbitmq_user = os.environ.get("RABBITMQ_USER", "guest")
    rabbitmq_pass = os.environ.get("RABBITMQ_PASS", "guest")
    rabbitmq_queue = os.environ.get("RABBITMQ_QUEUE", "homelab.alerts")

    credentials = pika.PlainCredentials(rabbitmq_user, rabbitmq_pass)

    msg = {
        "title": args.title,
        "message": args.message,
        "channels": [c.strip() for c in args.channels.split(",") if c.strip()],
    }

    if args.ntfy_topic:
        msg["ntfy_topic"] = args.ntfy_topic
    if args.mattermost_channel:
        msg["mattermost_channel"] = args.mattermost_channel
    if args.priority:
        msg["priority"] = args.priority
    if args.gotify_app:
        msg["gotify_app"] = args.gotify_app
    if args.pushover_device:
        msg["pushover_device"] = args.pushover_device
    if args.pushover_app:
        msg["pushover_app"] = args.pushover_app

    connection = pika.BlockingConnection(
        pika.ConnectionParameters(
            host=rabbitmq_host, port=rabbitmq_port, credentials=credentials
        )
    )
    channel = connection.channel()
    channel.queue_declare(queue=rabbitmq_queue, durable=True)
    channel.basic_publish(
        exchange="",
        routing_key=rabbitmq_queue,
        body=json.dumps(msg),
        properties=pika.BasicProperties(delivery_mode=2),  # make message persistent
    )
    print(f"Sent: {msg}")
    connection.close()


if __name__ == "__main__":
    main()