PluralKit/Myriad/Rest/Ratelimit/RatelimitHeaders.cs

using System;
using System.Globalization;
using System.Linq;
using System.Net.Http;

namespace Myriad.Rest.Ratelimit
{
    public record RatelimitHeaders
    {
        private const string LimitHeader = "X-RateLimit-Limit";
        private const string RemainingHeader = "X-RateLimit-Remaining";
        private const string ResetHeader = "X-RateLimit-Reset";
        private const string ResetAfterHeader = "X-RateLimit-Reset-After";
        private const string BucketHeader = "X-RateLimit-Bucket";
        private const string GlobalHeader = "X-RateLimit-Global";

        public bool Global { get; private set; }
        public int? Limit { get; private set; }
        public int? Remaining { get; private set; }
        public DateTimeOffset? Reset { get; private set; }
        public TimeSpan? ResetAfter { get; private set; }
        public string? Bucket { get; private set; }

        public DateTimeOffset? ServerDate { get; private set; }

        public bool HasRatelimitInfo =>
            Limit != null && Remaining != null && Reset != null && ResetAfter != null && Bucket != null;

        public RatelimitHeaders() { }

        public static RatelimitHeaders Parse(HttpResponseMessage response)
        {
            var headers = new RatelimitHeaders
            {
                ServerDate = response.Headers.Date,
                Limit = TryGetInt(response, LimitHeader),
                Remaining = TryGetInt(response, RemainingHeader),
                Bucket = TryGetHeader(response, BucketHeader)
            };


            var resetTimestamp = TryGetDouble(response, ResetHeader);
            if (resetTimestamp != null)
                headers.Reset = DateTimeOffset.FromUnixTimeMilliseconds((long)(resetTimestamp.Value * 1000));

            var resetAfterSeconds = TryGetDouble(response, ResetAfterHeader);
            if (resetAfterSeconds != null)
                headers.ResetAfter = TimeSpan.FromSeconds(resetAfterSeconds.Value);

            var global = TryGetHeader(response, GlobalHeader);
            if (global != null && bool.TryParse(global, out var globalBool))
                headers.Global = globalBool;

            return headers;
        }

        private static string? TryGetHeader(HttpResponseMessage response, string headerName)
        {
            if (!response.Headers.TryGetValues(headerName, out var values))
                return null;

            return values.FirstOrDefault();
        }

        private static int? TryGetInt(HttpResponseMessage response, string headerName)
        {
            var valueString = TryGetHeader(response, headerName);

            if (!int.TryParse(valueString, NumberStyles.Integer, CultureInfo.InvariantCulture, out var value))
                return null;

            return value;
        }

        private static double? TryGetDouble(HttpResponseMessage response, string headerName)
        {
            var valueString = TryGetHeader(response, headerName);

            if (!double.TryParse(valueString, NumberStyles.Float, CultureInfo.InvariantCulture, out var value))
                return null;

            return value;
        }
    }
}