// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.OpenApi.Models;
using Microsoft.OpenApi.Models.References;
 
namespace Microsoft.AspNetCore.OpenApi;
 
internal static class OpenApiDocumentExtensions
{
    /// <summary>
    /// Registers a <see cref="OpenApiSchema" /> into the top-level components store on the
    /// <see cref="OpenApiDocument" /> and returns a resolvable reference to it.
    /// </summary>
    /// <param name="document">The <see cref="OpenApiDocument"/> to register the schema onto.</param>
    /// <param name="schemaId">The ID that serves as the key for the schema in the schema store.</param>
    /// <param name="schema">The <see cref="OpenApiSchema" /> to register into the document.</param>
    /// <returns>An <see cref="OpenApiSchema"/> with a reference to the stored schema.</returns>
    public static OpenApiSchema AddOpenApiSchemaByReference(this OpenApiDocument document, string schemaId, OpenApiSchema schema)
    {
        document.Components ??= new();
        document.Components.Schemas ??= new Dictionary<string, OpenApiSchema>();
        document.Components.Schemas[schemaId] = schema;
        document.Workspace ??= new();
        var location = document.BaseUri + "/components/schemas/" + schemaId;
        document.Workspace.RegisterComponentForDocument(document, schema, location);
        return new OpenApiSchemaReference(schemaId, document);
    }
}