Function foldMapSafe

packages/printer/src/internal/docTree.ts:239–271

(
  self: DocTree.DocTree<A>,
  M: monoid.Monoid<M>,
  f: (a: A) => M
)

>(3, (self, M, f) => Effect.runSync(foldMapSafe(self, M, f)))

const foldMapSafe = <A, M>(
  self: DocTree.DocTree<A>,
  M: monoid.Monoid<M>,
  f: (a: A) => M
): Effect.Effect<M> => {
  switch (self._tag) {
    case "EmptyTree":
    case "CharTree":
    case "TextTree":
    case "LineTree": {
      return Effect.succeed(M.empty)
    }
    case "AnnotationTree": {
      return Effect.map(
        Effect.suspend(() => foldMapSafe(self.tree, M, f)),
        (that) => M.combine(f(self.annotation), that)
      )
    }
    case "ConcatTree": {
      if (Arr.isEmptyReadonlyArray(self.trees)) {
        return Effect.succeed(M.empty)
      }
      return Effect.map(
        Effect.forEach(self.trees, (tree) => foldMapSafe(tree, M, f)),
        (trees) => {
          const head = trees[0]
          const tail = trees.slice(1)
          return Arr.reduce(tail, head, M.combine)
        }
      )
    }
  }
}

// -----------------------------------------------------------------------------
// Rendering