web-check/server.js

64 lines
2.0 KiB
JavaScript

const express = require('express');
const awsServerlessExpress = require('aws-serverless-express');
const fs = require('fs');
const path = require('path');
const app = express();
// Read the '/api' directory and import each lambda function
const handlers = {};
fs.readdirSync(path.join(__dirname, '/api')).forEach(file => {
const route = `/api/${file.split('.')[0]}`; // remove .js extension
const handler = require(path.join(__dirname, '/api', file)).handler;
handlers[route] = handler;
app.get(route, async (req, res) => {
try {
const result = await new Promise((resolve, reject) => {
const cb = (err, response) => err ? reject(err) : resolve(response);
const promise = handler(req, {}, cb);
if (promise && typeof promise.then === 'function') {
promise.then(resolve).catch(reject);
}
});
res.status(result.statusCode).json(JSON.parse(result.body));
} catch (err) {
res.status(500).json({ error: err.message });
}
});
});
app.get('/api', async (req, res) => {
const results = {};
const url = req.query.url;
const handlerPromises = Object.entries(handlers).map(async ([route, handler]) => {
try {
const result = await new Promise((resolve, reject) => {
const cb = (err, response) => err ? reject(err) : resolve(response);
const promise = handler({ query: { url } }, {}, cb);
if (promise && typeof promise.then === 'function') {
promise.then(resolve).catch(reject);
}
});
results[route.slice(5)] = JSON.parse(result.body); // remove '/api/' prefix
} catch (err) {
results[route.slice(5)] = { error: err.message };
}
});
await Promise.all(handlerPromises);
res.json(results);
});
// Create serverless express server
const server = awsServerlessExpress
.createServer(app)
.listen(3001, () => {
console.log('Listening on port 3001');
});
exports.handler = (event, context) => {
console.log(`EVENT: ${JSON.stringify(event)}`);
awsServerlessExpress.proxy(server, event, context);
};