mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-05-12 02:20:28 +08:00
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import httpx
|
|
import time
|
|
from fastapi import APIRouter, Depends, Form, HTTPException
|
|
from typing import Annotated
|
|
from services.config import ConfigCenter, VERSION
|
|
from services.auth import get_current_active_user, User, has_users
|
|
from api.response import success
|
|
from services.vector_db import VectorDBService
|
|
router = APIRouter(prefix="/api/config", tags=["config"])
|
|
|
|
|
|
@router.get("/")
|
|
async def get_config(
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
key: str
|
|
):
|
|
value = await ConfigCenter.get(key)
|
|
return success({"key": key, "value": value})
|
|
|
|
|
|
@router.post("/")
|
|
async def set_config(
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
key: str = Form(...),
|
|
value: str = Form(...)
|
|
):
|
|
original_value = await ConfigCenter.get(key)
|
|
value_to_save = value
|
|
if key == "AI_EMBED_DIM":
|
|
try:
|
|
parsed_value = int(value)
|
|
except (TypeError, ValueError):
|
|
raise HTTPException(status_code=400, detail="AI_EMBED_DIM must be an integer")
|
|
if parsed_value <= 0:
|
|
raise HTTPException(status_code=400, detail="AI_EMBED_DIM must be greater than zero")
|
|
value_to_save = str(parsed_value)
|
|
|
|
await ConfigCenter.set(key, value_to_save)
|
|
|
|
if key == "AI_EMBED_DIM" and str(original_value) != value_to_save:
|
|
try:
|
|
service = VectorDBService()
|
|
await service.clear_all_data()
|
|
except Exception as exc:
|
|
raise HTTPException(status_code=500, detail=f"Failed to clear vector database: {exc}")
|
|
|
|
return success({"key": key, "value": value_to_save})
|
|
|
|
|
|
@router.get("/all")
|
|
async def get_all_config(
|
|
current_user: Annotated[User, Depends(get_current_active_user)]
|
|
):
|
|
configs = await ConfigCenter.get_all()
|
|
return success(configs)
|
|
|
|
|
|
@router.get("/status")
|
|
async def get_system_status():
|
|
system_info = {
|
|
"version": VERSION,
|
|
"title": await ConfigCenter.get("APP_NAME", "Foxel"),
|
|
"logo": await ConfigCenter.get("APP_LOGO", "/logo.svg"),
|
|
"is_initialized": await has_users(),
|
|
"app_domain": await ConfigCenter.get("APP_DOMAIN"),
|
|
"file_domain": await ConfigCenter.get("FILE_DOMAIN"),
|
|
}
|
|
return success(system_info)
|
|
|
|
|
|
latest_version_cache = {
|
|
"timestamp": 0,
|
|
"data": None
|
|
}
|
|
|
|
|
|
@router.get("/latest-version")
|
|
async def get_latest_version():
|
|
current_time = time.time()
|
|
if current_time - latest_version_cache["timestamp"] < 3600 and latest_version_cache["data"]:
|
|
return success(latest_version_cache["data"])
|
|
try:
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
resp = await client.get(
|
|
"https://api.github.com/repos/DrizzleTime/Foxel/releases/latest",
|
|
follow_redirects=True,
|
|
)
|
|
resp.raise_for_status()
|
|
data = resp.json()
|
|
version_info = {
|
|
"latest_version": data.get("tag_name"),
|
|
"body": data.get("body")
|
|
}
|
|
latest_version_cache["timestamp"] = current_time
|
|
latest_version_cache["data"] = version_info
|
|
return success(version_info)
|
|
except httpx.RequestError as e:
|
|
if latest_version_cache["data"]:
|
|
return success(latest_version_cache["data"])
|
|
return success({"latest_version": None, "body": None})
|