// Tooltip decorator for simple composition with other components

import React, { HTMLAttributes, ReactNode } from "react"
import hoistNonReactStatics from "hoist-non-react-statics";
import { Tooltip, TooltipProps } from "./tooltip";
import { isReactNode } from "../../utils/isReactNode";
import uniqueId from "lodash/uniqueId"

export interface TooltipDecoratorProps {
  tooltip?: ReactNode | Omit<TooltipProps, "htmlFor">;
}

export function withTooltip<T extends React.ComponentType<any>>(Target: T): T {
  const DecoratedComponent = class extends React.Component<HTMLAttributes<any> & TooltipDecoratorProps> {
    static displayName = `withTooltip(${Target.displayName || Target.name})`;

    protected tooltipId = uniqueId("tooltip_target_");

    render() {
      const { tooltip, ...targetProps } = this.props;

      if (tooltip) {
        const tooltipId = targetProps.id || this.tooltipId;
        const tooltipProps: TooltipProps = {
          htmlFor: tooltipId,
          following: true,
          ...(isReactNode(tooltip) ? { children: tooltip } : tooltip),
        };
        if (!tooltipProps.following) {
          targetProps.style = {
            position: "relative",
            ...(targetProps.style || {})
          }
        }
        targetProps.id = tooltipId;
        targetProps.children = (
          <>
            {targetProps.children}
            <Tooltip {...tooltipProps}/>
          </>
        )
      }
      return <Target {...targetProps as any}/>;
    }
  }

  return hoistNonReactStatics(DecoratedComponent, Target) as any;
}