update

src/oj/problem/composables/useFlowchart.ts

@@ -1,36 +1,90 @@
 import { ref } from "vue"
 import { useMessage } from "naive-ui"
+import { toRefs } from "vue"
 import { submitFlowchart, getCurrentProblemFlowchartSubmission } from "oj/api"
 import { useProblemStore } from "oj/store/problem"
 import { utoa } from "utils/functions"
 import { useMermaidConverter } from "./useMermaidConverter"
-import { useFlowchartEvaluation } from "./useFlowchartEvaluation"
+import {
+  useFlowchartWebSocket,
+  type FlowchartEvaluationUpdate,
+} from "shared/composables/websocket"
 
-export function useFlowchartSubmission() {
+export interface Evaluation {
+  score: number
+  grade: string
+  feedback: string
+  suggestions: string
+  criteria_details: {
+    [key: string]: { score: number; max: number; comment: string }
+  }
+}
+
+export function useFlowchart() {
   const message = useMessage()
   const problemStore = useProblemStore()
   const { problem } = toRefs(problemStore)
 
   const { convertToMermaid } = useMermaidConverter()
-  const {
-    evaluationResult,
-    submissionStatus,
-    loading,
-    connect,
-    disconnect,
-    subscribeToSubmission,
-    clearResult,
-    setLoading,
-    setEvaluationResult,
-  } = useFlowchartEvaluation()
 
-  // 提交次数
+  // 评估相关状态
+  const evaluation = ref<Evaluation | null>(null)
+  const loading = ref(false)
+
+  // 提交相关状态
   const submissionCount = ref(0)
-
-  // 存储流程图数据
   const myFlowchartZippedStr = ref("")
   const myMermaidCode = ref("")
 
+  // 处理 WebSocket 消息
+  const handleWebSocketMessage = (data: FlowchartEvaluationUpdate) => {
+    console.log("收到流程图评分更新:", data)
+
+    if (data.type === "flowchart_evaluation_completed") {
+      loading.value = false
+      evaluation.value = {
+        score: data.score || 0,
+        grade: data.grade || "",
+        feedback: data.feedback || "",
+        suggestions: data.suggestions || "",
+        criteria_details: data.criteria_details || {},
+      }
+      message.success(`流程图评分完成！得分: ${data.score}分 (${data.grade}级)`)
+    } else if (data.type === "flowchart_evaluation_failed") {
+      console.log("处理评分失败消息")
+      loading.value = false
+      message.error(`流程图评分失败: ${data.error}`)
+    } else {
+      console.log("未知的消息类型:", data.type)
+    }
+  }
+
+  // 创建 WebSocket 连接
+  const { connect, disconnect, subscribe } = useFlowchartWebSocket(
+    handleWebSocketMessage,
+  )
+
+  // 订阅提交更新
+  const subscribeToSubmission = (submissionId: string) => {
+    console.log("开始订阅WebSocket更新")
+    subscribe(submissionId)
+  }
+
+  // 清除结果
+  const clearResult = () => {
+    evaluation.value = null
+  }
+
+  // 设置加载状态
+  const setLoading = (value: boolean) => {
+    loading.value = value
+  }
+
+  // 设置评估结果
+  const setEvaluation = (result: Evaluation) => {
+    evaluation.value = result
+  }
+
   // 检查当前问题的流程图提交状态
   const checkCurrentSubmissionStatus = async () => {
     if (!problem.value?.id) return
@@ -43,12 +97,13 @@ export function useFlowchartSubmission() {
     if (submission && submission.status === 2) {
       myFlowchartZippedStr.value = data.submission.flowchart_data.data
       myMermaidCode.value = data.submission.mermaid_code
-      setEvaluationResult({
+      submissionCount.value += 1
+      setEvaluation({
         score: submission.ai_score,
         grade: submission.ai_grade,
         feedback: submission.ai_feedback,
         suggestions: submission.ai_suggestions,
-        criteriaDetails: submission.ai_criteria_details,
+        criteria_details: submission.ai_criteria_details,
       })
     }
   }
@@ -97,14 +152,20 @@ export function useFlowchartSubmission() {
   }
 
   return {
-    evaluationResult,
-    submissionStatus,
+    // 评估相关
+    evaluation,
     loading,
+    connect,
+    disconnect,
+    subscribeToSubmission,
+    clearResult,
+    setLoading,
+    setEvaluation,
+
+    // 提交相关
     submissionCount,
     myFlowchartZippedStr,
     myMermaidCode,
-    connect,
-    disconnect,
     checkCurrentSubmissionStatus,
     submitFlowchartData,
   }

src/oj/problem/composables/useFlowchartEvaluation.ts

@@ -1,97 +0,0 @@
-import { ref } from "vue"
-import { useMessage } from "naive-ui"
-import {
-  useFlowchartWebSocket,
-  type FlowchartEvaluationUpdate,
-} from "shared/composables/websocket"
-
-export interface EvaluationResult {
-  score?: number
-  grade?: string
-  feedback?: string
-  suggestions?: string
-  criteriaDetails?: any
-}
-
-export interface SubmissionStatus {
-  status: string
-  submission_id: string
-  created_time?: string
-}
-
-export function useFlowchartEvaluation() {
-  const message = useMessage()
-
-  const evaluationResult = ref<EvaluationResult | null>(null)
-  const submissionStatus = ref<SubmissionStatus | null>(null)
-  const loading = ref(false)
-
-  // 处理 WebSocket 消息
-  const handleWebSocketMessage = (data: FlowchartEvaluationUpdate) => {
-    console.log("收到流程图评分更新:", data)
-
-    if (data.type === "flowchart_evaluation_completed") {
-      loading.value = false
-      evaluationResult.value = {
-        score: data.score,
-        grade: data.grade,
-        feedback: data.feedback,
-      }
-      submissionStatus.value = null // 清除状态
-      message.success(`流程图评分完成！得分: ${data.score}分 (${data.grade}级)`)
-    } else if (data.type === "flowchart_evaluation_failed") {
-      console.log("处理评分失败消息")
-      loading.value = false
-      submissionStatus.value = null // 清除状态
-      message.error(`流程图评分失败: ${data.error}`)
-    } else {
-      console.log("未知的消息类型:", data.type)
-    }
-  }
-
-  // 创建 WebSocket 连接
-  const { connect, disconnect, subscribe } = useFlowchartWebSocket(
-    handleWebSocketMessage,
-  )
-
-  // 订阅提交更新
-  const subscribeToSubmission = (submissionId: string) => {
-    console.log("开始订阅WebSocket更新")
-    subscribe(submissionId)
-
-    // 设置评分状态显示
-    submissionStatus.value = {
-      status: "processing",
-      submission_id: submissionId,
-      created_time: new Date().toISOString(),
-    }
-  }
-
-  // 清除结果
-  const clearResult = () => {
-    evaluationResult.value = null
-    submissionStatus.value = null
-  }
-
-  // 设置加载状态
-  const setLoading = (value: boolean) => {
-    loading.value = value
-  }
-
-  // 设置评估结果
-  const setEvaluationResult = (result: EvaluationResult) => {
-    evaluationResult.value = result
-  }
-
-  return {
-    evaluationResult,
-    submissionStatus,
-    loading,
-    connect,
-    disconnect,
-    subscribeToSubmission,
-    clearResult,
-    setLoading,
-    setEvaluationResult,
-  }
-}