mirror of
https://github.com/DeviceFarmer/stf.git
synced 2026-04-20 12:55:26 +02:00
Support installation from URLs.
This commit is contained in:
@@ -1,12 +1,16 @@
|
||||
var http = require('http')
|
||||
var util = require('util')
|
||||
var fs = require('fs')
|
||||
|
||||
var express = require('express')
|
||||
var formidable = require('formidable')
|
||||
var Promise = require('bluebird')
|
||||
var ApkReader = require('adbkit-apkreader')
|
||||
var request = require('request')
|
||||
var temp = require('temp')
|
||||
|
||||
var logger = require('../../util/logger')
|
||||
var requtil = require('../../util/requtil')
|
||||
var Storage = require('../../util/storage')
|
||||
|
||||
module.exports = function(options) {
|
||||
@@ -19,46 +23,86 @@ module.exports = function(options) {
|
||||
app.set('case sensitive routing', true)
|
||||
app.set('trust proxy', true)
|
||||
|
||||
app.use(express.json())
|
||||
app.use(express.urlencoded())
|
||||
|
||||
storage.on('timeout', function(id) {
|
||||
log.info('Cleaning up inactive resource "%s"', id)
|
||||
})
|
||||
|
||||
function process(file) {
|
||||
log.info('Processing "%s"', file.path)
|
||||
|
||||
var reader = ApkReader.readFile(file.path)
|
||||
var manifest = reader.readManifestSync()
|
||||
var id = storage.store(file)
|
||||
|
||||
return {
|
||||
url: util.format(
|
||||
'http://%s:%s/api/v1/resources/%s'
|
||||
, options.publicIp
|
||||
, options.port
|
||||
, id
|
||||
)
|
||||
, manifest: manifest
|
||||
}
|
||||
}
|
||||
|
||||
function download(url) {
|
||||
var resolver = Promise.defer()
|
||||
var path = temp.path({
|
||||
dir: options.saveDir
|
||||
})
|
||||
|
||||
log.info('Downloading "%s" to "%s"', url, path)
|
||||
|
||||
function errorListener(err) {
|
||||
resolver.reject(err)
|
||||
}
|
||||
|
||||
function closeListener() {
|
||||
resolver.resolve({
|
||||
path: path
|
||||
})
|
||||
}
|
||||
|
||||
try {
|
||||
var dl = request(url)
|
||||
.pipe(fs.createWriteStream(path))
|
||||
.on('error', errorListener)
|
||||
.on('close', closeListener)
|
||||
}
|
||||
catch (err) {
|
||||
resolver.reject(err)
|
||||
}
|
||||
|
||||
return resolver.promise.finally(function() {
|
||||
dl.removeListener('error', errorListener)
|
||||
dl.removeListener('end', closeListener)
|
||||
})
|
||||
}
|
||||
|
||||
app.post('/api/v1/resources', function(req, res) {
|
||||
var form = Promise.promisifyAll(new formidable.IncomingForm())
|
||||
form.parseAsync(req)
|
||||
.spread(function(fields, files) {
|
||||
if (files.file) {
|
||||
try {
|
||||
var reader = ApkReader.readFile(files.file.path)
|
||||
var manifest = reader.readManifestSync()
|
||||
var id = storage.store(files.file)
|
||||
res.json(201, {
|
||||
success: true
|
||||
, url: util.format(
|
||||
'http://%s:%s/api/v1/resources/%s'
|
||||
, options.publicIp
|
||||
, options.port
|
||||
, id
|
||||
)
|
||||
, manifest: manifest
|
||||
})
|
||||
}
|
||||
catch (err) {
|
||||
log.error('ApkReader had an error', err.stack)
|
||||
res.json(500, {
|
||||
success: false
|
||||
})
|
||||
}
|
||||
return process(files.file)
|
||||
}
|
||||
else if (fields.url) {
|
||||
return download(fields.url).then(process)
|
||||
}
|
||||
else {
|
||||
res.json(400, {
|
||||
success: false
|
||||
})
|
||||
throw new requtil.ValidationError('"file" or "url" is required')
|
||||
}
|
||||
})
|
||||
.then(function(data) {
|
||||
data.success = true
|
||||
res.json(201, data)
|
||||
})
|
||||
.catch(requtil.ValidationError, function() {
|
||||
res.json(400, {
|
||||
success: false
|
||||
, error: 'ValidationError'
|
||||
})
|
||||
})
|
||||
.catch(function(err) {
|
||||
log.error('Failed to save resource: ', err.stack)
|
||||
res.json(500, {
|
||||
|
||||
Reference in New Issue
Block a user