// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Text;
using Microsoft.AspNetCore.Mvc.Formatters;
using Microsoft.Net.Http.Headers;
 
namespace BasicWebSite.Formatters;
 
public class CustomFormatter : TextOutputFormatter
{
    public string ContentType { get; private set; }
 
    public CustomFormatter(string contentType)
    {
        ContentType = contentType;
        SupportedMediaTypes.Add(MediaTypeHeaderValue.Parse(contentType));
        SupportedEncodings.Add(Encoding.GetEncoding("utf-8"));
    }
 
    public override bool CanWriteResult(OutputFormatterCanWriteContext context)
    {
        if (base.CanWriteResult(context))
        {
            var actionReturnString = context.Object as string;
            if (actionReturnString != null)
            {
                return true;
            }
        }
        return false;
    }
 
    public override async Task WriteResponseBodyAsync(OutputFormatterWriteContext context, Encoding selectedEncoding)
    {
        var response = context.HttpContext.Response;
        response.ContentType = ContentType + ";charset=utf-8";
        await response.WriteAsync(context.Object.ToString());
    }
}