Returns function for

index.d.ts

@@ -247,6 +247,6 @@ let mainWindow;
 });
 ```
 */
-declare function contextMenu(options?: contextMenu.Options): void;
+declare function contextMenu(options?: contextMenu.Options): Function;
 
 export = contextMenu;

index.js

@@ -28,8 +28,8 @@ const removeUnusedMenuItems = menuTemplate => {
 		});
 };
 
-const create = (win, options) => {
-	webContents(win).on('context-menu', (event, props) => {
+const contextMenu = (win, options) => {
+	return (event, props) => {
 		if (typeof options.shouldShowMenu === 'function' && options.shouldShowMenu(event, props) === false) {
 			return;
 		}
@@ -226,29 +226,48 @@ const create = (win, options) => {
 			*/
 			menu.popup(electron.remote ? electron.remote.getCurrentWindow() : win);
 		}
-	});
+	};
+};
+
+const create = (win, options) => {
+	const menu = contextMenu(win, options);
+	webContents(win).on('context-menu', menu);
+
+	return () => {
+		webContents(win).removeListener('context-menu', menu);
+	};
 };
 
 module.exports = (options = {}) => {
+	const disposeFunctions = [];
 	if (options.window) {
 		const win = options.window;
 
 		// When window is a webview that has not yet finished loading webContents is not available
 		if (webContents(win) === undefined) {
 			win.addEventListener('dom-ready', () => {
-				create(win, options);
+				disposeFunctions.push(create(win, options));
 			}, {once: true});
-			return;
+		} else {
+			disposeFunctions.push(create(win, options));
+		}
+	} else {
+		for (const win of (electron.BrowserWindow || electron.remote.BrowserWindow).getAllWindows()) {
+			disposeFunctions.push(create(win, options));
 		}
 
-		return create(win, options);
+		(electron.app || electron.remote.app).on('browser-window-created', (event, win) => {
+			disposeFunctions.push(create(win, options));
+		});
 	}
 
-	for (const win of (electron.BrowserWindow || electron.remote.BrowserWindow).getAllWindows()) {
-		create(win, options);
-	}
+	const dispose = () => {
+		for (const f of disposeFunctions) {
+			f();
+		}
+
+		disposeFunctions.length = 0;
+	};
 
-	(electron.app || electron.remote.app).on('browser-window-created', (event, win) => {
-		create(win, options);
-	});
+	return dispose;
 };

index.test-d.ts

@@ -1,4 +1,4 @@
 import {expectType} from 'tsd';
 import contextMenu = require('.');
 
-expectType<void>(contextMenu());
+expectType<Function>(contextMenu());

package.json

@@ -1,13 +1,12 @@
 {
-	"name": "electron-context-menu",
-	"version": "0.15.0",
+	"name": "pj-electron-context-menu",
+	"version": "0.15.4",
 	"description": "Context menu for your Electron app",
 	"license": "MIT",
-	"repository": "sindresorhus/electron-context-menu",
+	"repository": "panjiang/electron-context-menu",
 	"author": {
-		"name": "Sindre Sorhus",
-		"email": "sindresorhus@gmail.com",
-		"url": "sindresorhus.com"
+		"name": "Jiang Pan",
+		"email": "782590544@qq.com"
 	},
 	"scripts": {
 		"start": "electron fixture.js",

readme.md

@@ -1,5 +1,11 @@
 # electron-context-menu [![Build Status](https://travis-ci.org/sindresorhus/electron-context-menu.svg?branch=master)](https://travis-ci.org/sindresorhus/electron-context-menu)
 
+Fork from `electron-context-menu` for:
+
+- Returns `dispose` function for `removeListener('context-menu', *)`, in my case of using HMR(Hot Module Replacement).
+
+---
+
 > Context menu for your [Electron](https://electronjs.org) app
 
 <img src="screenshot.png" width="125" align="right">
@@ -12,7 +18,7 @@ You can use this module directly in both the main and renderer process.
 ## Install
 
 ```
-$ npm install electron-context-menu
+$ npm install pj-electron-context-menu
 ```
 
 *Requires Electron 4 or later.*
@@ -24,7 +30,7 @@ $ npm install electron-context-menu
 const {app, BrowserWindow} = require('electron');
 const contextMenu = require('electron-context-menu');
 
-contextMenu({
+const dispose = contextMenu({
 	prepend: (defaultActions, params, browserWindow) => [
 		{
 			label: 'Rainbow',
@@ -47,6 +53,10 @@ let mainWindow;
 	await app.whenReady();
 	mainWindow = new BrowserWindow();
 })();
+
+
+// You can dispose for unmount
+dispose()
 ```