@diplodoc/transform/src/transform/title.ts

A simple transformer of text in YFM (Yandex Flavored Markdown) to HTML

import Token from 'markdown-it/lib/token';

export = function extractTitle(tokens: Token[]) {
    let title = '',
        contentTokens = tokens,
        titleTokens: Token[] = [];

    if (Array.isArray(tokens) && tokens.length > 0) {
        if (tokens[0].type === 'heading_open' && tokens[0].tag === 'h1') {
            titleTokens = tokens[1].children || [];
            title = tokens[1].content;
            // cut out "heading_open", "inline" and "heading_close" tokens
            contentTokens = tokens.slice(3);
        }
    }

    return {
        titleTokens,
        title,
        tokens: contentTokens,
    };
};