# @Time    : 2023/8/14 03:59
# @Author  : Lan
# @File    : views.py
# @Software: PyCharm
from fastapi import APIRouter, Form, UploadFile, File, Depends, HTTPException
from apps.admin.depends import admin_required
from apps.base.models import FileCodes
from apps.base.pydantics import SelectFileModel
from apps.base.utils import get_expire_info, get_file_path_name, error_ip_limit, upload_ip_limit
from core.response import APIResponse
from core.settings import settings
from core.storage import file_storage
from core.utils import get_select_token

share_api = APIRouter(
    prefix='/share',
    tags=['分享'],
)


@share_api.post('/text/', dependencies=[Depends(admin_required)])
async def share_text(text: str = Form(...), expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), ip: str = Depends(upload_ip_limit)):
    expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
    await FileCodes.create(
        code=code,
        text=text,
        expired_at=expired_at,
        expired_count=expired_count,
        used_count=used_count,
        size=len(text),
        prefix='文本分享'
    )
    upload_ip_limit.add_ip(ip)
    return APIResponse(detail={
        'code': code,
    })


@share_api.post('/file/', dependencies=[Depends(admin_required)])
async def share_file(expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), file: UploadFile = File(...), ip: str = Depends(upload_ip_limit)):
    if file.size > settings.uploadSize:
        raise HTTPException(status_code=403, detail=f'文件大小超过限制，最大为{settings.uploadSize}字节')
    expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
    path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(file)
    await file_storage.save_file(file, save_path)
    await FileCodes.create(
        code=code,
        prefix=prefix,
        suffix=suffix,
        uuid_file_name=uuid_file_name,
        file_path=path,
        size=file.size,
        expired_at=expired_at,
        expired_count=expired_count,
        used_count=used_count,
    )
    upload_ip_limit.add_ip(ip)
    return APIResponse(detail={
        'code': code,
        'name': file.filename,
    })


async def get_code_file_by_code(code):
    file_code = await FileCodes.filter(code=code).first()
    if not file_code:
        return False, '文件不存在'
    if await file_code.is_expired():
        return False, '文件已过期',
    return True, file_code


@share_api.get('/select/')
async def get_code_file(code: str, ip: str = Depends(error_ip_limit)):
    has, file_code = await get_code_file_by_code(code)
    if not has:
        error_ip_limit.add_ip(ip)
        return APIResponse(code=404, detail=file_code)
    file_code.used_count += 1
    file_code.expired_count -= 1
    await file_code.save()
    return await file_storage.get_file_response(file_code)


@share_api.post('/select/')
async def select_file(data: SelectFileModel, ip: str = Depends(error_ip_limit)):
    has, file_code = await get_code_file_by_code(data.code)
    if not has:
        error_ip_limit.add_ip(ip)
        return APIResponse(code=404, detail=file_code)
    file_code.used_count += 1
    file_code.expired_count -= 1
    await file_code.save()
    return APIResponse(detail={
        'code': file_code.code,
        'name': file_code.prefix + file_code.suffix,
        'size': file_code.size,
        'text': file_code.text if file_code.text is not None else await file_storage.get_file_url(file_code),
    })


@share_api.get('/download')
async def download_file(key: str, code: str, ip: str = Depends(error_ip_limit)):
    is_valid = await get_select_token(code) == key
    if not is_valid:
        error_ip_limit.add_ip(ip)
    has, file_code = await get_code_file_by_code(code)
    if not file_code:
        return APIResponse(code=404, detail='文件不存在')
    if file_code.text:
        return APIResponse(detail=file_code.text)
    else:
        return await file_storage.get_file_response(file_code)