aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib/Tray.js
blob: 631342b24b4f9d4ad397e30f04d8b5a0602d08ac (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import { app, Tray, Menu } from 'electron';
import path from 'path';

const FILE_EXTENSION = process.platform === 'win32' ? 'ico' : 'png';
const INDICATOR_TRAY_PLAIN = 'tray';
const INDICATOR_TRAY_UNREAD = 'tray-unread';

function getAsset(type, asset) {
  return path.join(
    __dirname, '..', 'assets', 'images', type, process.platform, `${asset}.${FILE_EXTENSION}`,
  );
}

export default class TrayIcon {
  mainWindow = null;
  trayIcon = null;

  constructor(mainWindow) {
    this.mainWindow = mainWindow;
  }

  show() {
    this.trayIcon = new Tray(getAsset('tray', INDICATOR_TRAY_PLAIN));
    const trayMenuTemplate = [
      {
        label: 'Show Franz',
        click() {
          this.mainWindow.show();
        },
      }, {
        label: 'Quit Franz',
        click() {
          app.quit();
        },
      },
    ];

    const trayMenu = Menu.buildFromTemplate(trayMenuTemplate);
    this.trayIcon.setContextMenu(trayMenu);

    this.trayIcon.on('click', () => {
      this.mainWindow.show();
    });
  }

  hide() {
    if (this.trayIcon) {
      this.trayIcon.destroy();
      this.trayIcon = null;
    }
  }

  setIndicator(indicator) {
    if (!this.trayIcon) return;

    this.trayIcon.setImage(getAsset('tray', indicator !== 0 ? INDICATOR_TRAY_UNREAD : INDICATOR_TRAY_PLAIN));

    if (process.platform === 'darwin') {
      this.trayIcon.setPressedImage(
        getAsset('tray', `${indicator !== 0 ? INDICATOR_TRAY_UNREAD : INDICATOR_TRAY_PLAIN}-active`),
      );
    }
  }
}