78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
from fastapi import HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from app.models.user import User
|
|
from app.models.group import PluginGroup
|
|
from app.schemas.client import CheckUpdateData
|
|
from datetime import datetime, timezone
|
|
|
|
class GroupService:
|
|
def check_update(self, db: Session, username: str) -> CheckUpdateData:
|
|
user = db.query(User).filter(User.username == username).first()
|
|
if not user:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="用户不存在")
|
|
|
|
# Check expiration
|
|
is_expired = False
|
|
if user.expire_date:
|
|
now = datetime.now(timezone.utc)
|
|
expire_date = user.expire_date
|
|
if expire_date.tzinfo is None:
|
|
expire_date = expire_date.replace(tzinfo=timezone.utc)
|
|
if now > expire_date:
|
|
is_expired = True
|
|
|
|
# Get group and version
|
|
if not user.group_id:
|
|
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="用户未分配组")
|
|
|
|
group = db.query(PluginGroup).filter(PluginGroup.id == user.group_id).first()
|
|
if not group:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="用户所属组不存在")
|
|
|
|
# 允许无版本的情况，或者使用默认版本
|
|
version_file = group.current_version_file
|
|
if not version_file or version_file == '-':
|
|
# 尝试查找任意一个版本，或者返回空
|
|
# 根据用户要求 "现在默认只有一个了"
|
|
# 这里可以列出 archives 目录下的第一个文件
|
|
import os
|
|
try:
|
|
archives_dir = "archives"
|
|
files = [f for f in os.listdir(archives_dir) if f.endswith('.zip')]
|
|
if files:
|
|
files.sort(reverse=True) # Assuming newer versions sort higher
|
|
version_file = files[0]
|
|
except:
|
|
pass
|
|
|
|
if not version_file or version_file == '-':
|
|
# 如果还是没有，则抛出异常，或者返回空 update
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="当前组无可用版本")
|
|
|
|
# Construct download URL (assuming base URL or relative path)
|
|
# In a real scenario, this might be from config
|
|
download_url = f"/download/{version_file}"
|
|
|
|
# Extract version from filename roughly or use file metadata if available
|
|
# Assuming filename format: plugin_v1.0.2.zip
|
|
version = "unknown"
|
|
filename = version_file
|
|
if "v" in filename:
|
|
try:
|
|
# Simple extraction logic, can be improved
|
|
parts = filename.split("v")
|
|
if len(parts) > 1:
|
|
version_part = parts[1]
|
|
version = "v" + version_part.split(".zip")[0].split("_")[0]
|
|
except:
|
|
pass
|
|
|
|
return CheckUpdateData(
|
|
version=version,
|
|
download_url=download_url,
|
|
force_update=False,
|
|
is_expired=is_expired
|
|
)
|
|
|
|
group_service = GroupService()
|