AFFiNE-Mirror/libs/components/board-state/src/data/migrate.ts

/* eslint-disable @typescript-eslint/ban-ts-comment */
import {
    Decoration,
    TDDocument,
    TDShapeType,
} from '@toeverything/components/board-types';

export function migrate(document: TDDocument, newVersion: number): TDDocument {
    const { version = 0 } = document;

    if (!('assets' in document)) {
        document.assets = {};
    }

    // Remove unused assets when loading a document
    const assetIdsInUse = new Set<string>();

    Object.values(document.pages).forEach(page =>
        Object.values(page.shapes).forEach(shape => {
            const { parentId, children, assetId } = shape;

            if (assetId) {
                assetIdsInUse.add(assetId);
            }

            // Fix missing parent bug
            if (parentId !== page.id && !page.shapes[parentId]) {
                console.warn('Encountered a shape with a missing parent!');
                shape.parentId = page.id;
            }

            if (shape.type === TDShapeType.Group && children) {
                children.forEach(childId => {
                    if (!page.shapes[childId]) {
                        console.warn(
                            'Encountered a parent with a missing child!',
                            shape.id,
                            childId
                        );
                        children?.splice(children.indexOf(childId), 1);
                    }
                });

                // TODO: Remove the shape if it has no children
            }
        })
    );

    Object.keys(document.assets).forEach(assetId => {
        if (!assetIdsInUse.has(assetId)) {
            delete document.assets[assetId];
        }
    });

    if (version === newVersion) return document;

    // if (version < 14) {
    //     Object.values(document.pages).forEach(page => {
    //         Object.values(page.shapes)
    //             // .filter(shape => shape.type === TDShapeType.Text)
    //             .forEach(
    //                 shape =>
    //                     (shape as TextShape).style.font === FontStyle.Script
    //             );
    //     });
    // }

    // Lowercase styles, move binding meta to binding
    if (version <= 13) {
        Object.values(document.pages).forEach(page => {
            Object.values(page.bindings).forEach(binding => {
                // eslint-disable-next-line @typescript-eslint/no-explicit-any
                Object.assign(binding, (binding as any).meta);
            });

            Object.values(page.shapes).forEach(shape => {
                Object.entries(shape.style).forEach(([id, style]) => {
                    if (typeof style === 'string') {
                        // @ts-ignore
                        shape.style[id] = style.toLowerCase();
                    }
                });

                if (shape.type === TDShapeType.Arrow) {
                    if (shape.decorations) {
                        Object.entries(shape.decorations).forEach(
                            ([id, decoration]) => {
                                if ((decoration as unknown) === 'Arrow') {
                                    shape.decorations = {
                                        ...shape.decorations,
                                        [id]: Decoration.Arrow,
                                    };
                                }
                            }
                        );
                    }
                }
            });
        });
    }

    // Add document name and file system handle
    if (version <= 13.1) {
        document.name = 'New Document';
    }

    if (version < 15) {
        document.assets = {};
    }

    Object.values(document.pages).forEach(page => {
        Object.values(page.shapes).forEach(shape => {
            if (version < 15.2) {
                if (
                    shape.type === TDShapeType.Image ||
                    shape.type === TDShapeType.Video
                ) {
                    shape.style.isFilled = true;
                }
            }

            if (version < 15.3) {
                if (
                    shape.type === TDShapeType.Rectangle ||
                    shape.type === TDShapeType.Triangle ||
                    shape.type === TDShapeType.Ellipse ||
                    shape.type === TDShapeType.Arrow
                ) {
                    shape.label = (shape as any).text || '';
                    shape.labelPoint = [0.5, 0.5];
                }
            }
        });
    });

    // Cleanup
    Object.values(document.pageStates).forEach(pageState => {
        pageState.selectedIds = pageState.selectedIds.filter(id => {
            return document.pages[pageState.id].shapes[id] !== undefined;
        });
        pageState.bindingId = undefined;
        pageState.editingId = undefined;
        pageState.hoveredId = undefined;
        pageState.pointedId = undefined;
    });

    document.version = newVersion;

    return document;
}