formats.ts

import { getSemver, isObject } from '../utils';
import { schemas } from '@asyncapi/specs';

import type { Format } from '@stoplight/spectral-core';
import type { MaybeAsyncAPI } from '../types';

export class Formats extends Map<string, Format> {
  filterByMajorVersions(majorsToInclude: string[]): Formats {
    return new Formats([...this.entries()].filter(element => {return majorsToInclude.includes(element[0].split('.')[0]);}));
  }

  excludeByVersions(versionsToExclude: string[]): Formats {
    return new Formats([...this.entries()].filter(element => {return !versionsToExclude.includes(element[0]);}));
  }

  find(version: string): Format | undefined {
    return this.get(formatVersion(version));
  }

  formats(): Format[] {
    return [...this.values()];
  } 
}

export const AsyncAPIFormats = new Formats(Object.entries(schemas).reverse().map(([version]) => [version, createFormat(version)])); // reverse is used for giving newer versions a higher priority when matching

function isAsyncAPIVersion(versionToMatch: string, document: unknown): document is { asyncapi: string } & Record<string, unknown> {
  const asyncAPIDoc = document as MaybeAsyncAPI;
  if (!asyncAPIDoc) return false;

  const documentVersion = String(asyncAPIDoc.asyncapi);
  return isObject(document) && 'asyncapi' in document 
    && assertValidAsyncAPIVersion(documentVersion)
    && versionToMatch === formatVersion(documentVersion);
}
  
function assertValidAsyncAPIVersion(documentVersion: string): boolean {
  const semver = getSemver(documentVersion);
  const regexp = new RegExp(`^(${semver.major})\\.(${semver.minor})\\.(0|[1-9][0-9]*)$`); // eslint-disable-line security/detect-non-literal-regexp
  return regexp.test(documentVersion);
}

function createFormat(version: string): Format {
  const format: Format = (document: unknown): boolean =>
    isAsyncAPIVersion(version, document);
  
  const semver = getSemver(version);
  format.displayName = `AsyncAPI ${semver.major}.${semver.minor}.x`;

  return format; 
}

const formatVersion = function (version: string): string {
  const versionSemver = getSemver(version);
  return `${versionSemver.major}.${versionSemver.minor}.0`;
};