digitransit-ui/app/util/RouteContext.js

/* eslint-disable react/prop-types */
import React, { useMemo } from 'react';

const RouteContext = React.createContext(null);

const useRoute = () => {
  const context = React.useContext(RouteContext);
  if (!context) {
    throw new Error('useRoute must be used within a RouteContext.Provider');
  }
  return context;
};

const RouteContextProvider = ({ value, children }) => {
  const memoizedValue = useMemo(
    () => value,
    [value.match, value.props, value.error],
  );
  return (
    <RouteContext.Provider value={memoizedValue}>
      {children}
    </RouteContext.Provider>
  );
};

const withRouteContext = render => args => {
  const { Component, props, error, match } = args;
  if (Component && (props || error)) {
    const value = { match, props, error };
    return (
      <RouteContextProvider value={value}>
        {render ? (
          render(args)
        ) : (
          <Component {...props} match={match} error={error} />
        )}
      </RouteContextProvider>
    );
  }
  return null;
};

export { RouteContext, useRoute, withRouteContext };