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
}

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
            }
        });
    }

}

export default CSSAnimation;