mirror of https://gitee.com/bigwinds/arangodb
131 lines
3.2 KiB
JavaScript
131 lines
3.2 KiB
JavaScript
'use strict';
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2016 ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// 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.
|
|
///
|
|
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// @author Alan Plum
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
const _ = require('lodash');
|
|
const assert = require('assert');
|
|
const typeIs = require('type-is');
|
|
const mediaTyper = require('media-typer');
|
|
const requestParts = require('internal').requestParts;
|
|
|
|
|
|
exports.validateParams = function validateParams(typeDefs, rawParams) {
|
|
const params = {};
|
|
for (const entry of typeDefs) {
|
|
const name = entry[0];
|
|
const def = entry[1];
|
|
if (def.schema.isJoi) {
|
|
const result = def.schema.validate(rawParams[name]);
|
|
if (result.error) {
|
|
throw result.error;
|
|
}
|
|
params[name] = result.value;
|
|
}
|
|
}
|
|
return params;
|
|
};
|
|
|
|
|
|
exports.parseRequestBody = function parseRequestBody(def, req) {
|
|
let body = req.body;
|
|
|
|
if (!def.contentTypes) {
|
|
assert(!body || !body.length, 'Unexpected request body');
|
|
return null;
|
|
}
|
|
|
|
const indicatedType = req.get('content-type');
|
|
let actualType;
|
|
|
|
if (indicatedType) {
|
|
for (const candidate of def.contentTypes) {
|
|
if (typeIs.is(indicatedType, candidate)) {
|
|
actualType = candidate;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!actualType) {
|
|
actualType = def.contentTypes[0] || indicatedType;
|
|
}
|
|
|
|
const parsedType = mediaTyper.parse(actualType);
|
|
actualType = mediaTyper.format(_.pick(parsedType, ['type', 'subtype', 'suffix']));
|
|
|
|
if (parsedType.type === 'multipart') {
|
|
body = requestParts(req._raw);
|
|
}
|
|
|
|
let handler;
|
|
for (const entry of req.context.service.types.entries()) {
|
|
const key = entry[0];
|
|
const value = entry[1];
|
|
let match;
|
|
if (key instanceof RegExp) {
|
|
match = actualType.test(key);
|
|
} else if (typeof key === 'function') {
|
|
match = key(actualType);
|
|
} else {
|
|
match = typeIs.is(key, actualType);
|
|
}
|
|
if (match && value.fromClient) {
|
|
handler = value;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (handler) {
|
|
body = handler.fromClient(body, req, parsedType);
|
|
}
|
|
|
|
return body;
|
|
};
|
|
|
|
|
|
exports.validateRequestBody = function validateRequestBody(def, req) {
|
|
let body = req.body;
|
|
|
|
const schema = def.model && (def.model.schema || def.model);
|
|
if (!schema) {
|
|
return body;
|
|
}
|
|
|
|
if (schema.isJoi) {
|
|
const result = schema.validate(body);
|
|
|
|
if (result.error) {
|
|
result.error.message = result.error.message.replace(/^"value"/, '"request body"');
|
|
throw result.error;
|
|
}
|
|
|
|
body = result.value;
|
|
}
|
|
|
|
if (def.model && def.model.fromClient) {
|
|
body = def.model.fromClient(body);
|
|
}
|
|
|
|
return body;
|
|
};
|