Class OptionalImpl

packages/effect/src/Optic.ts:968–1121

export interface Traversal<in out S, in out A> extends Optional<S, ReadonlyArray<A>> {}

class OptionalImpl<S, A> implements Optional<S, A> {
  /** @internal */
  readonly node: Node
  readonly getResult: (s: S) => Result.Result<A, SchemaIssue.Issue>
  readonly replaceResult: (a: A, s: S) => Result.Result<S, SchemaIssue.Issue>
  constructor(
    node: Node,
    getResult: (s: S) => Result.Result<A, SchemaIssue.Issue>,
    replaceResult: (a: A, s: S) => Result.Result<S, SchemaIssue.Issue>
  ) {
    this.node = node
    this.getResult = getResult
    this.replaceResult = replaceResult
  }
  replace(a: A, s: S): S {
    return Result.getOrElse(this.replaceResult(a, s), () => s)
  }
  modify(f: (a: A) => A): (s: S) => S {
    return (s) => Result.getOrElse(Result.flatMap(this.getResult(s), (a) => this.replaceResult(f(a), s)), () => s)
  }
  compose(that: any): any {
    return make(compose(this.node, that.node))
  }
  key(key: PropertyKey): any {
    return make(compose(this.node, [new PathNode([key])]))
  }
  optionalKey(key: PropertyKey): any {
    return make(
      compose(
        this.node,
        primitiveNode(
          "Lens",
          (s) => s[key],
          (a, s) => {
            const copy = cloneShallow(s)
            if (a === undefined) {
              if (Array.isArray(copy) && typeof key === "number") {
                copy.splice(key, 1)
              } else {
                delete copy[key]
              }
            } else {
              InternalRecord.assignProperty(copy, key, a)
            }
            return copy
          }
        )
      )
    )
  }
  check(...checks: readonly [SchemaAST.Check<any>, ...Array<SchemaAST.Check<any>>]): any {
    return make(compose(this.node, [new CheckNode(checks)]))
  }
  refine<B extends A>(refinement: (a: A) => a is B, annotations?: Schema.Annotations.Filter): any {
    return make(compose(this.node, [new CheckNode([SchemaAST.makeFilterByGuard(refinement, annotations)])]))
  }
  tag(tag: string): any {
    const err = Result.fail(new SchemaIssue.InvalidValue({ expected: `${JSON.stringify(tag)} tag` }))