@sendoutcards/apollo-codegen-typescript/lib/printer.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const generator_1 = require("@babel/generator");
class Printer {
    constructor() {
        this.printQueue = [];
    }
    print() {
        return (this.printQueue.reduce((document, printable) => {
            if (typeof printable === 'string') {
                return document + printable;
            }
            else {
                const documentPart = generator_1.default(printable).code;
                return document + this.indentComments(documentPart);
            }
        }, '') + '\n');
    }
    enqueue(printable) {
        if (this.printQueue.length > 0) {
            this.printQueue.push('\n');
            this.printQueue.push('\n');
        }
        this.printQueue.push(printable);
    }
    printAndClear() {
        const output = this.print();
        this.printQueue = [];
        return output;
    }
    indentComments(documentPart) {
        const lines = documentPart.split('\n').filter(Boolean);
        let currentLine = 0;
        const newDocumentParts = [];
        let maxCommentColumn = 0;
        while (currentLine !== lines.length) {
            const currentLineContents = lines[currentLine];
            const commentColumn = currentLineContents.indexOf('//');
            if (commentColumn > 0) {
                if (maxCommentColumn < commentColumn) {
                    maxCommentColumn = commentColumn;
                }
                const [contents, comment] = currentLineContents.split('//');
                newDocumentParts.push({
                    main: contents.replace(/\s+$/g, ''),
                    comment: comment ? comment.trim() : null
                });
            }
            else {
                newDocumentParts.push({
                    main: currentLineContents,
                    comment: null
                });
            }
            currentLine++;
        }
        return newDocumentParts
            .reduce((memo, part) => {
            const { main, comment } = part;
            let line;
            if (comment !== null) {
                const spacesBetween = maxCommentColumn - main.length;
                line = `${main}${' '.repeat(spacesBetween)} // ${comment.trim()}`;
            }
            else {
                line = main;
            }
            return [...memo, line];
        }, [])
            .join('\n');
    }
}
exports.default = Printer;
//# sourceMappingURL=printer.js.map