fix: support for off() & addListener() EventEmitter aliases

electron · Sep 2, 2023 · 5bdff02 · 5bdff02
1 parent dd2d3f7
commit 5bdff02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/module-declaration.ts b/src/module-declaration.ts
@@ -185,7 +185,7 @@ export const generateModuleDeclaration = (
           listener = `(${args.join(`,\n${indent}`)}) => void`;
         }
 
-        for (let method of ['on', 'once', 'addListener', 'removeListener']) {
+        for (let method of ['on', 'off', 'once', 'addListener', 'removeListener']) {
           moduleAPI.push(`${method}(event: '${moduleEvent.name}', listener: ${listener}): this;`);
         }
       });
@@ -251,7 +251,7 @@ export const generateModuleDeclaration = (
   }
 
   const returnsThis = (moduleMethod: Pick<DocumentationBlock, 'name'>) =>
-    ['on', 'once', 'removeAllListeners', 'removeListener'].includes(moduleMethod.name);
+    ['on', 'off', 'once', 'addListener', 'removeAllListeners', 'removeListener'].includes(moduleMethod.name);
 
   const addMethod = (moduleMethod: MethodDocumentationBlock, prefix = '') => {
     utils.extendArray(