Function alterAnnotationSafe

packages/printer/src/internal/docStream.ts:236–289

(
  self: DocStream.DocStream<A>,
  f: (a: A) => Option.Option<B>,
  stack: List.List<AnnotationRemoval>
)

>(2, (self, f) => Effect.runSync(alterAnnotationSafe(self, f, List.nil())))

const alterAnnotationSafe = <A, B>(
  self: DocStream.DocStream<A>,
  f: (a: A) => Option.Option<B>,
  stack: List.List<AnnotationRemoval>
): Effect.Effect<DocStream.DocStream<B>> => {
  switch (self._tag) {
    case "CharStream": {
      return Effect.map(
        Effect.suspend(() => alterAnnotationSafe(self.stream, f, stack)),
        char(self.char)
      )
    }
    case "TextStream": {
      return Effect.map(
        Effect.suspend(() => alterAnnotationSafe(self.stream, f, stack)),
        text(self.text)
      )
    }
    case "LineStream": {
      return Effect.map(
        Effect.suspend(() => alterAnnotationSafe(self.stream, f, stack)),
        line(self.indentation)
      )
    }
    case "PushAnnotationStream": {
      const altered = f(self.annotation)
      if (Option.isSome(altered)) {
        return Effect.map(
          Effect.suspend(() => alterAnnotationSafe(self.stream, f, List.prepend(stack, DontRemove))),
          pushAnnotation(altered.value)
        )
      }
      return Effect.suspend(() => alterAnnotationSafe(self.stream, f, List.prepend(stack, Remove)))
    }
    case "PopAnnotationStream": {
      if (List.isNil(stack)) {
        return Effect.dieMessage(
          "BUG: DocStream.alterAnnotations - received empty stack to pop from" +
            " - please report an issue at https://github.com/Effect-TS/printer/issues"
        )
      }
      if (stack.head === DontRemove) {
        return Effect.map(
          Effect.suspend(() => alterAnnotationSafe(self.stream, f, stack.tail)),
          popAnnotation
        )
      }
      return Effect.suspend(() => alterAnnotationSafe(self.stream, f, stack.tail))
    }
    default: {
      return Effect.succeed(self as unknown as DocStream.DocStream<B>)
    }
  }
}

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