fix: Stringify a Document as a Document (fixes #576)

eemeli · Oct 13, 2024 · 438688f · 438688f
1 parent baaabd0
commit 438688f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/public-api.ts b/src/public-api.ts
@@ -3,6 +3,7 @@ import type { Reviver } from './doc/applyReviver.js'
 import { Document, Replacer } from './doc/Document.js'
 import { prettifyError, YAMLParseError } from './errors.js'
 import { warn } from './log.js'
+import { isDocument } from './nodes/identity.js'
 import type { Node, ParsedNode } from './nodes/Node.js'
 import type {
   CreateNodeOptions,
@@ -225,5 +226,6 @@ export function stringify(
     const { keepUndefined } = options ?? (replacer as CreateNodeOptions) ?? {}
     if (!keepUndefined) return undefined
   }
+  if (isDocument(value) && !_replacer) return value.toString(options)
   return new Document(value, _replacer, options).toString(options)
 }
diff --git a/tests/doc/stringify.ts b/tests/doc/stringify.ts
@@ -1405,6 +1405,12 @@ describe('YAML.stringify on ast Document', () => {
     const doc = YAML.parseDocument('null')
     expect(YAML.stringify(doc)).toBe('null\n')
   })
+
+  test('Document with comments', () => {
+    const src = 'a: 1\n\n# c\n'
+    const doc = YAML.parseDocument(src)
+    expect(YAML.stringify(doc)).toBe(src)
+  })
 })
 
 describe('flow collection padding', () => {