feat: Add custom header to fetch instance

trandaison · Jul 29, 2024 · 5af577d · 5af577d
1 parent b06924a
commit 5af577d
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 15 deletions.
diff --git a/src/module.ts b/src/module.ts
@@ -42,20 +42,24 @@ export default defineNuxtModule<AuthOptions>({
         url: "/login",
         method: "POST",
         property: "",
+        headers: undefined,
       },
-      logout: { url: "/logout", method: "DELETE" },
+      logout: { url: "/logout", method: "DELETE", headers: undefined },
       refresh: {
         url: "/refresh_tokens",
         method: "POST",
         property: "",
+        headers: undefined,
       },
       user: {
         url: "/me",
         method: "GET",
         property: "",
+        headers: undefined,
       },
-      signup: { url: "/signup", method: "POST" },
+      signup: { url: "/signup", method: "POST", headers: undefined },
     },
+    headers: undefined,
     token: {
       headerName: "Authorization",
       type: "Bearer",

diff --git a/src/runtime/services/Auth.ts b/src/runtime/services/Auth.ts
@@ -89,7 +89,8 @@ export class Auth implements AuthService {
       this.loginPromise ??= this.httpService.call<T>(
         this.config.endpoints.login.method,
         this.config.endpoints.login.url,
-        credentials
+        credentials,
+        { headers: this.config.endpoints.login.headers }
       );
       const res = await this.loginPromise;
       const data = this.getProperty(res, "login");
@@ -114,7 +115,7 @@ export class Auth implements AuthService {
         this.config.endpoints.user.method,
         this.config.endpoints.user.url,
         undefined,
-        { auth: true }
+        { auth: true, headers: this.config.endpoints.user.headers }
       );
       const res = await this.fetchUserPromise;
       const data = this.getProperty(res, "user");
@@ -133,7 +134,9 @@ export class Auth implements AuthService {
       if (callApi) {
         this.logoutPromise ??= this.httpService.call<unknown>(
           this.config.endpoints.logout.method,
-          this.config.endpoints.logout.url
+          this.config.endpoints.logout.url,
+          undefined,
+          { headers: this.config.endpoints.logout.headers }
         );
         await this.logoutPromise;
       }
@@ -150,7 +153,8 @@ export class Auth implements AuthService {
       this.refreshTokensPromise ??= this.httpService.call(
         this.config.endpoints.refresh.method,
         this.config.endpoints.refresh.url,
-        { [this.config.refreshToken.paramName]: this.storage.refreshToken() }
+        { [this.config.refreshToken.paramName]: this.storage.refreshToken() },
+        { headers: this.config.endpoints.refresh.headers }
       );
       const res = await this.refreshTokensPromise;
       const data = this.getProperty(res, "refresh");

diff --git a/src/runtime/services/HttpService.ts b/src/runtime/services/HttpService.ts
@@ -32,10 +32,6 @@ export default class HttpService {
     this.$fetch = $fetch.create({
       baseURL: baseUrl,
 
-      headers: {
-        Accept: "application/json",
-      },
-
       onRequest: this.onRequest.bind(this),
 
       onResponse: async (context) => {
@@ -54,12 +50,11 @@ export default class HttpService {
         }
 
         const { token } = await this.$auth.refreshTokens().catch(() => ({
-
-          token: undefined
-        }))
+          token: undefined,
+        }));
 
         if (!token) {
-          this.onAuthFailure(AuthStatus.Expired)
+          this.onAuthFailure(AuthStatus.Expired);
           return;
         }
 
@@ -91,7 +86,11 @@ export default class HttpService {
   }
 
   private onRequest({ options }: FetchContext) {
-    options.headers = new Headers(options.headers);
+    options.headers = new Headers({
+      Accept: "application/json",
+      ...this.$configs.headers,
+      ...options.headers,
+    });
     const authOption = options.auth ?? true;
     const token = this.$auth.accessToken();
     if (authOption !== false && token) {

diff --git a/src/types.ts b/src/types.ts
@@ -20,6 +20,7 @@ export interface EndpointOption {
   url: string;
   method?: string;
   property?: string;
+  headers?: Record<string, any>;
 }
 
 export interface AuthEndpointOptions {
@@ -69,6 +70,7 @@ export interface RedirectOptions {
 
 export interface AuthConfig {
   endpoints: AuthEndpointOptions;
+  headers?: Record<string, any>;
   token: TokenOptions;
   refreshToken: RefreshTokenOptions;
   redirect: RedirectOptions;
@@ -84,6 +86,7 @@ export interface AuthConfig {
 
 export interface AuthOptions {
   endpoints?: Partial<AuthEndpointOptions>;
+  headers?: Record<string, any>;
   token?: Partial<TokenOptions>;
   refreshToken?: Partial<RefreshTokenOptions>;
   cookie?: Partial<CookieOptions>;