fix: add setHeader to more classes

supabase · Aug 15, 2024 · 908b8e7 · 908b8e7
1 parent 1914280
commit 908b8e7
Show file tree

Hide file tree

Showing 6 changed files with 29 additions and 0 deletions.
diff --git a/packages/postgrest/lib/src/postgrest_builder.dart b/packages/postgrest/lib/src/postgrest_builder.dart
@@ -93,6 +93,12 @@ class PostgrestBuilder<T, S, R> implements Future<T> {
     );
   }
 
+  PostgrestBuilder<T, S, R> setHeader(String key, String value) {
+    return _copyWith(
+      headers: {..._headers, key: value},
+    );
+  }
+
   Future<T> _execute() async {
     final String? method = _method;
 

diff --git a/packages/postgrest/lib/src/postgrest_filter_builder.dart b/packages/postgrest/lib/src/postgrest_filter_builder.dart
@@ -485,6 +485,7 @@ class PostgrestFilterBuilder<T> extends PostgrestTransformBuilder<T> {
     return copyWithUrl(url);
   }
 
+  @override
   PostgrestFilterBuilder<T> setHeader(String key, String value) {
     return PostgrestFilterBuilder(
       _copyWith(headers: {..._headers, key: value}),

diff --git a/packages/postgrest/lib/src/postgrest_query_builder.dart b/packages/postgrest/lib/src/postgrest_query_builder.dart
@@ -257,6 +257,7 @@ class PostgrestQueryBuilder<T> extends RawPostgrestBuilder<T, T, T> {
     ));
   }
 
+  @override
   PostgrestQueryBuilder<T> setHeader(String key, String value) {
     return PostgrestQueryBuilder(
       url: _url,

diff --git a/packages/postgrest/lib/src/postgrest_transform_builder.dart b/packages/postgrest/lib/src/postgrest_transform_builder.dart
@@ -6,6 +6,13 @@ class PostgrestTransformBuilder<T> extends RawPostgrestBuilder<T, T, T> {
   PostgrestTransformBuilder<T> copyWithUrl(Uri url) =>
       PostgrestTransformBuilder(_copyWith(url: url));
 
+  @override
+  PostgrestTransformBuilder<T> setHeader(String key, String value) {
+    return PostgrestTransformBuilder(
+      _copyWith(headers: {..._headers, key: value}),
+    );
+  }
+
   /// Performs horizontal filtering with SELECT.
   ///
   /// ```dart

diff --git a/packages/postgrest/lib/src/raw_postgrest_builder.dart b/packages/postgrest/lib/src/raw_postgrest_builder.dart
@@ -41,6 +41,13 @@ class RawPostgrestBuilder<T, S, R> extends PostgrestBuilder<T, S, R> {
     ));
   }
 
+  @override
+  RawPostgrestBuilder<T, S, R> setHeader(String key, String value) {
+    return PostgrestFilterBuilder(
+      _copyWithType(headers: {..._headers, key: value}),
+    );
+  }
+
   /// Converts any response that comes from the server into a type-safe response.
   ///
   /// ```dart

diff --git a/packages/postgrest/lib/src/response_postgrest_builder.dart b/packages/postgrest/lib/src/response_postgrest_builder.dart
@@ -16,6 +16,13 @@ class ResponsePostgrestBuilder<T, S, R> extends PostgrestBuilder<T, S, R> {
           converter: builder._converter,
         );
 
+  @override
+  ResponsePostgrestBuilder<T, S, R> setHeader(String key, String value) {
+    return ResponsePostgrestBuilder(
+      _copyWith(headers: {..._headers, key: value}),
+    );
+  }
+
   /// Converts any response that comes from the server into a type-safe response.
   ///
   /// ```dart