yaml 1.3.2 → 1.6.0

package/dist/Document.js

@@ -9,7 +9,7 @@ var _addComment = _interopRequireDefault(require("./addComment"));
 
 var _Anchors = _interopRequireDefault(require("./Anchors"));
 
-var _Node = require("./cst/Node");
+var _constants = require("./constants");
 
 var _errors = require("./errors");
 
@@ -21,7 +21,7 @@ var _Alias = _interopRequireDefault(require("./schema/Alias"));
 
 var _Collection = _interopRequireWildcard(require("./schema/Collection"));
 
-var _Node2 = _interopRequireDefault(require("./schema/Node"));
+var _Node = _interopRequireDefault(require("./schema/Node"));
 
 var _Scalar = _interopRequireDefault(require("./schema/Scalar"));
 
@@ -33,14 +33,15 @@ function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { de
 
 function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
 
-const isCollectionItem = node => node && [_Node.Type.MAP_KEY, _Node.Type.MAP_VALUE, _Node.Type.SEQ_ITEM].includes(node.type);
+const isCollectionItem = node => node && [_constants.Type.MAP_KEY, _constants.Type.MAP_VALUE, _constants.Type.SEQ_ITEM].includes(node.type);
 
 class Document {
   constructor(options) {
-    this.anchors = new _Anchors.default();
+    this.anchors = new _Anchors.default(options.anchorPrefix);
     this.commentBefore = null;
     this.comment = null;
     this.contents = null;
+    this.directivesEndMarker = null;
     this.errors = [];
     this.options = options;
     this.schema = null;
@@ -118,12 +119,13 @@ class Document {
     if (!this.schema) this.schema = new _schema.default(Object.assign({}, this.getDefaults(), this.options));
   }
 
-  parse(node) {
+  parse(node, prevDoc) {
     if (this.options.keepCstNodes) this.cstNode = node;
     if (this.options.keepNodeTypes) this.type = 'DOCUMENT';
     const {
       directives = [],
       contents = [],
+      directivesEndMarker,
       error,
       valueRange
     } = node;
@@ -133,7 +135,26 @@ class Document {
       this.errors.push(error);
     }
 
+    this.parseDirectives(directives, prevDoc);
+    if (directivesEndMarker) this.directivesEndMarker = true;
+    this.range = valueRange ? [valueRange.start, valueRange.end] : null;
+    this.setSchema();
+    this.anchors._cstAliases = [];
+    this.parseContents(contents);
+    this.anchors.resolveNodes();
+
+    if (this.options.prettyErrors) {
+      for (const error of this.errors) if (error instanceof _errors.YAMLError) error.makePretty();
+
+      for (const warn of this.warnings) if (warn instanceof _errors.YAMLError) warn.makePretty();
+    }
+
+    return this;
+  }
+
+  parseDirectives(directives, prevDoc) {
     const directiveComments = [];
+    let hasDirectives = false;
     directives.forEach(directive => {
       const {
         comment,
@@ -143,11 +164,13 @@ class Document {
       switch (name) {
         case 'TAG':
           this.resolveTagDirective(directive);
+          hasDirectives = true;
           break;
 
         case 'YAML':
         case 'YAML:1.0':
           this.resolveYamlDirective(directive);
+          hasDirectives = true;
           break;
 
         default:
@@ -160,10 +183,24 @@ class Document {
 
       if (comment) directiveComments.push(comment);
     });
-    this.range = valueRange ? [valueRange.start, valueRange.end] : null;
-    this.setSchema();
-    this.anchors._cstAliases = [];
+
+    if (prevDoc && !hasDirectives && '1.1' === (this.version || prevDoc.version || this.options.version)) {
+      const copyTagPrefix = ({
+        handle,
+        prefix
+      }) => ({
+        handle,
+        prefix
+      });
+
+      this.tagPrefixes = prevDoc.tagPrefixes.map(copyTagPrefix);
+      this.version = prevDoc.version;
+    }
+
     this.commentBefore = directiveComments.join('\n') || null;
+  }
+
+  parseContents(contents) {
     const comments = {
       before: [],
       after: []
@@ -188,7 +225,7 @@ class Document {
       } else if (node.comment !== null) {
         const cc = contentNodes.length === 0 ? comments.before : comments.after;
         cc.push(node.comment);
-      } else if (node.type === _Node.Type.BLANK_LINE) {
+      } else if (node.type === _constants.Type.BLANK_LINE) {
         spaceBefore = true;
 
         if (contentNodes.length === 0 && comments.before.length > 0 && !this.commentBefore) {
@@ -233,8 +270,6 @@ class Document {
     }
 
     this.comment = comments.after.join('\n') || null;
-    this.anchors.resolveNodes();
-    return this;
   }
 
   resolveTagDirective(directive) {
@@ -331,21 +366,21 @@ class Document {
     }
 
     switch (type) {
-      case _Node.Type.BLOCK_FOLDED:
-      case _Node.Type.BLOCK_LITERAL:
-      case _Node.Type.QUOTE_DOUBLE:
-      case _Node.Type.QUOTE_SINGLE:
+      case _constants.Type.BLOCK_FOLDED:
+      case _constants.Type.BLOCK_LITERAL:
+      case _constants.Type.QUOTE_DOUBLE:
+      case _constants.Type.QUOTE_SINGLE:
         return _schema.default.defaultTags.STR;
 
-      case _Node.Type.FLOW_MAP:
-      case _Node.Type.MAP:
+      case _constants.Type.FLOW_MAP:
+      case _constants.Type.MAP:
         return _schema.default.defaultTags.MAP;
 
-      case _Node.Type.FLOW_SEQ:
-      case _Node.Type.SEQ:
+      case _constants.Type.FLOW_SEQ:
+      case _constants.Type.SEQ:
         return _schema.default.defaultTags.SEQ;
 
-      case _Node.Type.PLAIN:
+      case _constants.Type.PLAIN:
         return nonSpecific ? _schema.default.defaultTags.STR : null;
 
       default:
@@ -375,7 +410,7 @@ class Document {
       } = _ref;
 
       switch (node.context.src[start]) {
-        case _Node.Char.COMMENT:
+        case _constants.Char.COMMENT:
           {
             if (!node.commentHasRequiredWhitespace(start)) {
               const msg = 'Comments must be separated from other tokens by white space characters';
@@ -396,7 +431,7 @@ class Document {
           }
           break;
 
-        case _Node.Char.ANCHOR:
+        case _constants.Char.ANCHOR:
           if (hasAnchor) {
             const msg = 'A node can have at most one anchor';
             errors.push(new _errors.YAMLSemanticError(node, msg));
@@ -405,7 +440,7 @@ class Document {
           hasAnchor = true;
           break;
 
-        case _Node.Char.TAG:
+        case _constants.Char.TAG:
           if (hasTag) {
             const msg = 'A node can have at most one tag';
             errors.push(new _errors.YAMLSemanticError(node, msg));
@@ -430,7 +465,7 @@ class Document {
 
     let res;
 
-    if (node.type === _Node.Type.ALIAS) {
+    if (node.type === _constants.Type.ALIAS) {
       if (hasAnchor || hasTag) {
         const msg = 'An alias node must not specify any properties';
         errors.push(new _errors.YAMLSemanticError(node, msg));
@@ -449,18 +484,13 @@ class Document {
       res = new _Alias.default(src);
 
       anchors._cstAliases.push(res);
-
-      if (!src.resolved) {
-        const msg = 'Alias node contains a circular reference, which cannot be resolved as JSON';
-        this.warnings.push(new _errors.YAMLWarning(node, msg));
-      }
     } else {
       const tagName = this.resolveTagName(node);
 
       if (tagName) {
         res = schema.resolveNodeWithFallback(this, node, tagName);
       } else {
-        if (node.type !== _Node.Type.PLAIN) {
+        if (node.type !== _constants.Type.PLAIN) {
           const msg = `Failed to resolve ${node.type} node here`;
           errors.push(new _errors.YAMLSyntaxError(node, msg));
           return null;
@@ -539,21 +569,32 @@ class Document {
   }
 
   toJSON(arg) {
-    const cr = this.warnings.find(w => /circular reference/.test(w.message));
-    if (cr) throw new _errors.YAMLSemanticError(cr.source, cr.message);
-    const keep = this.options.keepBlobsInJSON && (typeof arg !== 'string' || !(this.contents instanceof _Scalar.default));
-    const mapAsMap = keep && !!this.options.mapAsMap;
-    return (0, _toJSON.default)(this.contents, arg, {
+    const {
+      keepBlobsInJSON,
+      mapAsMap,
+      maxAliasCount
+    } = this.options;
+    const keep = keepBlobsInJSON && (typeof arg !== 'string' || !(this.contents instanceof _Scalar.default));
+    const ctx = {
+      doc: this,
       keep,
-      mapAsMap
-    });
+      mapAsMap: keep && !!mapAsMap,
+      maxAliasCount
+    };
+    const anchorNames = Object.keys(this.anchors.map);
+    if (anchorNames.length > 0) ctx.anchors = anchorNames.map(name => ({
+      alias: [],
+      aliasCount: 0,
+      count: 1,
+      node: this.anchors.map[name]
+    }));
+    return (0, _toJSON.default)(this.contents, arg, ctx);
   }
 
   toString() {
     if (this.errors.length > 0) throw new Error('Document with errors cannot be stringified');
     this.setSchema();
     const lines = [];
-    if (this.commentBefore) lines.push(this.commentBefore.replace(/^/gm, '#'), '');
     let hasDirectives = false;
 
     if (this.version) {
@@ -577,7 +618,13 @@ class Document {
         hasDirectives = true;
       }
     });
-    if (hasDirectives) lines.push('---');
+    if (hasDirectives || this.directivesEndMarker) lines.push('---');
+
+    if (this.commentBefore) {
+      if (hasDirectives || !this.directivesEndMarker) lines.unshift('');
+      lines.unshift(this.commentBefore.replace(/^/gm, '#'));
+    }
+
     const ctx = {
       anchors: {},
       doc: this,
@@ -587,8 +634,8 @@ class Document {
     let contentComment = null;
 
     if (this.contents) {
-      if (this.contents instanceof _Node2.default) {
-        if (this.contents.spaceBefore && hasDirectives) lines.push('');
+      if (this.contents instanceof _Node.default) {
+        if (this.contents.spaceBefore && (hasDirectives || this.directivesEndMarker)) lines.push('');
         if (this.contents.commentBefore) lines.push(this.contents.commentBefore.replace(/^/gm, '#')); // top-level block scalars need to be indented if followed by a comment
 
         ctx.forceBlockIndent = !!this.comment;
@@ -648,7 +695,4 @@ _defineProperty(Document, "defaults", {
       prefix: _schema.default.defaultPrefix
     }]
   }
-});
-
-module.exports = exports.default;
-module.exports.default = exports.default;
+});

package/dist/constants.js

@@ -0,0 +1,34 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.Type = exports.Char = void 0;
+const Char = {
+  ANCHOR: '&',
+  COMMENT: '#',
+  TAG: '!',
+  DIRECTIVES_END: '-',
+  DOCUMENT_END: '.'
+};
+exports.Char = Char;
+const Type = {
+  ALIAS: 'ALIAS',
+  BLANK_LINE: 'BLANK_LINE',
+  BLOCK_FOLDED: 'BLOCK_FOLDED',
+  BLOCK_LITERAL: 'BLOCK_LITERAL',
+  COMMENT: 'COMMENT',
+  DIRECTIVE: 'DIRECTIVE',
+  DOCUMENT: 'DOCUMENT',
+  FLOW_MAP: 'FLOW_MAP',
+  FLOW_SEQ: 'FLOW_SEQ',
+  MAP: 'MAP',
+  MAP_KEY: 'MAP_KEY',
+  MAP_VALUE: 'MAP_VALUE',
+  PLAIN: 'PLAIN',
+  QUOTE_DOUBLE: 'QUOTE_DOUBLE',
+  QUOTE_SINGLE: 'QUOTE_SINGLE',
+  SEQ: 'SEQ',
+  SEQ_ITEM: 'SEQ_ITEM'
+};
+exports.Type = Type;

package/dist/cst/Alias.js

@@ -35,6 +35,4 @@ class Alias extends _Node.default {
 
 }
 
-exports.default = Alias;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = Alias;

package/dist/cst/BlankLine.js

@@ -5,17 +5,17 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = void 0;
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _constants = require("../constants");
+
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 class BlankLine extends _Node.default {
   constructor() {
-    super(_Node.Type.BLANK_LINE);
+    super(_constants.Type.BLANK_LINE);
   }
 
   get includesTrailingLines() {
@@ -49,6 +49,4 @@ class BlankLine extends _Node.default {
 
 }
 
-exports.default = BlankLine;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = BlankLine;

package/dist/cst/BlockValue.js

@@ -5,14 +5,14 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = exports.Chomp = void 0;
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _constants = require("../constants");
+
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 const Chomp = {
   CLIP: 'CLIP',
   KEEP: 'KEEP',
@@ -69,7 +69,7 @@ class BlockValue extends _Node.default {
     }
 
     const bi = indent + this.blockIndent;
-    const folded = this.type === _Node.Type.BLOCK_FOLDED;
+    const folded = this.type === _constants.Type.BLOCK_FOLDED;
     let atStart = true;
     let str = '';
     let sep = '';

package/dist/cst/Collection.js

@@ -6,18 +6,18 @@ Object.defineProperty(exports, "__esModule", {
 exports.grabCollectionEndComments = grabCollectionEndComments;
 exports.default = void 0;
 
+var _constants = require("../constants");
+
 var _BlankLine = _interopRequireDefault(require("./BlankLine"));
 
 var _CollectionItem = _interopRequireDefault(require("./CollectionItem"));
 
 var _Comment = _interopRequireDefault(require("./Comment"));
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
 function grabCollectionEndComments(node) {
@@ -32,7 +32,7 @@ function grabCollectionEndComments(node) {
   for (let i = len - 1; i >= 0; --i) {
     const n = cnode.items[i];
 
-    if (n.type === _Node.Type.COMMENT) {
+    if (n.type === _constants.Type.COMMENT) {
       // Keep sufficiently indented comments with preceding node
       const {
         indent,
@@ -40,7 +40,7 @@ function grabCollectionEndComments(node) {
       } = n.context;
       if (indent > 0 && n.range.start >= lineStart + indent) break;
       ci = i;
-    } else if (n.type === _Node.Type.BLANK_LINE) ci = i;else break;
+    } else if (n.type === _constants.Type.BLANK_LINE) ci = i;else break;
   }
 
   if (ci === -1) return null;
@@ -64,12 +64,12 @@ class Collection extends _Node.default {
     const ch = src[offset];
     if (!ch) return false;
     if (offset >= lineStart + indent) return true;
-    if (ch !== '#') return false;
+    if (ch !== '#' && ch !== '\n') return false;
     return Collection.nextContentHasIndent(src, offset, indent);
   }
 
   constructor(firstItem) {
-    super(firstItem.type === _Node.Type.SEQ_ITEM ? _Node.Type.SEQ : _Node.Type.MAP);
+    super(firstItem.type === _constants.Type.SEQ_ITEM ? _constants.Type.SEQ : _constants.Type.MAP);
 
     for (let i = firstItem.props.length - 1; i >= 0; --i) {
       if (firstItem.props[i].start < firstItem.context.lineStart) {
@@ -181,7 +181,7 @@ class Collection extends _Node.default {
         break;
       }
 
-      if (firstItem.type === _Node.Type.SEQ_ITEM !== (ch === '-')) {
+      if (firstItem.type === _constants.Type.SEQ_ITEM !== (ch === '-')) {
         let typeswitch = true;
 
         if (ch === '-') {

package/dist/cst/CollectionItem.js

@@ -5,16 +5,16 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = void 0;
 
+var _constants = require("../constants");
+
 var _errors = require("../errors");
 
 var _BlankLine = _interopRequireDefault(require("./BlankLine"));
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
 class CollectionItem extends _Node.default {
@@ -43,7 +43,7 @@ class CollectionItem extends _Node.default {
       atLineStart,
       lineStart
     } = context;
-    if (!atLineStart && this.type === _Node.Type.SEQ_ITEM) this.error = new _errors.YAMLSemanticError(this, 'Sequence items must not have preceding content on the same line');
+    if (!atLineStart && this.type === _constants.Type.SEQ_ITEM) this.error = new _errors.YAMLSemanticError(this, 'Sequence items must not have preceding content on the same line');
     const indent = atLineStart ? start - lineStart : context.indent;
 
     let offset = _Node.default.endOfWhiteSpace(src, start + 1);
@@ -77,7 +77,7 @@ class CollectionItem extends _Node.default {
       ch = src[offset];
     }
 
-    if (_Node.default.nextNodeIsIndented(ch, offset - (lineStart + indent), this.type !== _Node.Type.SEQ_ITEM)) {
+    if (_Node.default.nextNodeIsIndented(ch, offset - (lineStart + indent), this.type !== _constants.Type.SEQ_ITEM)) {
       this.node = parseNode({
         atLineStart,
         inCollection: false,
@@ -116,6 +116,4 @@ class CollectionItem extends _Node.default {
 
 }
 
-exports.default = CollectionItem;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = CollectionItem;

package/dist/cst/Comment.js

@@ -5,17 +5,17 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = void 0;
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _constants = require("../constants");
+
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 class Comment extends _Node.default {
   constructor() {
-    super(_Node.Type.COMMENT);
+    super(_constants.Type.COMMENT);
   }
   /**
    * Parses a comment line from the source
@@ -35,6 +35,4 @@ class Comment extends _Node.default {
 
 }
 
-exports.default = Comment;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = Comment;

package/dist/cst/Directive.js

@@ -5,14 +5,14 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = void 0;
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _constants = require("../constants");
+
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 class Directive extends _Node.default {
   static endOfDirective(src, offset) {
     let ch = src[offset];
@@ -31,7 +31,7 @@ class Directive extends _Node.default {
   }
 
   constructor() {
-    super(_Node.Type.DIRECTIVE);
+    super(_constants.Type.DIRECTIVE);
     this.name = null;
   }
 
@@ -77,6 +77,4 @@ class Directive extends _Node.default {
 
 }
 
-exports.default = Directive;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = Directive;

package/dist/cst/Document.js

@@ -5,6 +5,8 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = void 0;
 
+var _constants = require("../constants");
+
 var _errors = require("../errors");
 
 var _BlankLine = _interopRequireDefault(require("./BlankLine"));
@@ -15,12 +17,10 @@ var _Comment = _interopRequireDefault(require("./Comment"));
 
 var _Directive = _interopRequireDefault(require("./Directive"));
 
-var _Node = _interopRequireWildcard(require("./Node"));
+var _Node = _interopRequireDefault(require("./Node"));
 
 var _Range = _interopRequireDefault(require("./Range"));
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
-
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
 class Document extends _Node.default {
@@ -32,9 +32,11 @@ class Document extends _Node.default {
   }
 
   constructor() {
-    super(_Node.Type.DOCUMENT);
+    super(_constants.Type.DOCUMENT);
     this.directives = null;
     this.contents = null;
+    this.directivesEndMarker = null;
+    this.documentEndMarker = null;
   }
 
   parseDirectives(start) {
@@ -46,7 +48,7 @@ class Document extends _Node.default {
     let hasDirectives = false;
     let offset = start;
 
-    while (!_Node.default.atDocumentBoundary(src, offset, _Node.Char.DIRECTIVES_END)) {
+    while (!_Node.default.atDocumentBoundary(src, offset, _constants.Char.DIRECTIVES_END)) {
       offset = Document.startCommentOrEndBlankLine(src, offset);
 
       switch (src[offset]) {
@@ -103,7 +105,10 @@ class Document extends _Node.default {
       }
     }
 
-    if (src[offset]) return offset + 3;
+    if (src[offset]) {
+      this.directivesEndMarker = new _Range.default(offset, offset + 3);
+      return offset + 3;
+    }
 
     if (hasDirectives) {
       this.error = new _errors.YAMLSemanticError(this, 'Missing directives-end indicator line');
@@ -130,7 +135,7 @@ class Document extends _Node.default {
     let atLineStart = lineStart === start;
     this.valueRange = new _Range.default(offset);
 
-    while (!_Node.default.atDocumentBoundary(src, offset, _Node.Char.DOCUMENT_END)) {
+    while (!_Node.default.atDocumentBoundary(src, offset, _constants.Char.DOCUMENT_END)) {
       switch (src[offset]) {
         case '\n':
           if (atLineStart) {
@@ -190,6 +195,7 @@ class Document extends _Node.default {
     this.valueRange.end = offset;
 
     if (src[offset]) {
+      this.documentEndMarker = new _Range.default(offset, offset + 3);
       offset += 3;
 
       if (src[offset]) {
@@ -244,9 +250,11 @@ class Document extends _Node.default {
     this.directives.forEach(node => {
       offset = node.setOrigRanges(cr, offset);
     });
+    if (this.directivesEndMarker) offset = this.directivesEndMarker.setOrigRange(cr, offset);
     this.contents.forEach(node => {
       offset = node.setOrigRanges(cr, offset);
     });
+    if (this.documentEndMarker) offset = this.documentEndMarker.setOrigRange(cr, offset);
     return offset;
   }
 
@@ -260,7 +268,7 @@ class Document extends _Node.default {
     let str = directives.join('');
 
     if (contents.length > 0) {
-      if (directives.length > 0 || contents[0].type === _Node.Type.COMMENT) str += '---\n';
+      if (directives.length > 0 || contents[0].type === _constants.Type.COMMENT) str += '---\n';
       str += contents.join('');
     }
 
@@ -270,6 +278,4 @@ class Document extends _Node.default {
 
 }
 
-exports.default = Document;
-module.exports = exports.default;
-module.exports.default = exports.default;
+exports.default = Document;