youtube-music/src/plugins/discord/menu.ts

import prompt from 'custom-electron-prompt';

import { StatusDisplayType } from 'discord-api-types/v10';

import { discordService } from './main';
import { singleton } from '@/providers/decorators';
import promptOptions from '@/providers/prompt-options';
import { setMenuOptions } from '@/config/plugins';
import { t } from '@/i18n';

import type { MenuContext } from '@/types/contexts';
import type { DiscordPluginConfig } from './index';
import type { MenuTemplate } from '@/menu';

const registerRefreshOnce = singleton((refreshMenu: () => void) => {
  discordService?.registerRefreshCallback(refreshMenu);
});

const DiscordStatusDisplayTypeLabels: Record<StatusDisplayType, string> = {
  [StatusDisplayType.Name]:
    'plugins.discord.menu.set-status-display-type.submenu.pear-desktop',
  [StatusDisplayType.State]:
    'plugins.discord.menu.set-status-display-type.submenu.artist',
  [StatusDisplayType.Details]:
    'plugins.discord.menu.set-status-display-type.submenu.title',
};

export const onMenu = async ({
  window,
  getConfig,
  setConfig,
  refresh,
}: MenuContext<DiscordPluginConfig>): Promise<MenuTemplate> => {
  const config = await getConfig();
  registerRefreshOnce(refresh);

  return [
    {
      label: discordService?.isConnected()
        ? t('plugins.discord.menu.connected')
        : t('plugins.discord.menu.disconnected'),
      enabled: !discordService?.isConnected(),
      click: () => discordService?.connect(true),
    },
    {
      label: t('plugins.discord.menu.auto-reconnect'),
      type: 'checkbox',
      checked: config.autoReconnect,
      click(item: Electron.MenuItem) {
        setConfig({
          autoReconnect: item.checked,
        });
      },
    },
    {
      label: t('plugins.discord.menu.clear-activity'),
      click: () => discordService?.clearActivity(),
    },
    {
      label: t('plugins.discord.menu.clear-activity-after-timeout'),
      type: 'checkbox',
      checked: config.activityTimeoutEnabled,
      click(item: Electron.MenuItem) {
        setConfig({
          activityTimeoutEnabled: item.checked,
        });
      },
    },
    {
      label: t('plugins.discord.menu.play-on-pear-desktop'),
      type: 'checkbox',
      checked: config.playOnPearDesktop,
      click(item: Electron.MenuItem) {
        setConfig({
          playOnPearDesktop: item.checked,
        });
      },
    },
    {
      label: t('plugins.discord.menu.hide-github-button'),
      type: 'checkbox',
      checked: config.hideGitHubButton,
      click(item: Electron.MenuItem) {
        setConfig({
          hideGitHubButton: item.checked,
        });
      },
    },
    {
      label: t('plugins.discord.menu.hide-duration-left'),
      type: 'checkbox',
      checked: config.hideDurationLeft,
      click(item: Electron.MenuItem) {
        setConfig({
          hideDurationLeft: item.checked,
        });
      },
    },
    {
      label: t('plugins.discord.menu.set-inactivity-timeout'),
      click: () => setInactivityTimeout(window, config),
    },
    {
      label: t('plugins.discord.menu.set-status-display-type.label'),
      submenu: Object.values(StatusDisplayType)
        .filter(
          (v) => typeof StatusDisplayType[v as StatusDisplayType] !== 'number',
        )
        .map((statusDisplayType) => ({
          label: t(
            DiscordStatusDisplayTypeLabels[
              statusDisplayType as StatusDisplayType
            ],
          ),
          type: 'radio',
          checked: config.statusDisplayType === statusDisplayType,
          click() {
            setConfig({
              statusDisplayType: statusDisplayType as StatusDisplayType,
            });
          },
        })),
    },
  ];
};

async function setInactivityTimeout(
  win: Electron.BrowserWindow,
  options: DiscordPluginConfig,
) {
  const output = await prompt(
    {
      title: t('plugins.discord.prompt.set-inactivity-timeout.title'),
      label: t('plugins.discord.prompt.set-inactivity-timeout.label'),
      value: String(Math.round((options.activityTimeoutTime ?? 0) / 1e3)),
      type: 'counter',
      counterOptions: { minimum: 0, multiFire: true },
      width: 450,
      ...promptOptions(),
    },
    win,
  );

  if (output) {
    options.activityTimeoutTime = Math.round(~~output * 1e3);
    setMenuOptions('discord', options);
  }
}