/
index.ts
97 lines (89 loc) 路 2.41 KB
/
index.ts
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
import * as Ajv from 'ajv';
import * as draftSchema from 'ajv/lib/refs/json-schema-draft-04.json';
import { formats } from './formats';
import { OpenAPIV3 } from '../types';
import ajv = require('ajv');
export function createRequestAjv(
openApiSpec: OpenAPIV3.Document,
options: ajv.Options = {},
): Ajv.Ajv {
return createAjv(openApiSpec, options);
}
export function createResponseAjv(
openApiSpec: OpenAPIV3.Document,
options: ajv.Options = {},
): Ajv.Ajv {
return createAjv(openApiSpec, options, false);
}
function createAjv(
openApiSpec: OpenAPIV3.Document,
options: ajv.Options = {},
request = true,
): Ajv.Ajv {
const ajv = new Ajv({
...options,
schemaId: 'auto',
allErrors: true,
meta: draftSchema,
formats: { ...formats, ...options.formats },
unknownFormats: options.unknownFormats,
});
ajv.removeKeyword('propertyNames');
ajv.removeKeyword('contains');
ajv.removeKeyword('const');
if (request) {
ajv.removeKeyword('readOnly');
ajv.addKeyword('readOnly', {
modifying: true,
compile: sch => {
if (sch) {
return function validate(data, path, obj, propName) {
const isValid = !(sch === true && data != null);
delete obj[propName];
(<any>validate).errors = [
{
keyword: 'readOnly',
dataPath: path,
message: `is read-only`,
params: { readOnly: propName },
},
];
return isValid;
};
}
return () => true;
},
});
} else {
// response
ajv.removeKeyword('writeOnly');
ajv.addKeyword('writeOnly', {
modifying: true,
compile: sch => {
if (sch) {
return function validate(data, path, obj, propName) {
const isValid = !(sch === true && data != null);
(<any>validate).errors = [
{
keyword: 'writeOnly',
dataPath: path,
message: `is write-only`,
params: { writeOnly: propName },
},
];
return isValid;
};
}
return () => true;
},
});
}
if (openApiSpec.components?.schemas) {
Object.entries(openApiSpec.components.schemas).forEach(
([id, schema]: any[]) => {
ajv.addSchema(schema, `#/components/schemas/${id}`);
},
);
}
return ajv;
}