Function alterAnnotationsSafe

packages/printer/src/internal/docTree.ts:187–218

(
  self: DocTree.DocTree<A>,
  f: (a: A) => Iterable<B>
)

>(2, (self, f) => Effect.runSync(alterAnnotationsSafe(self, f)))

const alterAnnotationsSafe = <A, B>(
  self: DocTree.DocTree<A>,
  f: (a: A) => Iterable<B>
): Effect.Effect<DocTree.DocTree<B>> => {
  switch (self._tag) {
    case "EmptyTree": {
      return Effect.succeed(empty)
    }
    case "CharTree": {
      return Effect.succeed(char(self.char))
    }
    case "TextTree": {
      return Effect.succeed(text(self.text))
    }
    case "LineTree": {
      return Effect.succeed(line(self.indentation))
    }
    case "AnnotationTree": {
      return Arr.reduce(
        Arr.fromIterable(f(self.annotation)),
        Effect.suspend(() => alterAnnotationsSafe(self.tree, f)),
        (acc, b) => Effect.map(acc, annotation(b))
      )
    }
    case "ConcatTree": {
      return pipe(
        Effect.forEach(self.trees, (tree) => alterAnnotationsSafe(tree, f)),
        Effect.map(concat)
      )
    }
  }
}

/** @internal */
export const reAnnotate = dual<