WireMock.Net/src/WireMock.Net.Minimal/ResponseMessageBuilder.cs

// Copyright © WireMock.Net

using System.Net;
using WireMock.Admin.Mappings;
using WireMock.Constants;
using WireMock.Http;
using WireMock.Types;
using WireMock.Util;

namespace WireMock;

internal class ResponseMessageBuilder(IDateTimeUtils dateTimeUtils) : IResponseMessageBuilder
{
    private static readonly IDictionary<string, WireMockList<string>> ContentTypeJsonHeaders = new Dictionary<string, WireMockList<string>>
    {
        { HttpKnownHeaderNames.ContentType, new WireMockList<string> { WireMockConstants.ContentTypeJson } }
    };

    public ResponseMessage Create(HttpStatusCode statusCode, string? status, Guid? guid = null)
    {
        return Create((int)statusCode, status, null, guid);
    }

    public ResponseMessage Create(int statusCode, string? status, Guid? guid = null)
    {
        return Create(statusCode, status, null, guid);
    }

    public ResponseMessage Create(int statusCode, string? status, string? error, Guid? guid = null)
    {
        var response = new ResponseMessage
        {
            StatusCode = statusCode,
            Headers = ContentTypeJsonHeaders,
            DateTime = dateTimeUtils.UtcNow
        };

        if (status != null || error != null)
        {
            response.BodyData = new BodyData
            {
                DetectedBodyType = BodyType.Json,
                BodyAsJson = new StatusModel
                {
                    Guid = guid,
                    Status = status,
                    Error = error
                }
            };
        }

        return response;
    }

    public ResponseMessage Create(HttpStatusCode statusCode)
    {
        return new ResponseMessage
        {
            StatusCode = statusCode
        };
    }
}