# -*- coding: utf-8 -*-

import logging
import os
from typing import Optional

import urllib3
from fastapi import APIRouter, Request, Query, BackgroundTasks
from fastapi.responses import Response

from favicon_app.routes import favicon_service
from favicon_app.utils.file_util import FileUtil

urllib3.disable_warnings()
logging.captureWarnings(True)
logger = logging.getLogger(__name__)

_icon_root_path = favicon_service.icon_root_path
_default_icon_path = favicon_service.default_icon_path
_default_icon_content = favicon_service.default_icon_content

# 创建全局服务实例
_service = favicon_service.FaviconService()

# 创建FastAPI路由器
favicon_router = APIRouter(prefix="", tags=["favicon"])


@favicon_router.get('/icon/')
@favicon_router.get('/icon')
async def get_favicon(
        request: Request,
        bg_tasks: BackgroundTasks,
        url: Optional[str] = Query(None, description="网址：eg. https://www.baidu.com"),
        refresh: Optional[str] = Query(None, include_in_schema=False),
        sync: Optional[str] = Query('false', description="是否使用同步方式获取")
):
    """获取网站图标"""
    return await _service.get_favicon_handler(request, bg_tasks, url, refresh, sync)


@favicon_router.get('/icon/default')
async def get_default_icon(cache_time: int = Query(_service.time_of_1_days, include_in_schema=False)):
    """获取默认图标"""
    return Response(content=_default_icon_content,
                    media_type="image/png",
                    headers=_service.get_header("image/png", cache_time))


@favicon_router.get('/icon/count')
async def get_count():
    """获取统计数据"""
    return _service.get_count()


@favicon_router.get('/icon/referer', include_in_schema=False)
async def get_referrer(unique: Optional[str] = Query(None)):
    """获取请求来源信息，带unique参数时会进行去重处理"""
    content = 'None'
    path = os.path.join(_icon_root_path, 'data', 'referer.txt')

    if os.path.exists(path):
        try:
            content = FileUtil.read_file(path, mode='r') or 'None'

            if unique in ['true', '1', 'True']:
                lines = [line.strip() for line in content.split('\n') if line.strip()]
                unique_lines = list(set(lines))
                unique_content = '\n'.join(unique_lines)
                FileUtil.write_file(path, unique_content, mode='w')
                content = unique_content
        except Exception as e:
            logger.error(f"读取referer文件失败: {e}")
    return Response(content=content, media_type="text/plain")