ExportAssignment.ts

import { ts, SyntaxKind } from "../../../typescript";
import { Expression } from "../expression";
import { Statement } from "../statement";
import { ExportAssignmentStructure, ExportAssignmentSpecificStructure, StructureKind } from "../../../structures";
import { WriterFunction } from "../../../types";
import { callBaseGetStructure } from "../callBaseGetStructure";
import { callBaseSet } from "../callBaseSet";

export const ExportAssignmentBase = Statement;
export class ExportAssignment extends ExportAssignmentBase<ts.ExportAssignment> {
    /**
     * Gets if this is an export equals assignment.
     *
     * If this is false, then it's `export default`.
     */
    isExportEquals() {
        return this.compilerNode.isExportEquals || false;
    }

    /**
     * Sets if this is an export equals assignment or export default.
     * @param value - Whether it should be an export equals assignment.
     */
    setIsExportEquals(value: boolean) {
        if (this.isExportEquals() === value)
            return this;

        if (value)
            this.getFirstChildByKindOrThrow(SyntaxKind.DefaultKeyword).replaceWithText("=");
        else
            this.getFirstChildByKindOrThrow(SyntaxKind.EqualsToken).replaceWithText("default");

        return this;
    }

    /**
     * Gets the export assignment expression.
     */
    getExpression(): Expression {
        return this._getNodeFromCompilerNode(this.compilerNode.expression);
    }

    /**
     * Sets the expression of the export assignment.
     * @param textOrWriterFunction - Text or writer function to set as the export assignment expression.
     */
    setExpression(textOrWriterFunction: string | WriterFunction) {
        this.getExpression().replaceWithText(textOrWriterFunction, this._getWriterWithQueuedChildIndentation());
        return this;
    }

    /**
     * Sets the node from a structure.
     * @param structure - Structure to set the node with.
     */
    set(structure: Partial<ExportAssignmentStructure>) {
        callBaseSet(ExportAssignmentBase.prototype, this, structure);

        if (structure.expression != null)
            this.setExpression(structure.expression);
        if (structure.isExportEquals != null)
            this.setIsExportEquals(structure.isExportEquals);

        return this;
    }

    /**
     * Gets the structure equivalent to this node.
     */
    getStructure(): ExportAssignmentStructure {
        return callBaseGetStructure<ExportAssignmentSpecificStructure>(Statement.prototype, this, {
            kind: StructureKind.ExportAssignment,
            expression: this.getExpression().getText(),
            isExportEquals: this.isExportEquals()
        }) as any as ExportAssignmentStructure;
    }
}