electron.ts

import { app, Menu, nativeTheme, Tray } from 'electron'
import opener from 'opener'
import { openDashboardInBrowser, openEtherjotInBrowser } from './browser'
import { readConfigYaml } from './config'
import { runLauncher } from './launcher'
import { BeeManager } from './lifecycle'
import { getCurrentNetwork, toggleNetwork } from './network'
import { createNotification } from './notify'
import { getAssetPath, paths } from './path'

let tray: Tray

export function rebuildElectronTray() {
  if (!tray) {
    return
  }
  const contextMenu = Menu.buildFromTemplate([
    {
      label: 'Open Web UI',
      click: openDashboardInBrowser,
    },
    { type: 'separator' },
    {
      label: BeeManager.isRunning() ? 'Stop Bee' : 'Start Bee',
      click: () => {
        if (BeeManager.isRunning()) {
          BeeManager.stop()
        } else {
          runLauncher()
        }
      },
    },
    {
      label: getCurrentNetwork() === 'main' ? 'Switch to Testnet' : 'Switch to Mainnet',
      click: () => {
        toggleNetwork()
      },
    },
    { type: 'separator' },
    {
      type: 'submenu',
      label: 'Apps',
      submenu: [
        {
          label: 'Etherjot (demo)',
          click: openEtherjotInBrowser,
        },
      ],
    },
    { type: 'separator' },
    {
      label: 'Logs',
      click: async () => {
        opener(paths.log)
      },
    },
    {
      label: 'Data Directory',
      click: async () => {
        const config = readConfigYaml()
        if (typeof config['data-dir'] === 'string') {
          opener(config['data-dir'])
        }
      },
    },
    {
      label: 'Quit',
      click: async () => {
        BeeManager.stop()
        await BeeManager.waitForSigtermToFinish()
        app.quit()
      },
    },
  ])
  tray.setContextMenu(contextMenu)
}

function getTrayIcon() {
  if (process.platform === 'darwin') {
    // on macOS the resolution and dark/light is managed automatically
    return getAssetPath('trayTemplate.png')
  }

  const isDark = nativeTheme.shouldUseDarkColors

  if (isDark) {
    return getAssetPath('icon-inv.png')
  }

  return getAssetPath('icon.png')
}

export function runElectronTray() {
  const gotTheLock = app.requestSingleInstanceLock()

  if (!gotTheLock) {
    app.quit()
  } else {
    app.on('second-instance', () => {
      createNotification('Swarm is already running. Please close the previous instance first.')
    })
  }

  app.whenReady().then(() => {
    if (app.dock) {
      app.dock.setIcon(getAssetPath('icon.png'))
      app.dock.hide()
    }
    tray = new Tray(getTrayIcon())
    rebuildElectronTray()
  })
}