more bodyAsMap helpers

tekartik · Aug 16, 2024 · a3ab7e7 · a3ab7e7
1 parent 02203ab
commit a3ab7e7
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/http/lib/http_utils.dart b/http/lib/http_utils.dart
@@ -4,7 +4,8 @@ export 'src/http_utils.dart'
         TekartikHttpRequestExtension,
         TekartikHttpResponseExtension,
         TekartikStreamedResponseExtension,
-        TekartikTkHttpRequestExtension;
+        TekartikTkHttpRequestExtension,
+        HttpClientResponseExtension;
 export 'src/utils.dart'
     show
         httpDataAsMapOrNull,

diff --git a/http/lib/src/http_utils.dart b/http/lib/src/http_utils.dart
@@ -94,3 +94,14 @@ extension TekartikHttpResponseExtension on http.Response {
         reasonPhrase: reasonPhrase);
   }
 }
+
+/// Common response extension.
+extension HttpClientResponseExtension on HttpClientResponse {
+  /// Get the body as a map.
+  Map<String, Object?> get bodyAsMap => bodyAsMapOrNull!;
+
+  /// Get the body as a map or null.
+  Map<String, Object?>? get bodyAsMapOrNull {
+    return httpDataAsMapOrNull(body);
+  }
+}
diff --git a/http/test/http_utils_test.dart b/http/test/http_utils_test.dart
@@ -39,6 +39,10 @@ void main() {
       HttpRequest? httpRequest;
       expect(httpRequest?.bodyAsMapOrNull, null);
       expect(httpRequest?.bodyAsMap, null);
+
+      HttpClientResponse? clientResponse;
+      expect(clientResponse?.bodyAsMapOrNull, null);
+      expect(clientResponse?.bodyAsMap, null);
     });
   });
 }