semi-design/packages/semi-ui/_cssAnimation/index.tsx

import React, { CSSProperties, ReactNode } from 'react';
import { isEqual, noop } from "lodash";


interface AnimationEventsNeedBind {
    onAnimationStart: (e: React.AnimationEvent) => void
    onAnimationEnd: (e: React.AnimationEvent) => void

    [key: string]: (e: any) => void;
}

interface AnimationProps {
    startClassName?: string;
    endClassName?: string;
    children: ({}: {
        animationClassName: string,
        animationStyle: CSSProperties,
        animationEventsNeedBind: AnimationEventsNeedBind
    }) => ReactNode
    animationState: "enter" | "leave"
    onAnimationEnd?: () => void;
    onAnimationStart?: () => void;
}

interface AnimationState {
    currentClassName: string
    extraStyle: CSSProperties
}


class CSSAnimation extends React.Component<AnimationProps, AnimationState> {
    constructor(props) {
        super(props);
        this.state = {
            currentClassName: this.props.startClassName,
            extraStyle: {}
        };
    }

    componentDidUpdate(prevProps: Readonly<AnimationProps>, prevState: Readonly<AnimationState>, snapshot?: any) {
        const changedKeys = Object.keys(this.props).filter(key => !isEqual(this.props[key], prevProps[key]));
        if (changedKeys.includes("animationState")) {
        }
        if (changedKeys.includes("startClassName")) {
            this.setState({
                currentClassName: this.props.startClassName,
                extraStyle: {}
            }, this.props.onAnimationStart ?? noop);
        }

    }

    handleAnimationStart = () => {
        this.props.onAnimationStart?.();
    }


    handleAnimationEnd = () => {
        this.setState({
            currentClassName: this.props.endClassName,
            extraStyle: {}
        }, () => {
            this.props.onAnimationEnd?.();
        });
    }


    render() {
        return this.props.children({
            animationClassName: this.state.currentClassName ?? "",
            animationStyle: this.state.extraStyle,
            animationEventsNeedBind: {
                onAnimationStart: this.handleAnimationStart,
                onAnimationEnd: this.handleAnimationEnd
            }
        });
    }
}


const mergeAnimationFunction = (eventHandleFunctions: AnimationEventsNeedBind[]) => {
    //merge function in objects
    const mergedFunction = {};
    eventHandleFunctions.forEach(eventHandleFunction => {
        Object.keys(eventHandleFunction).forEach(key => {
            if (mergedFunction[key]) {
                const oldFunction = mergedFunction[key];
                mergedFunction[key] = (e) => {
                    eventHandleFunction[key](e);
                    oldFunction(e);
                };
            } else {
                mergedFunction[key] = eventHandleFunction[key];
            }
        });
    });
    return mergedFunction;
};

export { mergeAnimationFunction };
export default CSSAnimation;