From 14bee1ade88cf81ef93818352b0b68b3490d56d1 Mon Sep 17 00:00:00 2001
From: cfngc4594 <cfngc4594@gmail.com>
Date: Fri, 21 Mar 2025 23:58:50 +0800
Subject: [PATCH] feat(store): add problem store for managing editor state

---
 src/stores/problem-store.ts | 66 +++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 src/stores/problem-store.ts

diff --git a/src/stores/problem-store.ts b/src/stores/problem-store.ts
new file mode 100644
index 0000000..6939315
--- /dev/null
+++ b/src/stores/problem-store.ts
@@ -0,0 +1,66 @@
+import type {
+  EditorLanguage,
+  EditorLanguageConfig,
+  LanguageServerConfig,
+  Template,
+} from "@prisma/client";
+import type { editor } from "monaco-editor";
+import { createStore } from "zustand/vanilla";
+import { createJSONStorage, persist } from "zustand/middleware";
+import type { MonacoLanguageClient } from "monaco-languageclient";
+
+export type ProblemState = {
+  hydrated: boolean;
+  editor: editor.IStandaloneCodeEditor | null;
+  monacoLanguageClient: MonacoLanguageClient | null;
+  globalLang: EditorLanguage;
+  currentLang: EditorLanguage;
+  currentValue: string;
+  problemId: string;
+  templates: Template[];
+  editorLanguageConfigs: EditorLanguageConfig[];
+  languageServerConfigs: LanguageServerConfig[];
+};
+
+export type ProblemActions = {
+  setHydrated: (value: boolean) => void;
+  setEditor: (editor: editor.IStandaloneCodeEditor) => void;
+  setMonacoLanguageClient: (client: MonacoLanguageClient | null) => void;
+  setGlobalLang: (lang: EditorLanguage) => void;
+  setCurrentLang: (lang: EditorLanguage) => void;
+  setCurrentValue: (value: string) => void;
+};
+
+export type ProblemStore = ProblemState & ProblemActions;
+
+export const createProblemStore = (initState: ProblemState) => {
+  return createStore<ProblemStore>()(
+    persist(
+      (set) => ({
+        ...initState,
+        setHydrated: (value) => set({ hydrated: value }),
+        setEditor: (editor) => set({ editor }),
+        setMonacoLanguageClient: (client) => set({ monacoLanguageClient: client }),
+        setGlobalLang: (lang) => set({ globalLang: lang }),
+        setCurrentLang: (lang) => set({ currentLang: lang }),
+        setCurrentValue: (value) => set({ currentValue: value }),
+      }),
+      {
+        name: "problem-store",
+        storage: createJSONStorage(() => localStorage),
+        partialize: (state) => ({
+          globalLang: state.globalLang,
+        }),
+        onRehydrateStorage: () => {
+          return (state, error) => {
+            if (error) {
+              console.error("An error happened during hydration", error);
+            } else if (state) {
+              state.setHydrated(true);
+            }
+          };
+        },
+      }
+    )
+  );
+};