refactor: simplify code

Vanessa219 · Mar 23, 2020 · aafb6f8 · aafb6f8
1 parent ff9e1a8
commit aafb6f8
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/src/ts/toolbar/setToolbar.ts b/src/ts/toolbar/setToolbar.ts
@@ -47,13 +47,6 @@ export const disableToolbar = (toolbar: { [key: string]: HTMLElement }, names: s
 };
 
 const CLASS_DISABLED = 'vditor-menu--disabled'
-function setEnabled(e: HTMLElement, v: boolean){
-  if(v){
-    e.classList.remove(CLASS_DISABLED)
-  } else {
-    e.classList.add(CLASS_DISABLED)
-  }
-}
 
 /** 
  * 设置工具栏内容启用/禁用状态。
@@ -67,7 +60,7 @@ export function setToolbarEnabled(
 ): void {
   for (const [k, v] of Object.entries(toolbar)) {
     if (!names || names.includes(k)) {
-      setEnabled(v, enabled);
+      v.classList.toggle(CLASS_DISABLED, !enabled);
     }
   }
 }