abp-next-admin2/vueJs/src/store/modules/permission.ts

import { VuexModule, Module, Mutation, Action, getModule } from 'vuex-module-decorators'
import { RouteConfig } from 'vue-router'
import { asyncRoutes, constantRoutes } from '@/router'
import store from '@/store'
import { AbpModule } from '@/store/modules/abp'

const hasPermission = (roles: string[], route: RouteConfig) => {
  if (route.meta && route.meta.roles) {
    return roles.some(role => route.meta.roles.includes(role))
  } else {
    return true
  }
}

export const filterAsyncRoutes = (routes: RouteConfig[], roles: string[]) => {
  const res: RouteConfig[] = []
  routes.forEach(route => {
    const r = { ...route }
    if (hasPermission(roles, r)) {
      if (r.children) {
        r.children = filterAsyncRoutes(r.children, roles)
      }
      res.push(r)
    }
  })
  return res
}

export interface IPermissionState {
  routes: RouteConfig[]
  dynamicRoutes: RouteConfig[]
}

@Module({ dynamic: true, store, name: 'permission' })
class Permission extends VuexModule implements IPermissionState {
  public routes: RouteConfig[] = []
  public dynamicRoutes: RouteConfig[] = []
  public authorizedPermissions: string[] = []

  @Mutation
  private SET_ROUTES(routes: RouteConfig[]) {
    this.routes = constantRoutes.concat(routes)
    this.dynamicRoutes = routes
  }

  @Mutation
  private SET_AUTHPERMISSIONS(permissions: Array<string>) {
    this.authorizedPermissions = permissions
  }

  @Action
  public async GetPermissions() {
    const authPermissions = new Array<string>()
    const grantedPolicies = AbpModule.configuration.auth.grantedPolicies
    if (grantedPolicies) {
      Object.keys(grantedPolicies).forEach(key => {
        if (grantedPolicies[key]) {
          authPermissions.push(key)
        }
      })
    }
    if (authPermissions.length === 0) {
      // 防止没有任何权限无限刷新页面
      this.SET_AUTHPERMISSIONS(['guest'])
    } else {
      this.SET_AUTHPERMISSIONS(authPermissions)
    }
  }

  @Action
  public async GenerateRoutes() {
    await this.GetPermissions()
    // 没必要再针对admin角色授权,改成全部后台授权
    // if (this.authorizedPermissions.includes('admin')) {
    //   accessedRoutes = asyncRoutes
    // } else {
    //   accessedRoutes = filterAsyncRoutes(asyncRoutes, this.authorizedPermissions)
    // }
    const accessedRoutes = filterAsyncRoutes(asyncRoutes, this.authorizedPermissions)
    this.SET_ROUTES(accessedRoutes)
  }

  @Action ResetPermissions() {
    this.SET_AUTHPERMISSIONS([])
  }

  @Action ResetRoutes() {
    this.SET_ROUTES([])
  }
}

export const PermissionModule = getModule(Permission)