mirror of https://github.com/jkjoy/sunpeiwen.git
89 lines
3.6 KiB
TypeScript
89 lines
3.6 KiB
TypeScript
import type AjvCore from "../core"
|
|
import type {AnyValidateFunction, SourceCode} from "../types"
|
|
import type {SchemaEnv} from "../compile"
|
|
import {UsedScopeValues, UsedValueState, ValueScopeName, varKinds} from "../compile/codegen/scope"
|
|
import {_, nil, _Code, Code, getProperty} from "../compile/codegen/code"
|
|
|
|
export default function standaloneCode(
|
|
ajv: AjvCore,
|
|
refsOrFunc?: {[K in string]?: string} | AnyValidateFunction
|
|
): string {
|
|
if (!ajv.opts.code.source) {
|
|
throw new Error("moduleCode: ajv instance must have code.source option")
|
|
}
|
|
const {_n} = ajv.scope.opts
|
|
return typeof refsOrFunc == "function"
|
|
? funcExportCode(refsOrFunc.source)
|
|
: refsOrFunc !== undefined
|
|
? multiExportsCode<string>(refsOrFunc, getValidate)
|
|
: multiExportsCode<SchemaEnv>(ajv.schemas, (sch) =>
|
|
sch.meta ? undefined : ajv.compile(sch.schema)
|
|
)
|
|
|
|
function getValidate(id: string): AnyValidateFunction {
|
|
const v = ajv.getSchema(id)
|
|
if (!v) throw new Error(`moduleCode: no schema with id ${id}`)
|
|
return v
|
|
}
|
|
|
|
function funcExportCode(source?: SourceCode): string {
|
|
const usedValues: UsedScopeValues = {}
|
|
const n = source?.validateName
|
|
const vCode = validateCode(usedValues, source)
|
|
return `"use strict";${_n}module.exports = ${n};${_n}module.exports.default = ${n};${_n}${vCode}`
|
|
}
|
|
|
|
function multiExportsCode<T extends SchemaEnv | string>(
|
|
schemas: {[K in string]?: T},
|
|
getValidateFunc: (schOrId: T) => AnyValidateFunction | undefined
|
|
): string {
|
|
const usedValues: UsedScopeValues = {}
|
|
let code = _`"use strict";`
|
|
for (const name in schemas) {
|
|
const v = getValidateFunc(schemas[name] as T)
|
|
if (v) {
|
|
const vCode = validateCode(usedValues, v.source)
|
|
code = _`${code}${_n}exports${getProperty(name)} = ${v.source?.validateName};${_n}${vCode}`
|
|
}
|
|
}
|
|
return `${code}`
|
|
}
|
|
|
|
function validateCode(usedValues: UsedScopeValues, s?: SourceCode): Code {
|
|
if (!s) throw new Error('moduleCode: function does not have "source" property')
|
|
if (usedState(s.validateName) === UsedValueState.Completed) return nil
|
|
setUsedState(s.validateName, UsedValueState.Started)
|
|
|
|
const scopeCode = ajv.scope.scopeCode(s.scopeValues, usedValues, refValidateCode)
|
|
const code = new _Code(`${scopeCode}${_n}${s.validateCode}`)
|
|
return s.evaluated ? _`${code}${s.validateName}.evaluated = ${s.evaluated};${_n}` : code
|
|
|
|
function refValidateCode(n: ValueScopeName): Code | undefined {
|
|
const vRef = n.value?.ref
|
|
if (n.prefix === "validate" && typeof vRef == "function") {
|
|
const v = vRef as AnyValidateFunction
|
|
return validateCode(usedValues, v.source)
|
|
} else if ((n.prefix === "root" || n.prefix === "wrapper") && typeof vRef == "object") {
|
|
const {validate, validateName} = vRef as SchemaEnv
|
|
if (!validateName) throw new Error("ajv internal error")
|
|
const def = ajv.opts.code.es5 ? varKinds.var : varKinds.const
|
|
const wrapper = _`${def} ${n} = {validate: ${validateName}};`
|
|
if (usedState(validateName) === UsedValueState.Started) return wrapper
|
|
const vCode = validateCode(usedValues, validate?.source)
|
|
return _`${wrapper}${_n}${vCode}`
|
|
}
|
|
return undefined
|
|
}
|
|
|
|
function usedState(name: ValueScopeName): UsedValueState | undefined {
|
|
return usedValues[name.prefix]?.get(name)
|
|
}
|
|
|
|
function setUsedState(name: ValueScopeName, state: UsedValueState): void {
|
|
const {prefix} = name
|
|
const names = (usedValues[prefix] = usedValues[prefix] || new Map())
|
|
names.set(name, state)
|
|
}
|
|
}
|
|
}
|