zpark-ems/core/backend/app/api/v1/settings.py

from fastapi import APIRouter, Depends
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from pydantic import BaseModel
from app.core.database import get_db
from app.core.deps import get_current_user, require_roles
from app.models.user import User
from app.models.setting import SystemSetting
from app.services.audit import log_audit

router = APIRouter(prefix="/settings", tags=["系统设置"])

# Default settings — used when keys are missing from DB
DEFAULTS: dict[str, str] = {
    "platform_name": "天普零碳园区智慧能源管理平台",
    "data_retention_days": "365",
    "alarm_auto_resolve_minutes": "30",
    "simulator_interval_seconds": "15",
    "notification_email_enabled": "false",
    "notification_email_smtp": "",
    "report_auto_schedule_enabled": "false",
    "timezone": "Asia/Shanghai",
    # AI Model Settings
    "ai_enabled": "false",
    "ai_provider": "stepfun",  # stepfun, zhipu
    "ai_api_base_url": "https://api.stepfun.com/step_plan/v1",
    "ai_api_key": "1UVGFlMG9zaGrRvRATpBNdjKotLio6x9t6lKRKdxwYD3mEkLU2Itb30yb1rvzWRGs",
    "ai_model_name": "step-2-16k",
    "ai_temperature": "0.7",
    "ai_max_tokens": "2000",
    "ai_context_length": "8000",
    "ai_fallback_enabled": "true",
    "ai_fallback_provider": "zhipu",
    "ai_fallback_api_base_url": "https://open.bigmodel.cn/api/coding/paas/v4",
    "ai_fallback_api_key": "0b5fe625dfd64836bfd42cc9608aed42.wnQngOvi7EkAWjyn",
    "ai_fallback_model_name": "codegeex-4",
    "ai_system_prompt": "你是一个专业的光伏电站智能运维助手。你的任务是分析光伏电站的设备运行数据、告警信息和历史趋势，提供专业的诊断分析和运维建议。请用中文回答，结构清晰，重点突出。",
    "ai_diagnostic_prompt": "请分析以下光伏设备的运行数据，给出诊断报告：\n\n设备信息：{device_info}\n运行数据：{metrics}\n告警记录：{alarms}\n\n请按以下结构输出：\n## 运行概况\n## 问题诊断\n## 建议措施\n## 风险预警",
    "ai_insight_prompt": "请根据以下电站运行数据，生成运营洞察报告：\n\n电站概况：{station_info}\n关键指标：{kpis}\n近期告警：{recent_alarms}\n\n请给出3-5条关键洞察和建议。",
}


class SettingsUpdate(BaseModel):
    platform_name: str | None = None
    data_retention_days: int | None = None
    alarm_auto_resolve_minutes: int | None = None
    simulator_interval_seconds: int | None = None
    notification_email_enabled: bool | None = None
    notification_email_smtp: str | None = None
    report_auto_schedule_enabled: bool | None = None
    timezone: str | None = None
    ai_enabled: bool | None = None
    ai_provider: str | None = None
    ai_api_base_url: str | None = None
    ai_api_key: str | None = None
    ai_model_name: str | None = None
    ai_temperature: float | None = None
    ai_max_tokens: int | None = None
    ai_context_length: int | None = None
    ai_fallback_enabled: bool | None = None
    ai_fallback_provider: str | None = None
    ai_fallback_api_base_url: str | None = None
    ai_fallback_api_key: str | None = None
    ai_fallback_model_name: str | None = None
    ai_system_prompt: str | None = None
    ai_diagnostic_prompt: str | None = None
    ai_insight_prompt: str | None = None


def _mask_key(key: str) -> str:
    if not key or len(key) < 8:
        return "****"
    return "*" * (len(key) - 4) + key[-4:]


async def _get_raw_settings(db: AsyncSession) -> dict[str, str]:
    """Return merged settings dict WITHOUT masking (for internal use)."""
    result = await db.execute(select(SystemSetting))
    db_settings = {s.key: s.value for s in result.scalars().all()}
    return {**DEFAULTS, **db_settings}


@router.get("")
async def get_settings(
    db: AsyncSession = Depends(get_db),
    user: User = Depends(get_current_user),
):
    """Return all platform settings as a flat dict."""
    merged = await _get_raw_settings(db)
    # Cast types for frontend
    return {
        "platform_name": merged["platform_name"],
        "data_retention_days": int(merged["data_retention_days"]),
        "alarm_auto_resolve_minutes": int(merged["alarm_auto_resolve_minutes"]),
        "simulator_interval_seconds": int(merged["simulator_interval_seconds"]),
        "notification_email_enabled": merged["notification_email_enabled"] == "true",
        "notification_email_smtp": merged["notification_email_smtp"],
        "report_auto_schedule_enabled": merged["report_auto_schedule_enabled"] == "true",
        "timezone": merged["timezone"],
        # AI settings
        "ai_enabled": merged["ai_enabled"] == "true",
        "ai_provider": merged["ai_provider"],
        "ai_api_base_url": merged["ai_api_base_url"],
        "ai_api_key": _mask_key(merged["ai_api_key"]),
        "ai_model_name": merged["ai_model_name"],
        "ai_temperature": float(merged["ai_temperature"]),
        "ai_max_tokens": int(merged["ai_max_tokens"]),
        "ai_context_length": int(merged["ai_context_length"]),
        "ai_fallback_enabled": merged["ai_fallback_enabled"] == "true",
        "ai_fallback_provider": merged["ai_fallback_provider"],
        "ai_fallback_api_base_url": merged["ai_fallback_api_base_url"],
        "ai_fallback_api_key": _mask_key(merged["ai_fallback_api_key"]),
        "ai_fallback_model_name": merged["ai_fallback_model_name"],
        "ai_system_prompt": merged["ai_system_prompt"],
        "ai_diagnostic_prompt": merged["ai_diagnostic_prompt"],
        "ai_insight_prompt": merged["ai_insight_prompt"],
    }


@router.put("")
async def update_settings(
    data: SettingsUpdate,
    db: AsyncSession = Depends(get_db),
    user: User = Depends(require_roles("admin")),
):
    """Update platform settings (admin only)."""
    updates = data.model_dump(exclude_unset=True)
    changed_keys = []
    for key, value in updates.items():
        str_value = str(value).lower() if isinstance(value, bool) else str(value)
        result = await db.execute(select(SystemSetting).where(SystemSetting.key == key))
        setting = result.scalar_one_or_none()
        if setting:
            setting.value = str_value
        else:
            db.add(SystemSetting(key=key, value=str_value))
        changed_keys.append(key)

    await log_audit(
        db, user.id, "update", "system",
        detail=f"更新系统设置: {', '.join(changed_keys)}",
    )

    return {"message": "设置已更新"}


@router.post("/test-ai")
async def test_ai_connection(
    db: AsyncSession = Depends(get_db),
    user: User = Depends(require_roles("admin")),
):
    """Test AI model connection."""
    from app.services.llm_service import test_connection
    settings = await _get_raw_settings(db)
    result = await test_connection(settings)
    return result