fix(fe): change language's storekey to include problemId and contestId (

#2151)
skkuding · Oct 10, 2024 · 20453e9 · 20453e9
1 parent a1febb4
commit 20453e9
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 14 deletions.
diff --git a/apps/frontend/components/EditorHeader.tsx b/apps/frontend/components/EditorHeader.tsx
@@ -62,7 +62,7 @@ export default function Editor({
   contestId,
   templateString
 }: ProblemEditorProps) {
-  const { language, setLanguage } = useLanguageStore()
+  const { language, setLanguage } = useLanguageStore(problem.id, contestId)()
   const { code, setCode } = createCodeStore((state) => state)
   const testResultStore = useContext(TestResultsContext)
   if (!testResultStore) throw new Error('TestResultsContext is not provided')

diff --git a/apps/frontend/components/EditorResizablePanel.tsx b/apps/frontend/components/EditorResizablePanel.tsx
@@ -38,7 +38,7 @@ export default function EditorMainResizablePanel({
 }: ProblemEditorProps) {
   const pathname = usePathname()
   const base = contestId ? `/contest/${contestId}` : ''
-  const { language, setLanguage } = useLanguageStore()
+  const { language, setLanguage } = useLanguageStore(problem.id, contestId)()
   const testResultStore = useContext(TestResultsContext)
   if (!testResultStore) throw new Error('TestResultsContext is not provided')
   const { testResults } = useStore(testResultStore)
@@ -123,6 +123,8 @@ export default function EditorMainResizablePanel({
             >
               <ScrollArea className="h-full bg-[#121728]">
                 <CodeEditorInEditorResizablePanel
+                  problemId={problem.id}
+                  contestId={contestId}
                   enableCopyPaste={enableCopyPaste}
                 />
                 <ScrollBar orientation="horizontal" />
@@ -148,13 +150,17 @@ export default function EditorMainResizablePanel({
 }
 
 interface CodeEditorInEditorResizablePanelProps {
+  problemId: number
+  contestId?: number
   enableCopyPaste: boolean
 }
 
 function CodeEditorInEditorResizablePanel({
+  problemId,
+  contestId,
   enableCopyPaste
 }: CodeEditorInEditorResizablePanelProps) {
-  const { language } = useLanguageStore()
+  const { language } = useLanguageStore(problemId, contestId)()
   const { code, setCode } = createCodeStore()
 
   return (

diff --git a/apps/frontend/stores/editor.ts b/apps/frontend/stores/editor.ts
@@ -8,19 +8,22 @@ interface LanguageStore {
   setLanguage: (language: Language) => void
 }
 
-export const useLanguageStore = create(
-  persist<LanguageStore>(
-    (set) => ({
-      language: 'C',
-      setLanguage: (language) => {
-        set({ language })
+export const useLanguageStore = (problemId: number, contestId?: number) => {
+  const languageKey = `${problemId}${contestId ? `_${contestId}` : ''}_language`
+  return create(
+    persist<LanguageStore>(
+      (set) => ({
+        language: 'C',
+        setLanguage: (language) => {
+          set({ language })
+        }
+      }),
+      {
+        name: languageKey
       }
-    }),
-    {
-      name: 'language'
-    }
+    )
   )
-)
+}
 interface CodeState {
   code: string
   setCode: (code: string) => void