mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-05-07 03:22:40 +08:00
6b2ada0b42
- Updated import statements across multiple modules to use relative imports for better encapsulation. - Consolidated and organized the `__init__.py` files in various domain packages to expose necessary classes and functions. - Improved code readability and maintainability by grouping related imports and removing unused ones. - Ensured consistent import patterns across the domain, enhancing the overall structure of the codebase.
129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
from typing import Annotated, List, Optional
|
|
|
|
from fastapi import APIRouter, Depends, Request
|
|
|
|
from api.response import success
|
|
from domain.audit import AuditAction, audit
|
|
from domain.auth import User, get_current_active_user
|
|
from .service import ShareService
|
|
from .types import (
|
|
ShareCreate,
|
|
ShareInfo,
|
|
ShareInfoWithPassword,
|
|
SharePassword,
|
|
)
|
|
from models.database import UserAccount
|
|
|
|
public_router = APIRouter(prefix="/api/s", tags=["Share - Public"])
|
|
router = APIRouter(prefix="/api/shares", tags=["Share - Management"])
|
|
|
|
|
|
@router.post("", response_model=ShareInfoWithPassword)
|
|
@audit(
|
|
action=AuditAction.SHARE,
|
|
description="创建分享链接",
|
|
body_fields=["name", "paths", "expires_in_days", "access_type"],
|
|
)
|
|
async def create_share(
|
|
request: Request,
|
|
payload: ShareCreate,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
user_account = await UserAccount.get(id=current_user.id)
|
|
share = await ShareService.create_share_link(
|
|
user=user_account,
|
|
name=payload.name,
|
|
paths=payload.paths,
|
|
expires_in_days=payload.expires_in_days,
|
|
access_type=payload.access_type,
|
|
password=payload.password,
|
|
)
|
|
share_info = ShareInfo.from_orm(share).model_dump()
|
|
if payload.access_type == "password" and payload.password:
|
|
share_info["password"] = payload.password
|
|
return share_info
|
|
|
|
|
|
@router.get("", response_model=List[ShareInfo])
|
|
@audit(action=AuditAction.READ, description="获取我的分享列表")
|
|
async def get_my_shares(
|
|
request: Request, current_user: Annotated[User, Depends(get_current_active_user)]
|
|
):
|
|
user_account = await UserAccount.get(id=current_user.id)
|
|
shares = await ShareService.get_user_shares(user=user_account)
|
|
return [ShareInfo.from_orm(s) for s in shares]
|
|
|
|
|
|
@router.delete("/expired")
|
|
@audit(action=AuditAction.DELETE, description="删除已过期分享")
|
|
async def delete_expired_shares(
|
|
request: Request,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
user_account = await UserAccount.get(id=current_user.id)
|
|
deleted_count = await ShareService.delete_expired_shares(user=user_account)
|
|
return success({"deleted_count": deleted_count})
|
|
|
|
|
|
@router.delete("/{share_id}")
|
|
@audit(action=AuditAction.DELETE, description="删除分享链接")
|
|
async def delete_share(
|
|
share_id: int,
|
|
request: Request,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
user_account = await UserAccount.get(id=current_user.id)
|
|
await ShareService.delete_share_link(user=user_account, share_id=share_id)
|
|
return success(msg="分享已取消")
|
|
|
|
|
|
@public_router.post("/{token}/verify")
|
|
@audit(
|
|
action=AuditAction.SHARE,
|
|
description="校验分享密码",
|
|
body_fields=["password"],
|
|
redact_fields=["password"],
|
|
)
|
|
async def verify_password(request: Request, token: str, payload: SharePassword):
|
|
await ShareService.verify_share_password(token, payload.password)
|
|
return success(msg="验证成功")
|
|
|
|
|
|
@public_router.get("/{token}/ls")
|
|
@audit(action=AuditAction.SHARE, description="浏览分享内容")
|
|
async def list_share_content(
|
|
request: Request, token: str, path: str = "/", password: Optional[str] = None
|
|
):
|
|
share = await ShareService.ensure_share_access(token, password)
|
|
content = await ShareService.get_shared_item_details(share, path)
|
|
return success(
|
|
{
|
|
"path": path,
|
|
"entries": content.get("items", []),
|
|
"pagination": {
|
|
"total": content.get("total", 0),
|
|
"page": content.get("page", 1),
|
|
"page_size": content.get("page_size", 1),
|
|
"pages": content.get("pages", 1),
|
|
},
|
|
}
|
|
)
|
|
|
|
|
|
@public_router.get("/{token}")
|
|
@audit(action=AuditAction.SHARE, description="获取分享信息")
|
|
async def get_share_info(request: Request, token: str):
|
|
share = await ShareService.get_share_by_token(token)
|
|
return success(ShareInfo.from_orm(share))
|
|
|
|
|
|
@public_router.get("/{token}/download")
|
|
@audit(action=AuditAction.DOWNLOAD, description="下载分享文件")
|
|
async def download_shared_file(
|
|
token: str,
|
|
path: str,
|
|
request: Request,
|
|
password: Optional[str] = None,
|
|
):
|
|
return await ShareService.stream_shared_file(token, path, request.headers.get("Range"), password)
|