youtube-music/plugins/styled-bars/back.js

const {injectCSS} = require('../utils');
const {Menu} = require('electron');
const path = require('path');
const electronLocalshortcut = require("electron-localshortcut");
const config = require('../../config');
var { mainMenuTemplate } = require("../../menu");

//override menu template for custom menu
const originTemplate = mainMenuTemplate;
mainMenuTemplate = function (winHook) {
	//get template
	let template = originTemplate(winHook);
	//fix checkbox and roles
	fixCheck(template);
	//return as normal
	return template;
}
//win hook for fixing menu
let win;

//check that menu doesn't get created twice
let isFixed = false;

module.exports = winImport => {
	win = winImport;
	// css for custom scrollbar + disable drag area(was causing bugs)
	injectCSS(win.webContents, path.join(__dirname, 'style.css'));
	win.on('ready-to-show', () => {
		//build new menu (once)		
		if(!isFixed) {
			let template = mainMenuTemplate(win);
			let menu = Menu.buildFromTemplate(template);
			Menu.setApplicationMenu(menu);
			isFixed = true;
		}
		
		//register keyboard shortcut && hide menu if hideMenu is enabled
		if (config.get('options.hideMenu')) {
			win.webContents.send('updateMenu', null); 
			let enabled=  false;
			electronLocalshortcut.register(win, 'Esc', () => {
				if(enabled) {
					win.webContents.send('updateMenu', null); 
					enabled = false;
				} else {
					win.webContents.send('updateMenu', true); 
					enabled =  true;
				}
			});
		} 
	});
};

//go over each item in menu
function fixCheck(ogTemplate){
	for(let position in ogTemplate) {
		let item = ogTemplate[position];
		//apply function on submenu
		if(item.submenu != null) {
			fixCheck(item.submenu);
		} 
		//change onClick of checkbox+radio
		else if(item.type === 'checkbox' || item.type === 'radio') {
			let ogOnclick = item.click;
			item.click = (itemClicked) => {
				ogOnclick(itemClicked);
				checkCheckbox(itemClicked);
			};
		}  
		//customize roles
		else if (item.role != null) {
			fixRoles(item)
		}
	}
}

function fixRoles(MenuItem) {
	switch (MenuItem.role) {
		case 'reload':
			MenuItem['label'] = 'Reload';
			MenuItem.click = () => {win.webContents.reload();}
			break;
		case 'forceReload' :
			MenuItem.label = 'Force Reload';
			MenuItem.click = () => {win.webContents.reloadIgnoringCache();}
			break;
		case 'zoomIn':
			MenuItem.label = 'Zoom In';
			MenuItem.click = () => {win.webContents.setZoomLevel(win.webContents.getZoomLevel() + 1);}
			break;
		case 'zoomOut':
			MenuItem.label = 'Zoom Out';
			MenuItem.click = () => {win.webContents.setZoomLevel(win.webContents.getZoomLevel() - 1);}
			break;
		case 'resetZoom':
			MenuItem.label = 'Reset Zoom';
			MenuItem.click = () => {win.webContents.setZoomLevel(0);}
			break;
		default : 
			console.log(MenuItem.role +' was not expected');
	}
	delete MenuItem.role;
}

function checkCheckbox(item) {
	//check item
	item.checked = !item.checked;
	//update menu
	win.webContents.send('updateMenu', true); 
}