Function reAnnotateSafe

packages/printer/src/internal/docStream.ts:297–333

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

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

const reAnnotateSafe = <A, B>(
  self: DocStream.DocStream<A>,
  f: (a: A) => B
): Effect.Effect<DocStream.DocStream<B>> => {
  switch (self._tag) {
    case "CharStream": {
      return Effect.map(
        Effect.suspend(() => reAnnotateSafe(self.stream, f)),
        char(self.char)
      )
    }
    case "TextStream": {
      return Effect.map(
        Effect.suspend(() => reAnnotateSafe(self.stream, f)),
        text(self.text)
      )
    }
    case "LineStream": {
      return Effect.map(
        Effect.suspend(() => reAnnotateSafe(self.stream, f)),
        line(self.indentation)
      )
    }
    case "PushAnnotationStream": {
      return Effect.map(
        Effect.suspend(() => reAnnotateSafe(self.stream, f)),
        pushAnnotation(f(self.annotation))
      )
    }
    case "PopAnnotationStream": {
      return Effect.suspend(() => reAnnotateSafe(self.stream, f))
    }
    default: {
      return Effect.succeed(self as unknown as DocStream.DocStream<B>)
    }
  }
}

/** @internal */
export const unAnnotate = <A>(self: DocStream.DocStream<A>): DocStream.DocStream<never> =>