import {
  type OpenAPIReference,
  type OpenAPIReferenceable,
  type OpenAPIExtension,
  type OpenAPIParameter,
  type OpenAPIExample,
  type OpenAPIRequestBody,
  type OpenAPIResponse,
  type OpenAPIHeader,
  type OpenAPICallback,
} from 'ya-open-api-types';
import { type JsonValue } from 'type-fest';
import {
  type WhookAPICallbackDefinition,
  type WhookAPIExampleDefinition,
  type WhookAPIHeaderDefinition,
  type WhookAPIParameterDefinition,
  type WhookAPIRequestBodyDefinition,
  type WhookAPIResponseDefinition,
  type WhookAPISchemaDefinition,
} from '../types/openapi.js';
import { type ExpressiveJSONSchema } from 'ya-json-schema-types';

export function refersTo<
  T extends
    | WhookAPISchemaDefinition<unknown>
    | WhookAPIParameterDefinition<unknown>
    | WhookAPIExampleDefinition<JsonValue>
    | WhookAPIHeaderDefinition
    | WhookAPIResponseDefinition
    | WhookAPIRequestBodyDefinition
    | WhookAPICallbackDefinition,
>(resource: T) {
  return {
    $ref: `#/components/${
      'schema' in resource
        ? 'schemas'
        : 'parameter' in resource
          ? 'parameters'
          : 'header' in resource
            ? 'headers'
            : 'response' in resource
              ? 'responses'
              : 'requestBody' in resource
                ? 'requestBodies'
                : 'callback' in resource
                  ? 'callbacks'
                  : 'examples'
    }/${resource.name}`,
  } as T extends WhookAPISchemaDefinition
    ? ExpressiveJSONSchema
    : OpenAPIReference<
        T extends WhookAPIParameterDefinition<unknown>
          ? OpenAPIParameter<ExpressiveJSONSchema, OpenAPIExtension>
          : T extends WhookAPIExampleDefinition<infer U>
            ? OpenAPIExample<OpenAPIExample<U>>
            : T extends WhookAPIHeaderDefinition
              ? OpenAPIHeader<ExpressiveJSONSchema, OpenAPIExtension>
              : T extends WhookAPIResponseDefinition
                ? OpenAPIResponse<ExpressiveJSONSchema, OpenAPIExtension>
                : T extends WhookAPIRequestBodyDefinition
                  ? OpenAPIRequestBody<ExpressiveJSONSchema, OpenAPIExtension>
                  : T extends WhookAPICallbackDefinition
                    ? OpenAPICallback<ExpressiveJSONSchema, OpenAPIExtension>
                    : OpenAPIReferenceable<
                        ExpressiveJSONSchema,
                        OpenAPIExtension
                      >
      >;
}