import logging
import time

from ipware import get_client_ip
from user_agents import parse

from blog.documents import ELASTICSEARCH_ENABLED, ElaspedTimeDocumentManager

logger = logging.getLogger(__name__)


class OnlineMiddleware(object):
    """
    mk:
    在线用户中间件类，用于记录页面加载时间和用户访问信息

    该中间件会在每个请求处理前后记录时间，计算页面渲染耗时，
    并将相关信息存储到Elasticsearch中（如果启用的话）。

    Args:
        get_response: Django中间件的get_response回调函数
    """
    def __init__(self, get_response=None):
        self.get_response = get_response
        super().__init__()

    def __call__(self, request):
        """
        mk:
        中间件调用方法，处理请求并记录页面加载时间

        记录请求开始时间，处理请求，计算耗时，并将相关信息存储到Elasticsearch。
        同时替换响应内容中的加载时间占位符。

        Args:
            request: Django HttpRequest对象，包含当前请求信息

        Returns:
            HttpResponse: 处理后的HTTP响应对象
        """
        ''' page render time '''
        start_time = time.time()
        response = self.get_response(request)
        http_user_agent = request.META.get('HTTP_USER_AGENT', '')
        ip, _ = get_client_ip(request)
        user_agent = parse(http_user_agent)
        if not response.streaming:
            try:
                # mk:计算页面渲染耗时
                cast_time = time.time() - start_time
                if ELASTICSEARCH_ENABLED:
                    # mk:将耗时转换为毫秒并四舍五入到小数点后两位
                    time_taken = round((cast_time) * 1000, 2)
                    url = request.path
                    from django.utils import timezone
                    # mk:创建Elasticsearch文档记录
                    ElaspedTimeDocumentManager.create(
                        url=url,
                        time_taken=time_taken,
                        log_datetime=timezone.now(),
                        useragent=user_agent,
                        ip=ip)
                # mk:替换响应内容中的加载时间占位符
                response.content = response.content.replace(
                    b'<!!LOAD_TIMES!!>', str.encode(str(cast_time)[:5]))
            except Exception as e:
                logger.error("Error OnlineMiddleware: %s" % e)

        return response