From 40d895d8c7856e901a66e01c9ce320576ba98eba Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Fri, 8 May 2026 04:58:06 -0600
Subject: [PATCH] add csrf

---
 src/oj/ai/components/RankDistributionChart.vue |  4 +++-
 src/oj/ai/components/SolvedTable.vue           |  7 ++++++-
 src/oj/ai/components/TimeActivityHeatmap.vue   |  4 +++-
 src/oj/api.ts                                  |  6 +++++-
 src/oj/problem/components/SubmissionResult.vue | 17 +++++++++++++++--
 src/oj/store/ai.ts                             | 12 ++++++++++--
 6 files changed, 42 insertions(+), 8 deletions(-)

diff --git a/src/oj/ai/components/RankDistributionChart.vue b/src/oj/ai/components/RankDistributionChart.vue
index eb5bce0..19e3ffe 100644
--- a/src/oj/ai/components/RankDistributionChart.vue
+++ b/src/oj/ai/components/RankDistributionChart.vue
@@ -1,7 +1,9 @@
 <template>
   <n-card title="同期解题排名分布" size="small" v-if="show">
     <template #header-extra>
-      <n-text depth="3" style="font-size: 12px">了解同期解题速度和竞争力</n-text>
+      <n-text depth="3" style="font-size: 12px">
+        了解同期解题速度和竞争力
+        </n-text>
     </template>
     <div style="height: 300px">
       <Pie :data="data" :options="options" />
diff --git a/src/oj/ai/components/SolvedTable.vue b/src/oj/ai/components/SolvedTable.vue
index 965c735..c3e0b42 100644
--- a/src/oj/ai/components/SolvedTable.vue
+++ b/src/oj/ai/components/SolvedTable.vue
@@ -96,7 +96,12 @@ const columns: DataTableColumn<SolvedProblem>[] = [
   {
     title: () =>
       h(NTooltip, null, {
-        trigger: () => h("span", { style: "cursor:help; border-bottom: 1px dashed" }, "等级"),
+        trigger: () =>
+          h(
+            "span",
+            { style: "cursor:help; border-bottom: 1px dashed" },
+            "等级",
+          ),
         default: () =>
           h("div", null, [
             h("div", null, "基于同时段排名的百分位："),
diff --git a/src/oj/ai/components/TimeActivityHeatmap.vue b/src/oj/ai/components/TimeActivityHeatmap.vue
index e220060..9711864 100644
--- a/src/oj/ai/components/TimeActivityHeatmap.vue
+++ b/src/oj/ai/components/TimeActivityHeatmap.vue
@@ -1,7 +1,9 @@
 <template>
   <n-card title="时间活跃度分析" size="small" v-if="show">
     <template #header-extra>
-      <n-text depth="3" style="font-size: 12px">基于 AC 时间，发现解题高峰时段</n-text>
+      <n-text depth="3" style="font-size: 12px">
+        基于 AC 时间，发现解题高峰时段
+      </n-text>
     </template>
     <div style="height: 300px">
       <Bar :data="data" :options="options" />
diff --git a/src/oj/api.ts b/src/oj/api.ts
index 4f0e44d..6deac03 100644
--- a/src/oj/api.ts
+++ b/src/oj/api.ts
@@ -291,7 +291,11 @@ export function getAIDetailData(start: string, end: string, username?: string) {
   return http.get("ai/detail", { params: { start, end, username } })
 }
 
-export function getAIDurationData(end: string, duration: string, username?: string) {
+export function getAIDurationData(
+  end: string,
+  duration: string,
+  username?: string,
+) {
   return http.get("ai/duration", { params: { end, duration, username } })
 }
 
diff --git a/src/oj/problem/components/SubmissionResult.vue b/src/oj/problem/components/SubmissionResult.vue
index b39835c..02776cd 100644
--- a/src/oj/problem/components/SubmissionResult.vue
+++ b/src/oj/problem/components/SubmissionResult.vue
@@ -1,6 +1,10 @@
 <script setup lang="ts">
 import { JUDGE_STATUS, SubmissionStatus } from "utils/constants"
-import { submissionMemoryFormat, submissionTimeFormat } from "utils/functions"
+import {
+  getCSRFToken,
+  submissionMemoryFormat,
+  submissionTimeFormat,
+} from "utils/functions"
 import type { Submission } from "utils/types"
 import SubmissionResultTag from "shared/components/SubmissionResultTag.vue"
 import { useProblemStore } from "oj/store/problem"
@@ -61,9 +65,18 @@ async function fetchHint(submissionId: string) {
   hintError.value = ""
 
   try {
+    const headers: Record<string, string> = {
+      "Content-Type": "application/json",
+    }
+
+    const csrfToken = getCSRFToken()
+    if (csrfToken) {
+      headers["X-CSRFToken"] = csrfToken
+    }
+
     const response = await fetch("/api/ai/hint", {
       method: "POST",
-      headers: { "Content-Type": "application/json" },
+      headers,
       body: JSON.stringify({ submission_id: submissionId }),
     })
 
diff --git a/src/oj/store/ai.ts b/src/oj/store/ai.ts
index a324f7a..a42887f 100644
--- a/src/oj/store/ai.ts
+++ b/src/oj/store/ai.ts
@@ -29,7 +29,11 @@ export const useAIStore = defineStore("ai", () => {
   const mdContent = ref("")
 
   async function fetchDetailsData(start: string, end: string) {
-    const res = await getAIDetailData(start, end, targetUsername.value || undefined)
+    const res = await getAIDetailData(
+      start,
+      end,
+      targetUsername.value || undefined,
+    )
     detailsData.start = res.data.start
     detailsData.end = res.data.end
     detailsData.solved = res.data.solved
@@ -42,7 +46,11 @@ export const useAIStore = defineStore("ai", () => {
   }
 
   async function fetchDurationData(end: string, duration: string) {
-    const res = await getAIDurationData(end, duration, targetUsername.value || undefined)
+    const res = await getAIDurationData(
+      end,
+      duration,
+      targetUsername.value || undefined,
+    )
     durationData.value = res.data
   }