From f23f8f53e605140d27c45fb2ad7da71f06b834cb Mon Sep 17 00:00:00 2001
From: juetan <contact@juetan.cn>
Date: Tue, 7 Nov 2023 21:12:26 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E8=B7=AF=E7=94=B1?=
 =?UTF-8?q?=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env                                    | 10 ++--
 src/pages/_layout/components/menu.vue   |  6 ++-
 src/pages/system/menu/index.vue         |  6 +--
 src/router/guards/auth.ts               | 68 +++++++++++++++++++++++++
 src/router/guards/guard-auth.ts         | 31 -----------
 src/router/guards/guard-progress.ts     | 31 -----------
 src/router/guards/guard-title.ts        |  9 ----
 src/router/guards/progress.ts           | 30 +++++++++++
 src/router/guards/title.ts              | 12 +++++
 src/router/index.ts                     |  3 +-
 src/router/menus/index.ts               |  2 +-
 src/router/router/index.ts              | 33 ++++--------
 src/router/router/util.ts               | 17 +++++++
 src/router/routes/base.ts               | 14 ++---
 src/router/routes/{index.ts => page.ts} |  4 +-
 src/store/menu/index.ts                 | 12 ++++-
 src/types/env.d.ts                      |  9 +++-
 src/utils/listToTree.ts                 | 26 ++++++++++
 18 files changed, 205 insertions(+), 118 deletions(-)
 create mode 100644 src/router/guards/auth.ts
 delete mode 100644 src/router/guards/guard-auth.ts
 delete mode 100644 src/router/guards/guard-progress.ts
 delete mode 100644 src/router/guards/guard-title.ts
 create mode 100644 src/router/guards/progress.ts
 create mode 100644 src/router/guards/title.ts
 create mode 100644 src/router/router/util.ts
 rename src/router/routes/{index.ts => page.ts} (91%)

diff --git a/.env b/.env
index 25e9cd6..5c319d4 100644
--- a/.env
+++ b/.env
@@ -5,8 +5,12 @@
 VITE_TITLE = 绝弹项目管理
 # 网站副标题
 VITE_SUBTITLE = 快速开发web应用的模板工具
-# 接口前缀 说明：参见 axios 的 baseURL
+# 接口前缀：参见 axios 的 baseURL
 VITE_API = /
+# 首页路径
+VITE_HOME_PATH = /home/home
+# 路由模式：web(路径) hash(锚点)
+VITE_HISTORY = web
 
 # =====================================================================================
 # 开发设置
@@ -19,9 +23,5 @@ VITE_PORT = 3020
 VITE_PROXY = http://127.0.0.1:3030/
 # API文档 说明：需返回符合 OPENAPI 规范的json内容
 VITE_OPENAPI = http://127.0.0.1:3030/openapi.json
-
-# =====================================================================================
-# 构建设置
-# =====================================================================================
 # 文件后缀 说明：设为dev时会优先加载index.dev.vue文件，否则回退至index.vue文件
 VITE_EXTENSION = dev
\ No newline at end of file
diff --git a/src/pages/_layout/components/menu.vue b/src/pages/_layout/components/menu.vue
index 98b5226..9f83356 100644
--- a/src/pages/_layout/components/menu.vue
+++ b/src/pages/_layout/components/menu.vue
@@ -10,7 +10,7 @@ export default defineComponent({
     watch(
       () => route.path,
       () => {
-        selectedKeys.value = route.matched.map((i) => i.path);
+        selectedKeys.value = route.matched.map((i) => i.aliasOf?.path ?? i.path);
       },
       { immediate: true }
     );
@@ -31,7 +31,9 @@ export default defineComponent({
         const icon = route.icon ? () => <i class={route.icon} /> : null;
         const node: any = route.children?.length ? (
           <>
-            <div class="px-2"><a-divider margin={6} class="!border-slate-100"></a-divider></div>
+            <div class="px-2">
+              <a-divider margin={6} class="!border-slate-100"></a-divider>
+            </div>
             {this.renderItem(route?.children)}
           </>
         ) : (
diff --git a/src/pages/system/menu/index.vue b/src/pages/system/menu/index.vue
index 33a402a..2864e57 100644
--- a/src/pages/system/menu/index.vue
+++ b/src/pages/system/menu/index.vue
@@ -7,11 +7,11 @@
 <script setup lang="tsx">
 import { api } from "@/api";
 import { createColumn, updateColumn, useAniTable } from "@/components";
-import { MenuTypes, MenuType } from "@/constants/menu";
-import { flatedMenus } from "@/router";
+import { MenuType, MenuTypes } from "@/constants/menu";
+import { flatMenus } from "@/router";
 import { listToTree } from "@/utils/listToTree";
 
-const menuArr = flatedMenus.map((i) => ({ label: i.title, value: i.id }));
+const menuArr = flatMenus.map((i) => ({ label: i.title, value: i.id }));
 
 const expanded = ref(false);
 const toggleExpand = () => {
diff --git a/src/router/guards/auth.ts b/src/router/guards/auth.ts
new file mode 100644
index 0000000..0fdd9cb
--- /dev/null
+++ b/src/router/guards/auth.ts
@@ -0,0 +1,68 @@
+import { api } from "@/api";
+import { store, useUserStore } from "@/store";
+import { useMenuStore } from "@/store/menu";
+import { treeFind } from "@/utils/listToTree";
+import { Notification } from "@arco-design/web-vue";
+import { Router } from "vue-router";
+import { menus } from "../menus";
+import { APP_HOME_NAME } from "../routes/base";
+import { APP_ROUTE_NAME, routes } from "../routes/page";
+
+const WHITE_LIST = ["/:all(.*)*"];
+const UNSIGNIN_LIST = ["/login"];
+
+/**
+ * 权限守卫
+ * @param to 路由
+ * @description store不能放在外面，否则 pinia-plugin-peristedstate 插件会失效
+ * @returns
+ */
+export function useAuthGuard(router: Router) {
+  api.expireHandler = () => {
+    const userStore = useUserStore(store);
+    const redirect = router.currentRoute.value.path;
+    userStore.clearUser();
+    router.push({ path: "/login", query: { redirect } });
+  };
+
+  router.beforeEach(async function (to) {
+    const userStore = useUserStore(store);
+    const menuStore = useMenuStore(store);
+    if (to.meta.auth?.some((i) => i === "*")) {
+      return true;
+    }
+    if (WHITE_LIST.includes(to.path)) {
+      return true;
+    }
+    if (UNSIGNIN_LIST.includes(to.path)) {
+      if (!userStore.accessToken) {
+        return true;
+      }
+      Notification.warning({
+        title: "跳转提示",
+        content: `您已登陆，如需重新登陆请退出后再操作!`,
+      });
+      return false;
+    }
+    if (!userStore.accessToken) {
+      return { path: "/login", query: { redirect: to.path } };
+    }
+    if (!menuStore.menus.length) {
+      menuStore.setMenus(menus);
+      menuStore.setHome(import.meta.env.VITE_HOME_PATH);
+      for (const route of routes) {
+        router.addRoute(route);
+      }
+      const home = treeFind(routes, (i) => i.path === menuStore.home);
+      if (home) {
+        const route = { ...home, name: APP_HOME_NAME, alias: "/" };
+        router.removeRoute(home.name!);
+        router.addRoute(APP_ROUTE_NAME, route);
+        return router.replace(to.path);
+      }
+    }
+    return true;
+  });
+
+  return router;
+}
diff --git a/src/router/guards/guard-auth.ts b/src/router/guards/guard-auth.ts
deleted file mode 100644
index 11f40ce..0000000
--- a/src/router/guards/guard-auth.ts
+++ /dev/null
@@ -1,31 +0,0 @@
-import { store, useUserStore } from "@/store";
-import { Notification } from "@arco-design/web-vue";
-import { NavigationGuardWithThis } from "vue-router";
-
-const WHITE_LIST = ["/:all(.*)*"];
-const UNSIGNIN_LIST = ["/login"];
-
-// store不能放在外面，否则 pinia-plugin-peristedstate 插件会失效
-export const authGuard: NavigationGuardWithThis<undefined> = async function (to) {
-  const userStore = useUserStore(store);
-  if (to.meta.auth?.some((i) => i === "*")) {
-    return true;
-  }
-  if (WHITE_LIST.includes(to.path)) {
-    return true;
-  }
-  if (UNSIGNIN_LIST.includes(to.path)) {
-    if (userStore.accessToken) {
-      Notification.warning({
-        title: "跳转提示",
-        content: `提示：您已登陆，如需重新登陆请退出后再操作!`,
-      });
-      return false;
-    }
-    return true;
-  }
-  if (!userStore.accessToken) {
-    return { path: "/login", query: { redirect: to.path } };
-  }
-  return true;
-};
diff --git a/src/router/guards/guard-progress.ts b/src/router/guards/guard-progress.ts
deleted file mode 100644
index 645b31d..0000000
--- a/src/router/guards/guard-progress.ts
+++ /dev/null
@@ -1,31 +0,0 @@
-import { NProgress } from "@/libs/nprogress";
-import { useAppStore } from "@/store";
-import { NavigationGuardWithThis, NavigationHookAfter } from "vue-router";
-
-const routeMap = new Map<string, boolean>();
-
-const before: NavigationGuardWithThis<undefined> = function (to) {
-  NProgress.start();
-  if (routeMap.get(to.fullPath)) {
-    return true;
-  }
-  const appStore = useAppStore();
-  appStore.setPageLoading(true);
-};
-
-const after: NavigationHookAfter = function (to) {
-  NProgress.done();
-  if (routeMap.get(to.fullPath)) {
-    return;
-  }
-  const appStore = useAppStore();
-  setTimeout(() => {
-    appStore.setPageLoading(false);
-    routeMap.set(to.fullPath, true);
-  }, 500);
-};
-
-export const progressGuard = {
-  before,
-  after,
-};
diff --git a/src/router/guards/guard-title.ts b/src/router/guards/guard-title.ts
deleted file mode 100644
index 72be640..0000000
--- a/src/router/guards/guard-title.ts
+++ /dev/null
@@ -1,9 +0,0 @@
-import { store, useAppStore } from "@/store";
-import { NavigationHookAfter } from "vue-router";
-
-export const titleGuard: NavigationHookAfter = function (to) {
-  const appStore = useAppStore(store);
-  const title = to.meta.title || appStore.title;
-  const subtitle = appStore.subtitle;
-  document.title = `${title} | ${subtitle}`;
-};
diff --git a/src/router/guards/progress.ts b/src/router/guards/progress.ts
new file mode 100644
index 0000000..6a5d724
--- /dev/null
+++ b/src/router/guards/progress.ts
@@ -0,0 +1,30 @@
+import { NProgress } from "@/libs/nprogress";
+import { useAppStore } from "@/store";
+import { Router } from "vue-router";
+
+const routeMap = new Map<string, boolean>();
+
+export function useProgressGard(router: Router) {
+  router.beforeEach(function (to) {
+    NProgress.start();
+    if (routeMap.get(to.fullPath)) {
+      return true;
+    }
+    const appStore = useAppStore();
+    appStore.setPageLoading(true);
+  });
+
+  router.afterEach(function (to) {
+    NProgress.done();
+    if (routeMap.get(to.fullPath)) {
+      return;
+    }
+    const appStore = useAppStore();
+    setTimeout(() => {
+      appStore.setPageLoading(false);
+      routeMap.set(to.fullPath, true);
+    }, 500);
+  });
+
+  return router;
+}
diff --git a/src/router/guards/title.ts b/src/router/guards/title.ts
new file mode 100644
index 0000000..0520d8d
--- /dev/null
+++ b/src/router/guards/title.ts
@@ -0,0 +1,12 @@
+import { store, useAppStore } from "@/store";
+import { Router } from "vue-router";
+
+export function useTitleGuard(router: Router) {
+  router.beforeEach(function (to) {
+    const appStore = useAppStore(store);
+    const title = to.meta.title || appStore.title;
+    const subtitle = appStore.subtitle;
+    document.title = `${title} | ${subtitle}`;
+  });
+  return router;
+}
diff --git a/src/router/index.ts b/src/router/index.ts
index 30eb2d0..2ea8aa7 100644
--- a/src/router/index.ts
+++ b/src/router/index.ts
@@ -1,3 +1,4 @@
 export * from "./menus";
 export * from "./router";
-export * from "./routes";
+export * from "./routes/page";
+
diff --git a/src/router/menus/index.ts b/src/router/menus/index.ts
index 28b51f9..4ec207e 100644
--- a/src/router/menus/index.ts
+++ b/src/router/menus/index.ts
@@ -1,5 +1,5 @@
 import { RouteRecordRaw } from "vue-router";
-import { appRoutes } from "../routes";
+import { appRoutes } from "../routes/page";
 
 /**
  * 菜单项类型
diff --git a/src/router/router/index.ts b/src/router/router/index.ts
index 246e4f7..ba79091 100644
--- a/src/router/router/index.ts
+++ b/src/router/router/index.ts
@@ -1,42 +1,29 @@
-import { createRouter, createWebHashHistory } from "vue-router";
-import { authGuard } from "../guards/guard-auth";
-import { progressGuard } from "../guards/guard-progress";
-import { titleGuard } from "../guards/guard-title";
-import { routes } from "../routes";
+import { createRouter } from "vue-router";
+import { useAuthGuard } from "../guards/auth";
+import { useProgressGard } from "../guards/progress";
+import { useTitleGuard } from "../guards/title";
 import { baseRoutes } from "../routes/base";
-import { api } from "@/api";
-import { store, useUserStore } from "@/store";
+import { historyMode } from "./util";
 
 /**
  * 路由实例
  */
 export const router = createRouter({
-  history: createWebHashHistory(),
-  routes: [...baseRoutes, ...routes],
+  history: historyMode(),
+  routes: [...baseRoutes],
 });
 
 /**
  * 进度条守卫
  */
-router.beforeEach(progressGuard.before);
-router.afterEach(progressGuard.after);
+useProgressGard(router);
 
 /**
  * 权限守卫
  */
-router.beforeEach(authGuard);
+useAuthGuard(router);
 
 /**
  * 标题守卫
  */
-router.afterEach(titleGuard);
-
-/**
- * 设置令牌过期处理函数
- */
-api.expireHandler = () => {
-  const userStore = useUserStore(store);
-  const redirect = router.currentRoute.value.path;
-  userStore.clearUser();
-  router.push({ path: "/login", query: { redirect } });
-};
+useTitleGuard(router);
diff --git a/src/router/router/util.ts b/src/router/router/util.ts
new file mode 100644
index 0000000..2977d24
--- /dev/null
+++ b/src/router/router/util.ts
@@ -0,0 +1,17 @@
+import { createWebHashHistory, createWebHistory } from "vue-router";
+
+/**
+ * 模式映射
+ */
+const HistoryMap = {
+  web: createWebHistory,
+  hash: createWebHashHistory,
+};
+
+/**
+ * 路由模式
+ */
+export function historyMode() {
+  const mode = HistoryMap[import.meta.env.VITE_HISTORY];
+  return mode();
+}
diff --git a/src/router/routes/base.ts b/src/router/routes/base.ts
index 5c55ffa..6b129e3 100644
--- a/src/router/routes/base.ts
+++ b/src/router/routes/base.ts
@@ -1,14 +1,14 @@
-import { useMenuStore } from "@/store/menu";
 import { RouteRecordRaw } from "vue-router";
 
+export const APP_HOME_NAME = "__APP_HOME__";
+
+/**
+ * 基本路由
+ */
 export const baseRoutes: RouteRecordRaw[] = [
   {
     path: "/",
-    redirect: (to) => {
-      const { home } = useMenuStore();
-      return {
-        name: home,
-      };
-    },
+    name: APP_HOME_NAME,
+    component: () => "Home Page",
   },
 ];
diff --git a/src/router/routes/index.ts b/src/router/routes/page.ts
similarity index 91%
rename from src/router/routes/index.ts
rename to src/router/routes/page.ts
index 32f42f6..8d39e38 100644
--- a/src/router/routes/index.ts
+++ b/src/router/routes/page.ts
@@ -1,8 +1,8 @@
 import generatedRoutes from "virtual:generated-pages";
 import { RouteRecordRaw } from "vue-router";
 
-const TOP_ROUTE_PREF = "_";
-const APP_ROUTE_NAME = "_layout";
+export const TOP_ROUTE_PREF = "_";
+export const APP_ROUTE_NAME = "_layout";
 
 /**
  * 转换路由
diff --git a/src/store/menu/index.ts b/src/store/menu/index.ts
index dda8ab6..2390cfc 100644
--- a/src/store/menu/index.ts
+++ b/src/store/menu/index.ts
@@ -1,12 +1,12 @@
-import { defineStore } from "pinia";
 import { MenuItem } from "@/router";
+import { defineStore } from "pinia";
 
 export const useMenuStore = defineStore({
   id: "menu",
   state: (): MenuStore => {
     return {
       menus: [],
-      home: "/",
+      home: "",
     };
   },
   actions: {
@@ -16,6 +16,14 @@ export const useMenuStore = defineStore({
     setMenus(menus: MenuItem[]) {
       this.menus = menus;
     },
+
+    /**
+     * 设置首页
+     * @param path 路径
+     */
+    setHome(path: string) {
+      this.home = path;
+    }
   },
 });
 
diff --git a/src/types/env.d.ts b/src/types/env.d.ts
index c7b8b84..9034496 100644
--- a/src/types/env.d.ts
+++ b/src/types/env.d.ts
@@ -1,4 +1,3 @@
-import 'vite';
 
 interface ImportMetaEnv {
   /**
@@ -25,6 +24,14 @@ interface ImportMetaEnv {
    * 开发服务器端口
    */
   VITE_PORT: number;
+  /**
+   * 首页路径
+   */
+  VITE_HOME_PATH: string;
+  /**
+   * 路由模式
+   */
+  VITE_HISTORY: "web" | "hash";
 }
 
 interface ImportMeta {
diff --git a/src/utils/listToTree.ts b/src/utils/listToTree.ts
index 543411c..a0dcd08 100644
--- a/src/utils/listToTree.ts
+++ b/src/utils/listToTree.ts
@@ -33,3 +33,29 @@ export function treeEach(tree: any[], fn: (item: any) => void, before = true) {
     !before && fn(item);
   }
 }
+
+/**
+ * 查找子项
+ * @param tree 树结构
+ * @param fn 函数
+ * @returns
+ */
+export function treeFind<T extends { children?: T[]; [key: string]: any } = any>(
+  tree: T[],
+  fn: (item: T) => boolean
+): T | null {
+  let data: T | null = null;
+  for (const item of tree) {
+    if (fn(item)) {
+      data = item;
+      break;
+    }
+    if (item.children) {
+      data = treeFind(item.children, fn);
+      if (data) {
+        break;
+      }
+    }
+  }
+  return data;
+}