super-productivity/electron/indicator.js

const electron = require('electron');
//const dbus = require('./dbus');
const fs = require('fs');
const moment = require('moment');
const errorHandler = require('./error-handler');

const GNOME_SHELL_EXT_MIN_VERSION = 2;

let tray;
let isIndicatorRunning = false;
let isGnomeShellExtensionRunning = false;

function init(params) {
  const IS_LINUX = params.IS_LINUX;
  const IS_GNOME = params.IS_GNOME;
  const showApp = params.showApp;
  const quitApp = params.quitApp;
  const IS_MAC = params.IS_MAC;
  const app = params.app;
  const ICONS_FOLDER = params.ICONS_FOLDER;
  const isGnomeShellExtensionInstalled = isGnomeShellInstalled(IS_LINUX, IS_GNOME);

  initAppListeners(app);
  initListeners(isGnomeShellExtensionInstalled);

  // if we have the gnome shell extension installed set up bus
  if (IS_GNOME && isGnomeShellExtensionInstalled) {
    // TODO
    //dbus.init({
    //  quitApp,
    //  showApp,
    //});
    isGnomeShellExtensionRunning = true;
    return;
  }

  // don't start anything on GNOME as the indicator might not be visible and
  // the quiting behaviour confusing
  else if (IS_GNOME) {
    return;
  }

  // otherwise create a regular tray icon
  else {
    // switch tray icon based on
    let trayIcoFile;
    if (IS_MAC) {
      trayIcoFile = 'tray-ico-dark.png'
    } else {
      trayIcoFile = 'tray-ico.png'
    }

    tray = new electron.Tray(ICONS_FOLDER + trayIcoFile);

    tray.setContextMenu(createContextMenu(showApp, quitApp));

    tray.on('click', () => {
      showApp();
    });
    isIndicatorRunning = true;
    return tray;
  }
}

function isGnomeShellInstalled(IS_LINUX, IS_GNOME) {
  // check if shell extension is installed
  let isGnomeShellExtInstalled = false;
  if (IS_LINUX && IS_GNOME) {
    const LINUX_HOME_DIR = process.env['HOME'];
    const EXTENSION_PATH = LINUX_HOME_DIR + '/.local/share/gnome-shell/extensions/indicator@johannes.super-productivity.com';

    if (fs.existsSync(EXTENSION_PATH)) {
      const metaData = fs.readFileSync(EXTENSION_PATH + '/metadata.json');
      const version = JSON.parse(metaData).version;
      if (version >= GNOME_SHELL_EXT_MIN_VERSION) {
        isGnomeShellExtInstalled = true;
      } else {
        errorHandler('Indicator: Outdated version ' + version + ' of Gnome Shell Extension installed. Please install at least version ' + GNOME_SHELL_EXT_MIN_VERSION + '.');
      }
    }
  }
  return isGnomeShellExtInstalled;
}

function initAppListeners(app) {
  if (tray) {
    app.on('before-quit', () => {
      if (tray) {
        tray.destroy();
      }
    });
  }
}

function initListeners(isGnomeShellExtInstalled) {
  electron.ipcMain.on('CHANGED_CURRENT_TASK', (ev, params) => {
    const currentTask = params.current;
    const lastActiveTaskTask = params.lastActiveTask;

    let msg;

    if (currentTask) {
      msg = createIndicatorStr(currentTask);
    }

    if (isGnomeShellExtInstalled) {
      // gnome indicator handling
      if (currentTask && currentTask.title) {
        dbus.setTask(currentTask.id, msg);
      } else if (!currentTask && lastActiveTaskTask && !lastActiveTaskTask.isDone) {
        const msg = createIndicatorStr(lastActiveTaskTask);
        dbus.setTask('PAUSED', msg);
      } else {
        dbus.setTask('NONE', 'NONE');
      }
    } else if (tray) {
      // tray handling
      if (currentTask && currentTask.title) {
        tray.setTitle(msg);
      } else {
        tray.setTitle('');
      }
    }
  });

  electron.ipcMain.on('IPC_EVENT_POMODORO_UPDATE', (ev, params) => {
    const isOnBreak = params.isOnBreak;
    const currentSessionTime = params.currentSessionTime;
    const currentSessionInitialTime = params.currentSessionInitialTime;

    if (isGnomeShellExtInstalled) {
      dbus.updatePomodoro(isOnBreak, currentSessionTime, currentSessionInitialTime);
    }
  });
}

function createIndicatorStr(task) {
  if (task && task.title) {
    let title = task.title;
    let timeStr = '';
    let msg;

    if (title.length > 50) {
      title = title.substring(0, 47) + '...';
    }

    if (task.timeSpent && task.timeSpent._data) {
      task.timeSpent = moment.duration(task.timeSpent._data);
      timeStr += parseInt(task.timeSpent.asMinutes()).toString();
    }
    task.timeEstimate = task.timeEstimate && moment.duration(task.timeEstimate._data);
    const timeEstimateAsMin = moment.duration(task.timeEstimate).asMinutes();
    if (task.timeEstimate && timeEstimateAsMin > 0) {
      timeStr += '/' + timeEstimateAsMin;
    }

    msg = title + ' | ' + timeStr + 'm ';
    return msg;
  }
}

function createContextMenu(showApp, quitApp) {
  return electron.Menu.buildFromTemplate([
    {
      label: 'Show App', click: showApp
    },
    {
      label: 'Quit', click: quitApp
    }
  ]);
}

function isRunning() {
  return isIndicatorRunning || isGnomeShellExtensionRunning;
}

module.exports = {
  init,
  isRunning,
};