fix: auto grants

Thinker-ljn · May 15, 2022 · f5b6a8f · f5b6a8f
1 parent 9c4eebc
commit f5b6a8f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/lib/tampermonkey/grant.ts b/src/lib/tampermonkey/grant.ts
@@ -28,10 +28,7 @@ export const grants = [
   'GM_info',
 ]
 
-const tampermonkeyGrantMapper = grants.reduce((mapper, grant) => {
-  mapper[grant] = true
-  return mapper
-}, {} as Record<string, true>)
+const tampermonkeyGrantSet = new Set(grants)
 
 const usedGrants: Set<string> = new Set()
 export const parserGrant: Plugin = {
@@ -42,13 +39,13 @@ export const parserGrant: Plugin = {
         walkFull(moduleInfo.ast, (node: any) => {
           if (node.type === 'CallExpression') {
             const calleeName = node.callee.name
-            if (calleeName && tampermonkeyGrantMapper[calleeName]) {
+            if (calleeName && tampermonkeyGrantSet.has(calleeName)) {
               usedGrants.add(calleeName)
             }
           }
           if (
             node.type === 'Identifier' &&
-            tampermonkeyGrantMapper[node.name]
+            tampermonkeyGrantSet.has(node.name)
           ) {
             usedGrants.add(node.name)
           }