Added count to ModeledResponse (#103)

* Added GetWithCount method

* Made improvements

* Changed existing get method to retrieve count, added count to modeledresponse

* Fixed issue when making non get requests

* Added tests

Author: DanielW093

Postgrest/Interfaces/IPostgrestTable.cs

@@ -116,8 +116,9 @@ IPostgrestTable<TModel> Filter<TCriterion>(Expression<Func<TModel, object>> pred
         /// Executes the query using the defined filters on the current instance.
         /// </summary>
         /// <param name="cancellationToken"></param>
+        /// <param name="countType"></param>
         /// <returns></returns>
-        Task<ModeledResponse<TModel>> Get(CancellationToken cancellationToken = default);
+        Task<ModeledResponse<TModel>> Get(CancellationToken cancellationToken = default, Constants.CountType countType = Constants.CountType.Estimated);
 
         /// <summary>
         /// Executes a BULK INSERT query using the defined query params on the current instance.

Postgrest/Responses/ModeledResponse.cs

@@ -24,6 +24,11 @@ namespace Supabase.Postgrest.Responses
 		/// A list of models in the response.
 		/// </summary>
 		public List<T> Models { get; } = new();
+
+		/// <summary>
+		/// The number of results matching the specified filters
+		/// </summary>
+		public int Count = 0;
 
 		/// <inheritdoc />
 		public ModeledResponse(BaseResponse baseResponse, JsonSerializerSettings serializerSettings, Func<Dictionary<string, string>>? getHeaders = null, bool shouldParse = true) : base(baseResponse.ClientOptions, baseResponse.ResponseMessage, baseResponse.Content)
@@ -72,6 +77,18 @@ public ModeledResponse(BaseResponse baseResponse, JsonSerializerSettings seriali
 				}
 			}
 
+			try
+			{
+				var countStr = baseResponse.ResponseMessage?.Content.Headers.GetValues("Content-Range")
+					.FirstOrDefault();
+				Count = int.Parse(countStr?.Split('/')[1] ?? throw new InvalidOperationException());
+			}
+			catch (Exception e)
+			{
+				Debugger.Instance.Log(this, e.Message);
+				Count = -1;
+			}
+
 			Debugger.Instance.Log(this, $"Response: [{baseResponse.ResponseMessage?.StatusCode}]\n" + $"Parsed Models <{typeof(T).Name}>:\n\t{JsonConvert.SerializeObject(Models)}\n");
 		}
 	}

Postgrest/Table.cs

@@ -628,10 +628,18 @@ public async Task<int> Count(CountType type, CancellationToken cancellationToken
         }
 
         /// <inheritdoc />
-        public Task<ModeledResponse<TModel>> Get(CancellationToken cancellationToken = default)
+        public Task<ModeledResponse<TModel>> Get(CancellationToken cancellationToken = default, CountType type = CountType.Estimated)
         {
-            var request = Send<TModel>(_method, null, null, cancellationToken);
+            var attr = type.GetAttribute<MapToAttribute>();
+            
+            var headers = new Dictionary<string, string>
+            {
+                { "Prefer", $"count={attr?.Mapping}" }
+            };
+            
+            var request = Send<TModel>(_method, null, headers, cancellationToken);
             Clear();
+            
             return request;
         }
 

PostgrestTests/ClientTests.cs

@@ -1041,6 +1041,24 @@ public async Task TestCountWithFilter()
             Assert.IsNotNull(resp);
         }
 
+        [TestMethod("response count")]
+        public async Task TestCountInResponse()
+        {
+            var client = new Client(BaseUrl);
+
+            var resp = await client.Table<User>().Get(default, CountType.Exact);
+            Assert.IsTrue(resp.Count > -1);
+        }
+        
+        [TestMethod("response count: with filter")]
+        public async Task TestCountInResponseWithFilter()
+        {
+            var client = new Client(BaseUrl);
+
+            var resp = await client.Table<User>().Filter("status", Operator.Equals, "ONLINE").Get(default, CountType.Exact);
+            Assert.IsTrue(resp.Count > -1);
+        }
+
         [TestMethod("support: int arrays")]
         public async Task TestSupportIntArraysAsLists()
         {