KGPython/controller/GraphStyleController.py

# controller/GraphStyleController.py
import json
from robyn import jsonify, Response
from app import app
from service.GraphStyleService import GraphStyleService


# --- 核心工具函数：解决乱码 ---
def create_response(status_code, data_dict):
    """
    统一响应格式封装，强制使用 UTF-8 防止中文乱码。
    """
    return Response(
        status_code=status_code,
        description=json.dumps(data_dict, ensure_ascii=False),
        headers={"Content-Type": "application/json; charset=utf-8"}
    )


@app.post("/api/graph/style/save")
async def save_style_config(request):
    """
    保存配置接口 - 增强防跑偏版
    逻辑：
    1. 如果 body 中包含 is_auto_save: true，则强制忽略 group_name，防止自动保存篡改归属。
    2. 如果是普通保存或移动，则正常传递 group_name。
    """
    try:
        body = request.json()

        config_id = body.get('id')
        canvas_name = body.get('canvas_name')
        current_label = body.get('current_label')
        styles = body.get('styles')

        # 核心改动：获取 group_name
        group_name = body.get('group_name')

        # 增加一个前端标识：如果是实时同步(防抖保存)，前端可以传这个字段
        is_auto_save = body.get('is_auto_save', False)

        if not all([canvas_name, current_label, styles]):
            return create_response(200, {"code": 400, "msg": "参数不完整"})

        # 如果是自动保存模式，显式清空 group_name，强制 Service 进入“仅更新样式”逻辑
        final_group_name = None if is_auto_save else group_name

        # 将处理后的参数传给 Service 层
        success = GraphStyleService.save_config(
            canvas_name=canvas_name,
            current_label=current_label,
            styles_dict=styles,
            group_name=final_group_name,
            config_id=config_id
        )

        if success:
            return create_response(200, {"code": 200, "msg": "操作成功"})
        else:
            return create_response(200, {"code": 500, "msg": "操作失败"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"系统异常: {str(e)}"})


@app.get("/api/graph/style/list/grouped")
async def get_grouped_style_list(request):
    """获取【分组嵌套】格式的配置列表"""
    try:
        data = GraphStyleService.get_grouped_configs()
        return create_response(200, {"code": 200, "data": data, "msg": "查询成功"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"查询异常: {str(e)}"})


@app.post("/api/graph/style/group/apply")
async def apply_style_group(request):
    """应用全案"""
    try:
        body = request.json()
        group_id = body.get('group_id')

        if not group_id:
            return create_response(200, {"code": 400, "msg": "缺少分组ID"})

        success = GraphStyleService.apply_group_all(group_id)
        if success:
            return create_response(200, {"code": 200, "msg": "方案已成功应用"})
        else:
            return create_response(200, {"code": 500, "msg": "应用全案失败"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"操作异常: {str(e)}"})


@app.post("/api/graph/style/group/set_default")
async def set_default_style_group(request):
    """设为系统初始默认方案"""
    try:
        body = request.json()
        group_id = body.get('group_id')

        if not group_id:
            return create_response(200, {"code": 400, "msg": "缺少分组ID"})

        success = GraphStyleService.set_default_group(group_id)
        if success:
            return create_response(200, {"code": 200, "msg": "已成功设为系统默认方案"})
        else:
            return create_response(200, {"code": 500, "msg": "设置默认方案失败"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"操作异常: {str(e)}"})


@app.get("/api/graph/style/groups")
async def get_group_names(request):
    """获取所有已存在的方案组列表"""
    try:
        data = GraphStyleService.get_group_list()
        return create_response(200, {"code": 200, "data": data, "msg": "查询成功"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"查询异常: {str(e)}"})


@app.get("/api/graph/style/list")
async def get_style_list(request):
    """获取原始扁平配置列表"""
    try:
        data = GraphStyleService.get_all_configs()
        return create_response(200, {"code": 200, "data": data, "msg": "查询成功"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"查询异常: {str(e)}"})


@app.post("/api/graph/style/delete")
async def delete_style_config(request):
    """删除单条画布配置"""
    try:
        body = request.json()
        config_id = body.get('id')

        if not config_id:
            return create_response(200, {"code": 400, "msg": "缺少ID"})

        success = GraphStyleService.delete_config(config_id)
        if success:
            return create_response(200, {"code": 200, "msg": "删除成功"})
        else:
            return create_response(200, {"code": 500, "msg": "删除失败"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"操作异常: {str(e)}"})


@app.post("/api/graph/style/group/delete")
async def delete_style_group(request):
    """删除整个方案组及其下属所有配置"""
    try:
        body = request.json()
        group_id = body.get('group_id')

        if not group_id:
            return create_response(200, {"code": 400, "msg": "缺少分组ID"})

        success = GraphStyleService.delete_group(group_id)
        if success:
            return create_response(200, {"code": 200, "msg": "方案组已彻底删除"})
        else:
            return create_response(200, {"code": 500, "msg": "方案组删除失败"})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"操作异常: {str(e)}"})


@app.post("/api/graph/style/batch_delete")
async def batch_delete_style(request):
    """批量删除配置接口"""
    try:
        body = request.json()
        config_ids = body.get('ids')

        if isinstance(config_ids, str):
            try:
                config_ids = json.loads(config_ids)
            except:
                pass

        if not config_ids or not isinstance(config_ids, list):
            return create_response(200, {"code": 400, "msg": "参数格式错误"})

        count = GraphStyleService.batch_delete_configs(config_ids)
        return create_response(200, {"code": 200, "msg": f"成功删除 {count} 条配置", "count": count})
    except Exception as e:
        return create_response(200, {"code": 500, "msg": f"批量删除异常: {str(e)}"})