import { AnyPrimitiveType, AnyResolvableType, Resolver, TypeResolver } from './TypeResolver';

type TypeResolverFunction<P extends AnyPrimitiveType, R extends AnyResolvableType> = Resolver<
  R['__shape__'],
  P['__shape__']
>;

export class TypeResolverBuilder {
  private primitives = new Set<AnyPrimitiveType>();
  private resolvables = new Set<AnyResolvableType>();
  private resolverMap: Record<string, Record<string, TypeResolverFunction<AnyPrimitiveType, AnyResolvableType>>> = {};
  static new() {
    return new TypeResolverBuilder();
  }

  addPrimitive<P extends AnyPrimitiveType>(primitive: P) {
    this.primitives.add(primitive);
    return this;
  }

  addResolver<P extends AnyPrimitiveType, R extends AnyResolvableType>(
    primitive: P,
    resolvable: R,
    resolver: TypeResolverFunction<P, R>,
  ) {
    this.primitives.add(primitive);
    this.resolvables.add(resolvable);
    this.resolverMap[primitive.title] ??= {};
    this.resolverMap[primitive.title][resolvable.title] = resolver as TypeResolverFunction<
      AnyPrimitiveType,
      AnyResolvableType
    >;
    return this;
  }

  build() {
    return new TypeResolver(
      Array.from(this.primitives.values()),
      Array.from(this.resolvables.values()),
      this.resolverMap,
    );
  }
}