filecodebox/apps/admin/views.py

# @Time    : 2023/8/14 14:38
# @Author  : Lan
# @File    : views.py
# @Software: PyCharm
import math
import os
import time

from fastapi import APIRouter, Depends, Form
from pydantic import BaseModel

from apps.admin.depends import admin_required
from apps.admin.pydantics import IDData
from apps.base.models import FileCodes, KeyValue
from apps.base.utils import get_expire_info, get_file_path_name
from core.response import APIResponse
from core.settings import settings, data_root
from core.storage import FileStorageInterface, storages

admin_api = APIRouter(
    prefix='/admin',
    tags=['管理'],
)


@admin_api.post('/login', dependencies=[Depends(admin_required)])
async def login():
    return APIResponse()


@admin_api.delete('/file/delete', dependencies=[Depends(admin_required)])
async def file_delete(data: IDData):
    file_storage: FileStorageInterface = storages[settings.file_storage]()
    file_code = await FileCodes.get(id=data.id)
    await file_storage.delete_file(file_code)
    await file_code.delete()
    return APIResponse()


@admin_api.get('/file/list', dependencies=[Depends(admin_required)])
async def file_list(page: float = 1, size: int = 10):
    return APIResponse(detail={
        'page': page,
        'size': size,
        'data': await FileCodes.all().limit(size).offset((math.ceil(page) - 1) * size),
        'total': await FileCodes.all().count(),
    })


@admin_api.get('/config/get', dependencies=[Depends(admin_required)])
async def get_config():
    return APIResponse(detail=settings.items())


@admin_api.patch('/config/update', dependencies=[Depends(admin_required)])
async def update_config(data: dict):
    admin_token = data.get('admin_token')
    for key, value in data.items():
        if key not in settings.default_config:
            continue
        if key in ['errorCount', 'errorMinute', 'max_save_seconds', 'onedrive_proxy', 'openUpload', 'port', 's3_proxy', 'uploadCount', 'uploadMinute', 'uploadSize']:
            data[key] = int(value)
        elif key in ['opacity']:
            data[key] = float(value)
        else:
            data[key] = value
    if admin_token is None or admin_token == '':
        return APIResponse(code=400, detail='管理员密码不能为空')
    await KeyValue.filter(key='settings').update(value=data)
    for k, v in data.items():
        settings.__setattr__(k, v)
    return APIResponse()


# 根据code获取文件
async def get_file_by_id(id):
    # 查询文件
    file_code = await FileCodes.filter(id=id).first()
    # 检查文件是否存在
    if not file_code:
        return False, '文件不存在'
    return True, file_code


@admin_api.get('/file/download', dependencies=[Depends(admin_required)])
async def file_download(id: int):
    file_storage: FileStorageInterface = storages[settings.file_storage]()
    has, file_code = await get_file_by_id(id)
    # 检查文件是否存在
    if not has:
        # 返回API响应
        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)


class LocalFileClass:
    def __init__(self, file):
        self.file = file
        self.path = data_root / 'local' / file
        self.ctime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(os.path.getctime(self.path)))
        self.size = os.path.getsize(self.path)

    async def read(self):
        return open(self.path, 'rb')

    async def write(self, data):
        with open(self.path, 'w') as f:
            f.write(data)

    async def delete(self):
        os.remove(self.path)

    async def exists(self):
        return os.path.exists(self.path)


@admin_api.get('/local/lists', dependencies=[Depends(admin_required)])
async def get_local_lists():
    files = []
    for file in os.listdir(data_root / 'local'):
        files.append(LocalFileClass(file))
    return APIResponse(detail=files)


class DeleteItem(BaseModel):
    filename: str


@admin_api.delete('/local/delete', dependencies=[Depends(admin_required)])
async def delete_local_file(item: DeleteItem):
    file = LocalFileClass(item.filename)
    if await file.exists():
        await file.delete()
        return APIResponse(detail='删除成功')
    return APIResponse(code=404, detail='文件不存在')


class ShareItem(BaseModel):
    expire_value: int
    expire_style: str = 'day'
    filename: str


class File:
    def __init__(self, file):
        self.file = file


@admin_api.post('/local/share', dependencies=[Depends(admin_required)])
async def share_local_file(item: ShareItem):
    file = LocalFileClass(item.filename)
    if not await file.exists():
        return APIResponse(code=404, detail='文件不存在')
    text = File(await file.read())
    expired_at, expired_count, used_count, code = await get_expire_info(item.expire_value, item.expire_style)
    # 获取文件路径和名称
    path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(item)
    # 保存文件
    file_storage: FileStorageInterface = storages[settings.file_storage]()
    await file_storage.save_file(text, save_path)
    # 创建一个新的FileCodes实例
    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,
    )
    # 返回API响应
    return APIResponse(detail={
        'code': code,
        'name': file.file,
    })