From 92020085ffedd54face1ccd1b8d3048f2a07b921 Mon Sep 17 00:00:00 2001
From: David Miguel <me@davidmiguel.com>
Date: Thu, 26 Oct 2023 09:04:35 +0200
Subject: [PATCH] Allow to define global headers included in every request

---
 lib/src/generators/client.dart | 41 +++++++++++++++++++++-------------
 1 file changed, 26 insertions(+), 15 deletions(-)

diff --git a/lib/src/generators/client.dart b/lib/src/generators/client.dart
index be3551c..286ff19 100644
--- a/lib/src/generators/client.dart
+++ b/lib/src/generators/client.dart
@@ -149,6 +149,7 @@ class $clientException implements io.HttpException {
     this.code,
     this.body,
   });
+  
   @override
   final String message;
   @override
@@ -180,31 +181,38 @@ class $clientException implements io.HttpException {
 // CLASS: $clientName
 // ==========================================
 
-/// Client for ${spec.info.title}
-/// 
-/// `baseUrl`: Override baseUrl URL - else defaults to server url defined in spec
+/// Client for ${spec.info.title} (v.${spec.info.version})
 /// 
-/// `client`: Override HTTP client to use for requests
+/// ${spec.info.description ?? 'No description provided'}
 class $clientName {
+  /// Creates a new $clientName instance.
+  /// 
+  /// - [$clientName.baseUrl] Override base URL (default: server url defined in spec)
+  /// - [$clientName.headers] Global headers to be sent with every request
+  /// - [$clientName.client] Override HTTP client to use for requests
   $clientName({
     $authInputCode
-    String? baseUrl,
+    this.baseUrl,
+    this.headers = const {},
     http.Client? client,
   }) : assert(
           baseUrl == null || baseUrl.startsWith('http'),
           'baseUrl must start with http',
-        ) {
-    // Ensure trailing slash is removed from baseUrl
-    this.baseUrl = baseUrl?.replaceAll(RegExp(r'/\$'), '');
-    // Create a retry client
-    this.client = RetryClient(client ?? http.Client());
-  }
-
-  /// User provided override for baseUrl URL
-  late final String? baseUrl;
+        ),
+        assert(
+          baseUrl == null || !baseUrl.endsWith('/'),
+          'baseUrl must not end with /',
+        ),
+        client = RetryClient(client ?? http.Client());
+
+  /// Override base URL (default: server url defined in spec)
+  final String? baseUrl;
+  
+  /// Global headers to be sent with every request
+  final Map<String, String> headers;
 
   /// HTTP client for requests
-  late final http.Client client;
+  final http.Client client;
   
   ${authVariables.isEmpty ? '' : '/// Authentication related variables'}
   ${authVariables.join('\n')}
@@ -290,6 +298,9 @@ class $clientName {
     if (responseType.isNotEmpty){
       headers['${HttpHeaders.acceptHeader}'] = responseType;
     }
+    
+    // Add global headers
+    headers.addAll(this.headers);
 
     // Build the request object
     late http.Response response;