addons.js

const { GLib } = imports.gi;
const Main = imports.ui.main;
const AggregateMenu = Main.panel.statusArea.aggregateMenu;

let castMenu;
let addonMenuItems = {};
let addonMenuSignals = {};
let timeouts = {};

function findCastToTv()
{
	let menuItems = AggregateMenu.menu._getMenuItems();
	let index = 0;

	while(index < menuItems.length)
	{
		if(
			menuItems[index].hasOwnProperty('extensionId')
			&& menuItems[index].extensionId === 'cast-to-tv'
		) {
			return menuItems[index];
		}

		index++;
	}

	return null;
}

function setLastMenuItem(extMenu, item, endOffset)
{
	if(!endOffset) endOffset = 0;

	let subMenuItems = extMenu.castSubMenu.menu._getMenuItems();
	let lastItemIndex = subMenuItems.length - 1;
	extMenu.castSubMenu.menu.moveMenuItem(item, lastItemIndex - endOffset);
}

function enableAddon(uuid)
{
	let addonName = uuid.split('@')[0];

	if(timeouts[addonName])
		return;

	/* Give main extension time to finish startup */
	timeouts[addonName] = GLib.timeout_add(GLib.PRIORITY_DEFAULT, 2500, () =>
	{
		timeouts[addonName] = null;

		castMenu = findCastToTv();
		let mainExtension = Main.extensionManager.lookup('cast-to-tv@rafostar.github.com');
		let addonExtension = Main.extensionManager.lookup(uuid);

		if(!castMenu || !mainExtension || !addonExtension)
			return GLib.SOURCE_REMOVE;

		let { helper, soup, shared } = mainExtension.imports;
		let { AddonMenuItem } = addonExtension.imports.widget;

		addonMenuItems[addonName] = new AddonMenuItem({
			helper: helper,
			soupClient: soup.client,
			shared: shared.module.exports,
			path: mainExtension.path
		});

		addonMenuSignals[addonName] = [
			addonMenuItems[addonName].connect('destroy', () =>
			{
				addonMenuSignals[addonName].forEach(signal =>
					addonMenuItems[addonName].disconnect(signal))

				addonMenuSignals[addonName] = null;
				addonMenuItems[addonName].destroyed = true;
			})
		];

		if(
			addonMenuItems[addonName].hasOwnProperty('hasExtApp')
			&& addonMenuItems[addonName].hasExtApp
		) {
			addonMenuSignals[addonName].push(
				addonMenuItems[addonName].connect('activate', () =>
					helper.startApp(addonExtension.path)
				)
			)
		}
		else
		{
			addonMenuSignals[addonName].push(
				addonMenuItems[addonName].connect('activate', () =>
					helper.closeOtherApps()
				)
			)
		}

		let castMenuItems = castMenu.castSubMenu.menu._getMenuItems();
		let insertIndex = castMenuItems.length - 1;

		let prevMenuItem = castMenuItems.find(item =>
		{
			if(
				item.hasOwnProperty('isDesktopStream')
				|| (castMenuItems.indexOf(item) > 2
				&& addonMenuItems[addonName].label.text < item.label.text)
			) {
				return true;
			}

			return false;
		});

		/* Desktop streaming should be last on the list (experimental feature) */
		if(prevMenuItem && !addonMenuItems[addonName].hasOwnProperty('isDesktopStream'))
			insertIndex = castMenuItems.indexOf(prevMenuItem);

		castMenu.castSubMenu.menu.addMenuItem(addonMenuItems[addonName], insertIndex);

		if(
			castMenu.hasOwnProperty('isServiceEnabled')
			&& castMenu.isServiceEnabled === false
		) {
			if(addonMenuItems[addonName].hasOwnProperty('actor'))
				addonMenuItems[addonName].actor.hide();
			else
				addonMenuItems[addonName].hide();
		}

		if(castMenu.hasOwnProperty('serviceMenuItem'))
			setLastMenuItem(castMenu, castMenu.serviceMenuItem);

		if(castMenu.hasOwnProperty('settingsMenuItem'))
			setLastMenuItem(castMenu, castMenu.settingsMenuItem);

		return GLib.SOURCE_REMOVE;
	});
}

function disableAddon(uuid)
{
	let addonName = uuid.split('@')[0];

	if(timeouts[addonName])
	{
		GLib.source_remove(timeouts[addonName]);
		timeouts[addonName] = null;
	}

	if(!addonMenuItems[addonName] || addonMenuItems[addonName].destroyed)
		return;

	/* No need to reorder menu items when locking screen,
	as whole cast menu will be destroyed then anyway */
	let lockingScreen = (
		Main.sessionMode.currentMode == 'unlock-dialog'
		|| Main.sessionMode.currentMode == 'lock-screen'
	);

	if(!lockingScreen && castMenu)
		setLastMenuItem(castMenu, addonMenuItems[addonName]);

	/* Force GUI refresh by hiding item before removal */
	if(addonMenuItems[addonName].hasOwnProperty('actor'))
		addonMenuItems[addonName].actor.hide();
	else
		addonMenuItems[addonName].hide();

	addonMenuItems[addonName].destroy();
	addonMenuItems[addonName] = null;
}