atom/src/command-installer.js

const path = require('path')
const fs = require('fs-plus')

module.exports =
class CommandInstaller {
  constructor (applicationDelegate) {
    this.applicationDelegate = applicationDelegate
  }

  initialize (appVersion) {
    this.appVersion = appVersion
  }

  getInstallDirectory () {
    return '/usr/local/bin'
  }

  getResourcesDirectory () {
    return process.resourcesPath
  }

  installShellCommandsInteractively () {
    const showErrorDialog = (error) => {
      this.applicationDelegate.confirm({
        message: 'Failed to install shell commands',
        detail: error.message
      }, () => {})
    }

    this.installAtomCommand(true, (error, atomCommandName) => {
      if (error) return showErrorDialog(error)
      this.installApmCommand(true, (error, apmCommandName) => {
        if (error) return showErrorDialog(error)
        this.applicationDelegate.confirm({
          message: 'Commands installed.',
          detail: `The shell commands \`${atomCommandName}\` and \`${apmCommandName}\` are installed.`
        }, () => {})
      })
    })
  }

  getCommandNameForChannel (commandName) {
    let channelMatch = this.appVersion.match(/beta|nightly/)
    let channel = channelMatch ? channelMatch[0] : ''

    switch (channel) {
      case 'beta':
        return `${commandName}-beta`
      case 'nightly':
        return `${commandName}-nightly`
      default:
        return commandName
    }
  }

  installAtomCommand (askForPrivilege, callback) {
    this.installCommand(
      path.join(this.getResourcesDirectory(), 'app', 'atom.sh'),
      this.getCommandNameForChannel('atom'),
      askForPrivilege,
      callback
    )
  }

  installApmCommand (askForPrivilege, callback) {
    this.installCommand(
      path.join(this.getResourcesDirectory(), 'app', 'apm', 'node_modules', '.bin', 'apm'),
      this.getCommandNameForChannel('apm'),
      askForPrivilege,
      callback
    )
  }

  installCommand (commandPath, commandName, askForPrivilege, callback) {
    if (process.platform !== 'darwin') return callback()

    const destinationPath = path.join(this.getInstallDirectory(), commandName)

    fs.readlink(destinationPath, (error, realpath) => {
      if (error && error.code !== 'ENOENT') return callback(error)
      if (realpath === commandPath) return callback(null, commandName)
      this.createSymlink(fs, commandPath, destinationPath, error => {
        if (error && error.code === 'EACCES' && askForPrivilege) {
          const fsAdmin = require('fs-admin')
          this.createSymlink(fsAdmin, commandPath, destinationPath, (error) => { callback(error, commandName) })
        } else {
          callback(error)
        }
      })
    })
  }

  createSymlink (fs, sourcePath, destinationPath, callback) {
    fs.unlink(destinationPath, (error) => {
      if (error && error.code !== 'ENOENT') return callback(error)
      fs.makeTree(path.dirname(destinationPath), (error) => {
        if (error) return callback(error)
        fs.symlink(sourcePath, destinationPath, callback)
      })
    })
  }
}