wmantly/vpn-p2p
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

host model uses the object validation and passed better errors

Browse Source
This commit is contained in:
William Mantly 2019-12-20 20:25:00 -05:00
parent a3deb62871
commit b3c668adbf
Signed by: wmantly
GPG Key ID: E1EEC7650BA97160
1 changed files with 83 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
nodejs/models/hosts.js
View File

@ -2,25 +2,56 @@
const {promisify} = require('util');
const client = require('../redis');
const {processKeys, ObjectValidateError} = require('../utils/object_validate');
const hostKeysMap = {
'host': {isRequired: true, type: 'string', min: 3, max: 500},
'ip': {isRequired: true, type: 'string', min: 3, max: 500},
'updated': {default: function(){return (new Date).getTime()}, always: true},
'username': {isRequired: true, type: 'string', always: true},
'targetport': {isRequired: true, type: 'number', min:0, max:65535},
'forcessl': {isRequired: false, default: true, type: 'boolean'},
'targetssl': {isRequired: false, default: false, type: 'boolean'},
}
async function getInfo(data){
try{
let info = await client.HGETALL('host_' + data.host);
if(info){
info['host'] = data.host;
return info
}
}catch(error){
throw error
}
let error = new Error('HostNotFound');
error.message = 'Host does not exists';
error.status = 404;
throw error;
}
async function exists(host){
try{
await getInfo({host})
return true
}catch(error){
return false;
}
}
async function listAll(){
try{
let hosts = await client.SMEMBERS('hosts');
return hosts;
}catch(error){
return new Error(error);
throw error;
}
}
async function listAllDetail(){
let out = [];
@ -31,34 +62,65 @@ async function listAllDetail(){
return out
}
async function add(data, edit){
try{
data = processKeys(hostKeysMap, data, edit);
async function add(data){
if(data.host && await exists(data.host)){
let error = new Error('HostNameUsed');
error.message = 'Host already exists';
error.status = 409;
throw error;
}else if(data.host){
await client.SADD('hosts', data.host);
await client.HSET('host_' + data.host, 'ip', data.ip);
await client.HSET('host_' + data.host, 'updated', (new Date).getTime());
await client.HSET('host_' + data.host, 'username', data.username);
await client.HSET('host_' + data.host, 'targetPort', data.targetPort);
if(data.forceSSL !== undefined){
await client.HSET('host_' + data.host, 'forcessl', !!data.forceSSL);
}
if(data.targetSSL !== undefined){
await client.HSET('host_' + data.host, 'targetssl', !!data.targetSSL);
}
for(let key of Object.keys(data)){
await client.HSET('host_' + data.host, key, data[key]);
}
return true;
} catch(error){
throw error
}
}
async function edit(data, host){
try{
let hostData = await getInfo({host});
if(data.host && data.host !== host){
data = Object.assign({}, hostData, data);
if(await add('hosts', hostData)) await remove({host});
}else{
data = processKeys(hostKeysMap, data, true);
// console.log('host edit data', data);
for(let key of Object.keys(data)){
await client.HSET('host_' + host, key, data[key]);
}
}
} catch(error){
throw error;
}
}
async function remove(data){
try{
await getInfo(data);
await client.SREM('hosts', data.host);
let count = await client.DEL('host_' + data.host);
for(let key of Object.keys(hostKeysMap)){
await client.HDEL('host_' + data.host, key);
}
return count;
} catch(error) {
return new Error(error);
throw error;
}
}
module.exports = {getInfo, listAll, listAllDetail, add, remove};
module.exports = {getInfo, listAll, listAllDetail, add, edit, remove};