import { useContext } from 'react';
import PropTypes from 'prop-types';
import invariant from 'fbjs/lib/invariant';
import { __RouterContext as RouterContext } from 'react-router';
import { createLocation } from 'history';

export const LinkPropTypes = {
  to: PropTypes.string,
  replace: PropTypes.bool,
  external: PropTypes.oneOfType([
    PropTypes.bool,
    PropTypes.exact({
      download: PropTypes.bool,
      target: PropTypes.oneOf(['self', 'blank']),
      rel: PropTypes.string,
    }),
  ]),
};

export default function useLink(to, replace, external) {
  if (external) {
    invariant(to, "'external' needs 'to' value.");
  }
  if (replace) {
    invariant(to, "'replace' needs 'to' value.");
  }

  if (!to) {
    return undefined;
  }

  // Element is link.
  const { history, location } = useContext(RouterContext);

  if (external) {
    const hrefAttrs = {
      download: false,
      target: '_blank',
      rel: 'noopener noreferrer',
    };

    if (typeof external === 'object') {
      Object.assign(hrefAttrs, {
        ...external,
        target: external.target ? `_${external.target}` : '_blank',
      });
    }

    return {
      props: {
        accessibilityRole: 'link',
        href: to,
        ...hrefAttrs,
      },
    };
  }

  const linkLocation =
    typeof to === 'string' ? createLocation(to, null, null, location) : to;
  const href = linkLocation ? history.createHref(linkLocation) : '';

  return {
    props: {
      accessibilityRole: 'link',
      href,
      onPress: (event) => {
        if (event.defaultPrevented) {
          return;
        }

        event.preventDefault();

        if (replace) {
          history.replace(to);
        } else {
          history.push(to);
        }
      },
    },
  };
}

@@ -260,30 +260,71 @@
 : PropTypes.
-string
+oneOfType(%5BPropTypes.string, PropTypes.object%5D)
 ,%0A  replace:

Update Link to prop type