-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.js
98 lines (88 loc) · 2.66 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
const express = require('express')
const yargs = require('yargs')
const queryCommand = require('./commands/queryCommand')
const { QUERY_ERROR_NAMES } = require('./lib/query/queryErrors')
const http = require('http')
const cors = require('cors')
const { FetchError } = require('node-fetch')
const app = express()
app.use(express.json())
app.use(cors())
const argv = yargs.option('port', {
demandOption: false,
describe: 'The port for the webserver to run on, otherwise the default is 8080',
type: 'number'
}).argv
const port = argv['port'] || 8080
const getToken = req => {
if (req.headers.authorization && req.headers.authorization.split(' ')[0] === 'Bearer') {
return req.headers.authorization.split(' ')[1]
} else {
return null
}
}
app.get('/coffee', (req, res) => {
res.status(418).send()
})
app.get('/tea', (req, res) => {
res.status(418).send()
})
app.post('/api/query', (req, res) => {
const params = req.body
const token = getToken(req)
const args = {
host: params.host,
model: params.modelId,
query: params.query,
token: token
}
res.setHeader('Content-Type', 'application/json')
queryCommand
.queryModel(args)
.then(output => {
console.log('Query execution successfully completed!')
res.status(200).send(output)
})
.catch(err => {
console.error(err)
if (
err.name === QUERY_ERROR_NAMES.ParseError ||
err.name === QUERY_ERROR_NAMES.SyntaxError ||
err.name === QUERY_ERROR_NAMES.UnknownPredicateError ||
err.name === QUERY_ERROR_NAMES.PropertyNotEnumerableError
) {
res.status(400).send({
status: 400,
error: err.name,
message: err.message
})
} else if (err instanceof FetchError) {
res.status(400).send({
status: 400,
error: 'Bad Request',
message: `A fetch error has occurred for domain ${argv['host']}`
})
} else if (err.message == 'Unauthorized') {
res.status(401).send({
status: 401,
error: 'Unauthorized',
message: `Request has been unauthorized, token may be invalid or missing`
})
} else if (err.message == 'Not Found') {
res.status(404).send({
status: 404,
error: 'Not Found',
message: `Model with id:${params.modelId} not found`
})
} else {
res.status(500).send({
status: 500,
error: 'Internal Server Error',
message: 'No message available'
})
}
})
})
http.createServer(app).listen(port)
console.log('Solidatus Query API Server started successfully!')
console.log(`Server is listening on port:${port}`)