OpenApiEnumVarNamesExtension.cs

Go to the documentation of this file.

using Microsoft.OpenApi;
using Microsoft.OpenApi.Interfaces;
using Microsoft.OpenApi.Models;
using Microsoft.OpenApi.Writers;
using System;

namespace Tgstation.Server.Host.Core
{
    sealed class OpenApiEnumVarNamesExtension : IOpenApiExtension
    {
        readonly Type enumType;

        private OpenApiEnumVarNamesExtension(Type enumType)
        {
            this.enumType = enumType ?? throw new ArgumentNullException(nameof(enumType));
        }

        public static void Apply(OpenApiSchema openApiSchema, Type enumType)
        {
            if (openApiSchema == null)
                throw new ArgumentNullException(nameof(openApiSchema));

            openApiSchema.Extensions.Add("x-enum-varnames", new OpenApiEnumVarNamesExtension(enumType));
        }

        public void Write(IOpenApiWriter writer, OpenApiSpecVersion specVersion)
        {
            if (writer == null)
                throw new ArgumentNullException(nameof(writer));

            if (specVersion != OpenApiSpecVersion.OpenApi3_0)
                throw new InvalidOperationException("This extension only applies to OpenAPI 3.0!");

            writer.WriteStartArray();
            foreach (var enumValue in Enum.GetValues(enumType))
                writer.WriteValue(enumValue.ToString());

            writer.WriteEndArray();
        }
    }
}