Merge pull request #382 from notion-dotnet/253-expose-retry-after-value

Expose Retry-After value

Author: KoditkarVedant

Src/Notion.Client/NotionApiException.cs

@@ -5,7 +5,7 @@
 namespace Notion.Client
 {
     [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")]
-    public sealed class NotionApiException : Exception
+    public class NotionApiException : Exception
     {
         public NotionApiException(HttpStatusCode statusCode, NotionAPIErrorCode? notionAPIErrorCode, string message)
             : this(statusCode, notionAPIErrorCode, message, null)
@@ -21,6 +21,11 @@ private NotionApiException(
             NotionAPIErrorCode = notionAPIErrorCode;
             StatusCode = statusCode;
 
+            InitializeData();
+        }
+
+        private void InitializeData()
+        {
             Data.Add("StatusCode", StatusCode);
             Data.Add("NotionApiErrorCode", NotionAPIErrorCode);
         }

Src/Notion.Client/NotionApiRateLimitException.cs

@@ -0,0 +1,22 @@
+using System;
+using System.Net;
+
+namespace Notion.Client
+{
+    public sealed class NotionApiRateLimitException : NotionApiException
+    {
+        public TimeSpan? RetryAfter { get; }
+
+        public NotionApiRateLimitException(
+            HttpStatusCode statusCode,
+            NotionAPIErrorCode? notionAPIErrorCode,
+            string message,
+            TimeSpan? retryAfter)
+            : base(statusCode, notionAPIErrorCode, message)
+        {
+            RetryAfter = retryAfter;
+
+            Data.Add("RetryAfter", retryAfter);
+        }
+    }
+}

Src/Notion.Client/RestClient/RestClient.cs

@@ -112,6 +112,17 @@ private static async Task<Exception> BuildException(HttpResponseMessage response
                 try
                 {
                     errorResponse = JsonConvert.DeserializeObject<NotionApiErrorResponse>(errorBody);
+
+                    if (errorResponse.ErrorCode == NotionAPIErrorCode.RateLimited)
+                    {
+                        var retryAfter = response.Headers.RetryAfter.Delta;
+                        return new NotionApiRateLimitException(
+                            response.StatusCode,
+                            errorResponse.ErrorCode,
+                            errorResponse.Message,
+                            retryAfter
+                        );
+                    }
                 }
                 catch (Exception ex)
                 {