Blame view

src/directives/permission.ts 1.08 KB
vben authored
1
2
3
4
5
6
7
/**
 * Global authority directive
 * Used for fine-grained control of component permissions
 * @Example v-auth="RoleEnum.TEST"
 */
import type { App, Directive, DirectiveBinding } from 'vue';
8
import projectSetting from '/@/settings/projectSetting';
陈文彬 authored
9
10
11
12
import { usePermission } from '/@/hooks/web/usePermission';
import { PermissionModeEnum } from '/@/enums/appEnum';

function isAuth(el: Element, binding: any) {
13
14
  const { hasPermission } = usePermission();
陈文彬 authored
15
16
17
  const value = binding.value;
  if (!value) return;
  if (!hasPermission(value)) {
vben authored
18
    el.parentNode?.removeChild(el);
陈文彬 authored
19
20
  }
}
vben authored
21
陈文彬 authored
22
function isBackMode() {
23
  return projectSetting.permissionMode === PermissionModeEnum.BACK;
陈文彬 authored
24
}
vben authored
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

const mounted = (el: Element, binding: DirectiveBinding<any>) => {
  if (isBackMode()) return;
  isAuth(el, binding);
};

const updated = (el: Element, binding: DirectiveBinding<any>) => {
  if (!isBackMode()) return;
  isAuth(el, binding);
};

const authDirective: Directive = {
  mounted,
  updated,
};
陈文彬 authored
41
export function setupPermissionDirective(app: App) {
vben authored
42
  app.directive('auth', authDirective);
陈文彬 authored
43
}
vben authored
44
45

export default authDirective;