import json
import os
import django

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "timetracker.settings")
django.setup()

from django.contrib.messages import constants as message_constants
from django.contrib.messages.storage.fallback import FallbackStorage
from django.http import HttpRequest, HttpResponse
from django.test import TestCase, override_settings

from games.htmx_middleware import HTMXMessagesMiddleware


def get_response_ok(request):
    return HttpResponse("OK")


class HtmxDetails:
    boosted = False
    current_url = ""
    target_id = ""


class HTMXMessagesMiddlewareTest(TestCase):
    def _build_request(self, htmx=True, message_level=None):
        """Build a request with FallbackStorage message backend."""
        request = HttpRequest()
        request.method = "GET"
        request.path = "/test"
        request.META = {"SERVER_NAME": "localhost", "SERVER_PORT": "80"}
        request.session = {}

        storage = FallbackStorage(request)
        if message_level is not None:
            storage._set_level(message_level)
        request._messages = storage

        if htmx:
            request.htmx = HtmxDetails()

        return request

    def test_htmx_request_with_messages_sends_hx_trigger(self):
        """HTMX request with messages should include HX-Trigger header."""
        request = self._build_request(htmx=True)
        request._messages.add(message_constants.SUCCESS, "Item saved")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        self.assertIn("HX-Trigger", response)
        data = json.loads(response["HX-Trigger"])
        self.assertIn("show-toast", data)
        self.assertEqual(data["show-toast"]["message"], "Item saved")
        self.assertEqual(data["show-toast"]["type"], "success")

    def test_htmx_request_with_error_message(self):
        """Error messages should map to 'error' toast type."""
        request = self._build_request(htmx=True)
        request._messages.add(message_constants.ERROR, "Something failed")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        data = json.loads(response["HX-Trigger"])
        self.assertEqual(data["show-toast"]["type"], "error")

    def test_htmx_request_with_success_message(self):
        """Success messages should map to 'success' toast type."""
        request = self._build_request(htmx=True)
        request._messages.add(message_constants.SUCCESS, "Saved successfully")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        data = json.loads(response["HX-Trigger"])
        self.assertEqual(data["show-toast"]["type"], "success")

    def test_non_htmx_request_also_sends_hx_trigger(self):
        """Non-HTMX requests should also include HX-Trigger header."""
        request = self._build_request(htmx=False)
        request._messages.add(message_constants.SUCCESS, "Hello")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        self.assertIn("HX-Trigger", response)
        data = json.loads(response["HX-Trigger"])
        self.assertIn("show-toast", data)
        self.assertEqual(data["show-toast"]["message"], "Hello")

    def test_htmx_request_without_messages_no_hx_trigger(self):
        """HTMX request without messages should not include HX-Trigger header."""
        request = self._build_request(htmx=True)
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        self.assertNotIn("HX-Trigger", response)

    def test_warning_message_maps_to_warning(self):
        """Warning messages should map to 'warning' toast type."""
        request = self._build_request(htmx=True)
        request._messages.add(message_constants.WARNING, "Warning message")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        data = json.loads(response["HX-Trigger"])
        self.assertEqual(data["show-toast"]["type"], "warning")

    @override_settings(DEBUG=True)
    def test_debug_message_maps_to_debug(self):
        """Debug messages should map to 'debug' toast type."""
        request = self._build_request(htmx=True, message_level=message_constants.DEBUG)
        request._messages.add(message_constants.DEBUG, "Debug info")
        middleware = HTMXMessagesMiddleware(get_response_ok)

        response = middleware(request)

        data = json.loads(response["HX-Trigger"])
        self.assertEqual(data["show-toast"]["type"], "debug")