1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-04 10:19:30 +02:00

Add an app for resizing images. Still needs rate limiting, and still trying to decide how to pass the correct URL to the app.

This commit is contained in:
Simo Kinnunen 2014-05-20 19:00:53 +09:00
parent 9e4dc269a2
commit e56d757cde
7 changed files with 158 additions and 1 deletions

View file

@ -500,6 +500,32 @@ program
}) })
}) })
program
.command('storage-plugin-image')
.description('start storage image plugin')
.option('-p, --port <port>'
, 'port (or $PORT)'
, Number
, process.env.PORT || 7100)
.option('-r, --storage-url <url>'
, 'URL to storage client'
, String)
.option('--cache-dir <dir>'
, 'where to cache images'
, String
, os.tmpdir())
.action(function(options) {
if (!options.storageUrl) {
this.missingArgument('--storage-url')
}
require('./roles/storage/plugins/image')({
port: options.port
, storageUrl: options.storageUrl
, cacheDir: options.cacheDir
})
})
program program
.command('migrate') .command('migrate')
.description('migrates the database to the latest version') .description('migrates the database to the latest version')

View file

@ -0,0 +1,47 @@
var http = require('http')
var util = require('util')
var express = require('express')
var Promise = require('bluebird')
var gm = require('gm')
var logger = require('../../../../util/logger')
var parseCrop = require('./param/crop')
var parseGravity = require('./param/gravity')
var get = require('./task/get')
var transform = require('./task/transform')
module.exports = function(options) {
var log = logger.createLogger('storage:plugins:image')
, app = express()
, server = http.createServer(app)
app.set('strict routing', true)
app.set('case sensitive routing', true)
app.set('trust proxy', true)
app.get('/api/v1/resources/:id/*', function(req, res) {
get(req.params.id, options)
.then(function(stream) {
return transform(stream, {
crop: parseCrop(req.query.crop)
, gravity: parseGravity(req.query.gravity)
})
})
.then(function(out) {
res.status(200)
out.pipe(res)
})
.catch(function(err) {
log.error('Unable to transform resource "%s"', req.params.id, err.stack)
res.status(500)
.json({
success: false
})
})
})
server.listen(options.port)
log.info('Listening on port %d', options.port)
}

View file

@ -0,0 +1,14 @@
var RE_CROP = /^([0-9]*)x([0-9]*)$/
module.exports = function(raw) {
var parsed
if (raw && (parsed = RE_CROP.exec(raw))) {
return {
width: +parsed[1] || 0
, height: +parsed[2] || 0
}
}
return null
}

View file

@ -0,0 +1,21 @@
var GRAVITY = {
northwest: 'NorthWest'
, north: 'North'
, northeast: 'NorthEast'
, west: 'West'
, center: 'Center'
, east: 'East'
, southwest: 'SouthWest'
, south: 'South'
, southeast: 'SouthEast'
}
module.exports = function(raw) {
var parsed
if (raw && (parsed = GRAVITY[raw])) {
return parsed
}
return null
}

View file

@ -0,0 +1,19 @@
var util = require('util')
var http = require('http')
var Promise = require('bluebird')
module.exports = function(id, options) {
return new Promise(function(resolve, reject) {
http.get(util.format('%sapi/v1/resources/%s', options.storageUrl, id))
.on('response', function(res) {
if (res.statusCode !== 200) {
reject(new Error(util.format('HTTP %d', res.statusCode)))
}
else {
resolve(res)
}
})
.on('error', reject)
})
}

View file

@ -0,0 +1,26 @@
var gm = require('gm')
var Promise = require('bluebird')
module.exports = function(stream, options) {
return new Promise(function(resolve, reject) {
var transform = gm(stream)
if (options.gravity) {
transform.gravity(options.gravity)
}
if (options.crop) {
transform.geometry(options.crop.width, options.crop.height, '^')
transform.crop(options.crop.width, options.crop.height, 0, 0)
}
transform.stream(function(err, stdout) {
if (err) {
reject(err)
}
else {
resolve(stdout)
}
})
})
}

View file

@ -14,14 +14,18 @@ util.inherits(Storage, events.EventEmitter)
Storage.prototype.store = function(file) { Storage.prototype.store = function(file) {
var id = uuid.v4() var id = uuid.v4()
this.set(id, file)
return id
}
Storage.prototype.set = function(id, file) {
this.files[id] = { this.files[id] = {
timeout: 600000 timeout: 600000
, lastActivity: Date.now() , lastActivity: Date.now()
, data: file , data: file
} }
return id return file
} }
Storage.prototype.remove = function(id) { Storage.prototype.remove = function(id) {