import logging
import queue
import select
import threading
import time
from dataclasses import dataclass
from datetime import datetime

import psycopg2.extensions
from django.db import connections

from mailer.engine import EMPTY_QUEUE_SLEEP, send_all
from mailer.models import Message

logger = logging.getLogger(__name__)


notify_q = queue.Queue()


CHANNEL = "django_mailer_new_message"


def postgres_send_loop():
    """
    Loop indefinitely, checking queue using NOTIFY/LISTEN and running send_mail(),
    and additional running every MAILER_EMPTY_QUEUE_SLEEP seconds.
    """
    # See https://www.psycopg.org/docs/advanced.html#asynchronous-notifications

    # Get a connection, for a few lower level operations.
    dj_conn = connections[Message.objects.db]
    if dj_conn.connection is None:
        dj_conn.connect()
    conn = dj_conn.connection  # psycopg2 connection

    # As per psycopg2 docs, we want autocommit for timely notifications
    conn.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)

    curs = conn.cursor()
    install_trigger(curs, dj_conn)
    curs.execute(f"LISTEN {CHANNEL};")
    logger.debug(f"Waiting for notifications on channel '{CHANNEL}'")

    # We have a single worker thread that runs send_all(). This means we are not
    # sending messages in parallel, which is deliberate - in many cases email
    # sending may be throttled and we are less likely to exceed quotas if we
    # send in serial rather than parallel.
    worker_thread = threading.Thread(target=worker, daemon=True).start()

    if EMPTY_QUEUE_SLEEP is not None:
        beat_thread = threading.Thread(target=beat, daemon=True).start()
    else:
        beat_thread = None

    SELECT_TIMEOUT = 5
    while True:
        if select.select([conn], [], [], SELECT_TIMEOUT) == ([], [], []):
            # timeout
            pass
        else:
            conn.poll()
            last = conn.notifies.pop()
            # We don't care about payload or how many NOTIFY there were,
            # we'll just run once.
            dropped = conn.notifies
            if dropped:
                logger.debug("Dropping notifications %r", dropped)
            conn.notifies.clear()

            if notify_q.empty():
                logger.debug("Putting %r on queue", last)
                notify_q.put(last)
            else:
                # notify_q is not empty.

                # Our worker thread always processes all Messages. If it still
                # has an item on the queue, it will process all remaining
                # messages next time it runs, so there is no point adding
                # another item to the non-empty queue - this will just cause
                # `send_all()` to run pointlessly.

                # This is quite important for efficiency - if we have a
                # transaction that inserts 100 items into the Message table,
                # once it commits the NOTIFY gets sent 100 times. Each one is
                # received individually by the `.poll()` call above. The first
                # `send_all()` command will deal with them all - we don't want
                # `send_all()` to thrash away doing nothing another 99 times
                # afterwards.
                logger.debug("Discarding item %r as work queue is not empty", last)

    # Clean up:
    worker_thread.join()
    if beat_thread is not None:
        beat_thread.join()


def install_trigger(curs, dj_conn):
    message_table_name = Message._meta.db_table
    curs.execute(
        f"""
    CREATE OR REPLACE FUNCTION django_mailer_notify_new_message()
    RETURNS TRIGGER AS $$
    BEGIN
      PERFORM pg_notify('{CHANNEL}', NEW.id::text);
      RETURN NEW;
    END;
    $$ LANGUAGE plpgsql;

        """
    )

    # "CREATE OR REPLACE TRIGGER" doesn't exist until Postgresql 14
    curs.execute(
        f"""
    DO $$
    BEGIN
      IF NOT EXISTS (SELECT 1 FROM pg_trigger WHERE tgname = 'django_mailer_message_notify_trigger') THEN
        CREATE TRIGGER django_mailer_message_notify_trigger
        AFTER INSERT ON {dj_conn.ops.quote_name(message_table_name)}
        FOR EACH ROW EXECUTE FUNCTION django_mailer_notify_new_message();
      END IF;
    END $$;
    """
    )


def worker():
    while True:
        item = notify_q.get()
        logger.debug("Working on %r", item)
        send_all()
        logger.debug("Finished %r", item)
        notify_q.task_done()


@dataclass
class Scheduled:
    now: datetime


def beat():
    while True:
        if notify_q.empty():
            scheduled = Scheduled(now=datetime.now())
            logger.debug("Putting scheduled item %r on queue", scheduled)
            notify_q.put(scheduled)
        time.sleep(EMPTY_QUEUE_SLEEP)
