@ts-graphviz/ast 2.0.5 → 2.0.6-next-d7ff421ec861ca8fdede1a6bdf256f3455fb9797

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @ts-graphviz/ast
 
+## 2.0.6-next-d7ff421ec861ca8fdede1a6bdf256f3455fb9797
+
+### Patch Changes
+
+- [#1218](https://github.com/ts-graphviz/ts-graphviz/pull/1218) [`d7ff421`](https://github.com/ts-graphviz/ts-graphviz/commit/d7ff421ec861ca8fdede1a6bdf256f3455fb9797) Thanks [@kamiazya](https://github.com/kamiazya)! - Fix AST printing not to be destroyed during stringify even in HTML-Like format
+
+- Updated dependencies [[`d7ff421`](https://github.com/ts-graphviz/ts-graphviz/commit/d7ff421ec861ca8fdede1a6bdf256f3455fb9797)]:
+  - @ts-graphviz/common@2.1.5-next-d7ff421ec861ca8fdede1a6bdf256f3455fb9797
+
 ## 2.0.5
 
 ### Patch Changes

package/lib/ast.cjs

@@ -34,95 +34,57 @@ class Builder {
   }
 }
 const createElement = Builder.prototype.createElement.bind(new Builder());
-const EOL = /\r?\n/;
-function joinBy(sep) {
-  return (value) => value.join(sep);
-}
-function wrapWith(wrapper) {
-  return (value) => wrapper + value + wrapper;
-}
-function wrapByPair(l, r) {
-  return (value) => l + value + r;
-}
-function leftPadWith(left) {
-  return (value) => left + value;
-}
-const escape = (value) => value.replace(/(?<!\\)"|[\r\n]/g, escapeReplacer);
-function escapeReplacer(match) {
-  switch (match) {
-    case "\r":
-      return "\\r";
-    case "\n":
-      return "\\n";
-    default:
-      return '\\"';
-  }
-}
-const splitByLine = (value) => value.split(EOL);
-const indent = (style, size, eol) => common.pipe(
-  splitByLine,
-  common.map(leftPadWith(style === "space" ? " ".repeat(size) : "\n")),
-  joinBy(eol)
-);
-const endOfLine = (eol) => {
-  switch (eol) {
-    case "crlf":
-      return "\r\n";
-    case "lf":
-      return "\n";
-  }
-};
 const AttributeListPrintPlugin = {
   match(ast) {
     return ast.type === "AttributeList";
   },
-  print(context, ast) {
+  *print(context, ast) {
     if (ast.children.length === 0) {
-      return `${ast.kind.toLocaleLowerCase()} [];`;
+      yield `${ast.kind.toLocaleLowerCase()} [];`;
+    } else {
+      yield `${ast.kind.toLocaleLowerCase()} [`;
+      yield* context.printChildren(ast.children);
+      yield "];";
     }
-    const eol = endOfLine(context.endOfLine);
-    return common.pipe(
-      common.map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`${ast.kind.toLocaleLowerCase()} [${eol}`, `${eol}];`)
-    )(ast.children);
   }
 };
 const AttributePrintPlugin = {
   match(ast) {
     return ast.type === "Attribute";
   },
-  print(context, ast) {
-    return `${context.print(ast.key)} = ${context.print(ast.value)};`;
+  *print(context, ast) {
+    yield* context.print(ast.key);
+    yield " = ";
+    yield* context.print(ast.value);
+    yield ";";
   }
 };
+const EOL_PATTERN = /\r?\n/;
+const paddingMap = {
+  Block: " * ",
+  Macro: "# ",
+  Slash: "// "
+};
 const CommentPrintPlugin = {
   match(ast) {
     return ast.type === "Comment";
   },
-  print(context, ast) {
-    const eol = endOfLine(context.endOfLine);
-    switch (ast.kind) {
-      case "Block":
-        return common.pipe(
-          splitByLine,
-          common.map(leftPadWith(" * ")),
-          joinBy(eol),
-          wrapByPair(`/**${eol}`, `${eol} */`)
-        )(ast.value);
-      case "Macro":
-        return common.pipe(
-          splitByLine,
-          common.map(leftPadWith("# ")),
-          joinBy(eol)
-        )(ast.value);
-      default:
-        return common.pipe(
-          splitByLine,
-          common.map(leftPadWith("// ")),
-          joinBy(eol)
-        )(ast.value);
+  *print(context, ast) {
+    const padding = paddingMap[ast.kind];
+    if (ast.kind === "Block") {
+      yield* ["/**", context.EOL];
+    }
+    const lines = ast.value.split(EOL_PATTERN);
+    const lineLength = lines.length;
+    for (let i = 0; i < lineLength; i++) {
+      yield padding;
+      yield lines[i];
+      if (i < lineLength - 1) {
+        yield context.EOL;
+      }
+    }
+    if (ast.kind === "Block") {
+      yield* [context.EOL, " */"];
     }
   }
 };
@@ -130,73 +92,74 @@ const DotPrintPlugin = {
   match(ast) {
     return ast.type === "Dot";
   },
-  print(context, ast) {
-    return ast.children.map(context.print).join(endOfLine(context.endOfLine));
+  *print(context, ast) {
+    yield* context.join(ast.children, context.EOL);
   }
 };
 const EdgePrintPlugin = {
   match(ast) {
     return ast.type === "Edge";
   },
-  print(context, ast) {
-    const targets = common.pipe(
-      common.map(context.print),
-      joinBy(context.directed ? " -> " : " -- ")
-    )(ast.targets);
+  *print(context, ast) {
+    yield* context.join(ast.targets, context.directed ? " -> " : " -- ");
     if (ast.children.length === 0) {
-      return `${targets};`;
+      yield ";";
+    } else {
+      yield " [";
+      yield* context.printChildren(ast.children);
+      yield "];";
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = common.pipe(
-      common.map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`[${eol}`, `${eol}];`)
-    )(ast.children);
-    return `${targets} ${contents}`;
   }
 };
 const GraphPrintPlugin = {
   match(ast) {
     return ast.type === "Graph";
   },
-  print(context, ast) {
+  *print(context, ast) {
     context.directed = ast.directed;
-    const parts = [];
     if (ast.strict) {
-      parts.push("strict");
+      yield "strict ";
     }
-    parts.push(ast.directed ? "digraph" : "graph");
+    yield ast.directed ? "digraph " : "graph ";
     if (ast.id) {
-      parts.push(context.print(ast.id));
+      yield* context.print(ast.id);
+      yield " ";
     }
-    if (ast.children.length === 0) {
-      return `${parts.join(" ")} {}`;
+    yield "{";
+    if (ast.children.length >= 1) {
+      yield* context.printChildren(ast.children);
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = common.pipe(
-      common.map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`{${eol}`, `${eol}}`)
-    )(ast.children);
-    return `${parts.join(" ")} ${contents}`;
+    yield "}";
   }
 };
-const quoteLiteralValue = common.pipe(escape, wrapWith('"'));
-const quoteHTMLLikeLiteralValue = wrapByPair("<", ">");
+const escape = (value) => value.replace(/(?<!\\)"|[\r\n]/g, escapeReplacer);
+const escapeMap = {
+  "\r": String.raw`\r`,
+  "\n": String.raw`\n`,
+  '"': String.raw`\"`
+};
+function escapeReplacer(match) {
+  return escapeMap[match];
+}
 const LiteralPrintPlugin = {
   match(ast) {
     return ast.type === "Literal";
   },
-  print(context, ast) {
+  *print(context, ast) {
     switch (ast.quoted) {
       case "html":
-        return quoteHTMLLikeLiteralValue(ast.value);
+        yield "<";
+        yield ast.value;
+        yield ">";
+        return;
       case true:
-        return quoteLiteralValue(ast.value);
+        yield '"';
+        yield escape(ast.value);
+        yield '"';
+        return;
       default:
-        return escape(ast.value);
+        yield ast.value;
+        return;
     }
   }
 };
@@ -204,68 +167,57 @@ const NodePrintPlugin = {
   match(ast) {
     return ast.type === "Node";
   },
-  print(context, ast) {
-    const id = context.print(ast.id);
-    if (ast.children.length === 0) {
-      return `${id};`;
+  *print(context, ast) {
+    yield* context.print(ast.id);
+    if (ast.children.length >= 1) {
+      yield " [";
+      yield* context.printChildren(ast.children);
+      yield "]";
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = common.pipe(
-      common.map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`[${eol}`, `${eol}];`)
-    )(ast.children);
-    return `${id} ${contents}`;
+    yield ";";
   }
 };
 const NodeRefGroupPrintPlugin = {
   match(ast) {
     return ast.type === "NodeRefGroup";
   },
-  print(context, ast) {
-    return common.pipe(
-      common.map(context.print),
-      joinBy(" "),
-      wrapByPair("{", "}")
-    )(ast.children);
+  *print(context, ast) {
+    yield "{";
+    yield* context.join(ast.children, " ");
+    yield "}";
   }
 };
 const NodeRefPrintPlugin = {
   match(ast) {
     return ast.type === "NodeRef";
   },
-  print(context, ast) {
-    const parts = [context.print(ast.id)];
+  *print(context, ast) {
+    yield* context.print(ast.id);
     if (ast.port) {
-      parts.push(context.print(ast.port));
+      yield ":";
+      yield* context.print(ast.port);
     }
     if (ast.compass) {
-      parts.push(context.print(ast.compass));
+      yield ":";
+      yield* context.print(ast.compass);
     }
-    return parts.join(":");
   }
 };
 const SubgraphPrintPlugin = {
   match(ast) {
     return ast.type === "Subgraph";
   },
-  print(context, ast) {
-    const parts = ["subgraph"];
+  *print(context, ast) {
+    yield "subgraph";
     if (ast.id) {
-      parts.push(context.print(ast.id));
+      yield " ";
+      yield* context.print(ast.id);
     }
-    if (ast.children.length === 0) {
-      return `${parts.join(" ")} {}`;
+    yield " {";
+    if (ast.children.length >= 1) {
+      yield* context.printChildren(ast.children);
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = common.pipe(
-      common.map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`{${eol}`, `${eol}}`)
-    )(ast.children);
-    return `${parts.join(" ")} ${contents}`;
+    yield "}";
   }
 };
 const defaultPlugins$2 = [
@@ -296,27 +248,61 @@ class Printer {
    * @returns The DOT string generated from the ASTNode.
    */
   print(ast) {
+    return Array.from(this.toChunks(ast)).join("");
+  }
+  toChunks(ast) {
     const plugins = [...this.#plugins];
     const {
       indentSize = 2,
       indentStyle = "space",
-      endOfLine: endOfLine2 = "lf"
+      endOfLine = "lf"
     } = this.options;
+    const EOL = endOfLine === "crlf" ? "\r\n" : "\n";
+    const PADDING = indentStyle === "space" ? " ".repeat(indentSize) : "	";
     const context = {
       directed: true,
-      indentSize,
-      indentStyle,
-      endOfLine: endOfLine2,
-      print(a) {
+      EOL,
+      *printChildren(children) {
+        yield* indent(function* () {
+          yield EOL;
+          yield* context.join(children, EOL);
+        });
+        yield EOL;
+      },
+      *print(a) {
         for (const plugin of plugins) {
           if (plugin.match(a)) {
-            return plugin.print(context, a);
+            yield* plugin.print(this, a);
+            return;
+          }
+        }
+        throw new Error(
+          `No matching plugin found for AST node: ${JSON.stringify(a)}`
+        );
+      },
+      *join(array, separator) {
+        const childrenLength = array.length;
+        for (let i = 0; i < childrenLength; i++) {
+          yield* context.print(array[i]);
+          if (i < childrenLength - 1) {
+            yield separator;
           }
         }
-        throw Error();
       }
     };
-    return context.print(ast);
+    return {
+      [Symbol.iterator]: function* () {
+        yield* context.print(ast);
+      }
+    };
+    function* indent(tokens) {
+      for (const token of tokens()) {
+        yield token;
+        if (token === EOL) {
+          yield PADDING;
+        }
+      }
+    }
   }
 }
 function stringify(ast, options) {

package/lib/ast.d.ts

@@ -736,15 +736,18 @@ declare interface OtherExpectation {
   * PrintContext interface provides an interface for printing an ASTNode with a set of options.
   * @group Convert AST to DOT
   */
- export declare interface PrintContext extends Required<PrintOptions> {
+ export declare interface PrintContext {
      /**
       * Indicates if the AST should be printed in a directed graph.
       */
      directed: boolean;
+     readonly EOL: string;
      /**
       * A function to print an ASTNode, taking in an ASTNode as an argument. Returns a string.
       */
-     print(ast: ASTNode): string;
+     print(ast: ASTNode): Iterable<string>;
+     printChildren(children: ASTNode[]): Iterable<string>;
+     join(children: ASTNode[], separator: string): Iterable<string>;
  }
 
  /**
@@ -764,6 +767,7 @@ declare interface OtherExpectation {
       * @returns The DOT string generated from the ASTNode.
       */
      print(ast: ASTNode): string;
+     private toChunks;
  }
 
  /**
@@ -809,7 +813,7 @@ declare interface OtherExpectation {
       * @returns printed string
       * @memberof PrintPlugin
       */
-     print(context: PrintContext, ast: T): string;
+     print(context: PrintContext, ast: T): Generator<string>;
  }
 
  /**

package/lib/ast.js

@@ -1,4 +1,4 @@
-import { pipe, map, isNodeModel, isForwardRefNode, createModelsContext } from "@ts-graphviz/common";
+import { isNodeModel, isForwardRefNode, createModelsContext } from "@ts-graphviz/common";
 class Builder {
   /**
    * Constructor of Builder
@@ -32,95 +32,57 @@ class Builder {
   }
 }
 const createElement = Builder.prototype.createElement.bind(new Builder());
-const EOL = /\r?\n/;
-function joinBy(sep) {
-  return (value) => value.join(sep);
-}
-function wrapWith(wrapper) {
-  return (value) => wrapper + value + wrapper;
-}
-function wrapByPair(l, r) {
-  return (value) => l + value + r;
-}
-function leftPadWith(left) {
-  return (value) => left + value;
-}
-const escape = (value) => value.replace(/(?<!\\)"|[\r\n]/g, escapeReplacer);
-function escapeReplacer(match) {
-  switch (match) {
-    case "\r":
-      return "\\r";
-    case "\n":
-      return "\\n";
-    default:
-      return '\\"';
-  }
-}
-const splitByLine = (value) => value.split(EOL);
-const indent = (style, size, eol) => pipe(
-  splitByLine,
-  map(leftPadWith(style === "space" ? " ".repeat(size) : "\n")),
-  joinBy(eol)
-);
-const endOfLine = (eol) => {
-  switch (eol) {
-    case "crlf":
-      return "\r\n";
-    case "lf":
-      return "\n";
-  }
-};
 const AttributeListPrintPlugin = {
   match(ast) {
     return ast.type === "AttributeList";
   },
-  print(context, ast) {
+  *print(context, ast) {
     if (ast.children.length === 0) {
-      return `${ast.kind.toLocaleLowerCase()} [];`;
+      yield `${ast.kind.toLocaleLowerCase()} [];`;
+    } else {
+      yield `${ast.kind.toLocaleLowerCase()} [`;
+      yield* context.printChildren(ast.children);
+      yield "];";
     }
-    const eol = endOfLine(context.endOfLine);
-    return pipe(
-      map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`${ast.kind.toLocaleLowerCase()} [${eol}`, `${eol}];`)
-    )(ast.children);
   }
 };
 const AttributePrintPlugin = {
   match(ast) {
     return ast.type === "Attribute";
   },
-  print(context, ast) {
-    return `${context.print(ast.key)} = ${context.print(ast.value)};`;
+  *print(context, ast) {
+    yield* context.print(ast.key);
+    yield " = ";
+    yield* context.print(ast.value);
+    yield ";";
   }
 };
+const EOL_PATTERN = /\r?\n/;
+const paddingMap = {
+  Block: " * ",
+  Macro: "# ",
+  Slash: "// "
+};
 const CommentPrintPlugin = {
   match(ast) {
     return ast.type === "Comment";
   },
-  print(context, ast) {
-    const eol = endOfLine(context.endOfLine);
-    switch (ast.kind) {
-      case "Block":
-        return pipe(
-          splitByLine,
-          map(leftPadWith(" * ")),
-          joinBy(eol),
-          wrapByPair(`/**${eol}`, `${eol} */`)
-        )(ast.value);
-      case "Macro":
-        return pipe(
-          splitByLine,
-          map(leftPadWith("# ")),
-          joinBy(eol)
-        )(ast.value);
-      default:
-        return pipe(
-          splitByLine,
-          map(leftPadWith("// ")),
-          joinBy(eol)
-        )(ast.value);
+  *print(context, ast) {
+    const padding = paddingMap[ast.kind];
+    if (ast.kind === "Block") {
+      yield* ["/**", context.EOL];
+    }
+    const lines = ast.value.split(EOL_PATTERN);
+    const lineLength = lines.length;
+    for (let i = 0; i < lineLength; i++) {
+      yield padding;
+      yield lines[i];
+      if (i < lineLength - 1) {
+        yield context.EOL;
+      }
+    }
+    if (ast.kind === "Block") {
+      yield* [context.EOL, " */"];
     }
   }
 };
@@ -128,73 +90,74 @@ const DotPrintPlugin = {
   match(ast) {
     return ast.type === "Dot";
   },
-  print(context, ast) {
-    return ast.children.map(context.print).join(endOfLine(context.endOfLine));
+  *print(context, ast) {
+    yield* context.join(ast.children, context.EOL);
   }
 };
 const EdgePrintPlugin = {
   match(ast) {
     return ast.type === "Edge";
   },
-  print(context, ast) {
-    const targets = pipe(
-      map(context.print),
-      joinBy(context.directed ? " -> " : " -- ")
-    )(ast.targets);
+  *print(context, ast) {
+    yield* context.join(ast.targets, context.directed ? " -> " : " -- ");
     if (ast.children.length === 0) {
-      return `${targets};`;
+      yield ";";
+    } else {
+      yield " [";
+      yield* context.printChildren(ast.children);
+      yield "];";
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = pipe(
-      map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`[${eol}`, `${eol}];`)
-    )(ast.children);
-    return `${targets} ${contents}`;
   }
 };
 const GraphPrintPlugin = {
   match(ast) {
     return ast.type === "Graph";
   },
-  print(context, ast) {
+  *print(context, ast) {
     context.directed = ast.directed;
-    const parts = [];
     if (ast.strict) {
-      parts.push("strict");
+      yield "strict ";
     }
-    parts.push(ast.directed ? "digraph" : "graph");
+    yield ast.directed ? "digraph " : "graph ";
     if (ast.id) {
-      parts.push(context.print(ast.id));
+      yield* context.print(ast.id);
+      yield " ";
     }
-    if (ast.children.length === 0) {
-      return `${parts.join(" ")} {}`;
+    yield "{";
+    if (ast.children.length >= 1) {
+      yield* context.printChildren(ast.children);
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = pipe(
-      map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`{${eol}`, `${eol}}`)
-    )(ast.children);
-    return `${parts.join(" ")} ${contents}`;
+    yield "}";
   }
 };
-const quoteLiteralValue = pipe(escape, wrapWith('"'));
-const quoteHTMLLikeLiteralValue = wrapByPair("<", ">");
+const escape = (value) => value.replace(/(?<!\\)"|[\r\n]/g, escapeReplacer);
+const escapeMap = {
+  "\r": String.raw`\r`,
+  "\n": String.raw`\n`,
+  '"': String.raw`\"`
+};
+function escapeReplacer(match) {
+  return escapeMap[match];
+}
 const LiteralPrintPlugin = {
   match(ast) {
     return ast.type === "Literal";
   },
-  print(context, ast) {
+  *print(context, ast) {
     switch (ast.quoted) {
       case "html":
-        return quoteHTMLLikeLiteralValue(ast.value);
+        yield "<";
+        yield ast.value;
+        yield ">";
+        return;
       case true:
-        return quoteLiteralValue(ast.value);
+        yield '"';
+        yield escape(ast.value);
+        yield '"';
+        return;
       default:
-        return escape(ast.value);
+        yield ast.value;
+        return;
     }
   }
 };
@@ -202,68 +165,57 @@ const NodePrintPlugin = {
   match(ast) {
     return ast.type === "Node";
   },
-  print(context, ast) {
-    const id = context.print(ast.id);
-    if (ast.children.length === 0) {
-      return `${id};`;
+  *print(context, ast) {
+    yield* context.print(ast.id);
+    if (ast.children.length >= 1) {
+      yield " [";
+      yield* context.printChildren(ast.children);
+      yield "]";
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = pipe(
-      map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`[${eol}`, `${eol}];`)
-    )(ast.children);
-    return `${id} ${contents}`;
+    yield ";";
   }
 };
 const NodeRefGroupPrintPlugin = {
   match(ast) {
     return ast.type === "NodeRefGroup";
   },
-  print(context, ast) {
-    return pipe(
-      map(context.print),
-      joinBy(" "),
-      wrapByPair("{", "}")
-    )(ast.children);
+  *print(context, ast) {
+    yield "{";
+    yield* context.join(ast.children, " ");
+    yield "}";
   }
 };
 const NodeRefPrintPlugin = {
   match(ast) {
     return ast.type === "NodeRef";
   },
-  print(context, ast) {
-    const parts = [context.print(ast.id)];
+  *print(context, ast) {
+    yield* context.print(ast.id);
     if (ast.port) {
-      parts.push(context.print(ast.port));
+      yield ":";
+      yield* context.print(ast.port);
     }
     if (ast.compass) {
-      parts.push(context.print(ast.compass));
+      yield ":";
+      yield* context.print(ast.compass);
     }
-    return parts.join(":");
   }
 };
 const SubgraphPrintPlugin = {
   match(ast) {
     return ast.type === "Subgraph";
   },
-  print(context, ast) {
-    const parts = ["subgraph"];
+  *print(context, ast) {
+    yield "subgraph";
     if (ast.id) {
-      parts.push(context.print(ast.id));
+      yield " ";
+      yield* context.print(ast.id);
     }
-    if (ast.children.length === 0) {
-      return `${parts.join(" ")} {}`;
+    yield " {";
+    if (ast.children.length >= 1) {
+      yield* context.printChildren(ast.children);
     }
-    const eol = endOfLine(context.endOfLine);
-    const contents = pipe(
-      map(context.print),
-      joinBy(eol),
-      indent(context.indentStyle, context.indentSize, eol),
-      wrapByPair(`{${eol}`, `${eol}}`)
-    )(ast.children);
-    return `${parts.join(" ")} ${contents}`;
+    yield "}";
   }
 };
 const defaultPlugins$2 = [
@@ -294,27 +246,61 @@ class Printer {
    * @returns The DOT string generated from the ASTNode.
    */
   print(ast) {
+    return Array.from(this.toChunks(ast)).join("");
+  }
+  toChunks(ast) {
     const plugins = [...this.#plugins];
     const {
       indentSize = 2,
       indentStyle = "space",
-      endOfLine: endOfLine2 = "lf"
+      endOfLine = "lf"
     } = this.options;
+    const EOL = endOfLine === "crlf" ? "\r\n" : "\n";
+    const PADDING = indentStyle === "space" ? " ".repeat(indentSize) : "	";
     const context = {
       directed: true,
-      indentSize,
-      indentStyle,
-      endOfLine: endOfLine2,
-      print(a) {
+      EOL,
+      *printChildren(children) {
+        yield* indent(function* () {
+          yield EOL;
+          yield* context.join(children, EOL);
+        });
+        yield EOL;
+      },
+      *print(a) {
         for (const plugin of plugins) {
           if (plugin.match(a)) {
-            return plugin.print(context, a);
+            yield* plugin.print(this, a);
+            return;
+          }
+        }
+        throw new Error(
+          `No matching plugin found for AST node: ${JSON.stringify(a)}`
+        );
+      },
+      *join(array, separator) {
+        const childrenLength = array.length;
+        for (let i = 0; i < childrenLength; i++) {
+          yield* context.print(array[i]);
+          if (i < childrenLength - 1) {
+            yield separator;
           }
         }
-        throw Error();
       }
     };
-    return context.print(ast);
+    return {
+      [Symbol.iterator]: function* () {
+        yield* context.print(ast);
+      }
+    };
+    function* indent(tokens) {
+      for (const token of tokens()) {
+        yield token;
+        if (token === EOL) {
+          yield PADDING;
+        }
+      }
+    }
   }
 }
 function stringify(ast, options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ts-graphviz/ast",
-  "version": "2.0.5",
+  "version": "2.0.6-next-d7ff421ec861ca8fdede1a6bdf256f3455fb9797",
   "description": "Graphviz AST(Abstract Syntax Tree) Utilities",
   "keywords": [],
   "homepage": "https://github.com/ts-graphviz/ts-graphviz#readme",
@@ -37,7 +37,7 @@
   "module": "lib/ast.js",
   "types": "lib/ast.d.ts",
   "dependencies": {
-    "@ts-graphviz/common": "^2.1.4"
+    "@ts-graphviz/common": "^2.1.5-next-d7ff421ec861ca8fdede1a6bdf256f3455fb9797"
   },
   "devDependencies": {
     "peggy": "^4.0.3",