From 9274326ac5dd7f77d676d96785f05bdc324a7ab0 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Mon, 17 Apr 2023 13:50:11 +0800
Subject: [PATCH] add cache.

---
 src/oj/composables/code.ts           |  4 +++-
 src/oj/problem/components/Editor.vue | 30 +++++++++++++++++++++++++---
 src/oj/problem/components/Form.vue   | 13 +++++++++++-
 src/oj/problem/detail.vue            |  4 ++++
 src/utils/constants.ts               |  2 +-
 5 files changed, 47 insertions(+), 6 deletions(-)

diff --git a/src/oj/composables/code.ts b/src/oj/composables/code.ts
index e1fd7f8..6448533 100644
--- a/src/oj/composables/code.ts
+++ b/src/oj/composables/code.ts
@@ -1,6 +1,8 @@
+import { STORAGE_KEY } from "~/utils/constants"
+import storage from "~/utils/storage"
 import { Code } from "~/utils/types"
 
 export const code = reactive<Code>({
   value: "",
-  language: "C",
+  language: storage.get(STORAGE_KEY.LANGUAGE) || "C",
 })
diff --git a/src/oj/problem/components/Editor.vue b/src/oj/problem/components/Editor.vue
index 98eb43e..e78ddfc 100644
--- a/src/oj/problem/components/Editor.vue
+++ b/src/oj/problem/components/Editor.vue
@@ -3,23 +3,47 @@ import { SOURCES } from "utils/constants"
 import { code } from "oj/composables/code"
 import { isDesktop } from "~/shared/composables/breakpoints"
 import { problem } from "oj/composables/problem"
+import storage from "~/utils/storage"
+import { LANGUAGE } from "~/utils/types"
 
 const Form = defineAsyncComponent(() => import("./Form.vue"))
 const CodeEditor = defineAsyncComponent(() => import("~/shared/CodeEditor.vue"))
 
-code.language = problem.value!.languages[0] || "C"
-code.value = problem.value!.template[code.language] || SOURCES[code.language]
+const route = useRoute()
+const contestID = !!route.params.contestID ? route.params.contestID : null
+
+const storageKey = computed(
+  () => `problem_${problem.value!._id}_contest_${contestID}`
+)
+
+onMounted(() => {
+  if (storage.get(storageKey.value)) {
+    code.value = storage.get(storageKey.value)
+  } else {
+    code.value =
+      problem.value!.template[code.language] || SOURCES[code.language]
+  }
+})
 
 const editorHeight = computed(() =>
   isDesktop.value ? "calc(100vh - 133px)" : "calc(100vh - 172px)"
 )
+
+function reset() {
+  storage.remove(storageKey.value)
+}
+
+function changeCode(v: string) {
+  storage.set(storageKey.value, v)
+}
 </script>
 
 <template>
   <n-space vertical>
-    <Form />
+    <Form @reset="reset" />
     <CodeEditor
       v-model="code.value"
+      @update:model-value="changeCode"
       :language="code.language"
       :height="editorHeight"
     />
diff --git a/src/oj/problem/components/Form.vue b/src/oj/problem/components/Form.vue
index 7212f97..41a89ec 100644
--- a/src/oj/problem/components/Form.vue
+++ b/src/oj/problem/components/Form.vue
@@ -5,15 +5,21 @@ import { problem } from "oj/composables/problem"
 import { isDesktop, isMobile } from "~/shared/composables/breakpoints"
 import { useUserStore } from "~/shared/store/user"
 import Submit from "./Submit.vue"
+import storage from "~/utils/storage"
+import { STORAGE_KEY } from "utils/constants"
+import { LANGUAGE } from "~/utils/types"
 
 const route = useRoute()
 const router = useRouter()
 const userStore = useUserStore()
 
+const emit = defineEmits(["reset"])
+
 watch(() => code.language, reset)
 
 function reset() {
   code.value = problem.value!.template[code.language] || SOURCES[code.language]
+  emit("reset")
 }
 
 function goSubmissions() {
@@ -62,14 +68,19 @@ function select(key: string) {
       break
   }
 }
+
+function changeLanguage(v: LANGUAGE) {
+  storage.set(STORAGE_KEY.LANGUAGE, v)
+}
 </script>
 
 <template>
   <n-space>
     <n-select
       class="language"
-      :size="isDesktop ? 'medium' : 'small'"
       v-model:value="code.language"
+      @update:value="changeLanguage"
+      :size="isDesktop ? 'medium' : 'small'"
       :options="options"
     />
     <Submit />
diff --git a/src/oj/problem/detail.vue b/src/oj/problem/detail.vue
index 2edcb5e..1298885 100644
--- a/src/oj/problem/detail.vue
+++ b/src/oj/problem/detail.vue
@@ -34,6 +34,10 @@ async function init() {
   }
 }
 onMounted(init)
+onBeforeUnmount(() => {
+  problem.value = null
+  errMsg.value = "无数据"
+})
 </script>
 
 <template>
diff --git a/src/utils/constants.ts b/src/utils/constants.ts
index b3872d0..61e5963 100644
--- a/src/utils/constants.ts
+++ b/src/utils/constants.ts
@@ -120,7 +120,7 @@ export const PROBLEM_PERMISSION = {
 
 export const STORAGE_KEY = {
   AUTHED: "authed",
-  PROBLEM_CODE: "problemCode",
+  LANGUAGE: "problemLanguage",
 }
 
 export const DIFFICULTY = {