import { ModuleExtractor } from '../module/ModuleExtractor';
import { TypescriptCommentExtractor } from '../comment/TypescriptCommentExtractor';
import { ExportAssignmentInfo } from './ExportAssignmentInfo';
import { ExportAssignment, SyntaxKind, SourceFile } from 'ts-morph';
export class ExportAssignmentExtractor {
    public extract(sourceFile: SourceFile): ExportAssignmentInfo[] | undefined {
        const result: ExportAssignmentInfo[] = [];
        sourceFile.forEachDescendant(node => {
            switch (node.getKind()) {
                case SyntaxKind.ExportAssignment:
                    const x = node as ExportAssignment;
                    const isExportDefault = !x.isExportEquals();
                    const trailingComments = new TypescriptCommentExtractor().extract(x.getTrailingCommentRanges());
                    const leadingComments = new TypescriptCommentExtractor().extract(x.getLeadingCommentRanges());
                    const hasComment = trailingComments.length !== 0 || leadingComments.length !== 0;
                    const expression: ExportAssignmentInfo = {
                        text: x.getText(),
                        hasComment: hasComment,
                        trailingComments: trailingComments.length === 0 ? undefined : trailingComments,
                        leadingComments: leadingComments.length === 0 ? undefined : leadingComments,
                        modules: new ModuleExtractor().extract(x),
                        isExportDefault: isExportDefault,
                    };
                    result.push(expression);
                    break;
            }
        });
        return result.length === 0 ? undefined : result;
    }
}