import {Context} from 'react';
import type {MatcherState} from 'expect';
import {
  matcherHint,
  printExpected,
  EXPECTED_COLOR as expectedColor,
  RECEIVED_COLOR as receivedColor,
} from 'jest-matcher-utils';

import type {Node} from '../types.ts';

import {assertIsNode, diffs, printType} from './utilities.ts';

export function toProvideReactContext<Type>(
  this: MatcherState,
  node: Node<any, any>,
  Context: Context<Type>,
  value?: Type,
) {
  assertIsNode(node, {
    expectation: 'toProvideReactContext',
    isNot: this.isNot,
  });

  const foundByType = node.findAll(Context.Provider);
  const foundByValue =
    value == null
      ? foundByType
      : foundByType.filter((element) =>
          Object.is(value, element.prop('value')),
        );

  const pass = foundByValue.length > 0;

  const message = pass
    ? () =>
        `${matcherHint('.not.toProvideReactContext')}\n\n` +
        `Expected the React element:\n  ${receivedColor(node.toString())}\n` +
        `Not to contain context provider:\n  ${expectedColor(
          printType(Context.Provider),
        )}\n${
          value ? `With value matching:\n  ${printExpected(value)}\n` : ''
        }` +
        `But ${foundByValue.length} matching ${printType(Context.Provider)}${
          foundByValue.length === 1 ? 's were' : ' was'
        } found.\n`
    : () =>
        `${
          `${matcherHint('.toProvideReactContext')}\n\n` +
          `Expected the React element:\n  ${receivedColor(node.toString())}\n` +
          `To contain context provider:\n  ${expectedColor(
            printType(Context.Provider),
          )}\n${
            value ? `With value matching:\n  ${printExpected(value)}\n` : ''
          }`
        }${
          foundByType.length === 0
            ? `But no matching ${printType(Context.Provider)}s were found.\n`
            : `But the ${
                foundByType.length === 1
                  ? 'found provider has'
                  : 'found provider have'
              } had different values:\n\n${diffs(
                foundByType,
                {value},
                this.expand,
              )}`
        }`;

  return {pass, message};
}