import ts from "typescript";
import { Context, NodeParser } from "./NodeParser";
import { BaseType } from "./Type/BaseType";
import { DefinitionType } from "./Type/DefinitionType";
import { NeverType } from "./Type/NeverType";

export class TopRefNodeParser implements NodeParser {
    public constructor(protected childNodeParser: NodeParser, protected fullName: string, protected topRef: boolean) {}
    // hack to change the fullName by 'createSchemaByNodeKind' for parsing multiple node
    // other option would be adding a 'fullName' field to BaseType
    public setFullName(fullName: string) {
        this.fullName = fullName;
    }

    public createType(node: ts.Node, context: Context): BaseType {
        const baseType = this.childNodeParser.createType(node, context);

        if (baseType === undefined) {
            return new NeverType();
        }

        if (this.topRef && !(baseType instanceof DefinitionType)) {
            return new DefinitionType(this.fullName, baseType);
        } else if (!this.topRef && baseType instanceof DefinitionType) {
            return baseType.getType();
        } else {
            return baseType;
        }
    }
}