Function tree

packages/printer/src/internal/docTree.ts:451–482

(
  parser: () => DocTreeParser<DocStream.DocStream<A>, DocTree.DocTree<A>>
)

}

const tree = <A>(
  parser: () => DocTreeParser<DocStream.DocStream<A>, DocTree.DocTree<A>>
): DocTreeParser<DocStream.DocStream<A>, DocTree.DocTree<A>> => {
  return parserFlatMap(nextToken<A>(), (token) => {
    switch (token._tag) {
      case "EmptyToken": {
        return parserSucceed(empty)
      }
      case "CharToken": {
        return parserSucceed(char<A>(token.char))
      }
      case "TextToken": {
        return parserSucceed(text<A>(token.text))
      }
      case "LineToken": {
        return parserSucceed(line<A>(token.indentation))
      }
      case "PushAnnotationToken": {
        return parserFlatMap(parser(), (annotatedContents) =>
          // Make sure to handle the subsequent pop annotation token

          parserMap(
            nextToken<A>(),
            () => annotation(annotatedContents, token.annotation)
          ))
      }
      case "PopAnnotationToken": {
        return () => Option.none()
      }
    }
  })
}

/** @internal */
export const parser = <A>(): DocTreeParser<DocStream.DocStream<A>, DocTree.DocTree<A>> =>