import React from "react";
import {
    TransitionMotion,
    spring,
    OpaqueConfig,
    TransitionStyle,
} from "@serprex/react-motion";
import cxs from "cxs";
import isEqual from "react-fast-compare";
import { requestTimeout, clearRequestTimeout, RequestTimeout } from "../utils";

type Props = {
    children?: (JSX.Element | string)[];
    interval: number | number[];
    delay: number;
    adjustingSpeed: number;
    springConfig: {
        stiffness: number;
        damping: number;
    };
    fade: boolean;
    mask: boolean;
    noWrap: boolean;
    className?: string;
    onChange?: Function;
};

type State = {
    elements: (JSX.Element | string | number | {} | undefined)[];
    currentEl: JSX.Element | string | number | {} | undefined;
    currentWordIndex: number;
    wordCount: number;
    currentInterval: number;
};

class TextLoop extends React.PureComponent<Props, State> {
    isUnMounting = false;

    tickDelay: RequestTimeout = 0;

    tickLoop: RequestTimeout = 0;

    wordBox: HTMLDivElement | null = null;

    static defaultProps: Props = {
        interval: 3000,
        delay: 0,
        adjustingSpeed: 150,
        springConfig: { stiffness: 340, damping: 30 },
        fade: true,
        mask: false,
        noWrap: true,
    };

    constructor(props: Props) {
        super(props);
        const elements = React.Children.toArray(props.children);

        this.state = {
            elements,
            currentEl: elements[0],
            currentWordIndex: 0,
            wordCount: 0,
            currentInterval: Array.isArray(props.interval)
                ? props.interval[0]
                : props.interval,
        } as State;
    }

    componentDidMount(): void {
        // Starts animation
        const { delay } = this.props;
        const { currentInterval, elements } = this.state;

        if (currentInterval > 0 && elements.length > 1) {
            this.tickDelay = requestTimeout(() => {
                this.tickLoop = requestTimeout(this.tick, currentInterval);
            }, delay);
        }
    }

    componentDidUpdate(prevProps: Props, prevState: State): void {
        const { interval, children, delay } = this.props as Props;
        const { currentWordIndex } = this.state;

        const currentInterval = Array.isArray(interval)
            ? interval[currentWordIndex % interval.length]
            : interval;

        if (prevState.currentInterval !== currentInterval) {
            this.clearTimeouts();

            if (currentInterval > 0 && React.Children.count(children) > 1) {
                this.tickDelay = requestTimeout(() => {
                    this.tickLoop = requestTimeout(this.tick, currentInterval);
                }, delay);
            } else {
                this.setState((state, props) => {
                    const { currentWordIndex: _currentWordIndex } = state;

                    return {
                        currentInterval: Array.isArray(props.interval)
                            ? props.interval[
                                  _currentWordIndex % props.interval.length
                              ]
                            : props.interval,
                    };
                });
            }
        }

        if (!isEqual(prevProps.children, children)) {
            // eslint-disable-next-line react/no-did-update-set-state
            this.setState({
                elements: React.Children.toArray(children),
            });
        }
    }

    componentWillUnmount(): void {
        this.isUnMounting = true;
        this.clearTimeouts();
    }

    clearTimeouts(): void {
        if (this.tickLoop != null) {
            clearRequestTimeout(this.tickLoop);
        }

        if (this.tickDelay != null) {
            clearRequestTimeout(this.tickDelay);
        }
    }

    // Fade out animation
    willLeave = (): { opacity: OpaqueConfig; translate: OpaqueConfig } => {
        const { height } = this.getDimensions();

        return {
            opacity: spring(this.getOpacity(), this.props.springConfig),
            translate: spring(-height, this.props.springConfig),
        };
    };

    // Fade in animation
    willEnter = (): { opacity: 0 | 1; translate: number } => {
        const { height } = this.getDimensions();

        return {
            opacity: this.getOpacity(),
            translate: height,
        };
    };

    tick = (): void => {
        if (!this.isUnMounting) {
            this.setState(
                (state, props) => {
                    const currentWordIndex =
                        (state.currentWordIndex + 1) % state.elements.length;

                    const currentEl = state.elements[currentWordIndex];
                    const updatedState = {
                        currentWordIndex,
                        currentEl,
                        wordCount: (state.wordCount + 1) % 1000, // just a safe value to avoid infinite counts,
                        currentInterval: Array.isArray(props.interval)
                            ? props.interval[
                                  currentWordIndex % props.interval.length
                              ]
                            : props.interval,
                    };
                    if (props.onChange) {
                        props.onChange(updatedState);
                    }

                    return updatedState;
                },
                () => {
                    if (this.state.currentInterval > 0) {
                        this.clearTimeouts();
                        this.tickLoop = requestTimeout(
                            this.tick,
                            this.state.currentInterval
                        );
                    }
                }
            );
        }
    };

    getOpacity(): 0 | 1 {
        return this.props.fade ? 0 : 1;
    }

    getDimensions(): ClientRect | DOMRect | { width: 0; height: 0 } {
        if (this.wordBox == null) {
            return {
                width: 0,
                height: 0,
            };
        }

        return this.wordBox.getBoundingClientRect();
    }

    wrapperStyles = cxs({
        ...(this.props.mask && { overflow: "hidden" }),
        ...{
            display: "inline-block",
            position: "relative",
            verticalAlign: "top",
        },
    });

    elementStyles = cxs({
        display: "inline-block",
        left: 0,
        top: 0,
        whiteSpace: this.props.noWrap ? "nowrap" : "normal",
    });

    getTransitionMotionStyles(): TransitionStyle[] {
        const { springConfig } = this.props;
        const { wordCount, currentEl } = this.state;

        return [
            {
                key: `step-${wordCount}`,
                data: {
                    currentEl,
                },
                style: {
                    opacity: spring(1, springConfig),
                    translate: spring(0, springConfig),
                },
            },
        ];
    }

    render(): JSX.Element {
        const { className = "" } = this.props;
        return (
            <div className={`${this.wrapperStyles} ${className}`}>
                <TransitionMotion
                    willLeave={this.willLeave}
                    willEnter={this.willEnter}
                    styles={this.getTransitionMotionStyles()}
                >
                    {(interpolatedStyles): JSX.Element => {
                        const { height, width } = this.getDimensions();

                        const parsedWidth =
                            this.wordBox == null ? "auto" : width;

                        const parsedHeight =
                            this.wordBox == null ? "auto" : height;

                        return (
                            <div
                                style={{
                                    transition: `width ${this.props.adjustingSpeed}ms linear`,
                                    height: parsedHeight,
                                    width: parsedWidth,
                                }}
                            >
                                {interpolatedStyles.map(config => (
                                    <div
                                        className={this.elementStyles}
                                        ref={(n: HTMLDivElement): void => {
                                            this.wordBox = n;
                                        }}
                                        key={config.key}
                                        style={{
                                            opacity: config.style.opacity,
                                            transform: `translateY(${config.style.translate}px)`,
                                            position:
                                                this.wordBox == null
                                                    ? "relative"
                                                    : "absolute",
                                        }}
                                    >
                                        {config.data.currentEl}
                                    </div>
                                ))}
                            </div>
                        );
                    }}
                </TransitionMotion>
            </div>
        );
    }
}

export default TextLoop;
