import React from 'react';
import { TriggerMockContext } from '../../../tests/shared/demoTestContext';

const { default: OriginPortal } = await vi.importActual<typeof import('rc-util/es/Portal')>('rc-util/es/Portal');

class MockPortal extends React.Component<{ children?: React.ReactNode }> {
  container: boolean;

  static contextType = TriggerMockContext;

  componentDidMount() {
    this.createContainer();
  }

  createContainer() {
    this.container = true;
    this.forceUpdate();
  }

  render() {
    const { children } = this.props;
    if (this.container) {
      return children;
    }
    return null;
  }
}

export default React.forwardRef((props: any, ref: any) => {
  const context = React.useContext(TriggerMockContext);

  if (context?.mock === false) {
    return <OriginPortal {...props} ref={ref} />;
  }

  return <MockPortal {...props} />;
});