mirror of
https://github.com/balena-io/balena-cli.git
synced 2024-12-23 15:32:22 +00:00
102 lines
2.7 KiB
CoffeeScript
102 lines
2.7 KiB
CoffeeScript
###
|
|
Copyright 2016 Balena
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
###
|
|
|
|
express = require('express')
|
|
path = require('path')
|
|
bodyParser = require('body-parser')
|
|
Promise = require('bluebird')
|
|
balena = require('balena-sdk').fromSharedOptions()
|
|
utils = require('./utils')
|
|
|
|
createServer = ({ port, isDev } = {}) ->
|
|
app = express()
|
|
app.use bodyParser.urlencoded
|
|
extended: true
|
|
|
|
app.set('view engine', 'ejs')
|
|
app.set('views', path.join(__dirname, 'pages'))
|
|
|
|
if isDev
|
|
app.use(express.static(path.join(__dirname, 'pages', 'static')))
|
|
|
|
server = app.listen(port)
|
|
|
|
return { app, server }
|
|
|
|
###*
|
|
# @summary Await for token
|
|
# @function
|
|
# @protected
|
|
#
|
|
# @param {Object} options - options
|
|
# @param {String} options.path - callback path
|
|
# @param {Number} options.port - http port
|
|
#
|
|
# @example
|
|
# server.awaitForToken
|
|
# path: '/auth'
|
|
# port: 9001
|
|
# .then (token) ->
|
|
# console.log(token)
|
|
###
|
|
exports.awaitForToken = (options) ->
|
|
{ app, server } = createServer(port: options.port)
|
|
|
|
return new Promise (resolve, reject) ->
|
|
closeServer = (errorMessage, successPayload) ->
|
|
server.close ->
|
|
if errorMessage
|
|
reject(new Error(errorMessage))
|
|
return
|
|
|
|
resolve(successPayload)
|
|
|
|
renderAndDone = ({ request, response, viewName, errorMessage, statusCode, token }) ->
|
|
return getContext(viewName)
|
|
.then (context) ->
|
|
response.status(statusCode || 200).render(viewName, context)
|
|
request.connection.destroy()
|
|
closeServer(errorMessage, token)
|
|
|
|
app.post options.path, (request, response) ->
|
|
token = request.body.token?.trim()
|
|
|
|
Promise.try ->
|
|
if not token
|
|
throw new Error('No token')
|
|
return utils.loginIfTokenValid(token)
|
|
.tap (loggedIn) ->
|
|
if not loggedIn
|
|
throw new Error('Invalid token')
|
|
.then ->
|
|
renderAndDone({ request, response, viewName: 'success', token })
|
|
.catch (error) ->
|
|
renderAndDone({
|
|
request, response, viewName: 'error',
|
|
statusCode: 401, errorMessage: error.message
|
|
})
|
|
|
|
app.use (request, response) ->
|
|
response.status(404).send('Not found')
|
|
closeServer('Unknown path or verb')
|
|
|
|
exports.getContext = getContext = (viewName) ->
|
|
if viewName is 'success'
|
|
return Promise.props
|
|
dashboardUrl: balena.settings.get('dashboardUrl')
|
|
|
|
return Promise.resolve({})
|