main.js

'use strict'
const { autoUpdater } = require('electron-updater')
const log = require('electron-log')
const {
  app,
  BrowserWindow,
  systemPreferences,
  Tray,
  ipcMain,
  shell,
  dialog
} = require('electron')
const path = require('path')
const url = require('url')
const Positioner = require('electron-positioner')
const tesla = require('./tesla-api')
const Store = require('electron-store')
const store = new Store()
const Poller = require('./poller')
const contextMenu = require('electron-context-menu')
const isDev = require('electron-is-dev')

// Logging
autoUpdater.logger = log
autoUpdater.logger.transports.file.level = 'info'
autoUpdater.allowPrerelease = store.get('betaReleases')
log.info('Nikola App starting...')

// Keep a global reference of the window object
let mainWindow

function createWindow() {
  // Create the browser window.
  mainWindow = new BrowserWindow({
    width: 300,
    height: 450,
    show: true,
    frame: false,
    title: 'Nikola',
    fullscreenable: false,
    resizable: false,
    transparent: true,
    titleBarStyle: 'customButtonsOnHover',
    webPreferences: {
      backgroundThrottling: false,
      nodeIntegration: true
    }
  })

  // tray stuff
  let tray

  if (process.platform === 'darwin') {
    if (systemPreferences.isDarkMode()) {
      tray = new Tray(path.join(__dirname, 'src', 'assets', 'img', 'dark.png'))
    } else {
      tray = new Tray(path.join(__dirname, 'src', 'assets', 'img', 'white.png'))
    }
    tray.setToolTip('Nikola')
  }

  if (process.platform !== 'darwin') {
    tray = new Tray(
      path.join(__dirname, 'src', 'assets', 'img', 'win_tray.png')
    )
  }

  // Don't show the app in the dock (OSX)
  if (process.platform === 'darwin') {
    app.dock.hide()
    // Main window behavior
    mainWindow.on('blur', () => {
      mainWindow.hide()
    })
  }

  // Show detached devtools (for development)
  if (isDev) {
    mainWindow.openDevTools({
      mode: 'detach'
    })
  }

  const positioner = new Positioner(mainWindow)
  let bounds = tray.getBounds()

  mainWindow.webContents.on('did-finish-load', () => {
    // Auto update features
    if (!isDev) {
      autoUpdater.checkForUpdates()
    }

    setInterval(() => {
      if (!isDev) {
        store.get('betaReleases')
          ? (autoUpdater.allowPrerelease = true)
          : (autoUpdater.allowPrerelease = false)
        autoUpdater.checkForUpdates()
      }
    }, 300000)

    const dialogOptions = {
      type: 'info',
      buttons: ['Restart and install', 'Not now'],
      message: 'A new Nikola version has been downloaded!'
    }

    autoUpdater.on('update-downloaded', info => {
      dialog.showMessageBox(dialogOptions, i =>
        i === 0 ? autoUpdater.quitAndInstall() : null
      )
    })

    autoUpdater.on('error', err => {
      log.error(err)
    })

    if (process.platform === 'darwin') {
      bounds = tray.getBounds()
      positioner.move('trayCenter', bounds)
    }

    mainWindow.webContents.send('platform', process.platform)
    mainWindow.show()
    // start login and init sequence
    const startLogin = async (authToken, loginEmailPw) => {
      try {
        if (!authToken) {
          authToken = await tesla.login(loginEmailPw)
          store.set('authToken', authToken)
        }
        mainWindow.webContents.send('login', true)
        await getTeslaData()
        startPoller()
      } catch (error) {
        log.error(error)
        store.delete('authToken')
        mainWindow.webContents.send('login-failed', JSON.stringify(error))
      }
    }

    // get tesla Data
    let authToken
    const getTeslaData = async () => {
      authToken = store.get('authToken')

      if (mainWindow.isVisible() && authToken) {
        try {
          let vehicle
          try {
            vehicle = await tesla.vehicle(authToken)
          } catch (error) {
            log.error(error)
            store.delete('authToken')
            store.delete('vehicleId')
            poller ? poller.removeAllListeners() : null
            mainWindow.webContents.send('login', false)
            return
          }

          store.set('vehicleId', vehicle.vehicleID)
          if (vehicle.state !== 'online') {
            mainWindow.webContents.send('tesla-data', {
              ...vehicle
            })
            await tesla.wakeUp(authToken, vehicle.vehicleID)
            return
          }
          const vehicleData = await tesla.vehicleData(
            authToken,
            vehicle.vehicleID
          )
          mainWindow.webContents.send('tesla-data', {
            model: vehicle.model,
            ...vehicleData
          })
          mainWindow.webContents.send('tesla-data-error', false)
        } catch (error) {
          log.error(error)
          mainWindow.webContents.send('tesla-data-error', true)
        }
      }
    }

    // Polling function
    let poller
    const startPoller = () => {
      // Set 10s timeout between polls
      poller = new Poller(10000)
      // Wait till the timeout sent our event to the EventEmitter
      poller.onPoll(async () => {
        await getTeslaData()
        poller.poll()
      })
      // Initial start
      if (store.get('authToken')) {
        poller.poll()
      }
    }

    // Error in Renderer
    ipcMain.on('errorInWindow', async (event, error) => {
      log.error('Window error', error)
    })

    // Actions
    ipcMain.on('login-attempt', async (event, loginEmailPw) => {
      startLogin(false, loginEmailPw)
    })

    // wait helper function
    async function wait(ms) {
      return new Promise(resolve => {
        setTimeout(resolve, ms)
      })
    }

    ipcMain.on('door', async (event, action) => {
      mainWindow.webContents.send('action-loading', action)
      try {
        if (action === 'door-unlock') {
          await tesla.unLockDoor(store.get('authToken'), store.get('vehicleId'))
        } else {
          await tesla.lockDoor(store.get('authToken'), store.get('vehicleId'))
        }
        await wait(500)
        await getTeslaData()

        mainWindow.webContents.send('action-loading', null)
      } catch (error) {
        log.error(error)
        mainWindow.webContents.send('action-error', `Error with ${action}`)
        mainWindow.webContents.send('action-loading', null)
      }
    })

    ipcMain.on('climate', async (event, action) => {
      mainWindow.webContents.send('action-loading', action)
      try {
        if (action === 'climate-on') {
          log.info('triggered climate start')
          await tesla.climateStart(
            store.get('authToken'),
            store.get('vehicleId')
          )
        } else {
          log.info('triggering climate stop')
          await tesla.climateStop(
            store.get('authToken'),
            store.get('vehicleId')
          )
        }
        await wait(500)
        await getTeslaData()

        mainWindow.webContents.send('action-loading', null)
      } catch (error) {
        log.error(error)
        mainWindow.webContents.send('action-error', `Error with ${action}`)
        mainWindow.webContents.send('action-loading', null)
      }
    })

    ipcMain.on('sentryMode', async (event, action) => {
      mainWindow.webContents.send('action-loading', action)
      try {
        if (action === 'sentry-on') {
          log.info('triggering Sentry On')
          await tesla.setSentryMode(
            store.get('authToken'),
            store.get('vehicleId'),
            true
          )
        } else {
          log.info('triggering Sentry Off')
          await tesla.setSentryMode(
            store.get('authToken'),
            store.get('vehicleId'),
            false
          )
        }
        await wait(500)
        await getTeslaData()

        mainWindow.webContents.send('action-loading', null)
      } catch (error) {
        log.error(error)
        mainWindow.webContents.send('action-error', `Error with ${action}`)
        mainWindow.webContents.send('action-loading', null)
      }
    })

    ipcMain.on('climateTemp', async (event, temp) => {
      mainWindow.webContents.send('action-loading', 'climate-temp')
      try {
        log.info('Changing temperature')
        await tesla.setTemps(
          store.get('authToken'),
          store.get('vehicleId'),
          temp
        )
        await wait(500)
        await getTeslaData()
        mainWindow.webContents.send('action-loading', null)
      } catch (error) {
        log.error(error)
        mainWindow.webContents.send('action-error', `Error with climate-temp`)
        mainWindow.webContents.send('action-loading', null)
      }
    })

    const setMenu = () => {
      contextMenu({
        menu: actions => [
          {
            label: `Nikola ${app.getVersion()}`,
            click() {
              shell.openExternal('https://github.com/geraldoramos/nikola')
            }
          },
          {
            label: 'Enable Beta Releases',
            type: 'checkbox',
            checked: store.get('betaReleases'),
            click: function(item) {
              if (store.get('betaReleases')) {
                store.set('betaReleases', false)
                return
              }
              store.set('betaReleases', true)
            }
          },
          actions.separator(),
          {
            label: 'Logout',
            enabled: store.get('authToken') ? true : false,
            click() {
              store.delete('authToken')
              store.delete('vehicleId')
              poller ? poller.removeAllListeners() : null
              mainWindow.webContents.send('login', false)
            }
          },
          {
            label: 'Quit',
            accelerator: 'Command+Q',
            click() {
              app.quit()
            }
          }
        ]
      })
    }
    setMenu()

    const currentLogin = store.get('authToken')
    if (currentLogin) {
      startLogin(currentLogin, false)
    }

    let firstshow = true
    mainWindow.on('show', async () => {
      if (firstshow) {
        firstshow = false
        return
      }
      getTeslaData()
    })
  })

  // position window to the tray area
  tray.setIgnoreDoubleClickEvents(true)
  tray.on('click', event => {
    log.info('click tray event')
    if (process.platform === 'darwin') {
      bounds = tray.getBounds()
      positioner.move('trayCenter', bounds)
    }
    mainWindow.isVisible() ? mainWindow.hide() : mainWindow.show()
  })

  // and load the index.html of the app.
  let indexPath
  if (isDev) {
    indexPath = url.format({
      protocol: 'http:',
      host: 'localhost:8080',
      pathname: 'index.html',
      slashes: true
    })
  } else {
    indexPath = url.format({
      protocol: 'file:',
      pathname: path.join(__dirname, 'dist', 'index.html'),
      slashes: true
    })
  }
  mainWindow.loadURL(indexPath)

  mainWindow.on('closed', function() {
    log.info('window closed event')
    mainWindow = null
  })
}

app.on('ready', createWindow)

app.on('window-all-closed', () => {
  log.info('window-all-closed event')
  if (process.platform !== 'darwin') {
    app.quit()
  }
})

app.on('activate', () => {
  log.info('app on activate event')
  if (mainWindow === null) {
    createWindow()
  }
})