compose-scalar.js 3.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. import { SCALAR, isScalar } from '../nodes/identity.js';
  2. import { Scalar } from '../nodes/Scalar.js';
  3. import { resolveBlockScalar } from './resolve-block-scalar.js';
  4. import { resolveFlowScalar } from './resolve-flow-scalar.js';
  5. function composeScalar(ctx, token, tagToken, onError) {
  6. const { value, type, comment, range } = token.type === 'block-scalar'
  7. ? resolveBlockScalar(token, ctx.options.strict, onError)
  8. : resolveFlowScalar(token, ctx.options.strict, onError);
  9. const tagName = tagToken
  10. ? ctx.directives.tagName(tagToken.source, msg => onError(tagToken, 'TAG_RESOLVE_FAILED', msg))
  11. : null;
  12. const tag = tagToken && tagName
  13. ? findScalarTagByName(ctx.schema, value, tagName, tagToken, onError)
  14. : token.type === 'scalar'
  15. ? findScalarTagByTest(ctx, value, token, onError)
  16. : ctx.schema[SCALAR];
  17. let scalar;
  18. try {
  19. const res = tag.resolve(value, msg => onError(tagToken ?? token, 'TAG_RESOLVE_FAILED', msg), ctx.options);
  20. scalar = isScalar(res) ? res : new Scalar(res);
  21. }
  22. catch (error) {
  23. const msg = error instanceof Error ? error.message : String(error);
  24. onError(tagToken ?? token, 'TAG_RESOLVE_FAILED', msg);
  25. scalar = new Scalar(value);
  26. }
  27. scalar.range = range;
  28. scalar.source = value;
  29. if (type)
  30. scalar.type = type;
  31. if (tagName)
  32. scalar.tag = tagName;
  33. if (tag.format)
  34. scalar.format = tag.format;
  35. if (comment)
  36. scalar.comment = comment;
  37. return scalar;
  38. }
  39. function findScalarTagByName(schema, value, tagName, tagToken, onError) {
  40. if (tagName === '!')
  41. return schema[SCALAR]; // non-specific tag
  42. const matchWithTest = [];
  43. for (const tag of schema.tags) {
  44. if (!tag.collection && tag.tag === tagName) {
  45. if (tag.default && tag.test)
  46. matchWithTest.push(tag);
  47. else
  48. return tag;
  49. }
  50. }
  51. for (const tag of matchWithTest)
  52. if (tag.test?.test(value))
  53. return tag;
  54. const kt = schema.knownTags[tagName];
  55. if (kt && !kt.collection) {
  56. // Ensure that the known tag is available for stringifying,
  57. // but does not get used by default.
  58. schema.tags.push(Object.assign({}, kt, { default: false, test: undefined }));
  59. return kt;
  60. }
  61. onError(tagToken, 'TAG_RESOLVE_FAILED', `Unresolved tag: ${tagName}`, tagName !== 'tag:yaml.org,2002:str');
  62. return schema[SCALAR];
  63. }
  64. function findScalarTagByTest({ directives, schema }, value, token, onError) {
  65. const tag = schema.tags.find(tag => tag.default && tag.test?.test(value)) || schema[SCALAR];
  66. if (schema.compat) {
  67. const compat = schema.compat.find(tag => tag.default && tag.test?.test(value)) ??
  68. schema[SCALAR];
  69. if (tag.tag !== compat.tag) {
  70. const ts = directives.tagString(tag.tag);
  71. const cs = directives.tagString(compat.tag);
  72. const msg = `Value may be parsed as either ${ts} or ${cs}`;
  73. onError(token, 'TAG_RESOLVE_FAILED', msg, true);
  74. }
  75. }
  76. return tag;
  77. }
  78. export { composeScalar };