Spaces:
Running
Running
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 | |