from typing import Any, Optional

import orjson
from django.http import HttpResponse

from owslib.ext.django.encoders import DjangoJSONEncoder


class ORJSONResponse(HttpResponse):
    def __init__(self, data: Any, option: Optional[int] = None, **kwargs: Any) -> None:
        kwargs.setdefault("content_type", "application/json")
        content = orjson.dumps(
            data, default=lambda obj: DjangoJSONEncoder().default(obj), option=option
        )
        super().__init__(content=content, **kwargs)