diff --git a/vite.config.ts b/vite.config.ts
index ad66b16..0c24682 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -5,100 +5,102 @@ import AutoImport from "unplugin-auto-import/vite"
 import { NaiveUiResolver } from "unplugin-vue-components/resolvers"
 import Components from "unplugin-vue-components/vite"
 import Markdown from "unplugin-vue-markdown/vite"
-import { defineConfig } from "vite"
+import { defineConfig, loadEnv } from "vite"
 
-const url = process.env.VITE_OJ_URL
+export default defineConfig(({ mode }) => {
+  const env = loadEnv(mode, process.cwd())
+  const url = env.VITE_OJ_URL
 
-const proxyConfig = {
-  target: url,
-  changeOrigin: true,
-  headers: { Referer: url },
-}
-
-export default defineConfig({
-  build: {
-    rollupOptions: {
-      output: {
-        manualChunks: {
-          chart: ["vue-chartjs", "chart.js"],
-          editor: [
-            "@wangeditor-next/editor",
-            "@wangeditor-next/editor-for-vue",
-          ],
-          cm: [
-            "vue-codemirror",
-            "codemirror",
-            "@codemirror/lang-cpp",
-            "@codemirror/lang-python",
-          ],
+  const proxyConfig = {
+    target: url,
+    changeOrigin: true,
+    headers: { Referer: url },
+  }
+  return {
+    build: {
+      rollupOptions: {
+        output: {
+          manualChunks: {
+            chart: ["vue-chartjs", "chart.js"],
+            editor: [
+              "@wangeditor-next/editor",
+              "@wangeditor-next/editor-for-vue",
+            ],
+            cm: [
+              "vue-codemirror",
+              "codemirror",
+              "@codemirror/lang-cpp",
+              "@codemirror/lang-python",
+            ],
+          },
         },
       },
     },
-  },
-  resolve: {
-    alias: {
-      "~": path.resolve(__dirname, "./src"),
-      utils: path.resolve(__dirname, "./src/utils"),
-      oj: path.resolve(__dirname, "./src/oj"),
-      admin: path.resolve(__dirname, "./src/admin"),
-      learn: path.resolve(__dirname, "./src/learn"),
-    },
-  },
-  plugins: [
-    Vue({ include: [/\.vue$/, /\.md$/] }),
-    AutoImport({
-      imports: [
-        "vue",
-        "vue-router",
-        "@vueuse/core",
-        "pinia",
-        {
-          "naive-ui": [
-            "useDialog",
-            "useMessage",
-            "useNotification",
-            "useLoadingBar",
-          ],
-        },
-        {
-          from: "naive-ui",
-          imports: [
-            "DataTableColumn",
-            "FormRules",
-            "FormItemRule",
-            "SelectOption",
-            "UploadCustomRequestOptions",
-            "UploadFileInfo",
-            "MenuOption",
-            "DropdownDividerOption",
-            "DropdownOption",
-          ],
-          type: true,
-        },
-      ],
-      dts: "./src/auto-imports.d.ts",
-    }),
-    Components({
-      resolvers: [NaiveUiResolver()],
-      dts: "./src/components.d.ts",
-    }),
-    Markdown({
-      async markdownItSetup(md) {
-        md.use(
-          await Shiki({
-            themes: {
-              light: "vitesse-light",
-              dark: "vitesse-dark",
-            },
-          }),
-        )
+    resolve: {
+      alias: {
+        "~": path.resolve(__dirname, "./src"),
+        utils: path.resolve(__dirname, "./src/utils"),
+        oj: path.resolve(__dirname, "./src/oj"),
+        admin: path.resolve(__dirname, "./src/admin"),
+        learn: path.resolve(__dirname, "./src/learn"),
       },
-    }),
-  ],
-  server: {
-    proxy: {
-      "/api": proxyConfig,
-      "/public": proxyConfig,
     },
-  },
+    plugins: [
+      Vue({ include: [/\.vue$/, /\.md$/] }),
+      AutoImport({
+        imports: [
+          "vue",
+          "vue-router",
+          "@vueuse/core",
+          "pinia",
+          {
+            "naive-ui": [
+              "useDialog",
+              "useMessage",
+              "useNotification",
+              "useLoadingBar",
+            ],
+          },
+          {
+            from: "naive-ui",
+            imports: [
+              "DataTableColumn",
+              "FormRules",
+              "FormItemRule",
+              "SelectOption",
+              "UploadCustomRequestOptions",
+              "UploadFileInfo",
+              "MenuOption",
+              "DropdownDividerOption",
+              "DropdownOption",
+            ],
+            type: true,
+          },
+        ],
+        dts: "./src/auto-imports.d.ts",
+      }),
+      Components({
+        resolvers: [NaiveUiResolver()],
+        dts: "./src/components.d.ts",
+      }),
+      Markdown({
+        async markdownItSetup(md) {
+          md.use(
+            await Shiki({
+              themes: {
+                light: "vitesse-light",
+                dark: "vitesse-dark",
+              },
+            }),
+          )
+        },
+      }),
+    ],
+    server: {
+      proxy: {
+        "/api": proxyConfig,
+        "/public": proxyConfig,
+      },
+    },
+  }
 })