balena-cli/lib/actions/os.coffee

154 lines
4.5 KiB
CoffeeScript
Raw Normal View History

_ = require('lodash-contrib')
fs = require('fs')
os = require('os')
2014-12-02 15:08:22 +00:00
async = require('async')
path = require('path')
mkdirp = require('mkdirp')
2015-01-08 12:04:37 +00:00
resin = require('resin-sdk')
visuals = require('resin-cli-visuals')
progressStream = require('progress-stream')
diskio = require('diskio')
commandOptions = require('./command-options')
2014-12-02 15:08:22 +00:00
exports.download =
signature: 'os download <id>'
description: 'download device OS'
help: '''
Use this command to download the device OS configured to a specific network.
Ethernet:
You can setup the device OS to use ethernet by setting the `--network` option to "ethernet".
Wifi:
You can setup the device OS to use wifi by setting the `--network` option to "wifi".
If you set "network" to "wifi", you will need to specify the `--ssid` and `--key` option as well.
2014-12-02 15:08:22 +00:00
By default, this command saved the downloaded image into a resin specific directory.
You can save it to a custom location by specifying the `--output` option.
2014-12-02 15:08:22 +00:00
Examples:
$ resin os download 91 --network ethernet
$ resin os download 91 --network wifi --ssid MyNetwork --key secreykey123
$ resin os download 91 --network ethernet --output ~/MyResinOS.zip
'''
options: [
commandOptions.network
commandOptions.wifiSsid
commandOptions.wifiKey
{
signature: 'output'
parameter: 'output'
description: 'output file'
alias: 'o'
}
]
permission: 'user'
action: (params, options, done) ->
osParams =
network: options.network
wifiSsid: options.ssid
wifiKey: options.key
appId: params.id
2014-12-02 15:08:22 +00:00
try
fileName = resin.models.os.generateCacheName(osParams)
catch error
return done(error)
2014-12-02 15:08:22 +00:00
outputFile = options.output or path.join(resin.settings.get('directories.os'), fileName)
2014-12-02 15:08:22 +00:00
async.waterfall [
2014-12-02 15:08:22 +00:00
(callback) ->
# We need to ensure this directory exists
mkdirp(path.dirname(outputFile), _.unary(callback))
(callback) ->
2015-01-29 18:59:57 +00:00
console.info("Destination file: #{outputFile}\n")
2014-12-02 15:08:22 +00:00
2015-01-29 18:59:57 +00:00
bar = new visuals.widgets.Progress('Downloading Device OS')
2014-12-02 15:08:22 +00:00
resin.models.os.download osParams, outputFile, callback, (state) ->
console.log(bar.tick(state.percentage, state.eta))
], (error) ->
return done(error) if error?
console.info("\nFinished downloading #{outputFile}")
return done(null, outputFile)
exports.install =
signature: 'os install <image> [device]'
description: 'write an operating system image to a device'
help: '''
Use this command to write an operating system image to a device.
Note that this command requires admin privileges.
If `device` is omitted, you will be prompted to select a device interactively.
Notice this command asks for confirmation interactively.
You can avoid this by passing the `--yes` boolean option.
You can quiet the progress bar by passing the `--quiet` boolean option.
You may have to unmount the device before attempting this operation.
See the `drives` command to get a list of all connected devices to your machine and their respective ids.
In Mac OS X:
$ sudo diskutil unmountDisk /dev/xxx
In GNU/Linux:
$ sudo umount /dev/xxx
Examples:
$ resin os install rpi.iso /dev/disk2
'''
options: [ commandOptions.yes ]
permission: 'user'
action: (params, options, done) ->
async.waterfall [
(callback) ->
return callback(null, params.device) if params.device?
visuals.patterns.selectDrive(callback)
(device, callback) ->
params.device = device
visuals.patterns.confirm(options.yes, "This will completely erase #{params.device}. Are you sure you want to continue?", callback)
(confirmed, callback) ->
return done() if not confirmed
imageFileSize = fs.statSync(params.image).size
if imageFileSize is 0
error = new Error("Invalid OS image: #{params.image}. The image is 0 bytes.")
return callback(error)
progress = progressStream
length: imageFileSize
time: 500
if not options.quiet
bar = new visuals.widgets.Progress('Writing Device OS')
progress.on 'progress', (status) ->
console.log(bar.tick(status.percentage, status.eta))
imageFileStream = fs.createReadStream(params.image).pipe(progress)
diskio.writeStream(params.device, imageFileStream, callback)
], (error) ->
if os.platform() is 'win32' and error? and (error.code is 'EPERM' or error.code is 'EACCES')
windosu = require('windosu')
# Need to escape everypath to avoid errors
resinWritePath = "\"#{path.join(__dirname, '..', '..', 'bin', 'resin-write')}\""
windosu.exec("node #{resinWritePath} \"#{params.image}\" \"#{params.device}\"")
else
return done(error)