| from __future__ import annotations |
|
|
| import random |
|
|
| from fastapi import APIRouter, Header, HTTPException |
| from pydantic import BaseModel |
|
|
| from api.support import require_admin, require_identity |
| from services.auth_service import auth_service |
| from services.config import config |
|
|
|
|
| class CheckinRequest(BaseModel): |
| mode: str = "checkin" |
|
|
|
|
| def create_router() -> APIRouter: |
| router = APIRouter() |
|
|
| @router.get("/api/admin/checkin") |
| async def get_admin_checkin(authorization: str | None = Header(default=None)): |
| require_admin(authorization) |
| return {**auth_service.checkin_admin_summary(), "ads": config.get().get("checkin_ads", [])} |
|
|
| @router.get("/api/checkin") |
| async def get_checkin(authorization: str | None = Header(default=None)): |
| identity = require_identity(authorization) |
| if identity.get("role") != "normal": |
| raise HTTPException(status_code=403, detail={"error": "仅普通用户可以签到"}) |
| try: |
| summary = auth_service.checkin_summary(str(identity.get("id") or "")) |
| except ValueError as exc: |
| raise HTTPException(status_code=404, detail={"error": str(exc)}) from exc |
| return { |
| **summary, |
| "credits": auth_service.credit_summary(identity), |
| "ads": config.get().get("checkin_ads", []), |
| "content": config.checkin_content, |
| "credit_valid_days": config.checkin_credit_valid_days, |
| } |
|
|
| @router.post("/api/checkin") |
| async def perform_checkin(body: CheckinRequest, authorization: str | None = Header(default=None)): |
| identity = require_identity(authorization) |
| if identity.get("role") != "normal": |
| raise HTTPException(status_code=403, detail={"error": "仅普通用户可以签到"}) |
| mode = "random" if body.mode == "random" else "checkin" |
| reward = random.randint(config.checkin_random_min, config.checkin_random_max) if mode == "random" else max(1, config.checkin_daily_credits) |
| try: |
| result = auth_service.perform_checkin(str(identity.get("id") or ""), mode=mode, reward=reward) |
| summary = auth_service.checkin_summary(str(identity.get("id") or "")) |
| except ValueError as exc: |
| raise HTTPException(status_code=409, detail={"error": str(exc)}) from exc |
| return { |
| **summary, |
| "result": result, |
| "random_range": {"min": config.checkin_random_min, "max": config.checkin_random_max}, |
| "credits": auth_service.credit_summary(identity), |
| "ads": config.get().get("checkin_ads", []), |
| "content": config.checkin_content, |
| "credit_valid_days": config.checkin_credit_valid_days, |
| } |
|
|
| return router |
|
|
