molstar
Version:
A comprehensive macromolecular library.
205 lines • 6.96 kB
JavaScript
/**
* Copyright (c) 2019-2020 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author David Sehnal <david.sehnal@gmail.com>
*/
import express from 'express';
import compression from 'compression';
import cors from 'cors';
import * as bodyParser from 'body-parser';
import * as fs from 'fs';
import * as path from 'path';
import { swaggerUiIndexHandler, swaggerUiAssetsHandler } from '../common/swagger-ui';
import { makeDir } from '../../mol-util/make-dir';
import { getConfig } from './config';
import { UUID } from '../../mol-util';
import { shortcutIconLink, getSchema } from './api-schema';
var Config = getConfig();
var app = express();
app.use(compression({ level: 6, memLevel: 9, chunkSize: 16 * 16384, filter: function () { return true; } }));
app.use(cors({ methods: ['GET', 'PUT'] }));
app.use(bodyParser.json({ limit: '20mb' }));
function createIndex() {
var fn = path.join(Config.working_folder, 'index.json');
if (fs.existsSync(fn))
return;
if (!fs.existsSync(Config.working_folder))
makeDir(Config.working_folder);
fs.writeFileSync(fn, '[]', 'utf-8');
}
function writeIndex(index) {
var fn = path.join(Config.working_folder, 'index.json');
if (!fs.existsSync(Config.working_folder))
makeDir(Config.working_folder);
fs.writeFileSync(fn, JSON.stringify(index, null, 2), 'utf-8');
}
function readIndex() {
var fn = path.join(Config.working_folder, 'index.json');
if (!fs.existsSync(fn))
return [];
return JSON.parse(fs.readFileSync(fn, 'utf-8'));
}
function validateIndex(index) {
if (index.length > Config.max_states) {
var deletes = [], newIndex = [];
var toDelete = index.length - Config.max_states;
for (var _i = 0, index_1 = index; _i < index_1.length; _i++) {
var e = index_1[_i];
if (!e.isSticky && deletes.length < toDelete) {
deletes.push(e);
}
else {
newIndex.push(e);
}
}
for (var _a = 0, deletes_1 = deletes; _a < deletes_1.length; _a++) {
var d = deletes_1[_a];
try {
fs.unlinkSync(path.join(Config.working_folder, d.id + '.json'));
}
catch (_b) { }
}
return newIndex;
}
return index;
}
function remove(id) {
var index = readIndex();
var i = 0;
for (var _i = 0, index_2 = index; _i < index_2.length; _i++) {
var e = index_2[_i];
if (e.id !== id) {
i++;
continue;
}
if (e.isSticky)
return;
try {
for (var j = i + 1; j < index.length; j++) {
index[j - 1] = index[j];
}
index.pop();
writeIndex(index);
}
catch (_a) { }
try {
fs.unlinkSync(path.join(Config.working_folder, e.id + '.json'));
}
catch (_b) { }
return;
}
}
// function clear() {
// let index = readIndex();
// for (const e of index) {
// try {
// fs.unlinkSync(path.join(Config.working_folder, e.id + '.json'));
// } catch { }
// }
// writeIndex([]);
// }
function mapPath(path) {
if (!Config.api_prefix)
return path;
return "/" + Config.api_prefix + "/" + path;
}
app.get(mapPath("/get/:id"), function (req, res) {
var id = req.params.id || '';
console.log('Reading', id);
if (id.length === 0 || id.indexOf('.') >= 0 || id.indexOf('/') >= 0 || id.indexOf('\\') >= 0) {
res.status(404);
res.end();
return;
}
fs.readFile(path.join(Config.working_folder, id + '.json'), 'utf-8', function (err, data) {
if (err) {
res.status(404);
res.end();
return;
}
res.writeHead(200, {
'Content-Type': 'application/json; charset=utf-8',
});
res.write(data);
res.end();
});
});
// app.get(mapPath(`/clear`), (req, res) => {
// clear();
// res.status(200);
// res.end();
// });
app.get(mapPath("/remove/:id"), function (req, res) {
remove((req.params.id || '').toLowerCase());
res.status(200);
res.end();
});
// app.get(mapPath(`/latest`), (req, res) => {
// const index = readIndex();
// const id: string = index.length > 0 ? index[index.length - 1].id : '';
// console.log('Reading', id);
// if (id.length === 0 || id.indexOf('.') >= 0 || id.indexOf('/') >= 0 || id.indexOf('\\') >= 0) {
// res.status(404);
// res.end();
// return;
// }
// fs.readFile(path.join(Config.working_folder, id + '.json'), 'utf-8', (err, data) => {
// if (err) {
// res.status(404);
// res.end();
// return;
// }
// res.writeHead(200, {
// 'Content-Type': 'application/json; charset=utf-8',
// });
// res.write(data);
// res.end();
// });
// });
app.get(mapPath("/list"), function (req, res) {
var index = readIndex();
res.writeHead(200, {
'Content-Type': 'application/json; charset=utf-8',
});
res.write(JSON.stringify(index, null, 2));
res.end();
});
app.post(mapPath("/set"), function (req, res) {
console.log('SET', req.query.name, req.query.description);
var index = readIndex();
validateIndex(index);
var name = (req.query.name || new Date().toUTCString()).substr(0, 50);
var description = (req.query.description || '').substr(0, 100);
index.push({ timestamp: +new Date(), id: UUID.createv4(), name: name, description: description });
var entry = index[index.length - 1];
var data = JSON.stringify({
id: entry.id,
name: name,
description: description,
data: req.body
});
fs.writeFile(path.join(Config.working_folder, entry.id + '.json'), data, { encoding: 'utf8' }, function () { return res.end(); });
writeIndex(index);
});
var schema = getSchema(Config);
app.get(mapPath('/openapi.json'), function (req, res) {
res.writeHead(200, {
'Content-Type': 'application/json; charset=utf-8',
'Access-Control-Allow-Origin': '*',
'Access-Control-Allow-Headers': 'X-Requested-With'
});
res.end(JSON.stringify(schema));
});
app.use(mapPath('/'), swaggerUiAssetsHandler());
app.get(mapPath('/'), swaggerUiIndexHandler({
openapiJsonUrl: mapPath('/openapi.json'),
apiPrefix: Config.api_prefix,
title: 'PluginState Server API',
shortcutIconLink: shortcutIconLink
}));
createIndex();
app.listen(Config.port);
console.log("Mol* PluginState Server");
console.log('');
console.log(JSON.stringify(Config, null, 2));
//# sourceMappingURL=index.js.map