diff --git a/apps/base/utils.py b/apps/base/utils.py index 72155cb67..1c6d59fa8 100644 --- a/apps/base/utils.py +++ b/apps/base/utils.py @@ -5,12 +5,12 @@ import datetime import uuid import os -from fastapi import UploadFile +from fastapi import UploadFile, HTTPException from apps.base.depends import IPRateLimit from apps.base.models import FileCodes from core.settings import settings -from core.utils import get_random_num, get_random_string +from core.utils import get_random_num, get_random_string, max_save_times_desc async def get_file_path_name(file: UploadFile): @@ -43,11 +43,24 @@ async def get_expire_info(expire_value: int, expire_style: str): :return: expired_at 过期时间, expired_count 可用次数, used_count 已用次数, code 随机码 """ expired_count, used_count, now, code = -1, 0, datetime.datetime.now(), None + if settings.max_save_seconds > 0: + max_timedelta = datetime.timedelta(seconds=settings.max_save_seconds) + detail = await max_save_times_desc(settings.max_save_seconds) + detail = f'保存时间超过限制,{detail[0]}' + else: + max_timedelta = datetime.timedelta(days=7) + detail = '保存时间超过限制,最长保存时间:7天' if expire_style == 'day': + if datetime.timedelta(days=expire_value) > max_timedelta: + raise HTTPException(status_code=403, detail=detail) expired_at = now + datetime.timedelta(days=expire_value) elif expire_style == 'hour': + if datetime.timedelta(hours=expire_value) > max_timedelta: + raise HTTPException(status_code=403, detail=detail) expired_at = now + datetime.timedelta(hours=expire_value) elif expire_style == 'minute': + if datetime.timedelta(minutes=expire_value) > max_timedelta: + raise HTTPException(status_code=403, detail=detail) expired_at = now + datetime.timedelta(minutes=expire_value) elif expire_style == 'count': expired_at = now + datetime.timedelta(days=1) diff --git a/core/settings.py b/core/settings.py index 3b24f26b3..17937293c 100644 --- a/core/settings.py +++ b/core/settings.py @@ -13,6 +13,7 @@ 'name': '文件快递柜-FileCodeBox', 'description': '开箱即用的文件快传系统', 'keywords': 'FileCodeBox,文件快递柜,口令传送箱,匿名口令分享文本,文件', + 'max_save_seconds': 0, 's3_access_key_id': '', 's3_secret_access_key': '', 's3_bucket_name': '', diff --git a/core/utils.py b/core/utils.py index cfaeea25d..bbcbc3f21 100644 --- a/core/utils.py +++ b/core/utils.py @@ -57,3 +57,35 @@ async def get_file_url(code: str): :return: """ return f'/share/download?key={await get_select_token(code)}&code={code}' + +async def max_save_times_desc(max_save_seconds: int): + """ + 获取最大保存时间的描述 + :param max_save_seconds: + :return: + """ + def gen_desc_zh(value: int, desc: str): + if value > 0: + return f'{value}{desc}' + else: + return '' + def gen_desc_en(value: int, desc: str): + if value > 0: + ret = f'{value} {desc}' + if value > 1: + ret += 's' + ret += ' ' + return ret + else: + return '' + max_timedelta = datetime.timedelta(seconds=max_save_seconds) + desc_zh, desc_en = '最长保存时间:', 'Max save time: ' + desc_zh += gen_desc_zh(max_timedelta.days, '天') + desc_en += gen_desc_en(max_timedelta.days, 'day') + desc_zh += gen_desc_zh(max_timedelta.seconds // 3600, '小时') + desc_en += gen_desc_en(max_timedelta.seconds // 3600, 'hour') + desc_zh += gen_desc_zh(max_timedelta.seconds % 3600 // 60, '分钟') + desc_en += gen_desc_en(max_timedelta.seconds % 3600 // 60, 'minute') + desc_zh += gen_desc_zh(max_timedelta.seconds % 60, '秒') + desc_en += gen_desc_en(max_timedelta.seconds % 60, 'second') + return desc_zh, desc_en diff --git a/main.py b/main.py index 4a2f1527c..6ab00845e 100644 --- a/main.py +++ b/main.py @@ -4,10 +4,11 @@ # @Software: PyCharm import asyncio import os +import re from fastapi import FastAPI from starlette.middleware.cors import CORSMiddleware -from starlette.responses import HTMLResponse +from starlette.responses import HTMLResponse, FileResponse from starlette.staticfiles import StaticFiles from tortoise.contrib.fastapi import register_tortoise @@ -15,6 +16,7 @@ from apps.admin.views import admin_api from core.settings import data_root, settings from core.tasks import delete_expire_files +from core.utils import max_save_times_desc app = FastAPI() @@ -26,7 +28,24 @@ allow_headers=["*"], ) -app.mount('/assets', StaticFiles(directory='./fcb-fronted/dist/assets'), name="assets") +@app.get('/assets/{file_path:path}') +async def assets(file_path: str): + if settings.max_save_seconds > 0: + if re.match(r'SendView-[\d|a-f|A-F]+\.js', file_path): + with open(f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: + # 删除永久保存选项 + content = f.read() + content = content.replace('_(c,{label:e(r)("send.expireData.forever"),value:"forever"},null,8,["label"]),', '') + return HTMLResponse(content=content, media_type='text/javascript') + if re.match(r'index-[\d|a-f|A-F]+\.js', file_path): + with open(f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: + # 更改本文描述 + desc_zh, desc_en = await max_save_times_desc(settings.max_save_seconds) + content = f.read() + content = content.replace('天数<7', desc_zh) + content = content.replace('Days <7', desc_en) + return HTMLResponse(content=content, media_type='text/javascript') + return FileResponse(f'./fcb-fronted/dist/assets/{file_path}') register_tortoise( app,