import {
  ChangeVariant,
  FactVariant,
  OpenApiKind,
} from '@useoptic/openapi-utilities';

export type NodeDetail<T extends OpenApiKind> = {
  before: FactVariant<T> | null;
  after: FactVariant<T> | null;
  change: ChangeVariant<T> | null;
};

type ContentType = string;
type StatusCode = string;

export type HeaderParameterNode = NodeDetail<OpenApiKind.HeaderParameter>;
export type PathParameterNode = NodeDetail<OpenApiKind.PathParameter>;
export type QueryParameterNode = NodeDetail<OpenApiKind.QueryParameter>;
export type CookieParameterNode = NodeDetail<OpenApiKind.CookieParameter>;

export type RequestNode = NodeDetail<OpenApiKind.Request> & {
  bodies: Map<ContentType, BodyNode>;
};

export type ResponseNode = NodeDetail<OpenApiKind.Response> & {
  statusCode: string;
  headers: Map<string, NodeDetail<OpenApiKind.ResponseHeader>>;
  bodies: Map<ContentType, BodyNode>;
};

export type BodyNode = NodeDetail<OpenApiKind.Body> & {
  fields: Map<string, NodeDetail<OpenApiKind.Field>>;
  schemas: Map<string, NodeDetail<OpenApiKind.Schema>>;
};

export type EndpointNode = NodeDetail<OpenApiKind.Operation> & {
  method: string;
  path: string;
  polymorphicSchemas: { before: Set<string>; after: Set<string> };

  headerParameters: Map<string, HeaderParameterNode>;
  pathParameters: Map<string, PathParameterNode>;
  queryParameters: Map<string, QueryParameterNode>;
  cookieParameters: Map<string, CookieParameterNode>;
  request: RequestNode;
  responses: Map<StatusCode, ResponseNode>;
};

export type OpenAPIFactNodes = {
  specification: NodeDetail<OpenApiKind.Specification>;
  endpoints: Map<string, EndpointNode>;
};