Refactor to use composer.compose (#252)

* refactor to use `composer.compose`

* update

Author: ota-meshi

src/yaml-cst-parse.ts

@@ -17,31 +17,17 @@ export function parseAllDocsToCST(ctx: Context): ParsedCSTDocs {
     keepSourceTokens: true,
     lineCounter,
   });
-  const cstNodes: CST.Token[] = [];
+  const cstNodes: CST.Token[] = [...parser.parse(ctx.code)];
   const nodes: Document.Parsed[] = [];
-
-  /**
-   * Process for Document
-   */
-  function processDoc(node: Document.Parsed) {
-    for (const error of node.errors) {
+  for (const doc of composer.compose(cstNodes)) {
+    for (const error of doc.errors) {
       throw ctx.throwError(error.message, error.pos[0]);
     }
     // ignore warns
     // for (const error of doc.warnings) {
     //     throw ctx.throwError(error.message, error.pos[0])
     // }
-    nodes.push(node);
-  }
-
-  for (const cst of parser.parse(ctx.code)) {
-    cstNodes.push(cst);
-    for (const doc of composer.next(cst)) {
-      processDoc(doc);
-    }
-  }
-  for (const doc of composer.end()) {
-    processDoc(doc);
+    nodes.push(doc);
   }
 
   return { nodes, cstNodes, streamInfo: composer.streamInfo() };