import { getMatchingStatusCodeRange } from '../../utils';

import type { Oas2Rule, Oas3Rule } from '../../visitors';
import type { UserContext } from '../../walk';
import type { Oas3Response } from '../../typings/openapi';
import type { Oas2Response } from '../../typings/swagger';

export const ResponseContainsHeader: Oas3Rule | Oas2Rule = (options) => {
  const names: Record<string, string[]> = options.names || {};
  return {
    Operation: {
      Response: {
        enter: (response: Oas2Response | Oas3Response, { report, location, key }: UserContext) => {
          const expectedHeaders =
            names[key] ||
            names[getMatchingStatusCodeRange(key)] ||
            names[getMatchingStatusCodeRange(key).toLowerCase()] ||
            [];
          for (const expectedHeader of expectedHeaders) {
            if (
              !response?.headers ||
              !Object.keys(response?.headers).some(
                (header) => header.toLowerCase() === expectedHeader.toLowerCase()
              )
            ) {
              report({
                message: `Response object must contain a "${expectedHeader}" header.`,
                location: location.child('headers').key(),
              });
            }
          }
        },
      },
    },
  };
};