uniqueItems.ts 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. import type {CodeKeywordDefinition, ErrorObject, KeywordErrorDefinition} from "../../types"
  2. import type {KeywordCxt} from "../../compile/validate"
  3. import {checkDataTypes, getSchemaTypes, DataType} from "../../compile/validate/dataType"
  4. import {_, str, Name} from "../../compile/codegen"
  5. import {useFunc} from "../../compile/util"
  6. import equal from "../../runtime/equal"
  7. export type UniqueItemsError = ErrorObject<
  8. "uniqueItems",
  9. {i: number; j: number},
  10. boolean | {$data: string}
  11. >
  12. const error: KeywordErrorDefinition = {
  13. message: ({params: {i, j}}) =>
  14. str`must NOT have duplicate items (items ## ${j} and ${i} are identical)`,
  15. params: ({params: {i, j}}) => _`{i: ${i}, j: ${j}}`,
  16. }
  17. const def: CodeKeywordDefinition = {
  18. keyword: "uniqueItems",
  19. type: "array",
  20. schemaType: "boolean",
  21. $data: true,
  22. error,
  23. code(cxt: KeywordCxt) {
  24. const {gen, data, $data, schema, parentSchema, schemaCode, it} = cxt
  25. if (!$data && !schema) return
  26. const valid = gen.let("valid")
  27. const itemTypes = parentSchema.items ? getSchemaTypes(parentSchema.items) : []
  28. cxt.block$data(valid, validateUniqueItems, _`${schemaCode} === false`)
  29. cxt.ok(valid)
  30. function validateUniqueItems(): void {
  31. const i = gen.let("i", _`${data}.length`)
  32. const j = gen.let("j")
  33. cxt.setParams({i, j})
  34. gen.assign(valid, true)
  35. gen.if(_`${i} > 1`, () => (canOptimize() ? loopN : loopN2)(i, j))
  36. }
  37. function canOptimize(): boolean {
  38. return itemTypes.length > 0 && !itemTypes.some((t) => t === "object" || t === "array")
  39. }
  40. function loopN(i: Name, j: Name): void {
  41. const item = gen.name("item")
  42. const wrongType = checkDataTypes(itemTypes, item, it.opts.strictNumbers, DataType.Wrong)
  43. const indices = gen.const("indices", _`{}`)
  44. gen.for(_`;${i}--;`, () => {
  45. gen.let(item, _`${data}[${i}]`)
  46. gen.if(wrongType, _`continue`)
  47. if (itemTypes.length > 1) gen.if(_`typeof ${item} == "string"`, _`${item} += "_"`)
  48. gen
  49. .if(_`typeof ${indices}[${item}] == "number"`, () => {
  50. gen.assign(j, _`${indices}[${item}]`)
  51. cxt.error()
  52. gen.assign(valid, false).break()
  53. })
  54. .code(_`${indices}[${item}] = ${i}`)
  55. })
  56. }
  57. function loopN2(i: Name, j: Name): void {
  58. const eql = useFunc(gen, equal)
  59. const outer = gen.name("outer")
  60. gen.label(outer).for(_`;${i}--;`, () =>
  61. gen.for(_`${j} = ${i}; ${j}--;`, () =>
  62. gen.if(_`${eql}(${data}[${i}], ${data}[${j}])`, () => {
  63. cxt.error()
  64. gen.assign(valid, false).break(outer)
  65. })
  66. )
  67. )
  68. }
  69. },
  70. }
  71. export default def