Class InferenceUnionBuilder

src/Inference.ts:35–74

}

class InferenceUnionBuilder extends UnionBuilder<NestedValueArray, NestedValueArray, any> {
    private _numValues?: number;

    constructor(
        typeBuilder: TypeGraphBuilder,
        typeName: string,
        isInferred: boolean,
        private readonly _typeInference: TypeInference,
        private readonly _cjson: CompressedJSON
    ) {
        super(typeBuilder, typeName, isInferred);
    }

    setNumValues = (n: number): void => {
        if (this._numValues !== undefined) {
            return panic("Can only set number of values once");
        }
        this._numValues = n;
    };

    protected makeEnum(enumCases: string[]): TypeRef | null {
        assert(enumCases.length > 0);
        const numValues = defined(this._numValues);
        if (numValues >= MIN_LENGTH_FOR_ENUM && enumCases.length < Math.sqrt(numValues)) {
            return this.typeBuilder.getEnumType(this.typeName, true, OrderedSet(enumCases));
        }
        return null;
    }

    protected makeClass(classes: NestedValueArray, maps: any[]): TypeRef {
        assert(maps.length === 0);
        return this._typeInference.inferClassType(this._cjson, this.typeName, this.isInferred, classes);
    }

    protected makeArray(arrays: NestedValueArray): TypeRef {
        return this.typeBuilder.getArrayType(
            this._typeInference.inferType(this._cjson, pluralize.singular(this.typeName), this.isInferred, arrays)
        );
    }
}

function canBeEnumCase(s: string): boolean {
    if (s.length === 0) return true; // FIXME: Do we really want this?