AFFiNE-Mirror/libs/components/editor-blocks/src/blocks/code/index.ts

import { Protocol } from '@toeverything/datasource/db-service';
import {
    AsyncBlock,
    BaseView,
    BlockEditor,
    HTML2BlockResult,
} from '@toeverything/framework/virgo';
import { unescape } from 'html-escaper';
import {
    Block2HtmlProps,
    commonBlock2HtmlContent,
} from '../../utils/commonBlockClip';
import { CodeView } from './CodeView';

export class CodeBlock extends BaseView {
    type = Protocol.Block.Type.code;
    public override selectable = true;
    public override editable = true;
    // View = CodeView;
    View = CodeView;
    override async onCreate(block: AsyncBlock): Promise<AsyncBlock> {
        if (!block.getProperty('text')) {
            await block.setProperty('text', {
                value: [{ text: '' }],
            });
        }
        return block;
    }

    override async html2block({
        element,
        editor,
    }: {
        element: Element;
        editor: BlockEditor;
    }): Promise<HTML2BlockResult> {
        // debugger;
        if (element.tagName === 'CODE') {
            return [
                {
                    type: this.type,
                    properties: {
                        text: {
                            value: [
                                {
                                    text: unescape(
                                        (element as HTMLElement).innerText
                                    ),
                                },
                            ],
                        },
                        lang:
                            element.classList[0] &&
                            element.classList[0].substr(9),
                    },
                    children: [],
                },
            ];
        }
        return null;
    }

    override async block2html(props: Block2HtmlProps) {
        return `<code>${await commonBlock2HtmlContent(props)}<code/>`;
    }
}