55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
|
import type {CodeKeywordDefinition, ErrorObject, KeywordErrorDefinition} from "../../types"
|
||
|
|
import type {KeywordCxt} from "../../compile/validate"
|
||
|
|
import {_, or, Name, Code} from "../../compile/codegen"
|
||
|
|
import {useFunc} from "../../compile/util"
|
||
|
|
import equal from "../../runtime/equal"
|
||
|
|
|
||
|
|
export type EnumError = ErrorObject<"enum", {allowedValues: any[]}, any[] | {$data: string}>
|
||
|
|
|
||
|
|
const error: KeywordErrorDefinition = {
|
||
|
|
message: "must be equal to one of the allowed values",
|
||
|
|
params: ({schemaCode}) => _`{allowedValues: ${schemaCode}}`,
|
||
|
|
}
|
||
|
|
|
||
|
|
const def: CodeKeywordDefinition = {
|
||
|
|
keyword: "enum",
|
||
|
|
schemaType: "array",
|
||
|
|
$data: true,
|
||
|
|
error,
|
||
|
|
code(cxt: KeywordCxt) {
|
||
|
|
const {gen, data, $data, schema, schemaCode, it} = cxt
|
||
|
|
if (!$data && schema.length === 0) throw new Error("enum must have non-empty array")
|
||
|
|
const useLoop = schema.length >= it.opts.loopEnum
|
||
|
|
let eql: Name | undefined
|
||
|
|
const getEql = (): Name => (eql ??= useFunc(gen, equal))
|
||
|
|
|
||
|
|
let valid: Code
|
||
|
|
if (useLoop || $data) {
|
||
|
|
valid = gen.let("valid")
|
||
|
|
cxt.block$data(valid, loopEnum)
|
||
|
|
} else {
|
||
|
|
/* istanbul ignore if */
|
||
|
|
if (!Array.isArray(schema)) throw new Error("ajv implementation error")
|
||
|
|
const vSchema = gen.const("vSchema", schemaCode)
|
||
|
|
valid = or(...schema.map((_x: unknown, i: number) => equalCode(vSchema, i)))
|
||
|
|
}
|
||
|
|
cxt.pass(valid)
|
||
|
|
|
||
|
|
function loopEnum(): void {
|
||
|
|
gen.assign(valid, false)
|
||
|
|
gen.forOf("v", schemaCode as Code, (v) =>
|
||
|
|
gen.if(_`${getEql()}(${data}, ${v})`, () => gen.assign(valid, true).break())
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
function equalCode(vSchema: Name, i: number): Code {
|
||
|
|
const sch = schema[i]
|
||
|
|
return typeof sch === "object" && sch !== null
|
||
|
|
? _`${getEql()}(${data}, ${vSchema}[${i}])`
|
||
|
|
: _`${data} === ${sch}`
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
export default def
|